318 lines
10 KiB
Python
318 lines
10 KiB
Python
"""Multi-Agent Service 的 Handoffs 集成
|
|
|
|
将 Agent Handoffs 功能集成到现有的 Multi-Agent 系统中
|
|
"""
|
|
import uuid
|
|
import time
|
|
from typing import Dict, Any, Optional, AsyncGenerator
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.services.agent_handoff import get_handoff_manager
|
|
from app.services.collaborative_orchestrator import CollaborativeOrchestrator
|
|
from app.schemas.multi_agent_schema import MultiAgentRunRequest
|
|
from app.core.logging_config import get_business_logger
|
|
from app.core.exceptions import BusinessException
|
|
from app.core.error_codes import BizCode
|
|
from app.services.multi_agent_service import MultiAgentService
|
|
|
|
logger = get_business_logger()
|
|
|
|
|
|
class MultiAgentHandoffsService:
|
|
"""Multi-Agent Handoffs 服务 - 扩展现有的 Multi-Agent Service"""
|
|
|
|
def __init__(self, db: Session, multi_agent_service:MultiAgentService):
|
|
"""初始化服务
|
|
|
|
Args:
|
|
db: 数据库会话
|
|
multi_agent_service: 现有的 MultiAgentService 实例
|
|
"""
|
|
self.db = db
|
|
self.multi_agent_service = multi_agent_service
|
|
self.handoff_manager = get_handoff_manager()
|
|
|
|
logger.info("Multi-Agent Handoffs 服务初始化完成")
|
|
|
|
async def run_with_handoffs(
|
|
self,
|
|
app_id: uuid.UUID,
|
|
request: MultiAgentRunRequest
|
|
) -> Dict[str, Any]:
|
|
"""运行支持 handoffs 的多 Agent 任务
|
|
|
|
Args:
|
|
app_id: 应用 ID
|
|
request: 运行请求
|
|
|
|
Returns:
|
|
执行结果
|
|
"""
|
|
start_time = time.time()
|
|
|
|
try:
|
|
# 1. 获取配置
|
|
config = self.multi_agent_service.get_config(app_id)
|
|
if not config:
|
|
raise BusinessException(
|
|
"多 Agent 配置不存在",
|
|
BizCode.RESOURCE_NOT_FOUND
|
|
)
|
|
|
|
# 2. 检查是否启用 handoffs
|
|
execution_config = config.execution_config or {}
|
|
print("="*50)
|
|
print(execution_config)
|
|
enable_handoffs = execution_config.get("enable_handoffs", False)
|
|
|
|
if not enable_handoffs:
|
|
# 降级到普通模式
|
|
logger.info("Handoffs 未启用，使用普通模式")
|
|
return await self.multi_agent_service.run(app_id, request)
|
|
|
|
# 3. 创建协作编排器
|
|
orchestrator = CollaborativeOrchestrator(
|
|
db=self.db,
|
|
config=config,
|
|
handoff_manager=self.handoff_manager
|
|
)
|
|
|
|
# 4. 执行协作
|
|
result = await orchestrator.execute_with_handoffs(
|
|
message=request.message,
|
|
conversation_id=str(request.conversation_id) if request.conversation_id else None,
|
|
user_id=request.user_id,
|
|
variables=request.variables
|
|
)
|
|
|
|
# 5. 增强结果
|
|
result["mode"] = "handoffs"
|
|
result["elapsed_time"] = time.time() - start_time
|
|
|
|
logger.info(
|
|
"Handoffs 执行完成",
|
|
extra={
|
|
"app_id": str(app_id),
|
|
"handoff_count": result.get("handoff_count", 0),
|
|
"final_agent": result.get("final_agent_id"),
|
|
"elapsed_time": result["elapsed_time"]
|
|
}
|
|
)
|
|
|
|
return result
|
|
|
|
except Exception as e:
|
|
logger.error(f"Handoffs 执行失败: {str(e)}")
|
|
|
|
# 降级到普通模式
|
|
logger.info("降级到普通模式")
|
|
return await self.multi_agent_service.run(app_id, request)
|
|
|
|
async def run_stream_with_handoffs(
|
|
self,
|
|
app_id: uuid.UUID,
|
|
request: MultiAgentRunRequest
|
|
) -> AsyncGenerator[str, None]:
|
|
"""流式运行支持 handoffs 的多 Agent 任务
|
|
|
|
Args:
|
|
app_id: 应用 ID
|
|
request: 运行请求
|
|
|
|
Yields:
|
|
SSE 格式的事件流
|
|
"""
|
|
try:
|
|
# 1. 获取配置
|
|
config = self.multi_agent_service.get_config(app_id)
|
|
if not config:
|
|
yield f"data: {{\"event\": \"error\", \"error\": \"配置不存在\"}}\n\n"
|
|
return
|
|
|
|
# 2. 检查是否启用 handoffs
|
|
execution_config = config.execution_config or {}
|
|
enable_handoffs = execution_config.get("enable_handoffs", False)
|
|
|
|
if not enable_handoffs:
|
|
# 降级到普通流式模式
|
|
async for event in self.multi_agent_service.run_stream(app_id, request):
|
|
yield event
|
|
return
|
|
|
|
# 3. 创建协作编排器
|
|
orchestrator = CollaborativeOrchestrator(
|
|
db=self.db,
|
|
config=config,
|
|
handoff_manager=self.handoff_manager
|
|
)
|
|
|
|
# 4. 流式执行
|
|
async for event in orchestrator.execute_stream_with_handoffs(
|
|
message=request.message,
|
|
conversation_id=str(request.conversation_id) if request.conversation_id else None,
|
|
user_id=request.user_id,
|
|
variables=request.variables
|
|
):
|
|
yield event
|
|
|
|
except Exception as e:
|
|
logger.error(f"流式 Handoffs 执行失败: {str(e)}")
|
|
yield f"data: {{\"event\": \"error\", \"error\": \"{str(e)}\"}}\n\n"
|
|
|
|
def get_handoff_history(
|
|
self,
|
|
conversation_id: str
|
|
) -> Optional[Dict[str, Any]]:
|
|
"""获取会话的 handoff 历史
|
|
|
|
Args:
|
|
conversation_id: 会话 ID
|
|
|
|
Returns:
|
|
Handoff 历史信息
|
|
"""
|
|
state = self.handoff_manager.get_state(conversation_id)
|
|
if not state:
|
|
return None
|
|
|
|
return {
|
|
"conversation_id": state.conversation_id,
|
|
"current_agent_id": state.current_agent_id,
|
|
"handoff_count": state.get_handoff_count(),
|
|
"handoff_history": [
|
|
{
|
|
"from_agent": h.from_agent_id,
|
|
"to_agent": h.to_agent_id,
|
|
"reason": h.reason,
|
|
"timestamp": h.timestamp.isoformat(),
|
|
"user_message": h.user_message,
|
|
"context_summary": h.context_summary
|
|
}
|
|
for h in state.handoff_history
|
|
],
|
|
"created_at": state.created_at.isoformat(),
|
|
"updated_at": state.updated_at.isoformat()
|
|
}
|
|
|
|
def clear_handoff_state(self, conversation_id: str):
|
|
"""清除会话的 handoff 状态
|
|
|
|
Args:
|
|
conversation_id: 会话 ID
|
|
"""
|
|
self.handoff_manager.clear_state(conversation_id)
|
|
logger.info(f"清除 handoff 状态: {conversation_id}")
|
|
|
|
async def test_handoff_routing(
|
|
self,
|
|
app_id: uuid.UUID,
|
|
message: str
|
|
) -> Dict[str, Any]:
|
|
"""测试 handoff 路由决策（不实际执行）
|
|
|
|
Args:
|
|
app_id: 应用 ID
|
|
message: 测试消息
|
|
|
|
Returns:
|
|
路由决策结果
|
|
"""
|
|
# 1. 获取配置
|
|
config = self.multi_agent_service.get_config(app_id)
|
|
if not config:
|
|
raise BusinessException(
|
|
"多 Agent 配置不存在",
|
|
BizCode.RESOURCE_NOT_FOUND
|
|
)
|
|
|
|
# 2. 解析 sub agents
|
|
sub_agents = {}
|
|
for agent_data in config.sub_agents:
|
|
agent_id = agent_data.get("agent_id")
|
|
if agent_id:
|
|
sub_agents[str(agent_id)] = {
|
|
"info": agent_data
|
|
}
|
|
|
|
# 3. 测试路由
|
|
test_conversation_id = f"test-{uuid.uuid4()}"
|
|
|
|
# 选择初始 Agent
|
|
initial_agent_id = None
|
|
message_lower = message.lower()
|
|
|
|
for agent_id, agent_data in sub_agents.items():
|
|
agent_info = agent_data.get("info", {})
|
|
capabilities = agent_info.get("capabilities", [])
|
|
role = agent_info.get("role", "")
|
|
|
|
keywords = capabilities + ([role] if role else [])
|
|
for keyword in keywords:
|
|
if keyword.lower() in message_lower:
|
|
initial_agent_id = agent_id
|
|
break
|
|
|
|
if initial_agent_id:
|
|
break
|
|
|
|
if not initial_agent_id:
|
|
initial_agent_id = next(iter(sub_agents.keys()))
|
|
|
|
# 4. 生成 handoff 工具
|
|
handoff_tools = self.handoff_manager.generate_handoff_tools(
|
|
initial_agent_id,
|
|
sub_agents
|
|
)
|
|
|
|
# 5. 检查是否需要 handoff
|
|
handoff_suggestion = self.handoff_manager.should_handoff(
|
|
conversation_id=test_conversation_id,
|
|
current_agent_id=initial_agent_id,
|
|
message=message,
|
|
available_agents=sub_agents
|
|
)
|
|
|
|
return {
|
|
"message": message,
|
|
"initial_agent_id": initial_agent_id,
|
|
"initial_agent_name": sub_agents[initial_agent_id]["info"].get("name", ""),
|
|
"available_handoff_tools": [
|
|
{
|
|
"name": tool.name,
|
|
"target_agent_id": tool.target_agent_id,
|
|
"target_agent_name": tool.target_agent_name,
|
|
"description": tool.description
|
|
}
|
|
for tool in handoff_tools
|
|
],
|
|
"handoff_suggestion": handoff_suggestion,
|
|
"total_agents": len(sub_agents)
|
|
}
|
|
|
|
|
|
# 使用示例
|
|
"""
|
|
from app.services.multi_agent_service import MultiAgentService
|
|
from app.services.multi_agent_handoffs_integration import MultiAgentHandoffsService
|
|
|
|
# 创建服务
|
|
multi_agent_service = MultiAgentService(db)
|
|
handoffs_service = MultiAgentHandoffsService(db, multi_agent_service)
|
|
|
|
# 运行 handoffs
|
|
result = await handoffs_service.run_with_handoffs(
|
|
app_id=app_id,
|
|
request=MultiAgentRunRequest(
|
|
message="帮我解方程然后写诗",
|
|
conversation_id=uuid.uuid4(),
|
|
user_id="user-123"
|
|
)
|
|
)
|
|
|
|
# 查看 handoff 历史
|
|
history = handoffs_service.get_handoff_history(str(result["conversation_id"]))
|
|
print(f"Handoff 次数: {history['handoff_count']}")
|
|
for h in history['handoff_history']:
|
|
print(f"{h['from_agent']} → {h['to_agent']}: {h['reason']}")
|
|
"""
|