'commit'

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)