diff --git a/dsLightRag/Volcengine/T2_CreateIndex.py b/dsLightRag/Volcengine/T2_CreateIndex.py index 2b57df04..78453a35 100644 --- a/dsLightRag/Volcengine/T2_CreateIndex.py +++ b/dsLightRag/Volcengine/T2_CreateIndex.py @@ -50,6 +50,33 @@ def create_memory_collection(collection_name, description="情感陪伴记忆库 logger.info(f"创建响应: {json.dumps(response, ensure_ascii=False, indent=2)}") logger.info(f"集合 '{collection_name}' 创建成功") + # 添加测试消息,触发索引构建 + logger.info("添加测试消息以初始化索引...") + # 生成唯一会话ID + session_id = f"test_session_{int(time.time())}" + # 构造测试消息(符合sys_event_v1类型要求) + test_messages = [{ + "role": "user", + "content": "initial test message", + "memory_type": "sys_event_v1", + "user_id": "test_user" + }] + # 构造元数据(包含API要求的必填字段) + test_metadata = { + "default_user_id": "test_user", # API必填字段 + "default_assistant_id": "test_assistant", # API必填字段 + "user_id": "test_user", + "session_id": session_id, + "time": int(time.time() * 1000) # 修正字段名并使用毫秒级时间戳 + } + # 使用正确的add_session方法添加会话 + memory_service.add_session( + collection_name=collection_name, + session_id=session_id, + messages=test_messages, + metadata=test_metadata + ) + # 等待集合就绪 - 修改为模拟chat.py的索引就绪检查机制 logger.info("等待集合初始化完成...") max_retries = 30 # 最多重试30次