192 lines
8.4 KiB
Python
192 lines
8.4 KiB
Python
from __future__ import annotations
|
||
|
||
import re
|
||
from typing import Any, Callable
|
||
|
||
|
||
def split_customer_text(msg: str) -> tuple[str, str]:
|
||
"""
|
||
把混合消息拆分为（客户真实文字, 系统订单块）。
|
||
平台有时把客户文字和系统订单通知拼在同一条消息里。
|
||
"""
|
||
order_marker = re.search(r"\[系统订单信息\]|\[系统通知\]", msg or "")
|
||
if order_marker:
|
||
customer_text = (msg or "")[: order_marker.start()].strip()
|
||
order_block = (msg or "")[order_marker.start() :].strip()
|
||
else:
|
||
customer_text = (msg or "").strip()
|
||
order_block = ""
|
||
return customer_text, order_block
|
||
|
||
|
||
def build_prompt(
|
||
*,
|
||
message: Any,
|
||
state: Any,
|
||
extract_image_url: Callable[[str], str],
|
||
shop_type_resolver: Callable[[str, str], str],
|
||
shop_persona_resolver: Callable[[str, str], str],
|
||
parse_order_info: Callable[[str], dict[str, str]],
|
||
build_order_instruction: Callable[[str, str], str],
|
||
) -> str:
|
||
"""构建提示词。"""
|
||
msg_content = message.msg
|
||
stage_info = f"【当前阶段】{state.stage}"
|
||
|
||
customer_text, order_block = split_customer_text(msg_content)
|
||
has_order = bool(order_block)
|
||
|
||
if has_order:
|
||
order = parse_order_info(order_block)
|
||
if order.get("order_id"):
|
||
state.last_order_id = order["order_id"]
|
||
stage_info += f"\n【订单号】{order['order_id']}"
|
||
if order.get("order_status"):
|
||
state.order_status = order["order_status"]
|
||
stage_info += f"\n【订单状态】{order['order_status']}"
|
||
if order.get("pay_status"):
|
||
stage_info += f"\n【支付状态】{order['pay_status']}"
|
||
if order.get("amount"):
|
||
stage_info += f"\n【订单金额】{order['amount']}元"
|
||
if order.get("quantity"):
|
||
stage_info += f"\n【数量】{order['quantity']}件"
|
||
if order.get("order_time"):
|
||
stage_info += f"\n【下单时间】{order['order_time']}"
|
||
if order.get("buyer_note"):
|
||
stage_info += f"\n【买家备注】{order['buyer_note']}"
|
||
|
||
if state.discount_count > 0:
|
||
stage_info += f"\n【客户压价次数】{state.discount_count}"
|
||
|
||
shop_type = shop_type_resolver(message.acc_id or "", message.goods_name or "")
|
||
shop_persona = shop_persona_resolver(message.acc_id or "", message.goods_name or "")
|
||
shop_hint = ""
|
||
try:
|
||
from config.config import CONFIG_DIR
|
||
import json
|
||
|
||
cfg_path = CONFIG_DIR / "shop_prompts.json"
|
||
if cfg_path.exists():
|
||
with open(cfg_path, "r", encoding="utf-8") as f:
|
||
cfg = json.load(f)
|
||
hints = cfg.get("type_hints", {})
|
||
shop_hint = hints.get(shop_type, "")
|
||
if not shop_hint and message.acc_id:
|
||
sh = cfg.get("shops", {}).get(message.acc_id, {})
|
||
shop_hint = sh.get("hint", "")
|
||
except Exception:
|
||
pass
|
||
|
||
prompt = f"""收到新消息：
|
||
{stage_info}
|
||
|
||
发送者: {message.from_name} ({message.from_id})
|
||
"""
|
||
if message.goods_name:
|
||
prompt += f"商品名称: {message.goods_name}\n"
|
||
if shop_hint:
|
||
prompt += f"\n{shop_hint}\n"
|
||
if shop_persona:
|
||
prompt += f"\n【店铺人设】{shop_persona}\n"
|
||
|
||
order_paid = False
|
||
order_unpaid = False
|
||
if has_order:
|
||
order = parse_order_info(order_block)
|
||
paid_kws = ["等待发货", "已付款", "付款成功", "买家已付款"]
|
||
unpaid_kws = ["等待买家付款", "待付款", "未付款"]
|
||
ps = order.get("pay_status", "")
|
||
os_ = order.get("order_status", "")
|
||
if any(kw in ps or kw in os_ for kw in paid_kws):
|
||
order_paid = True
|
||
elif any(kw in ps or kw in os_ for kw in unpaid_kws):
|
||
order_unpaid = True
|
||
|
||
progress_keywords = [
|
||
"安排了吗",
|
||
"安排好了吗",
|
||
"好了吗",
|
||
"做了吗",
|
||
"做好了吗",
|
||
"弄好了吗",
|
||
"好了没",
|
||
"做了没",
|
||
"什么时候好",
|
||
"多久好",
|
||
"进度",
|
||
"催一下",
|
||
"快点",
|
||
"什么时候能好",
|
||
"做完了吗",
|
||
]
|
||
|
||
if customer_text:
|
||
prompt += f"\n客户说：{customer_text}\n"
|
||
image_url = extract_image_url(customer_text)
|
||
price_keywords = ["多少钱", "多少", "价格", "几块", "怎么收费", "报个价"]
|
||
size_keywords = [
|
||
"尺寸",
|
||
"比例",
|
||
"宽",
|
||
"高",
|
||
"米",
|
||
"厘米",
|
||
"mm",
|
||
"cm",
|
||
"横版",
|
||
"竖版",
|
||
"2米",
|
||
"3米",
|
||
"改成",
|
||
"做成",
|
||
]
|
||
has_size_change = any(kw in customer_text.lower() for kw in [k.lower() for k in size_keywords])
|
||
|
||
if shop_type == "gemini_api":
|
||
prompt += "\n【Gemini API 店铺】客户问账号/pro/续费/套餐等，按 API 客服自然回复，不要求发图。"
|
||
elif image_url:
|
||
prompt += "\n客户在继续发图阶段：先确认“已收图”，并引导客户把图和要求一次发完；等客户明确“发完了/统一报价”后再统一报价。"
|
||
elif any(kw in customer_text for kw in price_keywords):
|
||
last_url = extract_image_url(msg_content)
|
||
if last_url:
|
||
prompt += "\n客户在询问价格：若客户已确认发完，则给总报价；若还在发图，先引导发完后统一报价。"
|
||
else:
|
||
prompt += "\n客户在询问价格但未发图：先简短承接（如“在看呢/收到”），不要机械连发；再自然引导对方发图。"
|
||
if has_size_change:
|
||
prompt += (
|
||
"\n⚠️ 尺寸改动场景：优先判断图片主体是否会被拉伸变形，"
|
||
"不是只看整张图宽高比。若会变形，要先提示“需要补图/扩边”，再给报价。"
|
||
)
|
||
elif has_size_change:
|
||
prompt += (
|
||
"\n客户在改尺寸/改比例：先按主体比例判断是否会变形，"
|
||
"不是只看整图比例。若目标尺寸会拉伸主体，先明确说明要补图（如上下补图/扩边）再报价。"
|
||
)
|
||
elif any(kw in customer_text for kw in progress_keywords):
|
||
if order_unpaid:
|
||
prompt += "\n⚠️【订单未付款】客户问安排进度，但订单还未付款。自然告知拍下付款后马上安排即可。"
|
||
elif order_paid:
|
||
prompt += "\n客户催单，订单已付款，自然回复在做了/快了之类。"
|
||
else:
|
||
prompt += "\n客户催单，查询当前处理状态后自然回复。"
|
||
elif any(kw in customer_text for kw in ["贵", "有点贵", "太贵", "算了", "便宜点", "少点", "打折", "贵哦"]):
|
||
prompt += "\n⚠️【客户嫌贵】客户已表达价格顾虑或要放弃。禁止追问「什么问题」「说清楚点」。\n→ 第一步：直接让价一次（如原价 20→15），话术自然。\n→ 第二步：如果让价后客户还是嫌贵，引导客户自己去注册星绘平台：\n - 每个话术都必须带网址 xinhui.cloud\n - 强调「自己去弄」「自己操作」「自己下单」\n - 示例：「嫌贵就去 xinhui.cloud 注册个号，自己有余额可以弄」「xinhui.cloud 去注册下，新用户送体验金，自己弄便宜」\n - 语气直接，让客户自己去操作」"
|
||
elif any(kw in customer_text for kw in ["擦边", "黄色", "色情", "大尺度", "性感图", "露点", "半裸"]):
|
||
prompt += "\n⚠️【拒绝】客户询问擦边/黄色/敏感内容。直接拒绝，不接单，不说「发图来看看」。自然回复如：这类不做/不接/做不了。"
|
||
else:
|
||
prompt += "\n根据客户说的内容自然回应，像真人聊天，不要套模板。"
|
||
|
||
if has_order:
|
||
order = parse_order_info(order_block)
|
||
order_instruction = build_order_instruction(order.get("pay_status", ""), order.get("order_status", ""))
|
||
if customer_text:
|
||
if not order_unpaid:
|
||
prompt += f"\n\n【背景参考-订单通知】{order_instruction}"
|
||
else:
|
||
prompt += f"\n\n{order_instruction}"
|
||
|
||
if not customer_text and not has_order:
|
||
prompt += f"\n消息内容: {msg_content}\n请按工作流规则回复。"
|
||
|
||
return prompt
|