297 lines
13 KiB
Python
Executable File
297 lines
13 KiB
Python
Executable File
"""
|
||
ExecutionAgent V2 - 集成StageToolRegistry
|
||
支持执行bbav20311分析阶段脚本
|
||
"""
|
||
import asyncio
|
||
from typing import Dict, List, Optional
|
||
from .base import AgentMessage, AgentState, BaseAgent
|
||
from gfa_core.mas.llm import LLMManager
|
||
from gfa_core.project_paths import WORKSPACE_ROOT
|
||
|
||
|
||
class ExecutionAgent(BaseAgent):
|
||
"""
|
||
任务执行Agent V2
|
||
使用StageToolRegistry执行实际的分析阶段
|
||
"""
|
||
name = "execution"
|
||
role = "Task Execution Specialist"
|
||
|
||
def __init__(self, llm_manager: LLMManager, stage_registry=None):
|
||
super().__init__(llm_manager)
|
||
self.stage_registry = stage_registry
|
||
|
||
async def execute(self, state: AgentState, message: AgentMessage) -> AgentState:
|
||
"""
|
||
执行分析任务
|
||
|
||
从state中提取任务信息，使用StageToolRegistry执行
|
||
"""
|
||
print(f"[ExecutionAgent] 开始执行，session_id: {state.session_id}")
|
||
|
||
# 获取任务信息
|
||
task_plan = state.results.get("task_plan", {})
|
||
stages_detail = task_plan.get("stages", [])
|
||
# 优先使用用户传入的 job_id，其次使用 session_id
|
||
job_id = state.results.get("job_id") or state.session_id or "default_job"
|
||
prefix = state.results.get("prefix", "")
|
||
|
||
print(f"[ExecutionAgent] task_plan: {task_plan}")
|
||
print(f"[ExecutionAgent] stages_detail: {stages_detail}")
|
||
print(f"[ExecutionAgent] job_id: {job_id}, prefix: {prefix}")
|
||
|
||
# 提取 stage_id 列表（支持 dict 或 string）
|
||
stages_to_run = []
|
||
for stage in stages_detail:
|
||
if isinstance(stage, dict):
|
||
stages_to_run.append(stage.get("id", "unknown"))
|
||
else:
|
||
stages_to_run.append(stage)
|
||
|
||
print(f"[ExecutionAgent] stages_to_run: {stages_to_run}")
|
||
|
||
# 获取意图信息（用于参数）
|
||
intent = state.results.get("results", {}).get("intent", {}) or state.results.get("intent", {})
|
||
print(f"[ExecutionAgent] intent: {intent}")
|
||
|
||
if not stages_to_run:
|
||
# 检查是否是因为PreFlight验证失败导致没有可执行的阶段
|
||
# preflight_result在task_plan中，key是"preflight_result"
|
||
preflight_result = task_plan.get("preflight_result", {})
|
||
task_plan_valid = task_plan.get("valid", True)
|
||
task_plan_error = task_plan.get("error", "")
|
||
|
||
if not task_plan_valid:
|
||
# PreFlight验证失败，返回错误而不是加载旧结果
|
||
error_msg = task_plan_error or "执行前验证失败"
|
||
print(f"[ExecutionAgent] ❌ PreFlight验证失败，无法执行: {error_msg}")
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": f"❌ 执行前验证失败: {error_msg}"
|
||
})
|
||
# AgentState使用results字典
|
||
state.results["execution_status"] = "error"
|
||
state.results["execution_error"] = error_msg
|
||
state.results["execution_results"] = []
|
||
return state
|
||
|
||
# 如果没有需要执行的阶段（可能都已完成了），从manifest加载已有结果
|
||
print(f"[ExecutionAgent] ✅ 没有需要执行的阶段（可能都已完成了），从manifest加载结果")
|
||
|
||
# 从manifest加载已完成阶段的结果
|
||
completed_results = []
|
||
job_id = state.results.get("job_id") or state.session_id
|
||
prefix = state.results.get("prefix", "")
|
||
|
||
try:
|
||
from pathlib import Path
|
||
import json
|
||
workspace = WORKSPACE_ROOT / job_id / "final"
|
||
if workspace.exists():
|
||
for manifest_file in workspace.glob(f"{prefix}.*.manifest.json"):
|
||
try:
|
||
with open(manifest_file, 'r') as f:
|
||
manifest = json.load(f)
|
||
stage = manifest.get("stage", "unknown")
|
||
completed_results.append({
|
||
"stage": stage,
|
||
"result": {
|
||
"status": manifest.get("status", "success"),
|
||
"manifest": manifest,
|
||
"manifest_path": str(manifest_file),
|
||
"message": f"✅ {stage.capitalize()} 完成（从manifest加载）"
|
||
}
|
||
})
|
||
print(f"[ExecutionAgent] 加载manifest: {manifest_file.name}, stage: {stage}")
|
||
except Exception as e:
|
||
print(f"[ExecutionAgent] 加载manifest失败: {e}")
|
||
except Exception as e:
|
||
print(f"[ExecutionAgent] 扫描manifest目录失败: {e}")
|
||
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": f"✅ 所有阶段已完成，加载 {len(completed_results)} 个已有结果"
|
||
})
|
||
state.results["execution_status"] = "success"
|
||
state.results["execution_results"] = completed_results
|
||
state.results["execution_error"] = None
|
||
return state
|
||
|
||
# 检查stage_registry是否可用
|
||
print(f"[ExecutionAgent] stage_registry: {self.stage_registry}")
|
||
if self.stage_registry is None:
|
||
print("[ExecutionAgent] ❌ Stage registry not available")
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": "⚠️ Stage registry not available, cannot execute analysis"
|
||
})
|
||
state.results["execution_status"] = "error"
|
||
state.results["execution_error"] = "Stage registry not initialized"
|
||
return state
|
||
|
||
print(f"[ExecutionAgent] ✅ Stage registry available, stages: {self.stage_registry.list_all()}")
|
||
|
||
# 执行每个阶段
|
||
execution_results = []
|
||
overall_status = "success"
|
||
|
||
print(f"[ExecutionAgent] 开始执行 {len(stages_to_run)} 个阶段")
|
||
|
||
for stage_id in stages_to_run:
|
||
print(f"[ExecutionAgent] 处理阶段: {stage_id}")
|
||
|
||
# 检查阶段是否存在
|
||
try:
|
||
stage_tool = self.stage_registry.get(stage_id)
|
||
print(f"[ExecutionAgent] stage_tool: {stage_tool}")
|
||
except Exception as e:
|
||
print(f"[ExecutionAgent] ❌ 获取 stage_tool 失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": f"❌ Stage '{stage_id}' 获取失败: {str(e)}"
|
||
})
|
||
overall_status = "error"
|
||
continue
|
||
|
||
if stage_tool is None:
|
||
print(f"[ExecutionAgent] ❌ Stage '{stage_id}' not found")
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": f"❌ Stage '{stage_id}' not found in registry"
|
||
})
|
||
overall_status = "error"
|
||
continue
|
||
|
||
# 通知开始执行
|
||
print(f"[ExecutionAgent] 🚀 开始执行 {stage_tool.name} ({stage_id})")
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": f"🚀 Starting {stage_tool.name} ({stage_id})..."
|
||
})
|
||
|
||
# 准备参数
|
||
extra_params = state.results.get("params", {}).copy()
|
||
extra_params.update(intent.get("params", {}))
|
||
print(f"[ExecutionAgent] extra_params: {extra_params}")
|
||
|
||
# 执行阶段
|
||
try:
|
||
print(f"[ExecutionAgent] 调用 stage_registry.execute({stage_id}, {job_id}, {prefix})")
|
||
# stage_registry.execute 现在已经是 async，直接 await
|
||
result = await self.stage_registry.execute(
|
||
stage_id=stage_id,
|
||
job_id=job_id,
|
||
prefix=prefix,
|
||
extra_params=extra_params
|
||
)
|
||
print(f"[ExecutionAgent] ✅ 执行完成，结果: {result}")
|
||
|
||
execution_results.append({
|
||
"stage": stage_id,
|
||
"result": result
|
||
})
|
||
|
||
if result.get("status") in ["success", "partial_success"]:
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": f"✅ {stage_tool.name} completed successfully"
|
||
})
|
||
else:
|
||
# 提取错误信息（优先使用error字段，否则使用message字段）
|
||
error_msg = result.get('error') or result.get('message', 'Unknown error')
|
||
|
||
# 从日志文件读取详细错误信息
|
||
stderr = result.get('stderr', '')
|
||
stdout = result.get('stdout', '')
|
||
|
||
# 如果stderr为空，尝试从日志文件读取
|
||
if not stderr and result.get('manifest_path'):
|
||
try:
|
||
from pathlib import Path
|
||
manifest_path = Path(result['manifest_path'])
|
||
logs_dir = manifest_path.parent.parent / "logs"
|
||
if logs_dir.exists():
|
||
import glob
|
||
log_files = glob.glob(str(logs_dir / f"*_{stage_id}.log"))
|
||
if log_files:
|
||
latest_log = max(log_files, key=lambda x: Path(x).stat().st_mtime)
|
||
with open(latest_log, 'r', encoding='utf-8', errors='ignore') as f:
|
||
log_content = f.read()
|
||
stderr = log_content[-2000:] # 最后2000字符
|
||
except Exception as e:
|
||
print(f"[ExecutionAgent] 读取日志文件失败: {e}")
|
||
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": f"❌ {stage_tool.name} failed: {error_msg}"
|
||
})
|
||
overall_status = "error"
|
||
# 更新最后一个execution_result，添加错误详情
|
||
if execution_results:
|
||
execution_results[-1]["error_details"] = {
|
||
"error": error_msg,
|
||
"stderr": stderr,
|
||
"stdout": stdout,
|
||
"message": result.get('message', '')
|
||
}
|
||
break
|
||
|
||
except Exception as e:
|
||
execution_results.append({
|
||
"stage": stage_id,
|
||
"result": {"status": "error", "error": str(e)}
|
||
})
|
||
state.messages.append({
|
||
"role": "execution",
|
||
"content": f"❌ Exception in {stage_tool.name}: {str(e)}"
|
||
})
|
||
overall_status = "error"
|
||
break
|
||
|
||
# 更新状态 - 保存到AgentState.results（供其他Agent使用）
|
||
state.results["execution_results"] = execution_results
|
||
state.results["execution_status"] = overall_status
|
||
if overall_status == "error":
|
||
for item in execution_results:
|
||
result = item.get("result", {}) if isinstance(item, dict) else {}
|
||
error_msg = result.get("error") or result.get("message")
|
||
if error_msg:
|
||
state.results["execution_error"] = error_msg
|
||
break
|
||
|
||
return state
|
||
|
||
async def execute_single_stage(self, state: AgentState, stage_id: str) -> dict:
|
||
"""
|
||
执行单个分析阶段
|
||
|
||
Args:
|
||
state: 当前Agent状态
|
||
stage_id: 阶段ID
|
||
|
||
Returns:
|
||
执行结果字典
|
||
"""
|
||
if self.stage_registry is None:
|
||
return {"status": "error", "error": "Stage registry not initialized"}
|
||
|
||
job_id = state.session_id or state.results.get("job_id", "default_job")
|
||
prefix = state.results.get("prefix", "")
|
||
params = state.results.get("params", {})
|
||
|
||
# stage_registry.execute 现在已经是 async，直接 await
|
||
return await self.stage_registry.execute(
|
||
stage_id=stage_id,
|
||
job_id=job_id,
|
||
prefix=prefix,
|
||
extra_params=params
|
||
)
|
||
|
||
def get_available_stages(self) -> List[str]:
|
||
"""获取可用的分析阶段列表"""
|
||
if self.stage_registry is None:
|
||
return []
|
||
return [s.id for s in self.stage_registry.list_all()]
|