[BUG]: gracefully handle compaction orchestrator errors (#2940)

chroma-core · Oct 11, 2024 · 4693b0f · 4693b0f
1 parent 01a8893
commit 4693b0f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rust/worker/src/execution/orchestration/compact.rs b/rust/worker/src/execution/orchestration/compact.rs
@@ -139,6 +139,8 @@ impl ChromaError for GetSegmentWritersError {
 enum CompactionError {
     #[error(transparent)]
     SystemTimeError(#[from] std::time::SystemTimeError),
+    #[error("Result channel dropped")]
+    ResultChannelDropped,
 }
 
 impl ChromaError for CompactionError {
@@ -519,7 +521,8 @@ impl CompactOrchestrator {
         let mut handle = self.system.clone().start_component(self);
         let result = rx.await;
         handle.stop();
-        result.unwrap()
+        result
+            .map_err(|_| Box::new(CompactionError::ResultChannelDropped) as Box<dyn ChromaError>)?
     }
 }