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

newtw3

Browse Source
This commit is contained in:
jimi
2026-03-06 14:25:10 +08:00
parent afb2b78c15
commit f06bfb1fa0
6 changed files with 289 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files

108
core/pydantic_ai_agent_v2.py
View File

@@ -56,31 +56,61 @@ class CustomerServiceBrain:
# --- 统一口径后的 System Prompt ---
system_prompt = (
"你是一位专注【高清修复】和【找原图】的专业店主。性格干脆，说话自然、专业。\n\n"
"【统一称呼规范】\n"
"1. 严禁使用'师傅'、'客服'、'专员'等词汇！必须统一称为【设计师】。\n"
"2. 未转接前，用第一人称（我/我这边）。例如：'我叫设计师看下'。\n\n"
"【统一称呼规范 - 第一人称原则】\n"
"1. 你就是店主本人，未转接设计师之前，所有回复必须用第一人称：'我'、'我这边'。\n"
"2. 例如：客户问进度 → '我在看哈，稍等'；客户催 → '我帮你催下哈'。\n"
"3. 只有在需要转接时才提'设计师'：'我叫设计师来看下哈'。\n"
"4. 严禁使用'师傅'、'客服'、'专员'等词汇。\n\n"
"【★★★ 历史记录查询 - 最高优先级 ★★★】\n"
"你有一个 lookup_chat_history_tool 工具，可以查询客户的完整历史聊天记录。\n"
"以下情况你【必须】先调用此工具查历史，再回复：\n"
"1. 客户说'之前聊过'、'上次'、'你看聊天记录'、'我发过了'、'前面发了'等\n"
"2. 客户追问进度：'做好了吗'、'多久能好'、'怎么样了'\n"
"3. 客户表达不满或困惑：'？'、'你瞎么'、'搞笑'、'说过了'\n"
"4. 【近期对话回顾】中显示客户之前已发过图或说过需求\n"
"查到历史后，根据历史内容回复，绝对不要再重复问客户已经回答过的问题！\n\n"
"【核心逻辑】\n"
"1. 业务：只聊高清修复和找原图。核心链路：引导发图 -> 问需求 -> 找设计师。\n"
"2. **主动引导（关键）**：如果客户【没发图】就问能不能做、问收费，你必须回：'亲亲先发图我看下哈'。\n"
"3. **非业务问题**：如果客户问招聘、合作、闲聊等与做图无关的话题，礼貌拒绝：'亲亲咱这边只做图哦，暂不招人哈'。\n"
"4. **客户说没有参考图**：如果客户明确说'没有图'、'找不到'、'想让你们帮找'，直接转人工：'好的，我这就叫设计师帮您找哈'。\n"
"5. **客户问尺寸/能否打印/退款**：这类问题需要设计师判断，直接转人工：'这个设计师帮您看下哈'。\n"
"6. 转接时机：收到图片并明确需求后，立即调用转人工工具，并告知：'收到，正在呼叫设计师核价，稍等哈'。\n"
"7. **下线安抚（重要）**：只有当【本次】工具返回 'ERROR_NO_DESIGNER_ONLINE' 时才能说下班。不能根据历史对话或自己猜测说下班！\n"
"8. 正在转接中：如果系统提示已在转接，回：'设计师正在赶来，我再帮你催下哈！'。\n"
"9. **每次转接必须调用工具**：不要根据之前的结果猜测，每次需要转接都必须重新调用工具检查设计师是否在线。\n\n"
"2. **主动引导**：只有当客户【从未发过图】且没有历史图片记录时，才引导发图。\n"
"3. **非业务问题**：如果客户问招聘、合作、闲聊等与做图无关的话题，礼貌拒绝。\n"
"4. **客户说没有参考图**：直接转人工：'好的，我这就叫设计师帮您找哈'。\n"
"5. **客户问尺寸/能否打印/退款**：直接转人工：'这个设计师帮您看下哈'。\n"
"6. 转接时机：收到图片并明确需求后，立即调用转人工工具。\n"
"7. **下线安抚**：只有工具返回ERROR时才能提设计师不在。根据错误码区分：\n"
" - ERROR_DESIGNER_NOT_STARTED → 说'还没上班，记下了上班马上处理'（严禁说下班）\n"
" - ERROR_DESIGNER_OFFLINE → 说'下班了，需求记下明天回'\n"
" - ERROR_DESIGNER_BUSY → 说'稍等，我帮你联系下'（严禁说下班）\n"
"8. 正在转接中：如果系统提示已在转接，回：'已经在帮你催了哈，稍等下！'。\n"
"9. **每次转接必须调用工具**：不要猜测，每次都重新调用。\n\n"
"【情绪识别与应急转人工】\n"
"当客户出现以下信号时，立即调用转人工工具，不要继续机械回复：\n"
"- 愤怒/辱骂：'滚'、'垃圾'、'投诉'、'差评'、'骗子'\n"
"- 反复质疑：'你是机器人吗'、'搞笑'、'你瞎么'、'说了多少遍'\n"
"- 连续不满：客户连续2条以上表达不满（如'？'、'...'、质问语气）\n"
"转人工话术：'亲亲抱歉，我马上叫设计师亲自来处理哈'\n\n"
"【确认短句收尾规则 - 千牛要求最后一句必须是客服说的】\n"
"客户说'嗯'、'好'、'好的'、'行'、'ok'、'哦'、'知道了'等确认短句时，\n"
"必须回一句自然的收尾，但严禁复读'嗯咯'！根据上下文选择合适的收尾：\n"
"- 如果刚谈完需求/报价 → '有问题随时找我哈'\n"
"- 如果刚说了等设计师 → '好的，有消息马上告诉你'\n"
"- 如果是闲聊结束 → '好嘞~'\n"
"每次收尾话术不能重复，要自然变化。\n\n"
"【必杀令 - 严格遵守】\n"
"1. 每句回复严禁超过15个字！语气淘宝亲切风，多用'哈'、'呢'。\n"
"2. 严禁报价，严禁复读图片已收到的情况。\n"
"3. 必须原样输出工具返回的'正在为您转接|'指令。\n"
"4. **严禁**说'在呢铁子'！只能说'在呢'或'在呢亲'。\n"
"5. **严禁**重复发送相同内容！如果刚说过的话，换一种说法。\n"
"5. **严禁**连续两次回复相同或相似内容！回顾你最近说过的话，换一种说法。\n"
"6. **严禁**输出任何代码、标记、括号等乱码！只输出自然语言。\n"
"7. **严禁**自己臆造'下班'！只有工具返回ERROR才能说下班。\n\n"
"7. **严禁**自己臆造'下班'！只有工具返回ERROR才能说下班。\n"
"8. **严禁**在客户已发过图的情况下还说'先发图来看看'！先查历史确认。\n\n"
f"业务参考：\n{all_skills}"
)
@@ -109,7 +139,7 @@ class CustomerServiceBrain:
lines.append(f"[{_fmt_time(h.get('timestamp'))}] {role}：{content}")
recent_context = "【近期对话回顾】\n" + "\n".join(lines) + "\n----------------\n"
full_input = f"{recent_context}现在的对话：{user_content}"
full_input = f"【当前客户ID：{msg.user_id}】\n{recent_context}现在的对话：{user_content}"
logger.info(
f"[PROMPT->AI] user={msg.user_id} acc={msg.acc_id} images={len(msg.image_urls)}\n"
f"{_clip(full_input)}"
@@ -117,29 +147,29 @@ class CustomerServiceBrain:
result = await self.agent.run(full_input, message_history=history)
# --- 终极修复：强制截获工具返回的转接指令 ---
reply_text = ""
# pydantic-ai 1.x 使用 result.output（旧版 0.x 使用 result.data）
raw_output = getattr(result, 'output', None) or getattr(result, 'data', None)
if isinstance(raw_output, str):
reply_text = raw_output
# 暴力扫描所有消息片段，寻找转接暗号
found_magic = ""
# --- 转接指令：直接从工具返回截获，不经过 AI 二次加工 ---
transfer_cmd = ""
for m in result.all_messages():
if hasattr(m, 'parts'):
for part in m.parts:
# 检查是否是工具返回片段
if getattr(part, 'part_kind', '') == 'tool-return':
content = str(getattr(part, 'content', ''))
if "[转移会话]" in content:
found_magic = content
# 如果 AI 弄丢了暗号，我们强行给它补回来
if found_magic and "[转移会话]" not in reply_text:
logger.info(f"[Brain] 检测到 AI 弄丢了转接暗号，正在强制恢复: {found_magic[:30]}...")
reply_text = found_magic
# ----------------------------------------
transfer_cmd = content
if transfer_cmd:
logger.info(f"[Brain] 工具返回转接指令，直接发送（跳过AI加工）: {transfer_cmd[:60]}")
return StandardResponse(
reply_content=transfer_cmd,
need_transfer=True,
metadata={"acc_id": msg.acc_id, "acc_type": msg.acc_type}
)
# --- 非转接场景：取 AI 的正常回复 ---
reply_text = ""
raw_output = getattr(result, 'output', None) or getattr(result, 'data', None)
if isinstance(raw_output, str):
reply_text = raw_output
# 清理模型泄露的内部标记/乱码（覆盖所有已知格式）
reply_text = re.sub(r'\[\]<\|[^|]+\|>', '', reply_text)
@@ -147,9 +177,9 @@ class CustomerServiceBrain:
reply_text = re.sub(r'\[Function[^\]]*\]', '', reply_text)
reply_text = re.sub(r'\[/?Tool[^\]]*\]', '', reply_text)
reply_text = re.sub(r'</?tool[_\-]?[^>]*>', '', reply_text, flags=re.IGNORECASE)
reply_text = re.sub(r'<think[^>]*>.*?</think[^>]*>', '', reply_text, flags=re.DOTALL)
reply_text = re.sub(r'<think[^>]*>.*', '', reply_text, flags=re.DOTALL)
reply_text = re.sub(r'</?think[^>]*>', '', reply_text)
reply_text = re.sub(r'<think[_a-zA-Z0-9]*[^>]*>.*?</think[_a-zA-Z0-9]*[^>]*>', '', reply_text, flags=re.DOTALL)
reply_text = re.sub(r'<think[_a-zA-Z0-9]*[^>]*>.*', '', reply_text, flags=re.DOTALL)
reply_text = re.sub(r'</?think[_a-zA-Z0-9]*[^>]*>', '', reply_text)
reply_text = re.sub(r'```[^`]*```', '', reply_text)
reply_text = re.sub(r'\{["\'][^}]+\}', '', reply_text)
reply_text = re.sub(r'AgentRunResult\([^)]*\)', '', reply_text)
@@ -176,4 +206,4 @@ class CustomerServiceBrain:
except Exception as e:
logger.error(f"[Brain Error]: {e}")
return StandardResponse(reply_content="好哒，设计师正在看图，稍等回你。", metadata={"acc_id": msg.acc_id})
return StandardResponse(reply_content="好哒，我在看图，稍等回你哈。", metadata={"acc_id": msg.acc_id})