From 828c0b47819510c2fe989f6d27b2f11a74ae5a1e Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Thu, 26 Jun 2025 14:27:24 +0800 Subject: [PATCH] 'commit' --- dsRag/Milvus/X6_search_near_data.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dsRag/Milvus/X6_search_near_data.py b/dsRag/Milvus/X6_search_near_data.py index a7e55b85..bf1a8858 100644 --- a/dsRag/Milvus/X6_search_near_data.py +++ b/dsRag/Milvus/X6_search_near_data.py @@ -52,16 +52,16 @@ search_params = { "metric_type": "L2", # 使用 L2 距离度量方式 "params": {"nprobe": MS_NPROBE} # 设置 IVF_FLAT 的 nprobe 参数 } -expr = "document_id == 'MATH_DATA_1'" # 这回我只想查找 document_id='MATH_DATA_2' 的数据 -#expr = "document_id == 'MATH_DATA_2'" # 这回我只想查找 document_id='MATH_DATA_2' 的数据 +# 哪些文档查询,哪些不查询,我说了算! +# expr = "document_id in ['MATH_DATA_1', 'MATH_DATA_2']" +expr = "document_id in ['MATH_DATA_1']" results = collection_manager.search( current_embedding, search_params, - expr=expr, # 新增条件表达式 + expr=expr, # 使用in操作符 limit=10 ) -#results = collection_manager.search(current_embedding, search_params, limit=10) # 返回 2 条结果 end_time = time.time() # 9. 输出查询结果