238 lines
8.7 KiB
Python
238 lines
8.7 KiB
Python
|
"""
|
|||
|
|
交互Agent调度器
|
|||
|
|
|
|||
|
|
接收企微用户消息，通过LLM识别意图，路由到对应处理器。
|
|||
|
|
|
|||
|
|
支持意图：
|
|||
|
|
- create_work_order: 创建工单（"帮我创建XX工单"）
|
|||
|
|
- query_alarm: 查询告警（"今天有多少告警"）
|
|||
|
|
- export_report: 导出报表（"导出本周告警报表"）
|
|||
|
|
- general_chat: 兜底闲聊
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import logging
|
|||
|
|
from datetime import datetime, timedelta, timezone
|
|||
|
|
from typing import Dict, Optional
|
|||
|
|
|
|||
|
|
from openai import AsyncOpenAI
|
|||
|
|
|
|||
|
|
from app.config import settings
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
INTENT_SYSTEM_PROMPT = """你是物业安防AI助手。根据用户消息识别意图，仅输出JSON。
|
|||
|
|
|
|||
|
|
可选意图：
|
|||
|
|
- create_work_order: 用户要创建工单或上报问题
|
|||
|
|
- query_alarm: 用户要查询告警数据或统计
|
|||
|
|
- export_report: 用户要导出报表或Excel
|
|||
|
|
- general_chat: 其他闲聊或无法识别
|
|||
|
|
|
|||
|
|
输出格式：{"intent":"...","params":{...}}
|
|||
|
|
|
|||
|
|
params说明：
|
|||
|
|
- create_work_order: {"title":"工单标题","description":"描述","priority":"low/medium/high/urgent"}
|
|||
|
|
- query_alarm: {"time_range":"today/week/month","alarm_type":"leave_post/intrusion/all"}
|
|||
|
|
- export_report: {"time_range":"today/week/month"}
|
|||
|
|
- general_chat: {"message":"友好的回复内容"}"""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AgentDispatcher:
|
|||
|
|
"""交互Agent调度器（单例）"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
self._client: Optional[AsyncOpenAI] = None
|
|||
|
|
self._enabled = False
|
|||
|
|
|
|||
|
|
def init(self, config):
|
|||
|
|
"""初始化Agent"""
|
|||
|
|
self._enabled = config.enabled and bool(config.llm_api_key)
|
|||
|
|
if self._enabled:
|
|||
|
|
self._client = AsyncOpenAI(
|
|||
|
|
api_key=config.llm_api_key,
|
|||
|
|
base_url=config.llm_base_url,
|
|||
|
|
)
|
|||
|
|
logger.info(f"交互Agent已启用: model={config.llm_model}")
|
|||
|
|
else:
|
|||
|
|
logger.info("交互Agent未启用（AGENT_ENABLED=false 或缺少 API Key）")
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def enabled(self) -> bool:
|
|||
|
|
return self._enabled
|
|||
|
|
|
|||
|
|
async def handle_message(self, user_id: str, content: str) -> str:
|
|||
|
|
"""处理用户消息，返回回复文本"""
|
|||
|
|
if not self._enabled:
|
|||
|
|
return "AI助手未启用，请联系管理员配置。"
|
|||
|
|
|
|||
|
|
# 1. 意图识别
|
|||
|
|
intent_result = await self._classify_intent(content)
|
|||
|
|
intent = intent_result.get("intent", "general_chat")
|
|||
|
|
params = intent_result.get("params", {})
|
|||
|
|
|
|||
|
|
logger.info(f"Agent意图识别: user={user_id}, intent={intent}, params={params}")
|
|||
|
|
|
|||
|
|
# 2. 路由到对应 handler
|
|||
|
|
handlers = {
|
|||
|
|
"create_work_order": self._handle_create_work_order,
|
|||
|
|
"query_alarm": self._handle_query_alarm,
|
|||
|
|
"export_report": self._handle_export_report,
|
|||
|
|
"general_chat": self._handle_general_chat,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
handler = handlers.get(intent, self._handle_general_chat)
|
|||
|
|
try:
|
|||
|
|
return await handler(user_id, params, content)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"Agent handler异常: intent={intent}, error={e}", exc_info=True)
|
|||
|
|
return "处理请求时出错，请稍后重试。"
|
|||
|
|
|
|||
|
|
async def _classify_intent(self, content: str) -> Dict:
|
|||
|
|
"""LLM意图分类"""
|
|||
|
|
try:
|
|||
|
|
resp = await self._client.chat.completions.create(
|
|||
|
|
model=settings.agent.llm_model,
|
|||
|
|
messages=[
|
|||
|
|
{"role": "system", "content": INTENT_SYSTEM_PROMPT},
|
|||
|
|
{"role": "user", "content": content},
|
|||
|
|
],
|
|||
|
|
timeout=settings.agent.llm_timeout,
|
|||
|
|
)
|
|||
|
|
text = resp.choices[0].message.content.strip()
|
|||
|
|
if "```" in text:
|
|||
|
|
text = text.split("```")[1]
|
|||
|
|
if text.startswith("json"):
|
|||
|
|
text = text[4:]
|
|||
|
|
text = text.strip()
|
|||
|
|
return json.loads(text)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"意图识别失败: {e}")
|
|||
|
|
return {"intent": "general_chat", "params": {"message": "抱歉，我暂时无法理解您的请求。"}}
|
|||
|
|
|
|||
|
|
async def _handle_create_work_order(self, user_id: str, params: Dict, raw: str) -> str:
|
|||
|
|
"""创建工单"""
|
|||
|
|
from app.services.work_order_service import get_work_order_service
|
|||
|
|
svc = get_work_order_service()
|
|||
|
|
|
|||
|
|
title = params.get("title", "")
|
|||
|
|
if not title:
|
|||
|
|
title = raw[:50]
|
|||
|
|
|
|||
|
|
order = svc.create_work_order(
|
|||
|
|
title=title,
|
|||
|
|
description=params.get("description", raw),
|
|||
|
|
priority=params.get("priority", "medium"),
|
|||
|
|
assignee_uid=user_id,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if order:
|
|||
|
|
priority_names = {"low": "低", "medium": "中", "high": "高", "urgent": "紧急"}
|
|||
|
|
p_name = priority_names.get(order.priority.value, "中")
|
|||
|
|
return (
|
|||
|
|
f"工单已创建\n"
|
|||
|
|
f"编号：{order.order_no}\n"
|
|||
|
|
f"标题：{order.title}\n"
|
|||
|
|
f"优先级：{p_name}\n"
|
|||
|
|
f"状态：待处理"
|
|||
|
|
)
|
|||
|
|
return "工单创建失败，请稍后重试"
|
|||
|
|
|
|||
|
|
async def _handle_query_alarm(self, user_id: str, params: Dict, raw: str) -> str:
|
|||
|
|
"""查询告警统计"""
|
|||
|
|
from app.services.alarm_event_service import get_alarm_event_service
|
|||
|
|
|
|||
|
|
svc = get_alarm_event_service()
|
|||
|
|
|
|||
|
|
# 解析时间范围
|
|||
|
|
time_range = params.get("time_range", "today")
|
|||
|
|
now = datetime.now(timezone.utc)
|
|||
|
|
if time_range == "week":
|
|||
|
|
start = now - timedelta(days=now.weekday())
|
|||
|
|
start = start.replace(hour=0, minute=0, second=0, microsecond=0)
|
|||
|
|
range_label = "本周"
|
|||
|
|
elif time_range == "month":
|
|||
|
|
start = now.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
|
|||
|
|
range_label = "本月"
|
|||
|
|
else:
|
|||
|
|
start = now.replace(hour=0, minute=0, second=0, microsecond=0)
|
|||
|
|
range_label = "今日"
|
|||
|
|
|
|||
|
|
alarm_type_filter = params.get("alarm_type")
|
|||
|
|
if alarm_type_filter == "all":
|
|||
|
|
alarm_type_filter = None
|
|||
|
|
|
|||
|
|
alarms, total = svc.get_alarms(
|
|||
|
|
alarm_type=alarm_type_filter,
|
|||
|
|
start_time=start,
|
|||
|
|
end_time=now,
|
|||
|
|
page=1,
|
|||
|
|
page_size=10000,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 按类型统计
|
|||
|
|
type_count = {}
|
|||
|
|
status_count = {"NEW": 0, "CONFIRMED": 0, "FALSE": 0, "CLOSED": 0}
|
|||
|
|
for a in alarms:
|
|||
|
|
type_count[a.alarm_type] = type_count.get(a.alarm_type, 0) + 1
|
|||
|
|
if a.alarm_status in status_count:
|
|||
|
|
status_count[a.alarm_status] += 1
|
|||
|
|
|
|||
|
|
type_names = {"leave_post": "人员离岗", "intrusion": "周界入侵"}
|
|||
|
|
type_lines = [f" {type_names.get(t, t)}: {c}条" for t, c in type_count.items()]
|
|||
|
|
|
|||
|
|
return (
|
|||
|
|
f"{range_label}告警统计\n"
|
|||
|
|
f"总计: {total}条\n"
|
|||
|
|
+ "\n".join(type_lines) + "\n"
|
|||
|
|
f"待处理: {status_count['NEW']}条\n"
|
|||
|
|
f"已处理: {status_count['CLOSED']}条\n"
|
|||
|
|
f"误报过滤: {status_count['FALSE']}条"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
async def _handle_export_report(self, user_id: str, params: Dict, raw: str) -> str:
|
|||
|
|
"""导出Excel报表"""
|
|||
|
|
from app.services.report_generator import generate_alarm_report
|
|||
|
|
from app.services.oss_storage import get_oss_storage
|
|||
|
|
|
|||
|
|
time_range = params.get("time_range", "week")
|
|||
|
|
result = generate_alarm_report(time_range=time_range)
|
|||
|
|
|
|||
|
|
if not result:
|
|||
|
|
range_names = {"today": "今日", "week": "本周", "month": "本月"}
|
|||
|
|
return f"{range_names.get(time_range, '今日')}暂无告警数据，无法生成报表。"
|
|||
|
|
|
|||
|
|
filename, file_bytes = result
|
|||
|
|
|
|||
|
|
# 上传到 COS 获取下载链接
|
|||
|
|
oss = get_oss_storage()
|
|||
|
|
try:
|
|||
|
|
object_key = oss.upload_file(
|
|||
|
|
file_bytes.read(),
|
|||
|
|
f"reports/{filename}",
|
|||
|
|
content_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
|||
|
|
)
|
|||
|
|
download_url = oss.get_presigned_url(object_key, expire=3600)
|
|||
|
|
return f"报表已生成\n文件：{filename}\n下载：{download_url}"
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"报表上传COS失败: {e}")
|
|||
|
|
return f"报表已生成（{filename}），但上传失败，请联系管理员。"
|
|||
|
|
|
|||
|
|
async def _handle_general_chat(self, user_id: str, params: Dict, raw: str) -> str:
|
|||
|
|
"""兜底回复"""
|
|||
|
|
msg = params.get("message", "")
|
|||
|
|
if msg:
|
|||
|
|
return msg
|
|||
|
|
return "您好，我是安防AI助手。可以帮您：\n1. 创建工单\n2. 查询告警统计\n3. 导出告警报表\n\n请直接描述您的需求。"
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 全局单例
|
|||
|
|
_agent_dispatcher: Optional[AgentDispatcher] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_agent_dispatcher() -> AgentDispatcher:
|
|||
|
|
global _agent_dispatcher
|
|||
|
|
if _agent_dispatcher is None:
|
|||
|
|
_agent_dispatcher = AgentDispatcher()
|
|||
|
|
return _agent_dispatcher
|