Jelajahi Sumber

Retrieval service optimization (#13849)

Charlie.Wei 2 bulan lalu
induk
melakukan
abe5aca3e2
1 mengubah file dengan 0 tambahan dan 2 penghapusan
  1. 0 2
      api/core/rag/datasource/retrieval_service.py

+ 0 - 2
api/core/rag/datasource/retrieval_service.py

@@ -301,7 +301,6 @@ class RetrievalService:
                                 DocumentSegment.id,
                                 DocumentSegment.content,
                                 DocumentSegment.answer,
-                                DocumentSegment.doc_metadata,
                             )
                         )
                         .first()
@@ -362,7 +361,6 @@ class RetrievalService:
                     record = {
                         "segment": segment,
                         "score": document.metadata.get("score"),  # type: ignore
-                        "segment_metadata": segment.doc_metadata,
                     }
                     records.append(record)