浏览代码

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

tags/1.7.2
Qiang Lee 2 个月前
父节点
当前提交
e9045a8838
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. 5
    0
      api/core/rag/datasource/vdb/tencent/tencent_vector.py

+ 5
- 0
api/core/rag/datasource/vdb/tencent/tencent_vector.py 查看文件

@@ -246,6 +246,10 @@ class TencentVector(BaseVector):
return self._get_search_res(res, score_threshold)

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:
return []
res = self._client.hybrid_search(
@@ -269,6 +273,7 @@ class TencentVector(BaseVector):
),
retrieve_vector=False,
limit=kwargs.get("top_k", 4),
filter=filter,
)
score_threshold = float(kwargs.get("score_threshold") or 0.0)
return self._get_search_res(res, score_threshold)

正在加载...
取消
保存