diff --git a/dsLightRag/Config/__pycache__/Config.cpython-310.pyc b/dsLightRag/Config/__pycache__/Config.cpython-310.pyc index 7d0935e4..0824b323 100644 Binary files a/dsLightRag/Config/__pycache__/Config.cpython-310.pyc and b/dsLightRag/Config/__pycache__/Config.cpython-310.pyc differ diff --git a/dsLightRag/Volcengine/__pycache__/VikingDBMemoryService.cpython-310.pyc b/dsLightRag/Volcengine/__pycache__/VikingDBMemoryService.cpython-310.pyc new file mode 100644 index 00000000..92e57fa8 Binary files /dev/null and b/dsLightRag/Volcengine/__pycache__/VikingDBMemoryService.cpython-310.pyc differ diff --git a/dsLightRag/Volcengine/chat.py b/dsLightRag/Volcengine/chat.py index 1b84e341..a58a1cd1 100644 --- a/dsLightRag/Volcengine/chat.py +++ b/dsLightRag/Volcengine/chat.py @@ -7,7 +7,12 @@ from volcenginesdkarkruntime import Ark from Config.Config import VOLC_ACCESSKEY, VOLC_SECRETKEY, VOLC_API_KEY from VikingDBMemoryService import VikingDBMemoryService - +""" +在记忆库准备好后,我们先模拟一段包含两轮的完整对话。 +对话结束后,把这段对话历史消息写入记忆库。然后再开启一个新话题,提出和刚才相关的问题, +AI 就能用刚写入的记忆来回答。 +注意:首次写入需要 3–5 分钟建立索引,这段时间内检索会报错。 +""" def initialize_services(): load_dotenv() ak = VOLC_ACCESSKEY @@ -170,8 +175,8 @@ def main(): try: memory_service, llm_client = initialize_services() collection_name = "emotional_support" - user_id = "xiaoming" - assistant_id = "assistant" + user_id = "xiaoming" # 用户ID:小明 + assistant_id = "assistant1" # 助手ID:助手1 ensure_collection_exists(memory_service, collection_name) except Exception as e: print(f"初始化失败: {e}")