fix external retrieval without segment id (#11423)

langgenius · Dec 6, 2024 · a8d32f9 · a8d32f9
1 parent 5093337
commit a8d32f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/core/app/apps/base_app_generate_response_converter.py b/api/core/app/apps/base_app_generate_response_converter.py
@@ -82,7 +82,7 @@ def _get_simple_metadata(cls, metadata: dict[str, Any]):
             for resource in metadata["retriever_resources"]:
                 updated_resources.append(
                     {
-                        "segment_id": resource["segment_id"],
+                        "segment_id": resource.get("segment_id", ""),
                         "position": resource["position"],
                         "document_name": resource["document_name"],
                         "score": resource["score"],