jimi
5e9590030d
Some checks failed
Pre-commit / run (ubuntu-latest) (push) Has been cancelled
Deploy Sphinx documentation to Pages / build_en (ubuntu-latest, 3.10) (push) Has been cancelled
Deploy Sphinx documentation to Pages / build_zh (ubuntu-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.12) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.12) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.12) (push) Has been cancelled
107 lines
4.3 KiB
Python
107 lines
4.3 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from .agents import AfterSalesAgent, PreSalesAgent, QuoteAgent, RiskAgent, RouterAgent
|
|
from .config import FAST_ROUTE_ENABLED
|
|
from .image_quote_analyzer import analyze_image_for_quote
|
|
from .models import Decision
|
|
from .state_machine import evolve_after_sales_state, migrate_state_schema
|
|
from .store import ConversationStore
|
|
|
|
|
|
class Orchestrator:
|
|
def __init__(self) -> None:
|
|
self.router = RouterAgent()
|
|
self.pre_sales = PreSalesAgent()
|
|
self.quote = QuoteAgent()
|
|
self.after_sales = AfterSalesAgent()
|
|
self.risk = RiskAgent()
|
|
self.store = ConversationStore()
|
|
|
|
@staticmethod
|
|
def _fast_route(context: dict[str, Any]) -> tuple[str, str] | None:
|
|
if not FAST_ROUTE_ENABLED:
|
|
return None
|
|
msg = str(context.get("msg", "") or "")
|
|
lower = msg.lower()
|
|
if bool(context.get("auto_quote_trigger")):
|
|
return "quote", "fast:auto_quote_trigger"
|
|
current_urls = context.get("current_image_urls") or []
|
|
if isinstance(current_urls, list) and len(current_urls) > 0:
|
|
return "quote", "fast:current_image"
|
|
if int(context.get("pending_images", 0) or 0) > 0 and any(k in lower for k in ("报价", "多少钱", "价格", "找得到", "找图", "能做", "有没有")):
|
|
return "quote", "fast:pending_image_and_pricing"
|
|
if any(k in lower for k in ("退款", "退钱", "不满意", "重做", "售后", "返工", "重发")):
|
|
return "after_sales", "fast:after_sales_keyword"
|
|
return None
|
|
|
|
async def decide(self, context: dict[str, Any]) -> tuple[str, Decision, dict[str, Any]]:
|
|
customer_key = context["customer_key"]
|
|
session = self.store.get_session(customer_key)
|
|
prev_state = migrate_state_schema(session.get("state", {}))
|
|
prev_route = session.get("route", "pre_sales")
|
|
|
|
# 统一改为语义决策：不走关键词意图/订单硬判定。
|
|
intent = "unknown"
|
|
order_status = "unknown"
|
|
|
|
merged_ctx = {
|
|
**context,
|
|
"session_state": prev_state,
|
|
"previous_route": prev_route,
|
|
"intent": intent,
|
|
"order_status": order_status,
|
|
}
|
|
|
|
fast = self._fast_route(merged_ctx)
|
|
if fast is not None:
|
|
route, route_reason = fast
|
|
else:
|
|
route, route_reason = await self.router.route(merged_ctx)
|
|
|
|
if route == "quote":
|
|
latest_image_url = str(context.get("latest_image_url", "") or "").strip()
|
|
if latest_image_url:
|
|
analysis = await analyze_image_for_quote(
|
|
image_url=latest_image_url,
|
|
customer_text=str(context.get("msg", "") or ""),
|
|
goods_name=str(context.get("goods_name", "") or ""),
|
|
)
|
|
merged_ctx["image_quote_analysis"] = analysis
|
|
decision = await self.quote.decide(merged_ctx)
|
|
elif route == "after_sales":
|
|
decision = await self.after_sales.decide(merged_ctx)
|
|
elif route == "risk":
|
|
decision = await self.risk.decide(merged_ctx)
|
|
else:
|
|
decision = await self.pre_sales.decide(merged_ctx)
|
|
|
|
merged_state = {**prev_state, **(decision.state_patch or {})}
|
|
if route == "quote":
|
|
analysis_obj = merged_ctx.get("image_quote_analysis")
|
|
if isinstance(analysis_obj, dict) and analysis_obj:
|
|
merged_state["last_image_quote_analysis"] = analysis_obj
|
|
new_state = evolve_after_sales_state(
|
|
merged_state,
|
|
route=route,
|
|
action=decision.action,
|
|
intent=intent,
|
|
order_status=order_status,
|
|
msg=str(context.get("msg", "") or ""),
|
|
)
|
|
|
|
self.store.upsert_session(customer_key, context.get("acc_id", ""), context.get("customer_id", ""), route, new_state)
|
|
self.store.append_event(
|
|
customer_key,
|
|
"decision",
|
|
{
|
|
"route": route,
|
|
"route_reason": route_reason,
|
|
"action": decision.action,
|
|
"reason": decision.reason,
|
|
"after_sales_stage": new_state.get("after_sales_stage", "new"),
|
|
},
|
|
)
|
|
return route, decision, new_state
|