Retrieval service optimization (#13849)

12 months ago · abe5aca3e2
parent bea10b4356
commit abe5aca3e2
1 changed files with 0 additions and 2 deletions
--- a/api/core/rag/datasource/retrieval_service.py
+++ b/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)