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

Fix: Apply Metadata Filters Correctly in Full-Text Search Mode for Tencent Cloud Vector Database (#23564)

Browse Source
This commit is contained in:
Qiang Lee
2025-08-07 20:36:06 +08:00
committed by GitHub
parent 55487ba0c6
commit e9045a8838
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/rag/datasource/vdb/tencent/tencent_vector.py
View File

@@ -246,6 +246,10 @@ class TencentVector(BaseVector):
return self._get_search_res(res, score_threshold) return self._get_search_res(res, score_threshold)
def search_by_full_text(self, query: str, **kwargs: Any) -> list[Document]: def search_by_full_text(self, query: str, **kwargs: Any) -> list[Document]:
document_ids_filter = kwargs.get("document_ids_filter")
filter = None
if document_ids_filter:
filter = Filter(Filter.In("metadata.document_id", document_ids_filter))
if not self._enable_hybrid_search: if not self._enable_hybrid_search:
return [] return []
res = self._client.hybrid_search( res = self._client.hybrid_search(
@@ -269,6 +273,7 @@ class TencentVector(BaseVector):
), ),
retrieve_vector=False, retrieve_vector=False,
limit=kwargs.get("top_k", 4), limit=kwargs.get("top_k", 4),
filter=filter,
) )
score_threshold = float(kwargs.get("score_threshold") or 0.0) score_threshold = float(kwargs.get("score_threshold") or 0.0)
return self._get_search_res(res, score_threshold) return self._get_search_res(res, score_threshold)