From d7c1f43b49f6b31197e41497be15f4d195bc72c4 Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Wed, 4 Dec 2024 16:13:23 +0800 Subject: [PATCH] fix tidb full-text-search vector missed (#11337) --- .../rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py b/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py index a38f84636e..cfd47aac5b 100644 --- a/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py +++ b/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py @@ -375,7 +375,6 @@ class TidbOnQdrantVector(BaseVector): for result in results: if result: document = self._document_from_scored_point(result, Field.CONTENT_KEY.value, Field.METADATA_KEY.value) - document.metadata["vector"] = result.vector documents.append(document) return documents @@ -394,6 +393,7 @@ class TidbOnQdrantVector(BaseVector): ) -> Document: return Document( page_content=scored_point.payload.get(content_payload_key), + vector=scored_point.vector, metadata=scored_point.payload.get(metadata_payload_key) or {}, )