'commit'

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次