测试任务：分配完整流程测试任务给所有将军

2026-03-22 17:38:23 +08:00
parent 9475a9e4c6
commit 443da4369d
22 changed files with 422 additions and 0 deletions
@@ -0,0 +1,203 @@
+#!/usr/bin/env python3
+"""
+赵云Agent启动脚本
+监听新任务，自动执行并报告
+"""
+import os
+import sys
+import time
+import json
+from datetime import datetime
+from typing import Optional, Dict, Any
+import subprocess
+import signal
+import atexit
+import logging
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+
+class AgentRunner:
+    """Agent运行器"""
+    
+    def __init__(self, agent_name: str = "zhaoyun"):
+        self.agent_name = agent_name
+        self.task_dir = "./management/tasks/pending/"
+        self.agent_dir = f"./management/agents/{agent_name}/"
+        
+        # 确保目录存在
+        os.makedirs(self.task_dir, exist_ok=True)
+        os.makedirs(self.agent_dir, exist_ok=True)
+        
+        self.running = False
+        
+        logger.info(f"Agent {agent_name} 初始化完成")
+    
+    def check_for_new_task(self) -> Optional[Dict[str, Any]]:
+        """检查是否有新任务"""
+        try:
+            files = os.listdir(self.task_dir)
+            for file in sorted(files):
+                if file.endswith('.json') and not file.startswith('.'):
+                    task_file = os.path.join(self.task_dir, file)
+                    with open(task_file, 'r', encoding='utf-8') as f:
+                        task = json.load(f)
+                    
+                    # 检查任务是否已经分配给我们
+                    if task.get('assigned_to') == self.agent_name:
+                        # 执行任务
+
+                        task_result = self.execute_task(task)
+                        # 移动任务到完成目录
+
+                        done_dir = "./management/tasks/done/"
+                        os.makedirs(done_dir, exist_ok=True)
+                        
+                        done_file = os.path.join(done_dir, file)
+                        os.rename(task_file, done_file)
+                        
+                        logger.info(f"任务 {file} 执行完成")
+                        
+                        return task_result
+            return None
+            
+        except Exception as e:
+            logger.error(f"检查新任务失败: {e}")
+            return None
+    
+    def execute_task(self, task: Dict[str, Any]) -> Dict[str, Any]:
+        """执行任务"""
+        try:
+            task_id = task.get('task_id', 'unknown')
+            logger.info(f"开始执行任务: {task_id}")
+            
+            # 根据任务类型执行
+
+            task_type = task.get('type', 'unknown')
+            
+            if task_type == 'data_download':
+                return self.execute_data_download_task(task)
+            elif task_type == 'research':
+                return self.execute_research_task(task)
+            elif task_type == 'analysis':
+                return self.execute_analysis_task(task)
+            else:
+                logger.warning(f"未知任务类型: {task_type}")
+                return {"status": "unknown_task", "task_id": task_id}
+                
+        except Exception as e:
+            logger.error(f"执行任务失败: {e}")
+            return {"status": "error", "error": str(e)}
+    
+    def execute_data_download_task(self, task: Dict[str, Any]) -> Dict[str, Any]:
+        """执行数据下载任务"""
+        try:
+            task_desc = task.get('description', '')
+            logger.info(f"执行数据下载任务: {task_desc}")
+            
+            # 简化版执行
+            result = {
+                "status": "completed",
+                "agent": self.agent_name,
+                "task_id": task.get('task_id', 'unknown'),
+                "executed_at": datetime.now().isoformat(),
+                "execution_time": time.time(),
+                "output": {
+                    "message": f"Agent {self.agent_name} 成功执行数据下载任务",
+                    "task_type": "data_download",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            
+            logger.info(f"数据下载任务执行完成")
+            return result
+    
+    def execute_research_task(self, task: Dict[str, Any]) -> Dict[str, Any]:
+        """执行调研任务"""
+        try:
+            task_desc = task.get('description', '')
+            logger.info(f"执行调研任务: {task_desc}")
+            
+            result = {
+                "status": "completed",
+                "agent": self.agent_name,
+                "task_id": task.get('task_id', 'unknown'),
+                "executed_at": datetime.now().isoformat(),
+                "execution_time": time.time(),
+                "output": {
+                    "message": f"Agent {self.agent_name} 成功执行调研任务",
+                    "task_type": "research",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            
+            logger.info(f"调研任务执行完成")
+            return result
+    
+    def execute_analysis_task(self, task: Dict[str, Any]) -> Dict[str, Any]:
+        """执行分析任务"""
+        try:
+            task_desc = task.get('description', '')
+            logger.info(f"执行分析任务: {task_desc}")
+            
+            result = {
+                "status": "completed",
+                "agent": self.agent_name,
+                "task_id": task.get('task_id', 'unknown'),
+                "executed_at": datetime.now().isoformat(),
+                "execution_time": time.time(),
+                "output": {
+                    "message": f"Agent {self.agent_name} 成功执行分析任务",
+                    "task_type": "analysis",
+                    "timestamp": datetime.now().isoformat()
+                }
+            }
+            
+            logger.info(f"分析任务执行完成")
+            return result
+    
+    def run(self):
+        """运行Agent"""
+        logger.info(f"Agent {self.agent_name} 启动运行...")
+        self.running = True
+        
+        try:
+            while self.running:
+                task = self.check_for_new_task()
+                if task:
+                    logger.info(f"发现新任务，正在执行...")
+                    
+                    # 添加延迟，避免过于频繁的检查
+                    time.sleep(30)
+                    
+        except KeyboardInterrupt:
+            logger.info(f"Agent {self.agent_name} 收到停止信号")
+        except Exception as e:
+            logger.error(f"Agent运行异常: {e}")
+        
+        self.running = False
+        logger.info(f"Agent {self.agent_name} 停止运行.")
+    
+    def stop(self):
+        """停止Agent"""
+        self.running = False
+        logger.info(f"Agent {self.agent_name} 正在停止...")
+
+
+def main():
+    """主函数"""
+    runner = AgentRunner()
+    
+    # 注册退出处理
+    atexit.register(runner.stop)
+    
+    try:
+        runner.run()
+    except Exception as e:
+        logger.error(f"Agent运行失败: {e}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()