53e0065e3e
配置 SSOT（TOML + .env） 统一错误契约 Auth 与事务边界 Redis / Celery 可靠性:业务 Redis（DB/0）与 Celery broker/backend（DB/1）显式拆分；连接池、sync client 可观测性（OpenTelemetry + LGTM）
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
"""WebSocket 下发话题 chips（opening / resume / 每轮助手回复后）。"""
|
||
|
||
from datetime import datetime, timezone
|
||
|
||
from app.agents.chat.prompts_conversation import build_topic_chips
|
||
from app.agents.state_schema import (
|
||
MemoirStateSchema,
|
||
interview_control_state,
|
||
narrative_coverage_state,
|
||
)
|
||
from app.core.config import settings
|
||
from app.core.logging import get_logger
|
||
from app.features.conversation.ws.connection_manager import manager
|
||
from app.features.conversation.ws.message_types import MessageType
|
||
from app.features.conversation.ws.profile_collector import get_missing_profile_fields
|
||
from app.features.user.models import User
|
||
from app.features.conversation.constants import chat
|
||
|
||
log = get_logger(__name__)
|
||
|
||
|
||
async def maybe_send_topic_chips_ws(
|
||
conversation_id: str,
|
||
*,
|
||
user: User,
|
||
memoir_state: MemoirStateSchema,
|
||
reason: str,
|
||
language: str,
|
||
) -> None:
|
||
"""资料齐备且开关开启时，按当前回忆录阶段下发 topic_suggestions。失败静默。"""
|
||
if not chat.topic_chips_enabled:
|
||
return
|
||
if get_missing_profile_fields(user):
|
||
return
|
||
stage = (memoir_state.current_stage or "").strip() or "childhood"
|
||
try:
|
||
narrative_state = narrative_coverage_state(memoir_state)
|
||
control_state = interview_control_state(memoir_state)
|
||
empty_slots = control_state.prompt_empty_slots_for_stage(
|
||
narrative_state, stage
|
||
)
|
||
chips = build_topic_chips(
|
||
stage,
|
||
empty_slots,
|
||
max_chips=chat.topic_chips_max,
|
||
language=language,
|
||
)
|
||
if not chips:
|
||
return
|
||
await manager.send_message(
|
||
conversation_id,
|
||
{
|
||
"type": MessageType.TOPIC_SUGGESTIONS,
|
||
"conversation_id": conversation_id,
|
||
"data": {
|
||
"reason": reason,
|
||
"stage": stage,
|
||
"suggestions": chips,
|
||
},
|
||
"timestamp": datetime.now(timezone.utc).isoformat(),
|
||
},
|
||
)
|
||
log.info(
|
||
"event=ws_topic_chips_sent reason={} conversation_id={} stage={} count={}",
|
||
reason,
|
||
conversation_id,
|
||
stage,
|
||
len(chips),
|
||
)
|
||
except Exception as e:
|
||
log.warning("发送话题 chips 失败: {}", e)
|