Parcourir la source

fix tidb metadata filter (#16237)

tags/1.1.1
Jyong il y a 7 mois
Parent
révision
a8879057c0
Aucun compte lié à l'adresse e-mail de l'auteur

+ 6
- 12
api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py Voir le fichier

@@ -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,

Chargement…
Annuler
Enregistrer