feat: 交互Agent + VLM优化 + 企微演示模式
- 新增交互Agent调度器（意图识别 + 工单/查询/报表/闲聊4个Handler） - 新增工单服务、Excel报表生成器、企微消息加解密模块 - VLM提示词优化（角色设定、≤25字描述、布尔值优先输出） - VLM降级策略（入侵默认放行、离岗默认拦截） - 企微演示模式（WECHAT_TEST_UIDS兜底 + SERVICE_BASE_URL修复） - 新增Agent回调路由和测试接口 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
237
app/services/agent_dispatcher.py
Normal file
237
app/services/agent_dispatcher.py
Normal file
@@ -0,0 +1,237 @@
|
||||
"""
|
||||
交互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
|
||||
Reference in New Issue
Block a user