完整实现 Swarm 多智能体协作系统
- 新增 CLIPluginAdapter 统一接口 (backend/app/core/agent_adapter.py) - 新增 LLM 服务层，支持 Anthropic/OpenAI/DeepSeek/Ollama (backend/app/services/llm_service.py) - 新增 Agent 执行引擎，支持文件锁自动管理 (backend/app/services/agent_executor.py) - 新增 NativeLLMAgent 原生 LLM 适配器 (backend/app/adapters/native_llm_agent.py) - 新增进程管理器 (backend/app/services/process_manager.py) - 新增 Agent 控制 API (backend/app/routers/agents_control.py) - 新增 WebSocket 实时通信 (backend/app/routers/websocket.py) - 更新前端 AgentsPage，支持启动/停止 Agent - 测试通过：Agent 启动、批量操作、栅栏同步 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Claude Code
17
backend/app/routers/__init__.py
Normal file
17
backend/app/routers/__init__.py
Normal file
@@ -0,0 +1,17 @@
|
||||
"""API 路由模块"""
|
||||
|
||||
from . import agents, locks, meetings, heartbeats, workflows, resources, roles, humans
|
||||
from . import agents_control, websocket
|
||||
|
||||
__all__ = [
|
||||
"agents",
|
||||
"locks",
|
||||
"meetings",
|
||||
"heartbeats",
|
||||
"workflows",
|
||||
"resources",
|
||||
"roles",
|
||||
"humans",
|
||||
"agents_control",
|
||||
"websocket"
|
||||
]
|
||||
166
backend/app/routers/agents.py
Normal file
166
backend/app/routers/agents.py
Normal file
@@ -0,0 +1,166 @@
|
||||
"""
|
||||
Agent 管理 API 路由
|
||||
"""
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from pydantic import BaseModel
|
||||
from typing import List, Optional
|
||||
import time
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
# 内存存储，实际应用应该使用持久化存储
|
||||
agents_db = {}
|
||||
|
||||
|
||||
class Agent(BaseModel):
|
||||
agent_id: str
|
||||
name: str
|
||||
role: str
|
||||
model: str
|
||||
description: Optional[str] = None
|
||||
status: str = "idle"
|
||||
created_at: float = 0
|
||||
|
||||
|
||||
class AgentCreate(BaseModel):
|
||||
agent_id: str
|
||||
name: str
|
||||
role: str = "developer"
|
||||
model: str
|
||||
description: Optional[str] = None
|
||||
|
||||
|
||||
# Agent状态存储
|
||||
agent_states_db = {}
|
||||
|
||||
@router.get("/")
|
||||
async def list_agents():
|
||||
"""获取所有 Agent 列表"""
|
||||
# 合并数据库和默认agent
|
||||
default_agents = [
|
||||
{
|
||||
"agent_id": "claude-001",
|
||||
"name": "Claude Code",
|
||||
"role": "developer",
|
||||
"model": "claude-opus-4.6",
|
||||
"status": "working",
|
||||
"description": "主开发 Agent",
|
||||
"created_at": time.time() - 86400
|
||||
},
|
||||
{
|
||||
"agent_id": "kimi-001",
|
||||
"name": "Kimi CLI",
|
||||
"role": "architect",
|
||||
"model": "kimi-k2",
|
||||
"status": "idle",
|
||||
"description": "架构设计 Agent",
|
||||
"created_at": time.time() - 72000
|
||||
},
|
||||
{
|
||||
"agent_id": "opencode-001",
|
||||
"name": "OpenCode",
|
||||
"role": "reviewer",
|
||||
"model": "opencode-v1",
|
||||
"status": "idle",
|
||||
"description": "代码审查 Agent",
|
||||
"created_at": time.time() - 36000
|
||||
}
|
||||
]
|
||||
|
||||
# 使用数据库中的agent（覆盖默认的）
|
||||
agents_map = {a["agent_id"]: a for a in default_agents}
|
||||
agents_map.update(agents_db)
|
||||
|
||||
return {"agents": list(agents_map.values())}
|
||||
|
||||
|
||||
@router.post("/register")
|
||||
async def register_agent(agent: AgentCreate):
|
||||
"""注册新 Agent"""
|
||||
agent_data = {
|
||||
"agent_id": agent.agent_id,
|
||||
"name": agent.name,
|
||||
"role": agent.role,
|
||||
"model": agent.model,
|
||||
"description": agent.description or "",
|
||||
"status": "idle",
|
||||
"created_at": time.time()
|
||||
}
|
||||
agents_db[agent.agent_id] = agent_data
|
||||
return agent_data
|
||||
|
||||
|
||||
@router.get("/{agent_id}")
|
||||
async def get_agent(agent_id: str):
|
||||
"""获取指定 Agent 信息"""
|
||||
if agent_id in agents_db:
|
||||
return agents_db[agent_id]
|
||||
raise HTTPException(status_code=404, detail="Agent not found")
|
||||
|
||||
|
||||
@router.delete("/{agent_id}")
|
||||
async def delete_agent(agent_id: str):
|
||||
"""删除 Agent"""
|
||||
if agent_id in agents_db:
|
||||
del agents_db[agent_id]
|
||||
return {"message": "Agent deleted"}
|
||||
raise HTTPException(status_code=404, detail="Agent not found")
|
||||
|
||||
|
||||
@router.get("/{agent_id}/state")
|
||||
async def get_agent_state(agent_id: str):
|
||||
"""获取 Agent 状态"""
|
||||
# 如果存在真实状态，返回真实状态
|
||||
if agent_id in agent_states_db:
|
||||
return agent_states_db[agent_id]
|
||||
|
||||
# 默认mock状态
|
||||
default_states = {
|
||||
"claude-001": {
|
||||
"agent_id": agent_id,
|
||||
"task": "修复用户登录bug",
|
||||
"progress": 65,
|
||||
"working_files": ["src/auth/login.py", "src/auth/jwt.py"],
|
||||
"status": "working",
|
||||
"last_update": time.time() - 120
|
||||
},
|
||||
"kimi-001": {
|
||||
"agent_id": agent_id,
|
||||
"task": "等待会议开始",
|
||||
"progress": 0,
|
||||
"working_files": [],
|
||||
"status": "waiting",
|
||||
"last_update": time.time() - 300
|
||||
},
|
||||
"opencode-001": {
|
||||
"agent_id": agent_id,
|
||||
"task": "代码审查",
|
||||
"progress": 30,
|
||||
"working_files": ["src/components/Button.tsx"],
|
||||
"status": "working",
|
||||
"last_update": time.time() - 60
|
||||
}
|
||||
}
|
||||
|
||||
return default_states.get(agent_id, {
|
||||
"agent_id": agent_id,
|
||||
"task": "空闲",
|
||||
"progress": 0,
|
||||
"working_files": [],
|
||||
"status": "idle",
|
||||
"last_update": time.time()
|
||||
})
|
||||
|
||||
|
||||
@router.post("/{agent_id}/state")
|
||||
async def update_agent_state(agent_id: str, data: dict):
|
||||
"""更新 Agent 状态"""
|
||||
agent_states_db[agent_id] = {
|
||||
"agent_id": agent_id,
|
||||
"task": data.get("task", ""),
|
||||
"progress": data.get("progress", 0),
|
||||
"working_files": data.get("working_files", []),
|
||||
"status": data.get("status", "idle"),
|
||||
"last_update": time.time()
|
||||
}
|
||||
return {"success": True}
|
||||
391
backend/app/routers/agents_control.py
Normal file
391
backend/app/routers/agents_control.py
Normal file
@@ -0,0 +1,391 @@
|
||||
"""
|
||||
Agent 控制 API 路由
|
||||
|
||||
提供 Agent 启动、停止、状态查询等控制接口
|
||||
"""
|
||||
|
||||
import logging
|
||||
import uuid
|
||||
from typing import Dict, List, Optional, Any
|
||||
from datetime import datetime
|
||||
|
||||
from fastapi import APIRouter, HTTPException, BackgroundTasks
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from ..services.process_manager import get_process_manager, AgentStatus
|
||||
from ..services.agent_registry import get_agent_registry
|
||||
from ..services.heartbeat import get_heartbeat_service
|
||||
from ..adapters.native_llm_agent import NativeLLMAgent, NativeLLMAgentFactory
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
router = APIRouter(prefix="/api/agents/control", tags=["agents-control"])
|
||||
|
||||
|
||||
# ========== 请求模型 ==========
|
||||
|
||||
|
||||
class StartAgentRequest(BaseModel):
|
||||
"""启动 Agent 请求"""
|
||||
agent_id: str = Field(..., description="Agent 唯一标识")
|
||||
name: Optional[str] = Field(None, description="Agent 显示名称")
|
||||
role: str = Field("developer", description="Agent 角色")
|
||||
model: str = Field("claude-sonnet-4.6", description="使用的模型")
|
||||
agent_type: str = Field("native_llm", description="Agent 类型")
|
||||
config: Dict[str, Any] = Field(default_factory=dict, description="额外配置")
|
||||
|
||||
|
||||
class StopAgentRequest(BaseModel):
|
||||
"""停止 Agent 请求"""
|
||||
agent_id: str = Field(..., description="Agent ID")
|
||||
graceful: bool = Field(True, description="是否优雅关闭")
|
||||
|
||||
|
||||
class ExecuteTaskRequest(BaseModel):
|
||||
"""执行任务请求"""
|
||||
agent_id: str = Field(..., description="Agent ID")
|
||||
task_description: str = Field(..., description="任务描述")
|
||||
context: Dict[str, Any] = Field(default_factory=dict, description="任务上下文")
|
||||
|
||||
|
||||
class CreateMeetingRequest(BaseModel):
|
||||
"""创建会议请求"""
|
||||
meeting_id: str = Field(..., description="会议 ID")
|
||||
title: str = Field(..., description="会议标题")
|
||||
attendees: List[str] = Field(..., description="参会 Agent ID 列表")
|
||||
|
||||
|
||||
class JoinMeetingRequest(BaseModel):
|
||||
"""加入会议请求"""
|
||||
agent_id: str = Field(..., description="Agent ID")
|
||||
meeting_id: str = Field(..., description="会议 ID")
|
||||
timeout: int = Field(300, description="等待超时时间（秒）")
|
||||
|
||||
|
||||
# ========== 响应模型 ==========
|
||||
|
||||
|
||||
class AgentControlResponse(BaseModel):
|
||||
"""Agent 控制响应"""
|
||||
success: bool
|
||||
agent_id: str
|
||||
status: str
|
||||
message: str
|
||||
|
||||
|
||||
class AgentStatusResponse(BaseModel):
|
||||
"""Agent 状态响应"""
|
||||
agent_id: str
|
||||
status: str
|
||||
is_alive: bool
|
||||
uptime: Optional[float] = None
|
||||
restart_count: int = 0
|
||||
|
||||
|
||||
class ProcessManagerSummary(BaseModel):
|
||||
"""进程管理器摘要"""
|
||||
total_agents: int
|
||||
running_agents: int
|
||||
running_agent_ids: List[str]
|
||||
status_counts: Dict[str, int]
|
||||
monitor_running: bool
|
||||
|
||||
|
||||
# ========== API 端点 ==========
|
||||
|
||||
|
||||
@router.post("/start", response_model=AgentControlResponse)
|
||||
async def start_agent(request: StartAgentRequest, background_tasks: BackgroundTasks):
|
||||
"""
|
||||
启动 Agent
|
||||
|
||||
启动一个新的 Agent 实例，支持两种类型:
|
||||
- native_llm: 原生 LLM Agent（异步任务）
|
||||
- process_wrapper: 进程包装 Agent（外部 CLI 工具）
|
||||
"""
|
||||
process_manager = get_process_manager()
|
||||
|
||||
# 检查是否已在运行
|
||||
if request.agent_id in process_manager.get_all_agents():
|
||||
existing = process_manager.get_agent_status(request.agent_id)
|
||||
if existing != AgentStatus.STOPPED:
|
||||
return AgentControlResponse(
|
||||
success=False,
|
||||
agent_id=request.agent_id,
|
||||
status=existing.value,
|
||||
message="Agent 已在运行"
|
||||
)
|
||||
|
||||
# 准备配置
|
||||
config = request.config.copy()
|
||||
config["name"] = request.name or request.agent_id.replace("-", " ").title()
|
||||
config["role"] = request.role
|
||||
config["model"] = request.model
|
||||
|
||||
# 启动 Agent
|
||||
success = await process_manager.start_agent(
|
||||
agent_id=request.agent_id,
|
||||
agent_type=request.agent_type,
|
||||
config=config
|
||||
)
|
||||
|
||||
if success:
|
||||
return AgentControlResponse(
|
||||
success=True,
|
||||
agent_id=request.agent_id,
|
||||
status=AgentStatus.RUNNING.value,
|
||||
message=f"Agent {request.agent_id} 启动成功"
|
||||
)
|
||||
else:
|
||||
raise HTTPException(status_code=500, detail="启动 Agent 失败")
|
||||
|
||||
|
||||
@router.post("/stop", response_model=AgentControlResponse)
|
||||
async def stop_agent(request: StopAgentRequest):
|
||||
"""停止 Agent"""
|
||||
process_manager = get_process_manager()
|
||||
|
||||
success = await process_manager.stop_agent(
|
||||
agent_id=request.agent_id,
|
||||
graceful=request.graceful
|
||||
)
|
||||
|
||||
if success:
|
||||
return AgentControlResponse(
|
||||
success=True,
|
||||
agent_id=request.agent_id,
|
||||
status=AgentStatus.STOPPED.value,
|
||||
message=f"Agent {request.agent_id} 已停止"
|
||||
)
|
||||
else:
|
||||
return AgentControlResponse(
|
||||
success=False,
|
||||
agent_id=request.agent_id,
|
||||
status=AgentStatus.UNKNOWN.value,
|
||||
message=f"停止 Agent 失败或 Agent 未运行"
|
||||
)
|
||||
|
||||
|
||||
@router.post("/restart", response_model=AgentControlResponse)
|
||||
async def restart_agent(agent_id: str):
|
||||
"""重启 Agent"""
|
||||
process_manager = get_process_manager()
|
||||
|
||||
success = await process_manager.restart_agent(agent_id)
|
||||
|
||||
if success:
|
||||
return AgentControlResponse(
|
||||
success=True,
|
||||
agent_id=agent_id,
|
||||
status=AgentStatus.RUNNING.value,
|
||||
message=f"Agent {agent_id} 重启成功"
|
||||
)
|
||||
else:
|
||||
raise HTTPException(status_code=500, detail="重启 Agent 失败")
|
||||
|
||||
|
||||
@router.get("/status/{agent_id}", response_model=AgentStatusResponse)
|
||||
async def get_agent_status(agent_id: str):
|
||||
"""获取 Agent 状态"""
|
||||
process_manager = get_process_manager()
|
||||
|
||||
status = process_manager.get_agent_status(agent_id)
|
||||
all_agents = process_manager.get_all_agents()
|
||||
|
||||
if agent_id in all_agents:
|
||||
process_info = all_agents[agent_id]
|
||||
return AgentStatusResponse(
|
||||
agent_id=agent_id,
|
||||
status=status.value,
|
||||
is_alive=process_info.is_alive,
|
||||
uptime=process_info.uptime,
|
||||
restart_count=process_info.restart_count
|
||||
)
|
||||
else:
|
||||
raise HTTPException(status_code=404, detail="Agent 不存在")
|
||||
|
||||
|
||||
@router.get("/list", response_model=List[AgentStatusResponse])
|
||||
async def list_agents():
|
||||
"""列出所有 Agent 状态"""
|
||||
process_manager = get_process_manager()
|
||||
heartbeat_service = get_heartbeat_service()
|
||||
|
||||
agents = []
|
||||
for agent_id, process_info in process_manager.get_all_agents().items():
|
||||
# 获取心跳信息
|
||||
heartbeat = await heartbeat_service.get_heartbeat(agent_id)
|
||||
|
||||
agents.append(AgentStatusResponse(
|
||||
agent_id=agent_id,
|
||||
status=process_info.status.value,
|
||||
is_alive=process_info.is_alive,
|
||||
uptime=process_info.uptime,
|
||||
restart_count=process_info.restart_count
|
||||
))
|
||||
|
||||
return agents
|
||||
|
||||
|
||||
@router.get("/summary", response_model=ProcessManagerSummary)
|
||||
async def get_summary():
|
||||
"""获取进程管理器摘要"""
|
||||
process_manager = get_process_manager()
|
||||
summary = process_manager.get_summary()
|
||||
|
||||
return ProcessManagerSummary(**summary)
|
||||
|
||||
|
||||
@router.post("/execute")
|
||||
async def execute_task(request: ExecuteTaskRequest):
|
||||
"""
|
||||
让 Agent 执行任务
|
||||
|
||||
Agent 会自动:
|
||||
1. 分析任务，识别需要的文件
|
||||
2. 获取文件锁
|
||||
3. 调用 LLM 执行任务
|
||||
4. 释放文件锁
|
||||
"""
|
||||
process_manager = get_process_manager()
|
||||
|
||||
# 检查 Agent 是否运行
|
||||
all_agents = process_manager.get_all_agents()
|
||||
if request.agent_id not in all_agents:
|
||||
raise HTTPException(status_code=404, detail="Agent 未运行")
|
||||
|
||||
process_info = all_agents[request.agent_id]
|
||||
if not process_info.is_alive:
|
||||
raise HTTPException(status_code=400, detail="Agent 未运行")
|
||||
|
||||
# 获取 Agent 实例
|
||||
agent = process_info.agent
|
||||
if not agent:
|
||||
raise HTTPException(status_code=500, detail="Agent 实例不可用")
|
||||
|
||||
# 创建任务
|
||||
from ..core.agent_adapter import Task
|
||||
task = Task(
|
||||
task_id=f"task_{uuid.uuid4().hex[:12]}",
|
||||
description=request.task_description,
|
||||
context=request.context
|
||||
)
|
||||
|
||||
# 执行任务
|
||||
result = await agent.execute(task)
|
||||
|
||||
return {
|
||||
"success": result.success,
|
||||
"output": result.output,
|
||||
"error": result.error,
|
||||
"metadata": result.metadata,
|
||||
"execution_time": result.execution_time
|
||||
}
|
||||
|
||||
|
||||
@router.post("/meeting/create")
|
||||
async def create_meeting(request: CreateMeetingRequest):
|
||||
"""创建协作会议"""
|
||||
from ..services.meeting_scheduler import get_meeting_scheduler
|
||||
scheduler = get_meeting_scheduler()
|
||||
|
||||
queue = await scheduler.create_meeting(
|
||||
request.meeting_id,
|
||||
request.title,
|
||||
request.attendees
|
||||
)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"meeting_id": request.meeting_id,
|
||||
"title": queue.title,
|
||||
"expected_attendees": queue.expected_attendees,
|
||||
"min_required": queue.min_required,
|
||||
"status": queue.status
|
||||
}
|
||||
|
||||
|
||||
@router.post("/meeting/join")
|
||||
async def join_meeting(request: JoinMeetingRequest):
|
||||
"""让 Agent 加入会议（栅栏同步）"""
|
||||
process_manager = get_process_manager()
|
||||
|
||||
# 检查 Agent 是否运行
|
||||
all_agents = process_manager.get_all_agents()
|
||||
if request.agent_id not in all_agents:
|
||||
raise HTTPException(status_code=404, detail="Agent 未运行")
|
||||
|
||||
process_info = all_agents[request.agent_id]
|
||||
agent = process_info.agent
|
||||
if not agent:
|
||||
raise HTTPException(status_code=500, detail="Agent 实例不可用")
|
||||
|
||||
# 加入会议
|
||||
result = await agent.join_meeting(request.meeting_id, request.timeout)
|
||||
|
||||
return {
|
||||
"agent_id": request.agent_id,
|
||||
"meeting_id": request.meeting_id,
|
||||
"result": result
|
||||
}
|
||||
|
||||
|
||||
@router.post("/shutdown-all")
|
||||
async def shutdown_all_agents():
|
||||
"""关闭所有 Agent"""
|
||||
process_manager = get_process_manager()
|
||||
await process_manager.shutdown_all()
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"message": "所有 Agent 已关闭"
|
||||
}
|
||||
|
||||
|
||||
@router.post("/batch-start")
|
||||
async def batch_start_agents(agents: List[StartAgentRequest]):
|
||||
"""
|
||||
批量启动 Agent
|
||||
|
||||
用于快速创建团队
|
||||
"""
|
||||
results = []
|
||||
process_manager = get_process_manager()
|
||||
|
||||
for agent_request in agents:
|
||||
config = agent_request.config.copy()
|
||||
config["name"] = agent_request.name or agent_request.agent_id.replace("-", " ").title()
|
||||
config["role"] = agent_request.role
|
||||
config["model"] = agent_request.model
|
||||
|
||||
success = await process_manager.start_agent(
|
||||
agent_id=agent_request.agent_id,
|
||||
agent_type=agent_request.agent_type,
|
||||
config=config
|
||||
)
|
||||
|
||||
results.append({
|
||||
"agent_id": agent_request.agent_id,
|
||||
"success": success,
|
||||
"status": AgentStatus.RUNNING.value if success else AgentStatus.CRASHED.value
|
||||
})
|
||||
|
||||
return {
|
||||
"results": results,
|
||||
"total": len(results),
|
||||
"successful": sum(1 for r in results if r["success"])
|
||||
}
|
||||
|
||||
|
||||
# 健康检查端点
|
||||
@router.get("/health")
|
||||
async def health_check():
|
||||
"""健康检查"""
|
||||
process_manager = get_process_manager()
|
||||
summary = process_manager.get_summary()
|
||||
|
||||
return {
|
||||
"status": "healthy",
|
||||
"running_agents": summary["running_agents"],
|
||||
"monitor_running": summary["monitor_running"]
|
||||
}
|
||||
47
backend/app/routers/heartbeats.py
Normal file
47
backend/app/routers/heartbeats.py
Normal file
@@ -0,0 +1,47 @@
|
||||
"""
|
||||
心跳管理 API 路由
|
||||
"""
|
||||
from fastapi import APIRouter
|
||||
from pydantic import BaseModel
|
||||
from typing import Dict
|
||||
import time
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
heartbeats_db = {}
|
||||
|
||||
|
||||
class Heartbeat(BaseModel):
|
||||
agent_id: str
|
||||
timestamp: float
|
||||
is_timeout: bool = False
|
||||
|
||||
|
||||
@router.get("/")
|
||||
async def list_heartbeats():
|
||||
"""获取所有 Agent 心跳"""
|
||||
return {
|
||||
"heartbeats": {
|
||||
"claude-001": {
|
||||
"agent_id": "claude-001",
|
||||
"timestamp": time.time() - 30,
|
||||
"is_timeout": False
|
||||
},
|
||||
"kimi-001": {
|
||||
"agent_id": "kimi-001",
|
||||
"timestamp": time.time() - 60,
|
||||
"is_timeout": False
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@router.post("/{agent_id}")
|
||||
async def update_heartbeat(agent_id: str):
|
||||
"""更新 Agent 心跳"""
|
||||
heartbeats_db[agent_id] = {
|
||||
"agent_id": agent_id,
|
||||
"timestamp": time.time(),
|
||||
"is_timeout": False
|
||||
}
|
||||
return {"success": True}
|
||||
236
backend/app/routers/humans.py
Normal file
236
backend/app/routers/humans.py
Normal file
@@ -0,0 +1,236 @@
|
||||
"""
|
||||
人类输入 API 路由
|
||||
"""
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from pydantic import BaseModel
|
||||
from typing import List, Optional
|
||||
|
||||
from app.services.human_input import get_human_input_service
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
# ========== 请求/响应模型 ==========
|
||||
|
||||
class TaskRequest(BaseModel):
|
||||
"""任务请求"""
|
||||
content: str
|
||||
from_user: str = "user001"
|
||||
priority: str = "medium"
|
||||
title: str = ""
|
||||
target_files: List[str] = []
|
||||
suggested_agent: str = ""
|
||||
urgent: bool = False
|
||||
|
||||
|
||||
class CommentRequest(BaseModel):
|
||||
"""评论请求"""
|
||||
meeting_id: str
|
||||
content: str
|
||||
from_user: str = "user001"
|
||||
comment_type: str = "proposal"
|
||||
priority: str = "normal"
|
||||
|
||||
|
||||
class ParticipantRegister(BaseModel):
|
||||
"""参与者注册"""
|
||||
user_id: str
|
||||
name: str
|
||||
role: str = ""
|
||||
avatar: str = "👤"
|
||||
|
||||
|
||||
class UserStatusUpdate(BaseModel):
|
||||
"""用户状态更新"""
|
||||
status: str
|
||||
current_focus: str = ""
|
||||
|
||||
|
||||
# ========== API 端点 ==========
|
||||
|
||||
@router.get("/summary")
|
||||
async def get_summary():
|
||||
"""获取人类输入服务摘要"""
|
||||
service = get_human_input_service()
|
||||
summary = await service.get_summary()
|
||||
return summary
|
||||
|
||||
|
||||
@router.post("/register")
|
||||
async def register_participant(request: ParticipantRegister):
|
||||
"""注册人类参与者"""
|
||||
service = get_human_input_service()
|
||||
await service.register_participant(
|
||||
request.user_id,
|
||||
request.name,
|
||||
request.role,
|
||||
request.avatar
|
||||
)
|
||||
return {"success": True, "user_id": request.user_id}
|
||||
|
||||
|
||||
@router.get("/participants")
|
||||
async def get_participants():
|
||||
"""获取所有参与者"""
|
||||
service = get_human_input_service()
|
||||
participants = await service.get_participants()
|
||||
return {
|
||||
"participants": [
|
||||
{
|
||||
"id": p.id,
|
||||
"name": p.name,
|
||||
"role": p.role,
|
||||
"status": p.status,
|
||||
"avatar": p.avatar
|
||||
}
|
||||
for p in participants
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@router.post("/tasks")
|
||||
async def add_task_request(request: TaskRequest):
|
||||
"""提交任务请求"""
|
||||
service = get_human_input_service()
|
||||
task_id = await service.add_task_request(
|
||||
from_user=request.from_user,
|
||||
content=request.content,
|
||||
priority=request.priority,
|
||||
title=request.title,
|
||||
target_files=request.target_files,
|
||||
suggested_agent=request.suggested_agent,
|
||||
urgent=request.urgent
|
||||
)
|
||||
return {"success": True, "task_id": task_id}
|
||||
|
||||
|
||||
@router.get("/tasks")
|
||||
async def get_pending_tasks(
|
||||
priority: Optional[str] = None,
|
||||
agent: Optional[str] = None
|
||||
):
|
||||
"""获取待处理任务"""
|
||||
service = get_human_input_service()
|
||||
tasks = await service.get_pending_tasks(
|
||||
priority_filter=priority,
|
||||
agent_filter=agent
|
||||
)
|
||||
return {
|
||||
"tasks": [
|
||||
{
|
||||
"id": t.id,
|
||||
"from_user": t.from_user,
|
||||
"timestamp": t.timestamp,
|
||||
"priority": t.priority,
|
||||
"type": t.type,
|
||||
"title": t.title,
|
||||
"content": t.content,
|
||||
"target_files": t.target_files,
|
||||
"suggested_agent": t.suggested_agent,
|
||||
"urgent": t.urgent,
|
||||
"is_urgent": t.is_urgent
|
||||
}
|
||||
for t in tasks
|
||||
],
|
||||
"count": len(tasks)
|
||||
}
|
||||
|
||||
|
||||
@router.get("/tasks/urgent")
|
||||
async def get_urgent_tasks():
|
||||
"""获取紧急任务"""
|
||||
service = get_human_input_service()
|
||||
tasks = await service.get_urgent_tasks()
|
||||
return {
|
||||
"tasks": [
|
||||
{
|
||||
"id": t.id,
|
||||
"from_user": t.from_user,
|
||||
"content": t.content,
|
||||
"title": t.title,
|
||||
"suggested_agent": t.suggested_agent
|
||||
}
|
||||
for t in tasks
|
||||
],
|
||||
"count": len(tasks)
|
||||
}
|
||||
|
||||
|
||||
@router.put("/tasks/{task_id}/processing")
|
||||
async def mark_task_processing(task_id: str):
|
||||
"""标记任务为处理中"""
|
||||
service = get_human_input_service()
|
||||
success = await service.mark_task_processing(task_id)
|
||||
if not success:
|
||||
raise HTTPException(status_code=404, detail="Task not found")
|
||||
return {"success": True}
|
||||
|
||||
|
||||
@router.put("/tasks/{task_id}/complete")
|
||||
async def mark_task_completed(task_id: str):
|
||||
"""标记任务为已完成"""
|
||||
service = get_human_input_service()
|
||||
success = await service.mark_task_completed(task_id)
|
||||
if not success:
|
||||
raise HTTPException(status_code=404, detail="Task not found")
|
||||
return {"success": True}
|
||||
|
||||
|
||||
@router.post("/comments")
|
||||
async def add_meeting_comment(request: CommentRequest):
|
||||
"""提交会议评论"""
|
||||
service = get_human_input_service()
|
||||
comment_id = await service.add_meeting_comment(
|
||||
from_user=request.from_user,
|
||||
meeting_id=request.meeting_id,
|
||||
content=request.content,
|
||||
comment_type=request.comment_type,
|
||||
priority=request.priority
|
||||
)
|
||||
return {"success": True, "comment_id": comment_id}
|
||||
|
||||
|
||||
@router.get("/comments")
|
||||
async def get_pending_comments(meeting_id: Optional[str] = None):
|
||||
"""获取待处理评论"""
|
||||
service = get_human_input_service()
|
||||
comments = await service.get_pending_comments(meeting_id)
|
||||
return {
|
||||
"comments": [
|
||||
{
|
||||
"id": c.id,
|
||||
"from_user": c.from_user,
|
||||
"meeting_id": c.meeting_id,
|
||||
"timestamp": c.timestamp,
|
||||
"type": c.type,
|
||||
"priority": c.priority,
|
||||
"content": c.content
|
||||
}
|
||||
for c in comments
|
||||
],
|
||||
"count": len(comments)
|
||||
}
|
||||
|
||||
|
||||
@router.put("/comments/{comment_id}/addressed")
|
||||
async def mark_comment_addressed(comment_id: str):
|
||||
"""标记评论为已处理"""
|
||||
service = get_human_input_service()
|
||||
success = await service.mark_comment_addressed(comment_id)
|
||||
if not success:
|
||||
raise HTTPException(status_code=404, detail="Comment not found")
|
||||
return {"success": True}
|
||||
|
||||
|
||||
@router.put("/users/{user_id}/status")
|
||||
async def update_user_status(user_id: str, request: UserStatusUpdate):
|
||||
"""更新用户状态"""
|
||||
service = get_human_input_service()
|
||||
success = await service.update_user_status(
|
||||
user_id,
|
||||
request.status,
|
||||
request.current_focus
|
||||
)
|
||||
if not success:
|
||||
raise HTTPException(status_code=404, detail="User not found")
|
||||
return {"success": True}
|
||||
88
backend/app/routers/locks.py
Normal file
88
backend/app/routers/locks.py
Normal file
@@ -0,0 +1,88 @@
|
||||
"""
|
||||
文件锁 API 路由
|
||||
"""
|
||||
from fastapi import APIRouter
|
||||
from pydantic import BaseModel
|
||||
from typing import List, Optional
|
||||
import time
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
locks_db = [
|
||||
{
|
||||
"file_path": "src/main.py",
|
||||
"agent_id": "claude-001",
|
||||
"agent_name": "Claude Code",
|
||||
"locked_at": time.time() - 3600
|
||||
},
|
||||
{
|
||||
"file_path": "src/utils.py",
|
||||
"agent_id": "kimi-001",
|
||||
"agent_name": "Kimi CLI",
|
||||
"locked_at": time.time() - 1800
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
class FileLock(BaseModel):
|
||||
file_path: str
|
||||
agent_id: str
|
||||
agent_name: str = ""
|
||||
locked_at: float
|
||||
|
||||
|
||||
def format_elapsed(locked_at: float) -> str:
|
||||
"""格式化已锁定时间"""
|
||||
elapsed = time.time() - locked_at
|
||||
if elapsed < 60:
|
||||
return f"{int(elapsed)}秒"
|
||||
elif elapsed < 3600:
|
||||
return f"{int(elapsed / 60)}分钟"
|
||||
else:
|
||||
return f"{elapsed / 3600:.1f}小时"
|
||||
|
||||
@router.get("/")
|
||||
async def list_locks():
|
||||
"""获取所有文件锁列表"""
|
||||
locks_with_display = []
|
||||
for lock in locks_db:
|
||||
lock_copy = lock.copy()
|
||||
lock_copy["elapsed_display"] = format_elapsed(lock["locked_at"])
|
||||
locks_with_display.append(lock_copy)
|
||||
return {"locks": locks_with_display}
|
||||
|
||||
|
||||
@router.post("/acquire")
|
||||
async def acquire_lock(lock: FileLock):
|
||||
"""获取文件锁"""
|
||||
# 检查是否已被锁定
|
||||
for existing in locks_db:
|
||||
if existing["file_path"] == lock.file_path:
|
||||
return {"success": False, "message": "File already locked"}
|
||||
|
||||
locks_db.append({
|
||||
"file_path": lock.file_path,
|
||||
"agent_id": lock.agent_id,
|
||||
"agent_name": lock.agent_name or lock.agent_id,
|
||||
"locked_at": time.time()
|
||||
})
|
||||
return {"success": True, "message": "Lock acquired"}
|
||||
|
||||
|
||||
@router.post("/release")
|
||||
async def release_lock(data: dict):
|
||||
"""释放文件锁"""
|
||||
file_path = data.get("file_path", "")
|
||||
agent_id = data.get("agent_id", "")
|
||||
global locks_db
|
||||
locks_db = [l for l in locks_db if not (l["file_path"] == file_path and l["agent_id"] == agent_id)]
|
||||
return {"success": True, "message": "Lock released"}
|
||||
|
||||
|
||||
@router.get("/check")
|
||||
async def check_lock(file_path: str):
|
||||
"""检查文件锁定状态"""
|
||||
for lock in locks_db:
|
||||
if lock["file_path"] == file_path:
|
||||
return {"file_path": file_path, "locked": True, "locked_by": lock["agent_id"]}
|
||||
return {"file_path": file_path, "locked": False}
|
||||
194
backend/app/routers/meetings.py
Normal file
194
backend/app/routers/meetings.py
Normal file
@@ -0,0 +1,194 @@
|
||||
"""
|
||||
会议管理 API 路由
|
||||
"""
|
||||
from fastapi import APIRouter
|
||||
from pydantic import BaseModel
|
||||
from typing import List, Optional
|
||||
from datetime import datetime
|
||||
import time
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
meetings_db = []
|
||||
|
||||
|
||||
class Meeting(BaseModel):
|
||||
meeting_id: str
|
||||
title: str
|
||||
status: str
|
||||
attendees: List[str]
|
||||
agenda: str
|
||||
progress_summary: str
|
||||
created_at: float
|
||||
|
||||
|
||||
class MeetingCreate(BaseModel):
|
||||
title: str
|
||||
agenda: str
|
||||
meeting_type: str = "design_review"
|
||||
attendees: List[str] = []
|
||||
|
||||
|
||||
@router.get("/")
|
||||
async def list_meetings():
|
||||
"""获取所有会议列表"""
|
||||
return {
|
||||
"meetings": [
|
||||
{
|
||||
"meeting_id": "meeting-001",
|
||||
"title": "架构设计评审",
|
||||
"status": "in_progress",
|
||||
"attendees": ["claude-001", "kimi-001"],
|
||||
"agenda": "讨论系统架构设计",
|
||||
"progress_summary": "50%",
|
||||
"created_at": time.time() - 7200
|
||||
},
|
||||
{
|
||||
"meeting_id": "meeting-002",
|
||||
"title": "代码审查会议",
|
||||
"status": "completed",
|
||||
"attendees": ["claude-001"],
|
||||
"agenda": "审查前端组件代码",
|
||||
"progress_summary": "100%",
|
||||
"created_at": time.time() - 86400
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@router.get("/today")
|
||||
async def list_today_meetings():
|
||||
"""获取今日会议"""
|
||||
today = datetime.now().strftime("%Y-%m-%d")
|
||||
return {
|
||||
"meetings": [
|
||||
{
|
||||
"meeting_id": "meeting-001",
|
||||
"title": "架构设计评审",
|
||||
"date": today,
|
||||
"status": "in_progress",
|
||||
"attendees": ["claude-001", "kimi-001"],
|
||||
"steps": [
|
||||
{"step_id": "step-1", "label": "收集想法", "status": "completed"},
|
||||
{"step_id": "step-2", "label": "讨论迭代", "status": "active"},
|
||||
{"step_id": "step-3", "label": "生成共识", "status": "pending"}
|
||||
],
|
||||
"discussions": [
|
||||
{
|
||||
"agent_id": "claude-001",
|
||||
"agent_name": "Claude Code",
|
||||
"content": "建议采用微服务架构",
|
||||
"timestamp": datetime.now().isoformat(),
|
||||
"step": "讨论迭代"
|
||||
}
|
||||
],
|
||||
"progress_summary": "50%",
|
||||
"consensus": ""
|
||||
},
|
||||
{
|
||||
"meeting_id": "meeting-002",
|
||||
"title": "代码审查会议",
|
||||
"date": today,
|
||||
"status": "completed",
|
||||
"attendees": ["claude-001"],
|
||||
"steps": [
|
||||
{"step_id": "step-1", "label": "代码检查", "status": "completed"},
|
||||
{"step_id": "step-2", "label": "问题讨论", "status": "completed"}
|
||||
],
|
||||
"discussions": [],
|
||||
"progress_summary": "100%",
|
||||
"consensus": "代码质量良好，可以合并"
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@router.post("/")
|
||||
async def create_meeting(meeting: MeetingCreate):
|
||||
"""创建新会议"""
|
||||
meeting_id = f"meeting-{int(time.time())}"
|
||||
meeting_data = {
|
||||
"meeting_id": meeting_id,
|
||||
"title": meeting.title,
|
||||
"status": "waiting",
|
||||
"attendees": meeting.attendees,
|
||||
"agenda": meeting.agenda,
|
||||
"progress_summary": "0%",
|
||||
"created_at": time.time()
|
||||
}
|
||||
meetings_db.append(meeting_data)
|
||||
return meeting_data
|
||||
|
||||
|
||||
@router.get("/{meeting_id}")
|
||||
async def get_meeting(meeting_id: str):
|
||||
"""获取会议详情"""
|
||||
for meeting in meetings_db:
|
||||
if meeting["meeting_id"] == meeting_id:
|
||||
return meeting
|
||||
# 返回模拟数据
|
||||
return {
|
||||
"meeting_id": meeting_id,
|
||||
"title": "测试会议",
|
||||
"status": "in_progress",
|
||||
"attendees": ["claude-001"],
|
||||
"agenda": "测试议程",
|
||||
"progress_summary": "50%",
|
||||
"created_at": time.time()
|
||||
}
|
||||
|
||||
|
||||
@router.post("/create")
|
||||
async def create_meeting_api(meeting: MeetingCreate):
|
||||
"""创建会议 API（前端使用的端点）"""
|
||||
return await create_meeting(meeting)
|
||||
|
||||
|
||||
@router.post("/{meeting_id}/join")
|
||||
async def join_meeting(meeting_id: str, data: dict):
|
||||
"""Agent 加入会议"""
|
||||
agent_id = data.get("agent_id", "")
|
||||
return {"success": True, "meeting_id": meeting_id, "agent_id": agent_id}
|
||||
|
||||
|
||||
@router.post("/{meeting_id}/discuss")
|
||||
async def add_discussion(meeting_id: str, data: dict):
|
||||
"""添加讨论内容"""
|
||||
return {"success": True, "meeting_id": meeting_id}
|
||||
|
||||
|
||||
@router.post("/{meeting_id}/finish")
|
||||
async def finish_meeting(meeting_id: str, data: dict):
|
||||
"""完成会议"""
|
||||
return {"success": True, "meeting_id": meeting_id}
|
||||
|
||||
|
||||
@router.post("/{meeting_id}/progress")
|
||||
async def update_progress(meeting_id: str, data: dict):
|
||||
"""更新进度"""
|
||||
return {"success": True, "meeting_id": meeting_id}
|
||||
|
||||
|
||||
@router.post("/record/create")
|
||||
async def create_meeting_record(data: dict):
|
||||
"""创建会议记录（前端使用的端点）"""
|
||||
meeting_id = f"meeting-{int(time.time())}"
|
||||
meeting_data = {
|
||||
"meeting_id": meeting_id,
|
||||
"title": data.get("title", "未命名会议"),
|
||||
"agenda": data.get("agenda", ""),
|
||||
"attendees": data.get("attendees", []),
|
||||
"status": "waiting",
|
||||
"progress_summary": "0%",
|
||||
"steps": data.get("steps", []),
|
||||
"discussions": [],
|
||||
"created_at": time.time()
|
||||
}
|
||||
meetings_db.append(meeting_data)
|
||||
return meeting_data
|
||||
|
||||
|
||||
@router.post("/record/{meeting_id}/discussion")
|
||||
async def add_meeting_discussion(meeting_id: str, data: dict):
|
||||
"""添加会议讨论（前端使用的端点）"""
|
||||
return {"success": True, "meeting_id": meeting_id, "discussion": data}
|
||||
60
backend/app/routers/resources.py
Normal file
60
backend/app/routers/resources.py
Normal file
@@ -0,0 +1,60 @@
|
||||
"""
|
||||
资源管理 API 路由
|
||||
"""
|
||||
from fastapi import APIRouter
|
||||
from pydantic import BaseModel
|
||||
from typing import List, Optional
|
||||
import time
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
class TaskRequest(BaseModel):
|
||||
agent_id: str
|
||||
task: str
|
||||
timeout: Optional[int] = 300
|
||||
|
||||
|
||||
class TaskParseRequest(BaseModel):
|
||||
task: str
|
||||
|
||||
|
||||
@router.post("/execute")
|
||||
async def execute_task(request: TaskRequest):
|
||||
"""执行任务"""
|
||||
return {
|
||||
"success": True,
|
||||
"message": f"任务 '{request.task}' 已执行",
|
||||
"files_locked": ["src/main.py"],
|
||||
"duration_seconds": 5.5
|
||||
}
|
||||
|
||||
|
||||
@router.get("/status")
|
||||
async def get_all_status():
|
||||
"""获取所有 Agent 状态"""
|
||||
return {
|
||||
"agents": [
|
||||
{
|
||||
"agent_id": "claude-001",
|
||||
"status": "working",
|
||||
"current_task": "开发功能",
|
||||
"progress": 75
|
||||
},
|
||||
{
|
||||
"agent_id": "kimi-001",
|
||||
"status": "idle",
|
||||
"current_task": "",
|
||||
"progress": 0
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@router.post("/parse-task")
|
||||
async def parse_task(request: TaskParseRequest):
|
||||
"""解析任务文件"""
|
||||
return {
|
||||
"task": request.task,
|
||||
"files": ["src/main.py", "src/utils.py"]
|
||||
}
|
||||
55
backend/app/routers/roles.py
Normal file
55
backend/app/routers/roles.py
Normal file
@@ -0,0 +1,55 @@
|
||||
"""
|
||||
角色分配 API 路由
|
||||
"""
|
||||
from fastapi import APIRouter
|
||||
from pydantic import BaseModel
|
||||
from typing import List, Dict
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
class RoleRequest(BaseModel):
|
||||
task: str
|
||||
|
||||
|
||||
class RoleAllocateRequest(BaseModel):
|
||||
task: str
|
||||
agents: List[str]
|
||||
|
||||
|
||||
@router.post("/primary")
|
||||
async def get_primary_role(request: RoleRequest):
|
||||
"""获取任务主要角色"""
|
||||
return {
|
||||
"task": request.task,
|
||||
"primary_role": "developer",
|
||||
"role_scores": {
|
||||
"developer": 0.8,
|
||||
"architect": 0.6,
|
||||
"qa": 0.4,
|
||||
"pm": 0.2
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@router.post("/allocate")
|
||||
async def allocate_roles(request: RoleAllocateRequest):
|
||||
"""分配角色"""
|
||||
allocation = {}
|
||||
for i, agent in enumerate(request.agents):
|
||||
roles = ["developer", "architect", "qa"]
|
||||
allocation[agent] = roles[i % len(roles)]
|
||||
|
||||
return {
|
||||
"task": request.task,
|
||||
"primary_role": "developer",
|
||||
"allocation": allocation
|
||||
}
|
||||
|
||||
|
||||
@router.post("/explain")
|
||||
async def explain_roles(request: RoleAllocateRequest):
|
||||
"""解释角色分配"""
|
||||
return {
|
||||
"explanation": f"基于任务 '{request.task}' 的分析，推荐了最适合的角色分配方案。"
|
||||
}
|
||||
392
backend/app/routers/websocket.py
Normal file
392
backend/app/routers/websocket.py
Normal file
@@ -0,0 +1,392 @@
|
||||
"""
|
||||
WebSocket 实时通信
|
||||
|
||||
提供 Agent 与服务器之间的实时双向通信
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
import asyncio
|
||||
from typing import Dict, Set, Optional, Any
|
||||
from datetime import datetime
|
||||
|
||||
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
|
||||
from pydantic import BaseModel
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
class ConnectionManager:
|
||||
"""
|
||||
WebSocket 连接管理器
|
||||
|
||||
管理 WebSocket 连接，支持：
|
||||
1. Agent 连接管理
|
||||
2. 消息广播
|
||||
3. 私信发送
|
||||
4. 心跳检测
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
# Agent 连接: {agent_id: WebSocket}
|
||||
self.agent_connections: Dict[str, WebSocket] = {}
|
||||
# 客户端连接: {client_id: WebSocket}
|
||||
self.client_connections: Dict[str, WebSocket] = {}
|
||||
# 连接元数据: {connection_id: {"type": "agent"|"client", "connected_at": datetime}}
|
||||
self.connection_metadata: Dict[str, Dict[str, Any]] = {}
|
||||
|
||||
async def connect_agent(self, websocket: WebSocket, agent_id: str):
|
||||
"""Agent 连接"""
|
||||
await websocket.accept()
|
||||
self.agent_connections[agent_id] = websocket
|
||||
self.connection_metadata[agent_id] = {
|
||||
"type": "agent",
|
||||
"connected_at": datetime.now()
|
||||
}
|
||||
logger.info(f"Agent 连接: {agent_id}")
|
||||
|
||||
# 发送欢迎消息
|
||||
await self.send_to_agent(agent_id, {
|
||||
"type": "connected",
|
||||
"agent_id": agent_id,
|
||||
"message": "连接成功"
|
||||
})
|
||||
|
||||
async def connect_client(self, websocket: WebSocket, client_id: str):
|
||||
"""客户端连接"""
|
||||
await websocket.accept()
|
||||
self.client_connections[client_id] = websocket
|
||||
self.connection_metadata[client_id] = {
|
||||
"type": "client",
|
||||
"connected_at": datetime.now()
|
||||
}
|
||||
logger.info(f"客户端连接: {client_id}")
|
||||
|
||||
# 发送欢迎消息
|
||||
await self.send_to_client(client_id, {
|
||||
"type": "connected",
|
||||
"client_id": client_id,
|
||||
"message": "连接成功"
|
||||
})
|
||||
|
||||
def disconnect_agent(self, agent_id: str):
|
||||
"""断开 Agent 连接"""
|
||||
if agent_id in self.agent_connections:
|
||||
del self.agent_connections[agent_id]
|
||||
if agent_id in self.connection_metadata:
|
||||
del self.connection_metadata[agent_id]
|
||||
logger.info(f"Agent 断开: {agent_id}")
|
||||
|
||||
def disconnect_client(self, client_id: str):
|
||||
"""断开客户端连接"""
|
||||
if client_id in self.client_connections:
|
||||
del self.client_connections[client_id]
|
||||
if client_id in self.connection_metadata:
|
||||
del self.connection_metadata[client_id]
|
||||
logger.info(f"客户端断开: {client_id}")
|
||||
|
||||
async def send_to_agent(self, agent_id: str, message: Dict) -> bool:
|
||||
"""发送消息给 Agent"""
|
||||
if agent_id in self.agent_connections:
|
||||
try:
|
||||
await self.agent_connections[agent_id].send_json(message)
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"发送消息给 Agent 失败: {agent_id}: {e}")
|
||||
self.disconnect_agent(agent_id)
|
||||
return False
|
||||
return False
|
||||
|
||||
async def send_to_client(self, client_id: str, message: Dict) -> bool:
|
||||
"""发送消息给客户端"""
|
||||
if client_id in self.client_connections:
|
||||
try:
|
||||
await self.client_connections[client_id].send_json(message)
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"发送消息给客户端失败: {client_id}: {e}")
|
||||
self.disconnect_client(client_id)
|
||||
return False
|
||||
return False
|
||||
|
||||
async def broadcast_to_agents(self, message: Dict):
|
||||
"""广播消息给所有 Agent"""
|
||||
failed_agents = []
|
||||
for agent_id, websocket in self.agent_connections.items():
|
||||
try:
|
||||
await websocket.send_json(message)
|
||||
except Exception as e:
|
||||
logger.error(f"广播消息失败: {agent_id}: {e}")
|
||||
failed_agents.append(agent_id)
|
||||
|
||||
# 清理失败的连接
|
||||
for agent_id in failed_agents:
|
||||
self.disconnect_agent(agent_id)
|
||||
|
||||
async def broadcast_to_clients(self, message: Dict):
|
||||
"""广播消息给所有客户端"""
|
||||
failed_clients = []
|
||||
for client_id, websocket in self.client_connections.items():
|
||||
try:
|
||||
await websocket.send_json(message)
|
||||
except Exception as e:
|
||||
logger.error(f"广播消息失败: {client_id}: {e}")
|
||||
failed_clients.append(client_id)
|
||||
|
||||
# 清理失败的连接
|
||||
for client_id in failed_clients:
|
||||
self.disconnect_client(client_id)
|
||||
|
||||
async def broadcast_to_all(self, message: Dict):
|
||||
"""广播消息给所有连接"""
|
||||
await self.broadcast_to_agents(message)
|
||||
await self.broadcast_to_clients(message)
|
||||
|
||||
def get_connected_agents(self) -> Set[str]:
|
||||
"""获取已连接的 Agent"""
|
||||
return set(self.agent_connections.keys())
|
||||
|
||||
def get_connected_clients(self) -> Set[str]:
|
||||
"""获取已连接的客户端"""
|
||||
return set(self.client_connections.keys())
|
||||
|
||||
def get_connection_count(self) -> Dict[str, int]:
|
||||
"""获取连接数量"""
|
||||
return {
|
||||
"agents": len(self.agent_connections),
|
||||
"clients": len(self.client_connections),
|
||||
"total": len(self.agent_connections) + len(self.client_connections)
|
||||
}
|
||||
|
||||
|
||||
# 全局连接管理器
|
||||
manager = ConnectionManager()
|
||||
|
||||
|
||||
# ========== WebSocket 端点 ==========
|
||||
|
||||
|
||||
@router.websocket("/ws/agent/{agent_id}")
|
||||
async def agent_websocket_endpoint(websocket: WebSocket, agent_id: str):
|
||||
"""
|
||||
Agent WebSocket 端点
|
||||
|
||||
Agent 连接后可以：
|
||||
1. 接收任务分配
|
||||
2. 发送状态更新
|
||||
3. 参与实时协作
|
||||
"""
|
||||
await manager.connect_agent(websocket, agent_id)
|
||||
|
||||
try:
|
||||
while True:
|
||||
# 接收来自 Agent 的消息
|
||||
data = await websocket.receive_json()
|
||||
await handle_agent_message(agent_id, data)
|
||||
|
||||
except WebSocketDisconnect:
|
||||
manager.disconnect_agent(agent_id)
|
||||
except Exception as e:
|
||||
logger.error(f"Agent WebSocket 错误: {agent_id}: {e}")
|
||||
manager.disconnect_agent(agent_id)
|
||||
|
||||
|
||||
@router.websocket("/ws/client/{client_id}")
|
||||
async def client_websocket_endpoint(websocket: WebSocket, client_id: str):
|
||||
"""
|
||||
客户端 WebSocket 端点
|
||||
|
||||
客户端连接后可以：
|
||||
1. 实时监控 Agent 状态
|
||||
2. 接收事件通知
|
||||
3. 发送控制指令
|
||||
"""
|
||||
await manager.connect_client(websocket, client_id)
|
||||
|
||||
# 发送初始状态
|
||||
await manager.send_to_client(client_id, {
|
||||
"type": "initial_state",
|
||||
"connected_agents": list(manager.get_connected_agents()),
|
||||
"timestamp": datetime.now().isoformat()
|
||||
})
|
||||
|
||||
try:
|
||||
while True:
|
||||
# 接收来自客户端的消息
|
||||
data = await websocket.receive_json()
|
||||
await handle_client_message(client_id, data)
|
||||
|
||||
except WebSocketDisconnect:
|
||||
manager.disconnect_client(client_id)
|
||||
except Exception as e:
|
||||
logger.error(f"客户端 WebSocket 错误: {client_id}: {e}")
|
||||
manager.disconnect_client(client_id)
|
||||
|
||||
|
||||
@router.websocket("/ws")
|
||||
async def public_websocket_endpoint(websocket: WebSocket):
|
||||
"""
|
||||
公共 WebSocket 端点
|
||||
|
||||
自动生成 client_id 的客户端连接
|
||||
"""
|
||||
import uuid
|
||||
client_id = f"client_{uuid.uuid4().hex[:12]}"
|
||||
await manager.connect_client(websocket, client_id)
|
||||
|
||||
try:
|
||||
while True:
|
||||
data = await websocket.receive_json()
|
||||
await handle_client_message(client_id, data)
|
||||
|
||||
except WebSocketDisconnect:
|
||||
manager.disconnect_client(client_id)
|
||||
except Exception as e:
|
||||
logger.error(f"公共 WebSocket 错误: {client_id}: {e}")
|
||||
manager.disconnect_client(client_id)
|
||||
|
||||
|
||||
# ========== 消息处理 ==========
|
||||
|
||||
|
||||
async def handle_agent_message(agent_id: str, data: Dict):
|
||||
"""处理来自 Agent 的消息"""
|
||||
message_type = data.get("type")
|
||||
|
||||
if message_type == "heartbeat":
|
||||
# Agent 心跳更新
|
||||
await broadcast_agent_status(agent_id, data)
|
||||
|
||||
elif message_type == "status_update":
|
||||
# Agent 状态更新
|
||||
await broadcast_agent_status(agent_id, data)
|
||||
|
||||
elif message_type == "task_progress":
|
||||
# 任务进度更新
|
||||
await broadcast_task_progress(agent_id, data)
|
||||
|
||||
elif message_type == "meeting_joined":
|
||||
# Agent 加入会议
|
||||
await broadcast_event({
|
||||
"type": "meeting_event",
|
||||
"event": "agent_joined",
|
||||
"agent_id": agent_id,
|
||||
"meeting_id": data.get("meeting_id"),
|
||||
"timestamp": datetime.now().isoformat()
|
||||
})
|
||||
|
||||
elif message_type == "meeting_proposal":
|
||||
# 会议提案
|
||||
await broadcast_event({
|
||||
"type": "meeting_event",
|
||||
"event": "proposal",
|
||||
"agent_id": agent_id,
|
||||
"meeting_id": data.get("meeting_id"),
|
||||
"content": data.get("content"),
|
||||
"timestamp": datetime.now().isoformat()
|
||||
})
|
||||
|
||||
else:
|
||||
# 其他消息类型
|
||||
await broadcast_event({
|
||||
"type": "agent_message",
|
||||
"agent_id": agent_id,
|
||||
"message_type": message_type,
|
||||
"data": data,
|
||||
"timestamp": datetime.now().isoformat()
|
||||
})
|
||||
|
||||
|
||||
async def handle_client_message(client_id: str, data: Dict):
|
||||
"""处理来自客户端的消息"""
|
||||
message_type = data.get("type")
|
||||
|
||||
if message_type == "subscribe_agents":
|
||||
# 客户端订阅 Agent 状态
|
||||
await manager.send_to_client(client_id, {
|
||||
"type": "subscription_confirmed",
|
||||
"subscription": "agents"
|
||||
})
|
||||
|
||||
elif message_type == "send_to_agent":
|
||||
# 发送消息给特定 Agent
|
||||
agent_id = data.get("agent_id")
|
||||
message = data.get("message")
|
||||
if agent_id:
|
||||
await manager.send_to_agent(agent_id, {
|
||||
"type": "client_message",
|
||||
"from_client": client_id,
|
||||
"message": message
|
||||
})
|
||||
|
||||
elif message_type == "broadcast":
|
||||
# 广播消息给所有 Agent
|
||||
message = data.get("message")
|
||||
await manager.broadcast_to_agents({
|
||||
"type": "broadcast",
|
||||
"from_client": client_id,
|
||||
"message": message
|
||||
})
|
||||
|
||||
|
||||
async def broadcast_agent_status(agent_id: str, data: Dict):
|
||||
"""广播 Agent 状态更新"""
|
||||
await manager.broadcast_to_clients({
|
||||
"type": "agent_status",
|
||||
"agent_id": agent_id,
|
||||
"data": data,
|
||||
"timestamp": datetime.now().isoformat()
|
||||
})
|
||||
|
||||
|
||||
async def broadcast_task_progress(agent_id: str, data: Dict):
|
||||
"""广播任务进度更新"""
|
||||
await manager.broadcast_to_clients({
|
||||
"type": "task_progress",
|
||||
"agent_id": agent_id,
|
||||
"task_id": data.get("task_id"),
|
||||
"progress": data.get("progress"),
|
||||
"message": data.get("message"),
|
||||
"timestamp": datetime.now().isoformat()
|
||||
})
|
||||
|
||||
|
||||
async def broadcast_event(event: Dict):
|
||||
"""广播事件"""
|
||||
await manager.broadcast_to_all(event)
|
||||
|
||||
|
||||
# ========== HTTP API ==========
|
||||
|
||||
|
||||
@router.get("/ws/connections")
|
||||
async def get_connections():
|
||||
"""获取当前连接信息"""
|
||||
return {
|
||||
"agents": list(manager.get_connected_agents()),
|
||||
"clients": list(manager.get_connected_clients()),
|
||||
"count": manager.get_connection_count()
|
||||
}
|
||||
|
||||
|
||||
@router.post("/ws/broadcast")
|
||||
async def broadcast_message(message: Dict):
|
||||
"""通过 HTTP 广播消息到所有连接"""
|
||||
await manager.broadcast_to_all({
|
||||
"type": "broadcast",
|
||||
"message": message,
|
||||
"timestamp": datetime.now().isoformat()
|
||||
})
|
||||
return {"success": True, "message": "消息已广播"}
|
||||
|
||||
|
||||
@router.post("/ws/send/{agent_id}")
|
||||
async def send_to_agent(agent_id: str, message: Dict):
|
||||
"""通过 HTTP 发送消息给特定 Agent"""
|
||||
success = await manager.send_to_agent(agent_id, message)
|
||||
if success:
|
||||
return {"success": True, "agent_id": agent_id}
|
||||
else:
|
||||
return {"success": False, "error": "发送失败或 Agent 未连接"}
|
||||
218
backend/app/routers/workflows.py
Normal file
218
backend/app/routers/workflows.py
Normal file
@@ -0,0 +1,218 @@
|
||||
"""
|
||||
工作流管理 API 路由
|
||||
"""
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from pydantic import BaseModel
|
||||
from typing import List, Optional, Dict, Any
|
||||
from pathlib import Path
|
||||
|
||||
from app.services.workflow_engine import get_workflow_engine
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
# ========== 请求/响应模型 ==========
|
||||
|
||||
class MeetingNode(BaseModel):
|
||||
"""工作流节点"""
|
||||
meeting_id: str
|
||||
title: str
|
||||
node_type: str = "meeting"
|
||||
attendees: List[str]
|
||||
depends_on: List[str] = []
|
||||
completed: bool = False
|
||||
on_failure: Optional[str] = None
|
||||
progress: Optional[str] = None
|
||||
|
||||
|
||||
class WorkflowDetail(BaseModel):
|
||||
"""工作流详情"""
|
||||
workflow_id: str
|
||||
name: str
|
||||
description: str
|
||||
status: str
|
||||
progress: str
|
||||
current_node: Optional[str] = None
|
||||
meetings: List[MeetingNode]
|
||||
|
||||
|
||||
class WorkflowSummary(BaseModel):
|
||||
"""工作流摘要"""
|
||||
workflow_id: str
|
||||
name: str
|
||||
status: str
|
||||
progress: str
|
||||
|
||||
|
||||
class JoinExecutionRequest(BaseModel):
|
||||
"""加入执行节点请求"""
|
||||
agent_id: str
|
||||
|
||||
|
||||
class JumpRequest(BaseModel):
|
||||
"""跳转请求"""
|
||||
target_meeting_id: str
|
||||
|
||||
|
||||
# ========== API 端点 ==========
|
||||
|
||||
@router.get("/files")
|
||||
async def list_workflow_files():
|
||||
"""获取工作流文件列表"""
|
||||
engine = get_workflow_engine()
|
||||
workflow_dir = Path(engine._storage.base_path) / engine.WORKFLOWS_DIR
|
||||
|
||||
if not workflow_dir.exists():
|
||||
return {"files": []}
|
||||
|
||||
yaml_files = list(workflow_dir.glob("*.yaml")) + list(workflow_dir.glob("*.yml"))
|
||||
|
||||
files = []
|
||||
for f in yaml_files:
|
||||
stat = f.stat()
|
||||
files.append({
|
||||
"name": f.name,
|
||||
"path": f"workflow/{f.name}",
|
||||
"size": stat.st_size,
|
||||
"modified": stat.st_mtime
|
||||
})
|
||||
|
||||
return {"files": files}
|
||||
|
||||
|
||||
@router.get("/list")
|
||||
async def list_workflows():
|
||||
"""获取已加载的工作流列表"""
|
||||
engine = get_workflow_engine()
|
||||
workflows = await engine.list_workflows()
|
||||
return {"workflows": workflows}
|
||||
|
||||
|
||||
@router.post("/start/{workflow_path:path}")
|
||||
async def start_workflow(workflow_path: str):
|
||||
"""
|
||||
启动工作流
|
||||
|
||||
加载 YAML 工作流文件并准备执行
|
||||
"""
|
||||
engine = get_workflow_engine()
|
||||
try:
|
||||
workflow = await engine.load_workflow(workflow_path)
|
||||
detail = await engine.get_workflow_detail(workflow.workflow_id)
|
||||
return detail
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/{workflow_id}")
|
||||
async def get_workflow(workflow_id: str):
|
||||
"""获取工作流详情"""
|
||||
engine = get_workflow_engine()
|
||||
detail = await engine.get_workflow_detail(workflow_id)
|
||||
if not detail:
|
||||
raise HTTPException(status_code=404, detail="Workflow not found")
|
||||
return detail
|
||||
|
||||
|
||||
@router.get("/{workflow_id}/status")
|
||||
async def get_workflow_status(workflow_id: str):
|
||||
"""获取工作流状态"""
|
||||
engine = get_workflow_engine()
|
||||
status = await engine.get_workflow_status(workflow_id)
|
||||
if not status:
|
||||
raise HTTPException(status_code=404, detail="Workflow not found")
|
||||
return status
|
||||
|
||||
|
||||
@router.get("/{workflow_id}/next")
|
||||
async def get_next_node(workflow_id: str):
|
||||
"""获取下一个待执行节点"""
|
||||
engine = get_workflow_engine()
|
||||
meeting = await engine.get_next_meeting(workflow_id)
|
||||
if not meeting:
|
||||
return {"meeting": None, "message": "Workflow completed"}
|
||||
return {
|
||||
"meeting": {
|
||||
"meeting_id": meeting.meeting_id,
|
||||
"title": meeting.title,
|
||||
"node_type": meeting.node_type,
|
||||
"attendees": meeting.attendees,
|
||||
"depends_on": meeting.depends_on
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@router.post("/{workflow_id}/complete/{meeting_id}")
|
||||
async def complete_node(workflow_id: str, meeting_id: str):
|
||||
"""标记节点完成"""
|
||||
engine = get_workflow_engine()
|
||||
success = await engine.complete_meeting(workflow_id, meeting_id)
|
||||
if not success:
|
||||
raise HTTPException(status_code=404, detail="Workflow or meeting not found")
|
||||
return {"success": True, "message": "Node completed"}
|
||||
|
||||
|
||||
@router.post("/{workflow_id}/join/{meeting_id}")
|
||||
async def join_execution_node(workflow_id: str, meeting_id: str, request: JoinExecutionRequest):
|
||||
"""
|
||||
Agent 加入执行节点
|
||||
|
||||
标记 Agent 已完成执行，当所有 Agent 都完成时返回 ready
|
||||
"""
|
||||
engine = get_workflow_engine()
|
||||
result = await engine.join_execution_node(workflow_id, meeting_id, request.agent_id)
|
||||
if result.get("status") == "error":
|
||||
raise HTTPException(status_code=400, detail=result.get("message"))
|
||||
return result
|
||||
|
||||
|
||||
@router.get("/{workflow_id}/execution/{meeting_id}")
|
||||
async def get_execution_node_status(workflow_id: str, meeting_id: str):
|
||||
"""获取执行节点状态"""
|
||||
engine = get_workflow_engine()
|
||||
status = await engine.get_execution_status(workflow_id, meeting_id)
|
||||
if not status:
|
||||
raise HTTPException(status_code=404, detail="Execution node not found")
|
||||
return status
|
||||
|
||||
|
||||
@router.post("/{workflow_id}/jump")
|
||||
async def jump_to_node(workflow_id: str, request: JumpRequest):
|
||||
"""
|
||||
强制跳转到指定节点
|
||||
|
||||
重置目标节点及所有后续节点的完成状态
|
||||
"""
|
||||
engine = get_workflow_engine()
|
||||
success = await engine.jump_to_node(workflow_id, request.target_meeting_id)
|
||||
if not success:
|
||||
raise HTTPException(status_code=404, detail="Target node not found")
|
||||
return {
|
||||
"success": True,
|
||||
"message": f"Jumped to {request.target_meeting_id}",
|
||||
"detail": await engine.get_workflow_detail(workflow_id)
|
||||
}
|
||||
|
||||
|
||||
@router.post("/{workflow_id}/fail/{meeting_id}")
|
||||
async def handle_node_failure(workflow_id: str, meeting_id: str):
|
||||
"""
|
||||
处理节点失败
|
||||
|
||||
根据 on_failure 配置跳转到指定节点
|
||||
"""
|
||||
engine = get_workflow_engine()
|
||||
target = await engine.handle_failure(workflow_id, meeting_id)
|
||||
if target:
|
||||
return {
|
||||
"success": True,
|
||||
"message": f"Jumped to {target} due to failure",
|
||||
"target": target,
|
||||
"detail": await engine.get_workflow_detail(workflow_id)
|
||||
}
|
||||
return {
|
||||
"success": True,
|
||||
"message": "No failure handler configured"
|
||||
}
|
||||
Reference in New Issue
Block a user