🐛 Fix offset for standalone content results

Signed-off-by: gkumbhat <[email protected]>
foundation-model-stack · Aug 23, 2024 · 6e0b9f9 · 6e0b9f9
1 parent 52bc871
commit 6e0b9f9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/orchestrator/unary.rs b/src/orchestrator/unary.rs
@@ -594,10 +594,14 @@ pub async fn detect_content(
     let results = chunks
         .into_iter()
         .zip(response)
-        .flat_map(|(_chunk, response)| {
+        .flat_map(|(chunk, response)| {
             response
                 .into_iter()
-                .filter_map(|resp| (resp.score >= threshold).then_some(resp))
+                .filter_map(|mut resp| {
+                    resp.start += chunk.offset;
+                    resp.end += chunk.offset;
+                    (resp.score >= threshold).then_some(resp)
+                })
                 .collect::<Vec<_>>()
         })
         .collect::<Vec<_>>();