|
|
|
|
@ -328,16 +328,13 @@ class TidbOnQdrantVector(BaseVector):
|
|
|
|
|
)
|
|
|
|
|
document_ids_filter = kwargs.get("document_ids_filter")
|
|
|
|
|
if document_ids_filter:
|
|
|
|
|
should_conditions = []
|
|
|
|
|
for document_id_filter in document_ids_filter:
|
|
|
|
|
should_conditions.append(
|
|
|
|
|
if filter.must:
|
|
|
|
|
filter.must.append(
|
|
|
|
|
models.FieldCondition(
|
|
|
|
|
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(
|
|
|
|
|
collection_name=self._collection_name,
|
|
|
|
|
query_vector=query_vector,
|
|
|
|
|
@ -382,16 +379,13 @@ class TidbOnQdrantVector(BaseVector):
|
|
|
|
|
)
|
|
|
|
|
document_ids_filter = kwargs.get("document_ids_filter")
|
|
|
|
|
if document_ids_filter:
|
|
|
|
|
should_conditions = []
|
|
|
|
|
for document_id_filter in document_ids_filter:
|
|
|
|
|
should_conditions.append(
|
|
|
|
|
if scroll_filter.must:
|
|
|
|
|
scroll_filter.must.append(
|
|
|
|
|
models.FieldCondition(
|
|
|
|
|
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(
|
|
|
|
|
collection_name=self._collection_name,
|
|
|
|
|
scroll_filter=scroll_filter,
|
|
|
|
|
|