Merge branch 'feat/support-knowledge-metadata' into deploy/dev

2025-03-11 15:12:21 +08:00 · 2025-03-11 15:12:21 +08:00 · b8520dd91c
commit b8520dd91c
parent 1246621cfd fd499a5518
1 changed files with 6 additions and 5 deletions
--- 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,