326 lines
12 KiB
Python
326 lines
12 KiB
Python
"""
|
||
Conductor Agent 服务
|
||
负责与 Conductor Agent 通信
|
||
"""
|
||
|
||
import logging
|
||
import httpx
|
||
import json
|
||
from typing import Dict, Any
|
||
from uuid import uuid4
|
||
|
||
from models.chat import A2ARequest, A2AMessage, A2AMessagePart
|
||
from config import settings
|
||
from services.chat_history_service import chat_history_service
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class ConductorService:
|
||
"""Conductor Agent 通信服务"""
|
||
|
||
def __init__(self):
|
||
self.base_url = settings.CONDUCTOR_AGENT_URL
|
||
self.timeout = settings.CONDUCTOR_TIMEOUT
|
||
self._request_id_counter = 1
|
||
|
||
def _generate_message_id(self) -> str:
|
||
"""生成消息 ID"""
|
||
return f"msg-{uuid4().hex[:16]}"
|
||
|
||
def _build_a2a_request(self, user_message: str, system_user_id: int, context_id: str) -> Dict[str, Any]:
|
||
"""构建 A2A 协议请求"""
|
||
# 在消息中注入用户ID，让 Agent 知道当前用户
|
||
message_with_user_id = f"[SYSTEM_USER_ID:{system_user_id}] {user_message}"
|
||
|
||
return {
|
||
"jsonrpc": "2.0",
|
||
"method": "message/send",
|
||
"params": {
|
||
"message": {
|
||
"context_id": context_id,
|
||
"role": "user",
|
||
"parts": [
|
||
{
|
||
"kind": "text",
|
||
"text": message_with_user_id
|
||
}
|
||
],
|
||
"message_id": self._generate_message_id()
|
||
}
|
||
},
|
||
"id": self._request_id_counter
|
||
}
|
||
|
||
async def _save_operation_record_if_present(self, content: str, system_user_id: int, context_id: str):
|
||
"""
|
||
尝试从响应内容中提取operation_record并保存
|
||
|
||
如果content中包含JSON格式的operation_record，提取并保存到数据库
|
||
"""
|
||
try:
|
||
# 尝试解析content为JSON
|
||
content_json = json.loads(content)
|
||
|
||
# 检查是否包含operation_record
|
||
if isinstance(content_json, dict) and "operation_record" in content_json:
|
||
operation_record = content_json["operation_record"]
|
||
|
||
# 更新记录中的用户ID和上下文ID（以确保一致性）
|
||
operation_record["system_user_id"] = system_user_id
|
||
operation_record["context_id"] = context_id
|
||
|
||
# 导入并使用设备操作服务
|
||
from services.device_operation_service import get_device_operation_service
|
||
device_operation_service = get_device_operation_service()
|
||
|
||
# 保存操作记录
|
||
success = device_operation_service.save_operation_record(operation_record)
|
||
|
||
if success:
|
||
device_type = operation_record.get('device_type', 'unknown')
|
||
action = operation_record.get('action', 'unknown')
|
||
logger.info(f"💾 已保存设备操作记录: device={device_type}, action={action}")
|
||
else:
|
||
logger.warning(f"⚠️ 保存设备操作记录失败")
|
||
|
||
except json.JSONDecodeError:
|
||
# content不是JSON格式，跳过
|
||
pass
|
||
except Exception as e:
|
||
logger.error(f"❌ 提取或保存操作记录时出错: {e}", exc_info=True)
|
||
|
||
def _extract_content_from_response(self, response: Dict[str, Any]) -> tuple[str, str, bool]:
|
||
"""
|
||
从 A2A 响应中提取内容
|
||
|
||
Returns:
|
||
tuple[str, str, bool]: (content, status, is_error)
|
||
- content: 响应内容
|
||
- status: 状态 (success, failed, error)
|
||
- is_error: 是否为错误
|
||
"""
|
||
try:
|
||
result = response.get("result", {})
|
||
|
||
# 先检查任务状态，如果失败直接返回错误
|
||
status = result.get("status", {})
|
||
state = status.get("state", "")
|
||
|
||
if state == "failed":
|
||
# 尝试从错误信息中提取详细错误
|
||
error_msg = "❌ 任务执行失败"
|
||
|
||
# 从history中查找错误信息
|
||
history = result.get("history", [])
|
||
for item in reversed(history):
|
||
if item.get("role") == "agent":
|
||
for part in item.get("parts", []):
|
||
if text := part.get("text"):
|
||
if "错误" in text or "失败" in text or "error" in text.lower():
|
||
error_msg = text
|
||
break
|
||
|
||
return error_msg, "failed", True
|
||
|
||
# 1. 从 artifacts 提取
|
||
artifacts = result.get("artifacts", [])
|
||
if artifacts:
|
||
contents = []
|
||
for artifact in artifacts:
|
||
for part in artifact.get("parts", []):
|
||
if text := part.get("text"):
|
||
contents.append(text)
|
||
if contents:
|
||
return "\n\n".join(contents), "success", False
|
||
|
||
# 2. 从 history 提取 agent 回复
|
||
history = result.get("history", [])
|
||
agent_messages = []
|
||
for item in history:
|
||
if item.get("role") == "agent":
|
||
for part in item.get("parts", []):
|
||
if text := part.get("text"):
|
||
agent_messages.append(text)
|
||
|
||
if agent_messages:
|
||
return "\n\n".join(agent_messages), "success", False
|
||
|
||
# 3. 如果已完成但没有内容
|
||
if state == "completed":
|
||
return "✅ 任务已完成", "success", False
|
||
|
||
# 4. 默认返回
|
||
return "已收到响应", "success", False
|
||
|
||
except Exception as e:
|
||
logger.error(f"提取响应内容失败: {e}", exc_info=True)
|
||
return f"处理响应时出错: {str(e)}", "error", True
|
||
|
||
async def send_message(self, user_message: str, system_user_id: int, context_id: str) -> Dict[str, Any]:
|
||
"""
|
||
发送消息到 Conductor Agent
|
||
|
||
Args:
|
||
user_message: 用户消息
|
||
system_user_id: 系统用户ID
|
||
context_id: 会话上下文ID
|
||
|
||
Returns:
|
||
包含响应内容的字典
|
||
"""
|
||
# 先保存用户消息到数据库（保存原始用户消息，不带系统前缀）
|
||
await chat_history_service.save_message(
|
||
system_user_id=system_user_id,
|
||
context_id=context_id,
|
||
role="user",
|
||
content=user_message,
|
||
status="success"
|
||
)
|
||
logger.info(f"💾 已保存用户消息: context={context_id}, content={user_message[:50]}...")
|
||
|
||
try:
|
||
# 构建请求（注入用户ID）
|
||
request_data = self._build_a2a_request(user_message, system_user_id, context_id)
|
||
|
||
logger.info(f"📤 发送消息到 Conductor Agent: {user_message[:50]}...")
|
||
|
||
# 发送请求
|
||
async with httpx.AsyncClient(timeout=self.timeout) as client:
|
||
response = await client.post(
|
||
f"{self.base_url}/",
|
||
json=request_data,
|
||
headers={"Content-Type": "application/json"}
|
||
)
|
||
|
||
response.raise_for_status()
|
||
response_data = response.json()
|
||
|
||
logger.info(f"📥 收到 Conductor Agent 响应")
|
||
|
||
# 递增请求ID
|
||
self._request_id_counter += 1
|
||
|
||
# 提取内容（带错误检测）
|
||
content, msg_status, is_error = self._extract_content_from_response(response_data)
|
||
|
||
# 提取任务ID和上下文ID
|
||
result = response_data.get("result", {})
|
||
task_id = result.get("id")
|
||
response_context_id = result.get("contextId", context_id)
|
||
|
||
# 尝试从content中提取operation_record并保存
|
||
await self._save_operation_record_if_present(content, system_user_id, response_context_id)
|
||
|
||
# 保存agent响应到数据库
|
||
await chat_history_service.save_message(
|
||
system_user_id=system_user_id,
|
||
context_id=response_context_id,
|
||
role="agent",
|
||
content=content,
|
||
task_id=task_id,
|
||
status=msg_status,
|
||
error_message=content if is_error else None
|
||
)
|
||
logger.info(f"💾 已保存Agent回复: context={response_context_id}, content={content[:50]}...")
|
||
|
||
# 如果是错误，返回错误状态
|
||
if is_error:
|
||
return {
|
||
"content": content,
|
||
"context_id": response_context_id,
|
||
"task_id": task_id,
|
||
"status": msg_status # failed 或 error
|
||
}
|
||
|
||
return {
|
||
"content": content,
|
||
"context_id": response_context_id,
|
||
"task_id": task_id,
|
||
"status": "success"
|
||
}
|
||
|
||
except httpx.TimeoutException as e:
|
||
error_msg = "⏱️ 请求超时，Agent 可能正在处理复杂任务"
|
||
logger.error(error_msg)
|
||
|
||
# 保存错误消息到数据库
|
||
await chat_history_service.save_message(
|
||
system_user_id=system_user_id,
|
||
context_id=context_id,
|
||
role="system",
|
||
content=error_msg,
|
||
status="error",
|
||
error_message=str(e)
|
||
)
|
||
|
||
raise Exception(error_msg)
|
||
|
||
except httpx.ConnectError as e:
|
||
error_msg = f"🔌 无法连接到 Conductor Agent，请确保服务已启动 ({self.base_url})"
|
||
logger.error(error_msg)
|
||
|
||
# 保存错误消息到数据库
|
||
await chat_history_service.save_message(
|
||
system_user_id=system_user_id,
|
||
context_id=context_id,
|
||
role="system",
|
||
content=error_msg,
|
||
status="error",
|
||
error_message=str(e)
|
||
)
|
||
|
||
raise Exception(error_msg)
|
||
|
||
except httpx.HTTPStatusError as e:
|
||
error_msg = f"❌ Agent 返回错误: {e.response.status_code}"
|
||
logger.error(error_msg)
|
||
|
||
# 保存错误消息到数据库
|
||
await chat_history_service.save_message(
|
||
system_user_id=system_user_id,
|
||
context_id=context_id,
|
||
role="system",
|
||
content=error_msg,
|
||
status="error",
|
||
error_message=str(e)
|
||
)
|
||
|
||
raise Exception(error_msg)
|
||
|
||
except Exception as e:
|
||
error_msg = f"💥 发送消息失败: {str(e)}"
|
||
logger.error(error_msg, exc_info=True)
|
||
|
||
# 保存错误消息到数据库
|
||
await chat_history_service.save_message(
|
||
system_user_id=system_user_id,
|
||
context_id=context_id,
|
||
role="system",
|
||
content=error_msg,
|
||
status="error",
|
||
error_message=str(e)
|
||
)
|
||
|
||
raise Exception(error_msg)
|
||
|
||
async def test_connection(self) -> bool:
|
||
"""测试与 Conductor Agent 的连接"""
|
||
try:
|
||
async with httpx.AsyncClient(timeout=5.0) as client:
|
||
response = await client.get(
|
||
f"{self.base_url}/.well-known/agent-card.json"
|
||
)
|
||
response.raise_for_status()
|
||
data = response.json()
|
||
logger.info(f"✅ Conductor Agent 连接正常: {data.get('name', 'Unknown')}")
|
||
return True
|
||
except Exception as e:
|
||
logger.error(f"❌ Conductor Agent 连接失败: {e}")
|
||
return False
|
||
|
||
|
||
# 创建全局服务实例
|
||
conductor_service = ConductorService()
|
||
|