feat: 添加混合 Agent 路由架构
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m21s

This commit is contained in:
2026-04-26 17:37:57 +08:00
parent 7a769fab14
commit 87fb32a967
4 changed files with 707 additions and 0 deletions

View File

@@ -0,0 +1,193 @@
# backend/app/agent/intent_classifier.py
from enum import Enum
from dataclasses import dataclass
from typing import Optional, Dict, Any
import sys
import os
# 添加项目路径
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
from app.model_services.chat_services import get_chat_service
class IntentType(Enum):
"""意图类型枚举"""
KNOWLEDGE = "knowledge" # 知识查询 → RAG
REALTIME = "realtime" # 实时数据 → 工具
ACTION = "action" # 执行操作 → 工具
CHITCHAT = "chitchat" # 闲聊 → 直接回答
CLARIFY = "clarify" # 需要澄清 → 反问用户
MIXED = "mixed" # 复杂任务 → React 循环
UNKNOWN = "unknown" # 未知意图
@dataclass
class IntentResult:
"""意图识别结果"""
intent_type: IntentType
confidence: float
reasoning: str
metadata: Dict[str, Any] = None
class IntentClassifier:
"""意图分类器"""
def __init__(self):
self.llm = get_chat_service()
self._intent_examples = self._build_examples()
def _build_examples(self) -> str:
"""构建少样本示例"""
return """
<示例>
用户: "公司的报销政策是什么?"
意图: knowledge
推理: 用户询问公司内部政策,需要查询知识库
用户: "帮我查一下订单 12345 的状态"
意图: realtime
推理: 需要查询实时订单数据
用户: "帮我申请退款,订单号 67890"
意图: action
推理: 需要执行退款操作
用户: "今天天气怎么样?"
意图: realtime
推理: 需要查询实时天气数据
用户: "帮我写一份邮件给客户,查询订单状态,然后附上退款政策"
意图: mixed
推理: 需要查询订单、查询政策、生成邮件,多步骤任务
用户: "你好"
意图: chitchat
推理: 简单寒暄
用户: "我想查点东西..."
意图: clarify
推理: 用户没有说清楚要查什么
</示例>
"""
async def classify(self, user_input: str, context: Optional[str] = None) -> IntentResult:
"""
分类用户意图
Args:
user_input: 用户输入
context: 对话上下文(可选)
Returns:
IntentResult
"""
prompt = self._build_classification_prompt(user_input, context)
try:
response = await self.llm.ainvoke(prompt)
result = self._parse_response(response.content)
return result
except Exception as e:
print(f"Intent classification error: {e}")
# 降级策略:默认返回 mixed走 React 循环
return IntentResult(
intent_type=IntentType.MIXED,
confidence=0.5,
reasoning="分类失败,走通用路径"
)
def _build_classification_prompt(self, user_input: str, context: Optional[str]) -> str:
"""构建分类提示词"""
context_part = f"\n对话上下文:\n{context}" if context else ""
return f"""
你是一个专业的意图识别助手。请分析用户的输入,判断其意图类型。
可选意图类型:
- knowledge: 用户询问知识、政策、文档等,需要查询知识库
- realtime: 用户需要查询实时数据(订单状态、天气、股票等)
- action: 用户需要执行某项操作(退款、下单、发送邮件等)
- chitchat: 用户只是闲聊、打招呼,不需要工具或检索
- clarify: 用户的问题不明确,需要追问澄清
- mixed: 复杂任务,需要多步骤处理(同时需要检索+工具)
{self._intent_examples}
用户输入: {user_input}
{context_part}
请按以下格式输出纯JSON:
{{
"intent": "knowledge|realtime|action|chitchat|clarify|mixed",
"confidence": 0.85,
"reasoning": "简要说明为什么这个意图"
}}
"""
def _parse_response(self, response: str) -> IntentResult:
"""解析 LLM 响应"""
import json
import re
# 尝试提取 JSON
json_match = re.search(r'\{[\s\S]*\}', response)
if json_match:
try:
data = json.loads(json_match.group())
return IntentResult(
intent_type=IntentType(data['intent']),
confidence=float(data['confidence']),
reasoning=data['reasoning']
)
except:
pass
# 降级策略:关键词匹配
return self._fallback_classify(response)
def _fallback_classify(self, user_input: str) -> IntentResult:
"""关键词匹配降级策略"""
keywords = {
IntentType.KNOWLEDGE: ['政策', '文档', '规定', '手册', '指南', '什么是', '怎么'],
IntentType.REALTIME: ['订单', '状态', '天气', '股票', '价格', '库存'],
IntentType.ACTION: ['退款', '取消', '发送', '申请', '修改', '删除'],
IntentType.CHITCHAT: ['你好', 'hi', 'hello', '', '早上好', '晚上好'],
}
for intent_type, words in keywords.items():
if any(word in user_input.lower() for word in words):
return IntentResult(
intent_type=intent_type,
confidence=0.7,
reasoning=f"关键词匹配: {', '.join(words)}"
)
# 默认走混合路径
return IntentResult(
intent_type=IntentType.MIXED,
confidence=0.5,
reasoning="无法明确分类,走通用路径"
)
async def batch_classify(self, inputs: list[str]) -> list[IntentResult]:
"""批量分类(带缓存)"""
# 可以添加缓存逻辑
results = []
for inp in inputs:
results.append(await self.classify(inp))
return results
# 全局实例
_classifier: Optional[IntentClassifier] = None
def get_intent_classifier() -> IntentClassifier:
"""获取意图分类器实例"""
global _classifier
if _classifier is None:
_classifier = IntentClassifier()
return _classifier