'commit'

dsRag/Milvus/Utils/MilvusCollectionManager.py

@@ -71,7 +71,7 @@ class MilvusCollectionManager:
             # 使用 Milvus 的 query 方法查询指定 ID 的记录
             results = self.collection.query(
                 expr=f"id == {id}",  # 查询条件
-                output_fields=["id", "document_id", "user_input", "timestamp"]  # 返回的字段
+                output_fields=["id", "tags", "user_input", "timestamp"]  # 返回的字段
             )
             if results:
                 return results[0]  # 返回第一条记录

dsRag/Milvus/X1_create_collection.py

@@ -32,7 +32,7 @@ fields = [
     FieldSchema(name="timestamp", dtype=DataType.VARCHAR, max_length=32),
     FieldSchema(name="embedding", dtype=DataType.FLOAT_VECTOR, dim=MS_DIMENSION)
 ]
-schema_description = "Chat records collection with document_id , user_input, and timestamp"
+schema_description = "Chat records collection with tags , user_input, and timestamp"
 
 # 6. 创建集合
 print(f"正在创建集合 '{collection_name}'...")

dsRag/Milvus/X6_search_near_data.py

@@ -54,8 +54,7 @@ search_params = {
 }
 # 哪些文档查询，哪些不查询，我说了算！
 # 这样的话，我就可以打多个标签了！
-# expr = "document_id in ['MATH_DATA_1', 'MATH_DATA_2']"
-expr = "tags['tags'] == 'MATH_DATA_1'"
+expr = "array_contains(tags['tags'], 'MATH_DATA_1')"
 results = collection_manager.search(
     current_embedding,
     search_params,