refactor: streamline segment casting and enhance output validation logic

langgenius · Feb 11, 2025 · 127aa4b · 127aa4b
1 parent 87b2614
commit 127aa4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/core/rag/datasource/retrieval_service.py b/api/core/rag/datasource/retrieval_service.py
@@ -354,10 +354,10 @@ def format_retrieval_documents(cls, documents: list[Document]) -> list[Retrieval
                         .first()
                     )
 
-                    segment = cast(DocumentSegment, segment)
                     if not segment:
                         continue
 
+                    segment = cast(DocumentSegment, segment)
                     include_segment_ids.add(segment.id)
                     record = {
                         "segment": segment,