wip

chroma-core · Mar 20, 2024 · 9248399 · 9248399
1 parent c19de4e
commit 9248399
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/rust/worker/src/execution/orchestration/hnsw.rs b/rust/worker/src/execution/orchestration/hnsw.rs
@@ -156,6 +156,14 @@ impl Handler<PullLogsOutput> for HnswQueryOrchestrator {
         ctx: &crate::system::ComponentContext<HnswQueryOrchestrator>,
     ) {
         self.state = ExecutionState::Dedupe;
+        match self.result_channel.take() {
+            Some(tx) => {
+                let _ = tx.send("done".to_string());
+            }
+            None => {
+                // Log an error
+            }
+        }
         // TODO: implement the remaining state transitions and operators
         // The query orchestrator kills itself in the last state
     }

diff --git a/rust/worker/src/lib.rs b/rust/worker/src/lib.rs
@@ -34,7 +34,7 @@ pub async fn query_service_entrypoint() {
         }
     };
 
-    let mut system: system::System = system::System::new();
+    let system: system::System = system::System::new();
 
     // TODO: configurable
     let dispatcher =