Просмотр исходного кода

fix: establish connection after starting new Milvus instance & update migration script to use backup db

guantao 1 час назад
Родитель
Сommit
015f7292fa
2 измененных файлов с 9 добавлено и 1 удалено
  1. 8 0
      knowhub/vector_store.py
  2. 1 1
      migrate_knowledge.py

+ 8 - 0
knowhub/vector_store.py

@@ -41,6 +41,14 @@ class MilvusStore:
             try:
                 default_server.start()
                 print(f"[Milvus] Milvus Lite 启动成功 (端口 {default_server.listen_port})")
+                # 启动后建立连接
+                connections.connect(
+                    alias="default",
+                    host='127.0.0.1',
+                    port=default_server.listen_port,
+                    timeout=5
+                )
+                print(f"[Milvus] 已连接到新启动的实例")
             except Exception as e:
                 print(f"[Milvus] 启动失败: {e}")
                 # 尝试连接到可能已经在运行的实例

+ 1 - 1
migrate_knowledge.py

@@ -22,7 +22,7 @@ async def migrate_knowledge():
     """迁移知识数据"""
 
     # 源数据库路径
-    source_db = Path.home() / "main_agent/knowhub/knowhub.db"
+    source_db = Path("/root/knowhub_backup_20260309_204451.db")
     if not source_db.exists():
         print(f"❌ 源数据库不存在: {source_db}")
         return