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

fix tidb metadata filter

Browse Source
This commit is contained in:
jyong 2025-03-19 19:16:37 +08:00
parent f6404f93ca
commit 2ff931d82a
1 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -328,16 +328,13 @@ 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 = [] if filter.must:
for document_id_filter in document_ids_filter: filter.must.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.MatchAny(any=document_ids_filter),
) )
) )
if should_conditions:
filter.should = should_conditions # type: ignore
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,
@ -382,16 +379,13 @@ 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 = [] if scroll_filter.must:
for document_id_filter in document_ids_filter: scroll_filter.must.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.MatchAny(any=document_ids_filter),
) )
) )
if should_conditions:
scroll_filter.should = should_conditions # type: ignore
response = self._client.scroll( response = self._client.scroll(
collection_name=self._collection_name, collection_name=self._collection_name,
scroll_filter=scroll_filter, scroll_filter=scroll_filter,