From fd499a55185c2ed1f88f6296f0821a9e2aa5ff89 Mon Sep 17 00:00:00 2001 From: jyong <718720800@qq.com> Date: Tue, 11 Mar 2025 15:12:02 +0800 Subject: [PATCH] fix text split --- .../vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) 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 9fcbc2ecbc..b7e45cf054 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 @@ -328,12 +328,13 @@ class TidbOnQdrantVector(BaseVector): ) document_ids_filter = kwargs.get("document_ids_filter") if document_ids_filter: - filter.must.append( - models.FieldCondition( - key="metadata.document_id", - match=models.MatchAny(any=document_ids_filter), + for document_id_filter in document_ids_filter: + filter.should.append( + models.FieldCondition( + key="metadata.document_id", + match=models.MatchValue(value=document_id_filter), + ) ) - ) results = self._client.search( collection_name=self._collection_name, query_vector=query_vector,