From abe5aca3e217a5575dd733cbd9d94e3107e68ea4 Mon Sep 17 00:00:00 2001
From: "Charlie.Wei" <luowei@cvte.com>
Date: Mon, 17 Feb 2025 18:22:36 +0800
Subject: [PATCH] Retrieval service optimization (#13849)

---
 api/core/rag/datasource/retrieval_service.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/api/core/rag/datasource/retrieval_service.py b/api/core/rag/datasource/retrieval_service.py
index 07a9055944..725df69fd0 100644
--- 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)