Simon
141 lines
5.2 KiB
Python
141 lines
5.2 KiB
Python
# =============================================================================
|
||
# 企微IT智能服务台 — 阶段5 自动化 Dify 客户端
|
||
# =============================================================================
|
||
# 说明：Dify 承担意图识别与 AI 编排。
|
||
# 1. 意图识别：根据员工消息判断命中哪个自动化场景
|
||
# （password_reset / software_install / virus_dispose / terminal_locate）
|
||
# 2. AI 编排（可选）：生成处置方案草案
|
||
#
|
||
# 认证：Dify 开放 API 使用 Bearer Token（API Key）。
|
||
# 配置：AUTOMATION_DIFY_BASE_URL / AUTOMATION_DIFY_API_KEY（来自 settings）。
|
||
#
|
||
# 容错：若 Dify 未配置或无结构化输出，detect_intent 走关键词兜底，
|
||
# 保证 P0 四个场景在无真实 Dify 环境下也能演示闭环。
|
||
# =============================================================================
|
||
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import logging
|
||
import re
|
||
from typing import Any, Dict, Optional
|
||
|
||
from app.config import settings
|
||
from app.integrations.base import BaseClient, BaseClientError
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class DifyClient(BaseClient):
|
||
"""Dify API 客户端（意图识别 / AI 编排）。"""
|
||
|
||
system_name = "dify"
|
||
|
||
def __init__(
|
||
self,
|
||
api_key: str,
|
||
base_url: str,
|
||
timeout: Optional[float] = None,
|
||
audit=None,
|
||
):
|
||
super().__init__(base_url=base_url, timeout=timeout, audit=audit)
|
||
self.api_key = api_key
|
||
|
||
def _headers(self) -> Dict[str, str]:
|
||
return {
|
||
"Authorization": f"Bearer {self.api_key}",
|
||
"Content-Type": "application/json",
|
||
}
|
||
|
||
async def chat_completions(
|
||
self,
|
||
query: str,
|
||
user: str = "automation",
|
||
conversation_id: str = "",
|
||
response_mode: str = "blocking",
|
||
) -> Dict[str, Any]:
|
||
"""调用 Dify Chat 补全（兼容 OpenAI Chat Completions 格式）。
|
||
|
||
Returns:
|
||
Dict: {"answer": str, "conversation_id": str, ...}
|
||
"""
|
||
body = {
|
||
"inputs": {},
|
||
"query": query,
|
||
"user": user,
|
||
"response_mode": response_mode,
|
||
}
|
||
if conversation_id:
|
||
body["conversation_id"] = conversation_id
|
||
return await self.request(
|
||
"POST", "/v1/chat-messages",
|
||
json_data=body, headers=self._headers(), event="dify.chat",
|
||
)
|
||
|
||
async def detect_intent(self, message: str, employee_id: str = "") -> Dict[str, Any]:
|
||
"""意图识别：把员工消息发给 Dify，期望返回结构化场景意图。
|
||
|
||
解析策略：
|
||
1. 优先尝试从 answer 中解析 JSON（scenario_key + confidence）
|
||
2. 失败则走关键词兜底（无 Dify 结构化输出时也能跑通 P0）
|
||
|
||
Returns:
|
||
Dict: {"scenario_key": str|None, "confidence": float, "raw": str, "error": str}
|
||
"""
|
||
try:
|
||
data = await self.chat_completions(query=message, user=employee_id or "automation")
|
||
answer = data.get("answer", "")
|
||
except BaseClientError as e:
|
||
logger.warning(f"Dify 意图识别失败，转关键词兜底: {e}")
|
||
fb = self._fallback_intent(message)
|
||
fb["error"] = str(e)
|
||
return fb
|
||
|
||
intent = self._parse_intent(answer)
|
||
if intent["scenario_key"] is None:
|
||
# 关键词兜底
|
||
fb = self._fallback_intent(message)
|
||
fb["raw"] = answer
|
||
return fb
|
||
return intent
|
||
|
||
@staticmethod
|
||
def _parse_intent(answer: str) -> Dict[str, Any]:
|
||
"""尝试从 Dify 返回中解析 JSON 意图。"""
|
||
try:
|
||
m = re.search(r"\{.*\}", answer, re.DOTALL)
|
||
if m:
|
||
obj = json.loads(m.group(0))
|
||
return {
|
||
"scenario_key": obj.get("scenario_key"),
|
||
"confidence": float(obj.get("confidence", 0.0)),
|
||
"raw": answer,
|
||
}
|
||
except Exception:
|
||
pass
|
||
return {"scenario_key": None, "confidence": 0.0, "raw": answer}
|
||
|
||
@staticmethod
|
||
def _fallback_intent(message: str) -> Dict[str, Any]:
|
||
"""关键词兜底意图识别（无 Dify 结构化输出时使用）。"""
|
||
text = (message or "").lower()
|
||
rules = [
|
||
(("密码", "重置", "password", "忘密码", "修改密码"), "password_reset"),
|
||
(("安装", "软件", "install", "software", "wps", "office", "下载"), "software_install"),
|
||
(("病毒", "杀毒", "virus", "勒索", "木马", "火绒", "huorong"), "virus_dispose"),
|
||
(("定位", "终端", "电脑在哪", "locate", "terminal", "找电脑"), "terminal_locate"),
|
||
]
|
||
for keywords, key in rules:
|
||
if any(k in text for k in keywords):
|
||
return {"scenario_key": key, "confidence": 0.75, "raw": ""}
|
||
return {"scenario_key": None, "confidence": 0.0, "raw": ""}
|
||
|
||
|
||
async def get_dify_client(audit=None) -> Optional[DifyClient]:
|
||
"""从 settings 构建 Dify 客户端；未配置返回 None。"""
|
||
base_url = settings.automation_dify_base_url
|
||
api_key = settings.automation_dify_api_key
|
||
if not base_url or not api_key:
|
||
return None
|
||
return DifyClient(api_key=api_key, base_url=base_url, audit=audit)
|