fix type of "end"

Aleph-Alpha · Apr 12, 2024 · 1579578 · 1579578
1 parent 4b801ab
commit 1579578
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/intelligence_layer/use_cases/qa/multiple_chunk_retriever_qa.py b/src/intelligence_layer/use_cases/qa/multiple_chunk_retriever_qa.py
@@ -101,7 +101,9 @@ def _get_highlights_per_chunk(
                 if highlight.start < next_start and highlight.end > current_start:
                     highlights_with_indices_fixed = ScoredTextHighlight(
                         start=max(0, highlight.start - current_start),
-                        end=min(highlight.end - current_start, next_start),
+                        end=min(highlight.end - current_start, next_start)
+                        if isinstance(next_start, int)
+                        else highlight.end,
                         score=highlight.score,
                     )
                     current_overlaps.append(highlights_with_indices_fixed)