From 45470fd13d4db1bc707c1d7dbde43e31d5c47f12 Mon Sep 17 00:00:00 2001
From: gameloader <ggwqqo@163.com>
Date: Thu, 17 Apr 2025 11:26:17 +0800
Subject: [PATCH] change to milvus standlone

---
 config.py        | 1 +
 data_handling.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/config.py b/config.py
index 9a56d50..9354032 100644
--- a/config.py
+++ b/config.py
@@ -41,6 +41,7 @@ USER_ID_PREFIX = "user"
 MILVUS_PERSIST_BASE_DIR = Path("./milvus_user_data_openai")
 MILVUS_INDEX_PARAMS = {"index_type": "FLAT", "metric_type": "L2", "params": {}}
 MILVUS_SEARCH_PARAMS = {"metric_type": "L2", "params": {}}
+MILVUS_STAND_URI = "http://localhost:19530"
 
 # --- RAG Pipeline Configuration (保持不变) ---
 RETRIEVER_TOP_K = 3
diff --git a/data_handling.py b/data_handling.py
index 0bffc7c..64f8248 100644
--- a/data_handling.py
+++ b/data_handling.py
@@ -14,6 +14,7 @@ from config import (
     MILVUS_PERSIST_BASE_DIR,
     MILVUS_INDEX_PARAMS,
     MILVUS_SEARCH_PARAMS,
+    MILVUS_STAND_URI,
 )
 
 logger = logging.getLogger(__name__)  # Use logger
@@ -38,7 +39,7 @@ def initialize_milvus_lite(user_id: str) -> MilvusDocumentStore:
     print(f"Expecting Embedding Dimension (for first write): {OPENAI_EMBEDDING_DIM}")
 
     document_store = MilvusDocumentStore(
-        connection_args={"uri": milvus_uri},
+        connection_args={"uri": MILVUS_STAND_URI},
         collection_name=user_id,  # Default or customize
         index_params=MILVUS_INDEX_PARAMS,  # Pass index config
         search_params=MILVUS_SEARCH_PARAMS,  # Pass search config