Improve forum communication mechanism between agents.

utils/forum_reader.py

@@ -0,0 +1,162 @@
+"""
+Forum日志读取工具
+用于读取forum.log中的最新HOST发言
+"""
+
+import re
+from pathlib import Path
+from typing import Optional, List, Dict
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def get_latest_host_speech(log_dir: str = "logs") -> Optional[str]:
+    """
+    获取forum.log中最新的HOST发言
+    
+    Args:
+        log_dir: 日志目录路径
+        
+    Returns:
+        最新的HOST发言内容，如果没有则返回None
+    """
+    try:
+        forum_log_path = Path(log_dir) / "forum.log"
+        
+        if not forum_log_path.exists():
+            logger.debug("forum.log文件不存在")
+            return None
+            
+        with open(forum_log_path, 'r', encoding='utf-8', errors='ignore') as f:
+            lines = f.readlines()
+        
+        # 从后往前查找最新的HOST发言
+        host_speech = None
+        for line in reversed(lines):
+            # 匹配格式: [时间] [HOST] 内容
+            match = re.match(r'\[(\d{2}:\d{2}:\d{2})\]\s*\[HOST\]\s*(.+)', line)
+            if match:
+                _, content = match.groups()
+                # 处理转义的换行符，还原为实际换行
+                host_speech = content.replace('\\n', '\n').strip()
+                break
+        
+        if host_speech:
+            logger.info(f"找到最新的HOST发言，长度: {len(host_speech)}字符")
+        else:
+            logger.debug("未找到HOST发言")
+            
+        return host_speech
+        
+    except Exception as e:
+        logger.error(f"读取forum.log失败: {str(e)}")
+        return None
+
+
+def get_all_host_speeches(log_dir: str = "logs") -> List[Dict[str, str]]:
+    """
+    获取forum.log中所有的HOST发言
+    
+    Args:
+        log_dir: 日志目录路径
+        
+    Returns:
+        包含所有HOST发言的列表，每个元素是包含timestamp和content的字典
+    """
+    try:
+        forum_log_path = Path(log_dir) / "forum.log"
+        
+        if not forum_log_path.exists():
+            logger.debug("forum.log文件不存在")
+            return []
+            
+        with open(forum_log_path, 'r', encoding='utf-8', errors='ignore') as f:
+            lines = f.readlines()
+        
+        host_speeches = []
+        for line in lines:
+            # 匹配格式: [时间] [HOST] 内容
+            match = re.match(r'\[(\d{2}:\d{2}:\d{2})\]\s*\[HOST\]\s*(.+)', line)
+            if match:
+                timestamp, content = match.groups()
+                # 处理转义的换行符
+                content = content.replace('\\n', '\n').strip()
+                host_speeches.append({
+                    'timestamp': timestamp,
+                    'content': content
+                })
+        
+        logger.info(f"找到{len(host_speeches)}条HOST发言")
+        return host_speeches
+        
+    except Exception as e:
+        logger.error(f"读取forum.log失败: {str(e)}")
+        return []
+
+
+def get_recent_agent_speeches(log_dir: str = "logs", limit: int = 5) -> List[Dict[str, str]]:
+    """
+    获取forum.log中最近的Agent发言（不包括HOST）
+    
+    Args:
+        log_dir: 日志目录路径
+        limit: 返回的最大发言数量
+        
+    Returns:
+        包含最近Agent发言的列表
+    """
+    try:
+        forum_log_path = Path(log_dir) / "forum.log"
+        
+        if not forum_log_path.exists():
+            return []
+            
+        with open(forum_log_path, 'r', encoding='utf-8', errors='ignore') as f:
+            lines = f.readlines()
+        
+        agent_speeches = []
+        for line in reversed(lines):  # 从后往前读取
+            # 匹配格式: [时间] [AGENT_NAME] 内容
+            match = re.match(r'\[(\d{2}:\d{2}:\d{2})\]\s*\[(INSIGHT|MEDIA|QUERY)\]\s*(.+)', line)
+            if match:
+                timestamp, agent, content = match.groups()
+                # 处理转义的换行符
+                content = content.replace('\\n', '\n').strip()
+                agent_speeches.append({
+                    'timestamp': timestamp,
+                    'agent': agent,
+                    'content': content
+                })
+                if len(agent_speeches) >= limit:
+                    break
+        
+        agent_speeches.reverse()  # 恢复时间顺序
+        return agent_speeches
+        
+    except Exception as e:
+        logger.error(f"读取forum.log失败: {str(e)}")
+        return []
+
+
+def format_host_speech_for_prompt(host_speech: str) -> str:
+    """
+    格式化HOST发言，用于添加到prompt中
+    
+    Args:
+        host_speech: HOST发言内容
+        
+    Returns:
+        格式化后的内容
+    """
+    if not host_speech:
+        return ""
+    
+    return f"""
+### 论坛主持人最新总结
+以下是论坛主持人对各Agent讨论的最新总结和引导，请参考其中的观点和建议：
+
+{host_speech}
+
+---
+"""