Fix chunk offset (#47)

Signed-off-by: declark1 <daniel.clark@ibm.com>
foundation-model-stack · May 24, 2024 · f6e8eed · f6e8eed
1 parent 3e38689
commit f6e8eed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/orchestrator.rs b/src/orchestrator.rs
@@ -294,7 +294,7 @@ async fn handle_chunk_task(
                     .results
                     .into_iter()
                     .map(|token| Chunk {
-                        offset,
+                        offset: offset + token.start as usize,
                         text: token.text,
                     })
                     .collect::<Vec<_>>();