outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
jyong 2025-03-11 15:34:29 +08:00
commit f88de88c97
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py
View File

@ -328,13 +328,15 @@ class TidbOnQdrantVector(BaseVector):
) )
document_ids_filter = kwargs.get("document_ids_filter") document_ids_filter = kwargs.get("document_ids_filter")
if document_ids_filter: if document_ids_filter:
should_conditions = []
for document_id_filter in document_ids_filter: for document_id_filter in document_ids_filter:
filter.should.append( should_conditions.append(
models.FieldCondition( models.FieldCondition(
key="metadata.document_id", key="metadata.document_id",
match=models.MatchValue(value=document_id_filter), match=models.MatchValue(value=document_id_filter),
) )
) )
filter.should = should_conditions
results = self._client.search( results = self._client.search(
collection_name=self._collection_name, collection_name=self._collection_name,
query_vector=query_vector, query_vector=query_vector,