zuowei/tw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: improve routing logs and tuhui integration

Browse Source
This commit is contained in:
jimi
2026-03-08 17:34:56 +08:00
parent 39de916b89
commit 3a78eb304a
4 changed files with 214 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
core/pydantic_ai_agent_v2.py
View File

@@ -39,6 +39,36 @@ _HISTORY_LEAK_PATTERNS = [
r'(状态|金额|数量)[:：].*(状态|金额|数量)[:：]', # 状态:xxx 金额:xxx 连续出现
]
_FIND_ORIGINAL_INTENT_KEYWORDS = (
"找图",
"找原图",
"原图",
"素材",
"大图",
"源图",
)
_FIND_ORIGINAL_QUESTION_KEYWORDS = (
"有吗",
"有没",
"有没有",
"能找吗",
"找得到吗",
"能不能找到",
"能找到吗",
)
_REPAIR_INTENT_KEYWORDS = (
"修复",
"高清修复",
"高清",
"清晰",
"清楚",
"变清晰",
"修清楚",
"放大清晰",
)
def _clip(text: str, limit: int = 1200) -> str:
if text is None:
@@ -93,6 +123,31 @@ def _sanitize_reply_text(reply_text: str) -> str:
return text.strip()
def _normalize_text(text: Any) -> str:
return str(text or "").strip().lower()
def _infer_image_intent(current_text: str, history: Optional[List[dict]] = None) -> str:
text = _normalize_text(current_text)
recent_user_text = "\n".join(
_normalize_text(h.get("content", ""))
for h in (history or [])[-6:]
if h.get("role") == "user"
)
combined = f"{recent_user_text}\n{text}"
if any(k in combined for k in _REPAIR_INTENT_KEYWORDS):
return "repair"
if any(k in combined for k in _FIND_ORIGINAL_INTENT_KEYWORDS):
return "find_original"
if any(k in text for k in _FIND_ORIGINAL_QUESTION_KEYWORDS):
return "find_original"
return ""
class CustomerServiceBrain:
"""
重构后的单一 Agent 大脑：
@@ -147,9 +202,10 @@ class CustomerServiceBrain:
"3. **非业务问题**：如果客户问招聘、合作、闲聊等与做图无关的话题，礼貌拒绝。\n"
"4. **客户说没有参考图**：直接转人工：'好的，我这就叫设计师帮您找哈'。\n"
"5. **客户问尺寸/能否打印/退款**：直接转人工：'这个设计师帮您看下哈'。\n"
"6. **转接时机（严格两步）**：必须同时满足【有图】+【客户明确说了要找原图/修复/具体要求】才能转接。\n"
" 客户只发了图但没说需求 → 必须先问'亲亲这张是找原图还是修复哈？'\n"
" 客户说了'有吗'、'能找吗' → 这不算明确需求，要追问'是要找原图还是高清修复呢？'\n"
"6. **转接时机（严格两步）**：必须同时满足【有图】+【客户明确或可直接判断的需求】才能转接。\n"
" 客户只发了图但没说需求 → 先问'亲亲这张是找原图还是修复哈？'\n"
" 客户说了'有吗'、'能找吗'、'找图'、'找原图'、'有大图吗' → 直接按【找原图】意图处理，不要重复追问。\n"
" 客户说了'修复'、'高清'、'清晰点'、'放大清晰' → 直接按【高清修复】意图处理，不要重复追问。\n"
"7. **下线安抚**：只有工具返回ERROR时才能提设计师不在。根据错误码区分：\n"
" - ERROR_DESIGNER_NOT_STARTED → 说'还没上班，记下了上班马上处理'（严禁说下班）\n"
" - ERROR_DESIGNER_OFFLINE → 说'下班了，需求记下明天回'\n"
@@ -200,11 +256,27 @@ class CustomerServiceBrain:
image_count = max(len(msg.image_urls), 1)
if user_content.startswith("【系统：已收到图片消息"):
user_content = ""
user_content = (
f"【系统通知：客户已发送 {image_count} 张图片，图已收到不要再让客户发图。"
f"你现在必须先问客户：这张是找原图还是高清修复？有什么具体要求？"
f"等客户明确回答后才能转接，严禁跳过问需求直接转接！】\n{user_content}"
)
inferred_intent = _infer_image_intent(user_content, history)
if inferred_intent == "find_original":
logger.info(f"[Brain] 已根据客户表述推断为找原图意图: user={msg.user_id}")
user_content = (
f"【系统通知：客户已发送 {image_count} 张图片，图已收到不要再让客户发图。"
f"系统判断客户当前意图是【找原图】；像'有吗'、'能找吗'、'找图'都算找原图意图。"
f"不要再追问'找原图还是高清修复'，直接按找原图流程继续；如果信息足够就直接转接。】\n{user_content}"
)
elif inferred_intent == "repair":
logger.info(f"[Brain] 已根据客户表述推断为高清修复意图: user={msg.user_id}")
user_content = (
f"【系统通知：客户已发送 {image_count} 张图片，图已收到不要再让客户发图。"
f"系统判断客户当前意图是【高清修复】；像'修复'、'高清'、'清晰点'都算修复意图。"
f"不要再追问'找原图还是高清修复'，直接按高清修复流程继续；如果信息足够就直接转接。】\n{user_content}"
)
else:
user_content = (
f"【系统通知：客户已发送 {image_count} 张图片，图已收到不要再让客户发图。"
f"你现在必须先问客户：这张是找原图还是高清修复？有什么具体要求？"
f"等客户明确回答后才能转接，严禁跳过问需求直接转接！】\n{user_content}"
)
recent_context = ""
if history:

53
core/websocket_logger_setup.py
View File

@@ -1,6 +1,8 @@
import logging
import os
import subprocess
from datetime import datetime
from pathlib import Path
class _AnsiColorFormatter(logging.Formatter):
@@ -56,20 +58,67 @@ class _AnsiColorFormatter(logging.Formatter):
return f"{color}{msg}{self.RESET}"
_APP_VERSION = None
_LOG_RECORD_FACTORY_INSTALLED = False
def get_app_log_version() -> str:
global _APP_VERSION
if _APP_VERSION:
return _APP_VERSION
env_version = str(os.getenv("APP_VERSION", "")).strip()
if env_version:
_APP_VERSION = env_version
return _APP_VERSION
try:
repo_root = Path(__file__).resolve().parent.parent
git_version = subprocess.check_output(
["git", "-C", str(repo_root), "rev-parse", "--short", "HEAD"],
stderr=subprocess.DEVNULL,
text=True,
).strip()
except Exception:
git_version = ""
_APP_VERSION = git_version or "dev"
os.environ.setdefault("APP_VERSION", _APP_VERSION)
return _APP_VERSION
def install_log_record_factory():
global _LOG_RECORD_FACTORY_INSTALLED
if _LOG_RECORD_FACTORY_INSTALLED:
return
version = get_app_log_version()
old_factory = logging.getLogRecordFactory()
def record_factory(*args, **kwargs):
record = old_factory(*args, **kwargs)
record.app_version = getattr(record, "app_version", version)
return record
logging.setLogRecordFactory(record_factory)
_LOG_RECORD_FACTORY_INSTALLED = True
def setup_logger():
from logging.handlers import RotatingFileHandler
from config.config import LOG_DIR, LOG_MAX_BYTES, LOG_BACKUP_COUNT
install_log_record_factory()
logger = logging.getLogger("cs_agent")
if getattr(logger, "_cs_logger_configured", False):
return logger
logger.setLevel(logging.INFO)
logger.propagate = False
fmt = logging.Formatter("[%(asctime)s] %(message)s", datefmt="%H:%M:%S")
fmt = logging.Formatter("[v%(app_version)s][%(asctime)s] %(message)s", datefmt="%H:%M:%S")
use_color = (os.getenv("LOG_COLOR", "1").lower() in ("1", "true", "yes")) and not bool(os.getenv("NO_COLOR"))
ch = logging.StreamHandler()
ch.setFormatter(_AnsiColorFormatter("[%(asctime)s] %(message)s", datefmt="%H:%M:%S", use_color=use_color))
ch.setFormatter(_AnsiColorFormatter("[v%(app_version)s][%(asctime)s] %(message)s", datefmt="%H:%M:%S", use_color=use_color))
logger.addHandler(ch)
LOG_DIR.mkdir(exist_ok=True)