refactor: extract collection intent helpers from pydantic agent
This commit is contained in:
433
core/collection_intent_helpers.py
Normal file
433
core/collection_intent_helpers.py
Normal file
@@ -0,0 +1,433 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import random
|
||||
from typing import Any
|
||||
|
||||
|
||||
def classify_short_customer_text(text: str) -> str:
|
||||
"""
|
||||
短句分类器（状态机前置）：
|
||||
- finish_signal: 发图完成，可报价
|
||||
- progress_query: 追问进度/结果
|
||||
- ack: 简短确认
|
||||
- unknown: 未识别
|
||||
"""
|
||||
s = (text or "").strip()
|
||||
if not s:
|
||||
return "unknown"
|
||||
if len(s) > 8:
|
||||
return "unknown"
|
||||
|
||||
finish_kw = (
|
||||
"没了",
|
||||
"没有了",
|
||||
"就这",
|
||||
"就这张",
|
||||
"就这一张",
|
||||
"就这一个",
|
||||
"就一个",
|
||||
"先这些",
|
||||
"就这些",
|
||||
"发完了",
|
||||
"都发完了",
|
||||
)
|
||||
if any(k in s for k in finish_kw):
|
||||
return "finish_signal"
|
||||
|
||||
progress_kw = (
|
||||
"有吗",
|
||||
"有没",
|
||||
"有没有",
|
||||
"找到了吗",
|
||||
"找到了没",
|
||||
"没找到吗",
|
||||
"找到没",
|
||||
"找到没有",
|
||||
"进度",
|
||||
"结果",
|
||||
"多久好",
|
||||
"什么时候好",
|
||||
"好了没",
|
||||
"弄好了吗",
|
||||
"做了没",
|
||||
"高清",
|
||||
"发我",
|
||||
"重新发",
|
||||
"你重新发给我",
|
||||
)
|
||||
if any(k in s for k in progress_kw) or s in {"?", "？", "在吗", "人呢"}:
|
||||
return "progress_query"
|
||||
|
||||
ack_kw = ("嗯", "嗯嗯", "好", "好的", "行", "可以", "ok", "OK", "收到", "明白")
|
||||
if s in ack_kw:
|
||||
return "ack"
|
||||
return "unknown"
|
||||
|
||||
|
||||
def is_batch_finish_signal(text: str) -> bool:
|
||||
"""客户是否表达“图发完了，可以统一报价”"""
|
||||
if not text:
|
||||
return False
|
||||
if classify_short_customer_text(text) == "finish_signal":
|
||||
return True
|
||||
finish_keywords = [
|
||||
"发完了",
|
||||
"都发完了",
|
||||
"发齐了",
|
||||
"齐了",
|
||||
"先这些",
|
||||
"就这些",
|
||||
"全部",
|
||||
"一起报",
|
||||
"统一报价",
|
||||
"总共多少钱",
|
||||
"一共多少钱",
|
||||
"打包价",
|
||||
"总价",
|
||||
"报价吧",
|
||||
"报个总价",
|
||||
"给个总价",
|
||||
"没了",
|
||||
"没有了",
|
||||
"没图了",
|
||||
"就这",
|
||||
"就这张",
|
||||
"就这一张",
|
||||
"就这一个",
|
||||
"就一个",
|
||||
"先报吧",
|
||||
"报下价",
|
||||
"报个价",
|
||||
"可以报价了",
|
||||
"能报吗",
|
||||
]
|
||||
return any(k in text for k in finish_keywords)
|
||||
|
||||
|
||||
def is_cross_image_composite_intent(text: str) -> bool:
|
||||
"""
|
||||
识别多图跨图修改意图（A图元素放到B图）。
|
||||
例：A图的图案转到B图、这个图案放到另一张上。
|
||||
"""
|
||||
s = (text or "").strip()
|
||||
if not s:
|
||||
return False
|
||||
pair_marks = ("a图", "b图", "第一张", "第二张", "这张", "那张", "上一张", "另一张")
|
||||
op_kw = (
|
||||
"转到",
|
||||
"换到",
|
||||
"放到",
|
||||
"贴到",
|
||||
"移到",
|
||||
"套到",
|
||||
"合成",
|
||||
"融合",
|
||||
"替换到",
|
||||
"图案上去",
|
||||
"字放到",
|
||||
"元素放到",
|
||||
"logo放到",
|
||||
)
|
||||
return any(k in s.lower() for k in pair_marks) and any(k in s for k in op_kw)
|
||||
|
||||
|
||||
def is_batch_finish_intent(text: str, state: Any, has_incoming_urls: bool) -> bool:
|
||||
"""
|
||||
语义结束识别：
|
||||
- 显式口令：发完了/统一报价
|
||||
- 隐式意图：询价/砍价
|
||||
- 单图需求明确：如“这个门头上面的字做一下”可直接进入报价
|
||||
"""
|
||||
if not text:
|
||||
return False
|
||||
if is_batch_finish_signal(text):
|
||||
return True
|
||||
if has_incoming_urls:
|
||||
return False
|
||||
if not (getattr(state, "pending_image_urls", None) or []):
|
||||
return False
|
||||
|
||||
try:
|
||||
from utils.intent_analyzer import detect_intent_embedding, detect_intent_keywords
|
||||
|
||||
intent = detect_intent_embedding(text) or detect_intent_keywords(text)
|
||||
except Exception:
|
||||
intent = ""
|
||||
if intent in ("询价", "砍价"):
|
||||
return True
|
||||
|
||||
msg = (text or "").strip()
|
||||
if not msg:
|
||||
return False
|
||||
single_image_action_kw = (
|
||||
"做一下",
|
||||
"改一下",
|
||||
"处理一下",
|
||||
"就这张",
|
||||
"按这个做",
|
||||
"照这个做",
|
||||
"这个门头",
|
||||
"上面的字",
|
||||
"这个字",
|
||||
"这个图做",
|
||||
"能做吗",
|
||||
)
|
||||
multi_image_finish_kw = (
|
||||
"就这些",
|
||||
"就这几张",
|
||||
"按这几张",
|
||||
"这几张一起做",
|
||||
"一起做一下",
|
||||
"先按这些",
|
||||
"先按这几张",
|
||||
"直接报价",
|
||||
"现在报价",
|
||||
"看下报价",
|
||||
"先报个总价",
|
||||
"总价多少",
|
||||
"一起多少钱",
|
||||
"先做这几张",
|
||||
)
|
||||
hold_kw = ("还有", "再发", "先等", "稍后", "等会", "回头")
|
||||
image_count = len(getattr(state, "pending_image_urls", []) or [])
|
||||
if image_count == 1:
|
||||
if any(k in msg for k in single_image_action_kw) and not any(k in msg for k in hold_kw):
|
||||
return True
|
||||
elif image_count >= 2:
|
||||
if any(k in msg for k in multi_image_finish_kw) and not any(k in msg for k in hold_kw):
|
||||
return True
|
||||
if is_cross_image_composite_intent(msg) and not any(k in msg for k in hold_kw):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def is_related_image_followup_intent(text: str) -> bool:
|
||||
"""识别“新发的是上一张的截图/局部细节”的关联意图。"""
|
||||
s = (text or "").strip().lower()
|
||||
if not s:
|
||||
return False
|
||||
relation_kw = (
|
||||
"截图",
|
||||
"截屏",
|
||||
"局部",
|
||||
"细节",
|
||||
"放大",
|
||||
"裁剪",
|
||||
"同一张",
|
||||
"同一幅",
|
||||
"上一张",
|
||||
"上张",
|
||||
"前一张",
|
||||
"前面那张",
|
||||
"刚才那张",
|
||||
"这个是上面",
|
||||
"这个是那张",
|
||||
"补一张细节",
|
||||
"补个截图",
|
||||
)
|
||||
return any(k in s for k in relation_kw)
|
||||
|
||||
|
||||
def is_result_followup_query(text: str) -> bool:
|
||||
"""识别客户在找图流程中的结果/进度追问。"""
|
||||
if classify_short_customer_text(text) == "progress_query":
|
||||
return True
|
||||
s = (text or "").strip()
|
||||
if not s:
|
||||
return False
|
||||
followup_kw = (
|
||||
"找到了吗",
|
||||
"没找到吗",
|
||||
"找到没",
|
||||
"找到没有",
|
||||
"找到了没",
|
||||
"有吗",
|
||||
"有没",
|
||||
"有没有",
|
||||
"有结果吗",
|
||||
"结果呢",
|
||||
"进度",
|
||||
"多久好",
|
||||
"什么时候好",
|
||||
"好了没",
|
||||
"弄好了吗",
|
||||
"做了没",
|
||||
"你重新发",
|
||||
"重新发给我",
|
||||
"高清",
|
||||
"发我",
|
||||
)
|
||||
if any(k in s for k in followup_kw):
|
||||
return True
|
||||
return s in {"?", "？", "在吗", "人呢"}
|
||||
|
||||
|
||||
def build_collect_ack(count: int, related_followup: bool = False) -> str:
|
||||
if related_followup and count >= 2:
|
||||
related_templates = [
|
||||
"这张我收到了，看起来是上一张的截图/细节图，我按同一单一起处理。还有补充就继续发。",
|
||||
"收到，这张是关联补图我记上了（按同一需求处理）。你还有图就继续发。",
|
||||
"明白，这张是前图的局部截图，我会和前面那张一起算，不会分开漏掉。",
|
||||
]
|
||||
return random.choice(related_templates)
|
||||
if count <= 1:
|
||||
one_templates = [
|
||||
"这张收到啦，还有图就继续发，我一起给你看。",
|
||||
"图我看到了，后面还有就接着发，最后我一口价给你。",
|
||||
"收到这张了，你有其他图也发来，我统一帮你算。",
|
||||
"这张我先记上了，你那边还有的话接着发，我一起给你报。",
|
||||
"第1张收到，你继续发就行，发完我这边一次给你算清楚。",
|
||||
"这张没问题，我先收着。要是还有图，你直接连着发我就行。",
|
||||
"我先看到了这张，你后面还有就一起发来，我统一给你报价。",
|
||||
"这张图我已经记下了，后面有补充就继续甩过来哈。",
|
||||
]
|
||||
return random.choice(one_templates)
|
||||
templates = [
|
||||
"这几张我都收到了（现在{n}张）。还有的话继续发，我一起给你报。",
|
||||
"好嘞，先看到{n}张了。你可以继续发，或者直接说“就这些”我现在就报价。",
|
||||
"收到哈（共{n}张）。你还要补图就继续发，不补的话我现在也可以直接给价。",
|
||||
"我这边先收到了{n}张。你继续补图，或者直接说“按这些算”我就开始报。",
|
||||
"这波我已经记了{n}张，你要是还有就接着发，不补的话我立刻给总价。",
|
||||
"先看到{n}张图了，后面你看是继续发，还是直接让我现在报价都可以。",
|
||||
"好的，目前{n}张到位。你一句“就这些”，我马上给你打包价。",
|
||||
"图我都看到了（{n}张）。你还发我就继续收，不发我现在就给你报。",
|
||||
]
|
||||
return random.choice(templates).format(n=count)
|
||||
|
||||
|
||||
def build_collect_progress_reply(count: int) -> str:
|
||||
if count <= 1:
|
||||
templates = [
|
||||
"我这边在处理了，这张有结果我第一时间回你。",
|
||||
"在跟进中，这张一有进展我马上发你。",
|
||||
"这张我正在看，稍等我一会儿，结果出来就回你。",
|
||||
]
|
||||
return random.choice(templates)
|
||||
templates = [
|
||||
"我这边在按你这{n}张一起处理，有结果我立刻同步你。",
|
||||
"正在跟进这{n}张，出结果我第一时间发你，不会漏。",
|
||||
"进度在跑了（共{n}张），你稍等一下，我这边有结果马上回。",
|
||||
]
|
||||
return random.choice(templates).format(n=count)
|
||||
|
||||
|
||||
def build_collect_remind(count: int) -> str:
|
||||
if count <= 1:
|
||||
one_templates = [
|
||||
"这个要求我记住了。你还有图就继续发，不补图我就按这张给你报价。",
|
||||
"明白，这个需求我加上了。你继续发图也行，想直接报价也可以。",
|
||||
"我先记下这张。你如果是要我找图，不是做图，直接说一声，我按找图思路给你走。",
|
||||
"收到，这张我先按你的要求记好了。就做这一张的话，我现在直接给你报实价。",
|
||||
"你这要求我记下了，后面还有图就发，没有的话我现在直接算价。",
|
||||
"行，我按你这个要求来。继续补图也行，不补我就先报这张。",
|
||||
"这个点我懂了，你还要补图就接着发，不补我立刻给你报价。",
|
||||
"要求我已经加上了。你看是继续发，还是我现在直接报这张。",
|
||||
]
|
||||
return random.choice(one_templates)
|
||||
templates = [
|
||||
"需求我记下了（当前{n}张）。你继续补图，或者直接说“就这些”我现在报价。",
|
||||
"好，这个要求也加上了（现在{n}张）。不再补图的话我立刻给你打包价。",
|
||||
"收到（共{n}张）。你还发就继续，不发的话我现在就给总价。",
|
||||
"这个需求我加进去了（现在{n}张）。你继续发也行，直接报价也行。",
|
||||
"我这边都记好了（{n}张+需求）。你一句“先按这些算”，我马上报价。",
|
||||
"要求同步好了，目前{n}张。要补图继续发，不补图我现在就给你打包价。",
|
||||
"行，需求和图片我都收着了（{n}张）。你直接让我报价也可以。",
|
||||
"好的，这条需求也算进去了（共{n}张）。你看要不要我现在直接报。",
|
||||
]
|
||||
return random.choice(templates).format(n=count)
|
||||
|
||||
|
||||
def is_find_image_not_edit_conflict(text: str) -> bool:
|
||||
"""识别客户明确声明“要找图，不是做图”的冲突语义。"""
|
||||
s = (text or "").strip()
|
||||
if not s:
|
||||
return False
|
||||
find_kw = ("找图", "找原图", "找素材", "找同款")
|
||||
deny_edit_kw = ("不是让你做图", "不是做图", "不用做图", "不需要做图", "不是修图", "不用修图")
|
||||
return any(k in s for k in find_kw) and any(k in s for k in deny_edit_kw)
|
||||
|
||||
|
||||
def needs_clarification_in_collecting(text: str) -> bool:
|
||||
"""信息不足时先追问，不急着报价。"""
|
||||
s = (text or "").strip()
|
||||
if not s:
|
||||
return False
|
||||
short_non_vague_kw = (
|
||||
"？",
|
||||
"?",
|
||||
"没了",
|
||||
"没有了",
|
||||
"就这",
|
||||
"行",
|
||||
"好的",
|
||||
"ok",
|
||||
"报价",
|
||||
"找到了吗",
|
||||
"没找到吗",
|
||||
"找到没",
|
||||
"找到了没",
|
||||
"有吗",
|
||||
"有没",
|
||||
"有没有",
|
||||
"多久好",
|
||||
"什么时候好",
|
||||
"高清",
|
||||
)
|
||||
if len(s) <= 4:
|
||||
if any(k in s for k in short_non_vague_kw):
|
||||
return False
|
||||
return True
|
||||
vague_kw = (
|
||||
"这个也是",
|
||||
"一共几个图",
|
||||
"几个图",
|
||||
"啥意思",
|
||||
"没明白",
|
||||
"什么意思",
|
||||
"这个呢",
|
||||
"这个可以吗",
|
||||
"然后呢",
|
||||
"咋办",
|
||||
"怎么搞",
|
||||
)
|
||||
return any(k in s for k in vague_kw)
|
||||
|
||||
|
||||
def build_find_image_clarify_reply(state: Any) -> str:
|
||||
count = len(getattr(state, "pending_image_urls", []) or [])
|
||||
return (
|
||||
f"明白，你是要我帮你找图，不是做图。现在我这边先记了{count}张，"
|
||||
"你告诉我具体要找哪种：原图/同款/高清版，我按这个方向给你找。"
|
||||
)
|
||||
|
||||
|
||||
def build_not_understood_reply() -> str:
|
||||
"""信息不足时的澄清话术（随机）。"""
|
||||
templates = [
|
||||
"不好意思，不太懂你的意思，你再具体说下哈。",
|
||||
"抱歉我这边没完全理解，你可以换个说法再说一次吗？",
|
||||
"我有点没听明白，你是要找图还是要做图呀？",
|
||||
"不好意思我没抓到重点，你再补一句我就能接着处理。",
|
||||
"这句我理解得不太准，你再说具体一点我马上给你办。",
|
||||
"抱歉，这里我没太看懂。你是想让我找原图，还是按图处理？",
|
||||
"我这边还没完全明白你的意思，麻烦你再具体描述一下。",
|
||||
"不好意思，这条我没读懂，你再详细说一点我马上跟上。",
|
||||
]
|
||||
return random.choice(templates)
|
||||
|
||||
|
||||
def append_requirement(state: Any, text: str) -> None:
|
||||
"""追加需求并做去重/截断，减少上下文噪音。"""
|
||||
t = (text or "").strip()
|
||||
if not t:
|
||||
return
|
||||
t = t[:120]
|
||||
existing = list(getattr(state, "pending_requirements", []) or [])
|
||||
if existing and existing[-1] == t:
|
||||
return
|
||||
if t in existing[-5:]:
|
||||
return
|
||||
existing.append(t)
|
||||
if len(existing) > 20:
|
||||
existing = existing[-20:]
|
||||
state.pending_requirements = existing
|
||||
Reference in New Issue
Block a user