From f9d08a14490b014da1bfdf0e3adea72912a5224e Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Thu, 26 Jun 2025 14:45:46 +0800
Subject: [PATCH] 'commit'

---
 dsRag/Milvus/Utils/MilvusCollectionManager.py |   2 +-
 .../MilvusCollectionManager.cpython-310.pyc   | Bin 4425 -> 4418 bytes
 dsRag/Milvus/X1_create_collection.py          |   2 +-
 dsRag/Milvus/X6_search_near_data.py           |   3 +--
 4 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/dsRag/Milvus/Utils/MilvusCollectionManager.py b/dsRag/Milvus/Utils/MilvusCollectionManager.py
index 389b7e3a..fdfce8d7 100644
--- a/dsRag/Milvus/Utils/MilvusCollectionManager.py
+++ b/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]  # 返回第一条记录
diff --git a/dsRag/Milvus/Utils/__pycache__/MilvusCollectionManager.cpython-310.pyc b/dsRag/Milvus/Utils/__pycache__/MilvusCollectionManager.cpython-310.pyc
index b42a2a2bccdbe4750ed7e14e41cca9dc53253cf8..99ebc8ab8d5ac3f8dceeb302ab1d41a34f6fb80a 100644
GIT binary patch
delta 145
zcmX@9bV!LepO=@50SJoT#AK*%<aOg>WhqHaFP<FFCC?bOxt_~~i7{&O5uP+BZJ?rC
zyv3=BMadcQ1&Kw8xy41QAdR{}A_^u^lv-SxQ&OY>k^q?tA-F*-z0EninM{moldthl
WWz?FSBjCZU2^I$tW|Q{|bO8V|HzmCQ

delta 147
zcmX@4bW({opO=@50SM}z#boGh<aOiX;ZDg<F3nBND~ZocnH<3-&v<Kd1(ypG<E_bi
zdD8TBfoh|8i&GPek~8895{nXZi;Gl&VnzDM5=E)Sr8y-<nv(;0g&7Str}AbpF{)2K
e&p(w>dvdCPhkzDHhXIH%2N7H#b(1d$6afH^0V$sV

diff --git a/dsRag/Milvus/X1_create_collection.py b/dsRag/Milvus/X1_create_collection.py
index f1a33d1d..6d7372b9 100644
--- a/dsRag/Milvus/X1_create_collection.py
+++ b/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}'...")
diff --git a/dsRag/Milvus/X6_search_near_data.py b/dsRag/Milvus/X6_search_near_data.py
index 0546c5cd..37ec1c4a 100644
--- a/dsRag/Milvus/X6_search_near_data.py
+++ b/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,