fix changes after rebase

Author: codetheweb

rust/worker/src/execution/operators/apply_log_to_segment_writer.rs

@@ -1,5 +1,4 @@
-use std::sync::Arc;
-
+use super::materialize_logs::MaterializeLogOutput;
 use crate::execution::operator::Operator;
 use crate::segment::metadata_segment::MetadataSegmentError;
 use crate::segment::record_segment::ApplyMaterializedLogError;
@@ -9,11 +8,10 @@ use crate::segment::SegmentWriter;
 use async_trait::async_trait;
 use chroma_error::ChromaError;
 use chroma_error::ErrorCodes;
+use std::sync::Arc;
 use thiserror::Error;
 use tracing::Instrument;
 
-use super::materialize_logs::MaterializeLogOutput;
-
 #[derive(Error, Debug)]
 pub enum ApplyLogToSegmentWriterOperatorError {
     #[error("Preparation for log materialization failed {0}")]
@@ -88,7 +86,8 @@ impl<Writer: SegmentWriter + Send + Sync + Clone>
             .segment_writer
             .apply_materialized_log_chunk(materialized_chunk.clone())
             .instrument(tracing::trace_span!(
-                "Apply materialized logs to record segment"
+                "Apply materialized logs",
+                segment = input.segment_writer.get_name()
             ))
             .await
         {

rust/worker/src/execution/orchestration/compact.rs

@@ -41,6 +41,7 @@ use crate::system::ComponentHandle;
 use crate::system::Handler;
 use crate::system::ReceiverForMessage;
 use crate::system::System;
+use arrow::ipc::writer;
 use async_trait::async_trait;
 use chroma_blockstore::provider::BlockfileProvider;
 use chroma_error::ChromaError;
@@ -671,7 +672,14 @@ impl Handler<TaskResult<MaterializeLogOutput, MaterializeLogOperatorError>>
         let materialized_result = Arc::new(materialized_result);
 
         let (record_segment_writer, hnsw_segment_writer, metadata_segment_writer) =
-            self.get_segment_writers().await.unwrap(); // todo
+            match self.get_segment_writers().await {
+                Ok(writers) => writers,
+                Err(e) => {
+                    tracing::error!("Error getting segment writers: {:?}", e);
+                    terminate_with_error(self.result_channel.take(), e, ctx);
+                    return;
+                }
+            };
 
         self.dispatch_apply_log_to_segment_writer_task(
             record_segment_writer,