Retrieval service optimization (#13849)

This commit is contained in:
Charlie.Wei 2025-02-17 18:22:36 +08:00 committed by GitHub
parent bea10b4356
commit abe5aca3e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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