瀏覽代碼

Retrieval service optimization (#13849)

Charlie.Wei 2 月之前
父節點
當前提交
abe5aca3e2
共有 1 個文件被更改,包括 0 次插入2 次删除
  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)