'commit'

dsLightRag/Volcengine/T1_DropIndex.py

@@ -1,5 +1,5 @@
 import json
-from VikingDBMemoryService import VikingDBMemoryService
+from VikingDBMemoryService import VikingDBMemoryService, MEMORY_COLLECTION_NAME
 from Config.Config import VOLC_ACCESSKEY, VOLC_SECRETKEY
 
 def drop_existing_collection(collection_name):
@@ -34,5 +34,4 @@ def drop_existing_collection(collection_name):
 
 if __name__ == "__main__":
     # 可修改为目标集合名称
-    TARGET_COLLECTION = "emotional_support"
-    drop_existing_collection(TARGET_COLLECTION)
+    drop_existing_collection(MEMORY_COLLECTION_NAME)

dsLightRag/Volcengine/T2_CreateIndex.py

@@ -1,7 +1,7 @@
 import json
 
 from Config.Config import VOLC_ACCESSKEY, VOLC_SECRETKEY
-from VikingDBMemoryService import VikingDBMemoryService
+from VikingDBMemoryService import VikingDBMemoryService, MEMORY_COLLECTION_NAME
 from Volcengine.chat import wait_for_collection_ready
 
 
@@ -52,6 +52,4 @@ def create_memory_collection(collection_name, description="情感陪伴记忆库
         raise
 
 if __name__ == "__main__":
-    # 可修改为目标集合名称
-    TARGET_COLLECTION = "emotional_support"
-    create_memory_collection(TARGET_COLLECTION)
+    create_memory_collection(MEMORY_COLLECTION_NAME)

dsLightRag/Volcengine/VikingDBMemoryService.py

@@ -17,6 +17,8 @@ from volcenginesdkarkruntime import Ark
 
 from Config.Config import VOLC_SECRETKEY, VOLC_ACCESSKEY, VOLC_API_KEY
 
+# 记忆体集合名称
+MEMORY_COLLECTION_NAME="dsideal_collection"
 
 def initialize_services(ak=None, sk=None, ark_api_key=None):
     """初始化记忆数据库服务和LLM客户端"""