99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
|
"""
|
|||
|
|
聊天 API 路由
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
from uuid import uuid4
|
|||
|
|
from fastapi import APIRouter, HTTPException, status
|
|||
|
|
from fastapi.responses import JSONResponse
|
|||
|
|
|
|||
|
|
from models.chat import ChatRequest, ChatResponse, ErrorResponse
|
|||
|
|
from services.conductor_service import conductor_service
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
router = APIRouter()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post("/chat", response_model=ChatResponse)
|
|||
|
|
async def chat(request: ChatRequest):
|
|||
|
|
"""
|
|||
|
|
聊天接口
|
|||
|
|
|
|||
|
|
接收前端消息，转发给 Conductor Agent，返回 AI 回复
|
|||
|
|
同时将对话记录和设备操作记录存储到数据库
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
# 如果没有提供 context_id，生成一个新的
|
|||
|
|
context_id = request.context_id or f"session-{uuid4().hex[:16]}"
|
|||
|
|
|
|||
|
|
logger.info(f"💬 收到聊天请求: {request.query[:50]}... (user: {request.system_user_id}, context: {context_id})")
|
|||
|
|
|
|||
|
|
# 调用 Conductor Agent（传递用户ID）
|
|||
|
|
# conductor_service 已经处理了数据库存储逻辑
|
|||
|
|
result = await conductor_service.send_message(
|
|||
|
|
user_message=request.query,
|
|||
|
|
system_user_id=request.system_user_id,
|
|||
|
|
context_id=context_id
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 构建响应
|
|||
|
|
response = ChatResponse(
|
|||
|
|
content=result["content"],
|
|||
|
|
context_id=result["context_id"],
|
|||
|
|
task_id=result.get("task_id"),
|
|||
|
|
status=result["status"]
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 根据状态记录不同的日志
|
|||
|
|
if result["status"] == "success":
|
|||
|
|
logger.info(f"✅ 聊天响应成功: {len(result['content'])} 字符")
|
|||
|
|
else:
|
|||
|
|
logger.warning(f"⚠️ 聊天响应包含错误: status={result['status']}")
|
|||
|
|
|
|||
|
|
return response
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ 聊天请求失败: {e}", exc_info=True)
|
|||
|
|
# 注意：conductor_service 已经保存了错误到数据库
|
|||
|
|
raise HTTPException(
|
|||
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|||
|
|
detail={
|
|||
|
|
"error": "ChatError",
|
|||
|
|
"message": str(e)
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/chat/health")
|
|||
|
|
async def chat_health():
|
|||
|
|
"""检查聊天服务健康状态"""
|
|||
|
|
try:
|
|||
|
|
is_connected = await conductor_service.test_connection()
|
|||
|
|
|
|||
|
|
if is_connected:
|
|||
|
|
return {
|
|||
|
|
"status": "healthy",
|
|||
|
|
"conductor_agent": "connected",
|
|||
|
|
"url": conductor_service.base_url
|
|||
|
|
}
|
|||
|
|
else:
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
|
|||
|
|
content={
|
|||
|
|
"status": "unhealthy",
|
|||
|
|
"conductor_agent": "disconnected",
|
|||
|
|
"url": conductor_service.base_url
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"健康检查失败: {e}")
|
|||
|
|
return JSONResponse(
|
|||
|
|
status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
|
|||
|
|
content={
|
|||
|
|
"status": "error",
|
|||
|
|
"message": str(e)
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|