huanghai/dsProject
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

'commit'

Browse Source
This commit is contained in:
HuangHai
2025-09-07 08:02:23 +08:00
parent 1108011e79
commit adb4416ca2
3 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dsLightRag/Volcengine/T1_DropIndex.py
View File

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

6
dsLightRag/Volcengine/T2_CreateIndex.py
View File

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

2
dsLightRag/Volcengine/VikingDBMemoryService.py
View File

@@ -17,6 +17,8 @@ from volcenginesdkarkruntime import Ark
from Config.Config import VOLC_SECRETKEY, VOLC_ACCESSKEY, VOLC_API_KEY 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): def initialize_services(ak=None, sk=None, ark_api_key=None):
"""初始化记忆数据库服务和LLM客户端""" """初始化记忆数据库服务和LLM客户端"""