瀏覽代碼

fix tidb metadata filter (#16237)

tags/1.1.1
Jyong 7 月之前
父節點
當前提交
a8879057c0
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 6 行新增12 行删除
  1. 6
    12
      api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py

+ 6
- 12
api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_on_qdrant_vector.py 查看文件

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

Loading…
取消
儲存