Parcourir la source

增加测试milvus

lingmin_package@163.com il y a 4 mois
Parent
commit
362c152309
2 fichiers modifiés avec 13 ajouts et 2 suppressions
  1. 11 0
      README.md
  2. 2 2
      foundation/rag/vector/milvus_vector.py

+ 11 - 0
README.md

@@ -18,6 +18,17 @@
     pip install psycopg2-binary pgvector -i https://mirrors.aliyun.com/pypi/simple/
     pip install pymilvus sentence-transformers 
  
+
+
+
+### 向量数据库 milvus
+   - cd /home/cjb/lq_workspace/milvus
+   - docker-compose up -d
+   - 检查服务是否正常 http://192.168.0.3:9091/healthz
+   - 拉取并运行 Attu http://192.168.0.3:13000/#/connect
+   
+   
+
  
 ### 测试接口
 

+ 2 - 2
foundation/rag/vector/milvus_vector.py

@@ -208,7 +208,7 @@ class MilvusVectorManager(BaseVectorDB):
             # 执行搜索
             results = collection.search(
                 data=[query_embedding],
-                anns_field="embedding",
+                anns_field="vector",
                 param=search_params,
                 limit=top_k,
                 expr=filter_expr,
@@ -255,7 +255,7 @@ class MilvusVectorManager(BaseVectorDB):
             
             results = collection.search(
                 data=[query_embedding],
-                anns_field="embedding",
+                anns_field="vector",
                 param=search_params,
                 limit=top_k,
                 expr=filter_expr,