正则相关

backend/services/chat_service.py

@@ -172,6 +172,41 @@ class ChatService:
             logger.error(f"读取聊天失败 {role_name}/{chat_name}: {str(e)}")
             raise
     
+    def get_message(self, role_name: str, chat_name: str, floor: int) -> Dict:
+        """
+        获取指定楼层的消息
+        
+        Args:
+            role_name: 角色名称
+            chat_name: 聊天名称
+            floor: 楼层号
+            
+        Returns:
+            Dict: 消息数据，如果不存在则返回 None
+        """
+        chat_file = self.chat_dir / role_name / f"{chat_name}.jsonl"
+        
+        if not chat_file.exists():
+            return None
+        
+        try:
+            with open(chat_file, 'r', encoding='utf-8') as f:
+                lines = f.readlines()
+            
+            # 找到对应的消息行（floor + 1，因为第0行是header）
+            message_line_index = floor + 1
+            
+            if message_line_index >= len(lines):
+                return None
+            
+            # 解析并返回消息
+            msg_data = json.loads(lines[message_line_index])
+            return msg_data
+            
+        except Exception as e:
+            logger.error(f"获取消息失败 {role_name}/{chat_name}/{floor}: {str(e)}")
+            return None
+    
     def create_chat(self, role_name: str, chat_name: str, metadata: Dict = None) -> Dict:
         """
         创建新聊天