'commit'

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. 输出查询结果