71 lines
2.3 KiB
Python
71 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
|
|||
|
|
|
|||
|
|
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 settings.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=settings.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)
|