merge: integrate ZSY customer service and profile capabilities
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
import hashlib
|
||||
import json
|
||||
from collections.abc import Sequence
|
||||
from dataclasses import dataclass
|
||||
from datetime import UTC, datetime, timedelta
|
||||
from uuid import uuid4
|
||||
@@ -9,6 +10,7 @@ from sqlalchemy.exc import IntegrityError
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.core.contracts import AgentRequest, DomainEvent, RequestContext
|
||||
from app.core.conversation_privacy import sanitize_customer_service_message
|
||||
from app.core.errors import (
|
||||
ForbiddenAgentError,
|
||||
IdempotencyConflictError,
|
||||
@@ -21,7 +23,12 @@ from app.model.session import ConversationSession
|
||||
from app.repository.outbox_repository import OutboxRepository
|
||||
from app.service.advisor_rollout_service import AdvisorRolloutService
|
||||
from app.service.agent.bootstrap import get_agent_factory
|
||||
from app.service.agent.customer_service_routing import CustomerServiceIntentRouter
|
||||
from app.service.agent.factory import AgentFactory
|
||||
from app.service.customer_service_session_memory_service import (
|
||||
CustomerServiceSessionMemory,
|
||||
build_customer_service_session_memory,
|
||||
)
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
@@ -31,10 +38,95 @@ class RunAccepted:
|
||||
status: str = "queued"
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class CustomerServicePriorContext:
|
||||
"""受理阶段取得的短期上下文,仅包含脱敏内容和当前会话需要的最小轮次。"""
|
||||
|
||||
user_messages: tuple[str, ...] = ()
|
||||
session_context: tuple[str, ...] = ()
|
||||
|
||||
|
||||
def build_outbox_metadata(
|
||||
request: AgentRequest, prior_user_messages: Sequence[str], *,
|
||||
clarification_round: int = 0, session_context: Sequence[str] = (),
|
||||
) -> dict[str, object]:
|
||||
"""构造 Worker 使用的内部元数据,不信任外部传入的闲聊计数。"""
|
||||
metadata = request.metadata
|
||||
if request.agent_type == "customer_service":
|
||||
metadata = metadata.model_copy(update={
|
||||
"chitchat_streak": CustomerServiceIntentRouter.chitchat_streak(
|
||||
prior_user_messages, request.message
|
||||
),
|
||||
"clarification_round": clarification_round,
|
||||
"session_context": tuple(session_context[-6:]),
|
||||
})
|
||||
return metadata.model_dump(mode="json")
|
||||
|
||||
|
||||
class AgentRunApplicationService:
|
||||
def __init__(self, session: AsyncSession, factory: AgentFactory | None = None) -> None:
|
||||
def __init__(
|
||||
self, session: AsyncSession, factory: AgentFactory | None = None,
|
||||
session_memory: CustomerServiceSessionMemory | None = None,
|
||||
) -> None:
|
||||
self.session = session
|
||||
self.factory = factory if factory is not None else get_agent_factory()
|
||||
# 客服 Redis List 只保存当前会话的脱敏上下文,绝不接入长期客户记忆。
|
||||
self.session_memory = (
|
||||
session_memory
|
||||
if session_memory is not None
|
||||
else build_customer_service_session_memory()
|
||||
)
|
||||
|
||||
async def _read_customer_service_short_context(
|
||||
self, *, request: AgentRequest, user_id: int
|
||||
) -> CustomerServicePriorContext | None:
|
||||
"""读取 Redis 短期上下文;返回 ``None`` 时才表示需要在事务内回退 MySQL。"""
|
||||
memory_read = await self.session_memory.read(
|
||||
actor_id=str(user_id), session_id=request.session_id
|
||||
)
|
||||
if memory_read.degraded:
|
||||
return None
|
||||
context = tuple(
|
||||
f"{'用户' if turn.role == 'user' else '助手'}:"
|
||||
f"{sanitize_customer_service_message(turn.content)}"
|
||||
for turn in memory_read.turns[-6:]
|
||||
)
|
||||
return CustomerServicePriorContext(
|
||||
user_messages=memory_read.recent_user_messages[-3:], session_context=context,
|
||||
)
|
||||
|
||||
async def _load_mysql_prior_context(self, request: AgentRequest) -> CustomerServicePriorContext:
|
||||
"""Redis 故障时读取 MySQL 已脱敏历史;调用方必须已处于受理事务。"""
|
||||
mysql_messages = list(await self.session.scalars(
|
||||
select(ConversationMessage.content)
|
||||
.where(
|
||||
ConversationMessage.session_id == request.session_id,
|
||||
ConversationMessage.role == "user",
|
||||
)
|
||||
.order_by(ConversationMessage.id.desc())
|
||||
.limit(3)
|
||||
))
|
||||
user_messages = tuple(
|
||||
sanitize_customer_service_message(str(message))
|
||||
for message in reversed(mysql_messages)
|
||||
)
|
||||
return CustomerServicePriorContext(
|
||||
user_messages=user_messages,
|
||||
session_context=tuple(f"用户:{message}" for message in user_messages),
|
||||
)
|
||||
|
||||
async def _load_customer_service_prior_context(
|
||||
self, *, request: AgentRequest, user_id: int
|
||||
) -> CustomerServicePriorContext:
|
||||
"""为测试和非事务调用提供完整降级路径。
|
||||
|
||||
Redis 空 List 表示短期会话已过期或尚未写入,不能回退历史 MySQL;只有 Redis
|
||||
故障才允许降级回退,防止 30 分钟会话边界在不知情的情况下被历史消息绕过。
|
||||
"""
|
||||
short_context = await self._read_customer_service_short_context(
|
||||
request=request, user_id=user_id
|
||||
)
|
||||
return short_context or await self._load_mysql_prior_context(request)
|
||||
|
||||
async def accept(self, request: AgentRequest, context: RequestContext) -> RunAccepted:
|
||||
if request.agent_type == "advisor":
|
||||
@@ -50,9 +142,21 @@ class AgentRunApplicationService:
|
||||
created_at=datetime.now(UTC).replace(tzinfo=None),
|
||||
))
|
||||
raise
|
||||
# 客服原文不会进入会话或异步链路;安全关键词保留给后续路由生成风险提示。
|
||||
stored_request = request
|
||||
if request.agent_type == "customer_service":
|
||||
stored_request = request.model_copy(update={
|
||||
"message": sanitize_customer_service_message(request.message)
|
||||
})
|
||||
user_id = int(context.user_id)
|
||||
# Redis 是会话体验的可选依赖,不能在下方的 MySQL 行锁事务里等待网络 I/O。
|
||||
short_context = (
|
||||
await self._read_customer_service_short_context(request=request, user_id=user_id)
|
||||
if request.agent_type == "customer_service"
|
||||
else CustomerServicePriorContext()
|
||||
)
|
||||
request_hash = hashlib.sha256(
|
||||
json.dumps(request.model_dump(mode="json"), sort_keys=True).encode("utf-8")
|
||||
json.dumps(stored_request.model_dump(mode="json"), sort_keys=True).encode("utf-8")
|
||||
).hexdigest()
|
||||
now = datetime.now(UTC).replace(tzinfo=None)
|
||||
async with self.session.begin():
|
||||
@@ -92,10 +196,22 @@ class AgentRunApplicationService:
|
||||
raise RuntimeError("idempotency record has no run")
|
||||
return RunAccepted(run.run_id, run.trace_id, run.status)
|
||||
|
||||
clarification_round = session_row.clarification_round if session_row is not None else 0
|
||||
# Redis 短期会话与 MySQL 降级统一返回时间正序,连续闲聊计数不依赖存储实现。
|
||||
if short_context is None:
|
||||
prior_context = await self._load_mysql_prior_context(request)
|
||||
else:
|
||||
prior_context = short_context
|
||||
outbox_metadata = build_outbox_metadata(
|
||||
stored_request, prior_context.user_messages,
|
||||
clarification_round=clarification_round,
|
||||
session_context=prior_context.session_context,
|
||||
)
|
||||
|
||||
trace_id = context.trace_id
|
||||
message = ConversationMessage(
|
||||
session_id=request.session_id, customer_id=user_id, portal="api",
|
||||
role="user", content=request.message, trace_id=trace_id, created_at=now,
|
||||
role="user", content=stored_request.message, trace_id=trace_id, created_at=now,
|
||||
)
|
||||
self.session.add(message)
|
||||
await self.session.flush()
|
||||
@@ -121,7 +237,11 @@ class AgentRunApplicationService:
|
||||
await OutboxRepository(self.session).append(DomainEvent(
|
||||
event_id=str(uuid4()), event_type="agent.run_requested", aggregate_type="agent_run",
|
||||
aggregate_id=run_id, trace_id=trace_id,
|
||||
payload={"run_id": run_id, "metadata": request.metadata.model_dump(mode="json")},
|
||||
payload={
|
||||
"run_id": run_id,
|
||||
"actor_type": "visitor" if "visitor" in context.roles else "authenticated",
|
||||
"metadata": outbox_metadata,
|
||||
},
|
||||
occurred_at=now,
|
||||
))
|
||||
return RunAccepted(run_id, trace_id)
|
||||
|
||||
Reference in New Issue
Block a user