5ff495729e
- Add plan_interview_turn: emotion_first / memoir_push / follow_user_only - Inject hard directive block at top of guided system prompt - Pass stage_switched_this_turn from ChatOrchestrator after stage detection - Log interview_turn_plan for observability; add unit tests
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
"""Bundled parameters for chat system prompts (InterviewAgent)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Dict, List, Optional
|
|
|
|
from app.agents.chat.interview_turn_plan import InterviewTurnPlan
|
|
from app.agents.state_schema import KnownFact, PersonaThread
|
|
|
|
|
|
@dataclass
|
|
class ChatPromptContext:
|
|
"""访谈轮次构建 `get_guided_conversation_prompt` 所需的字段集合。"""
|
|
|
|
current_stage: str
|
|
empty_slots: List[str]
|
|
filled_slots: Dict[str, str]
|
|
all_stages_coverage: Optional[Dict[str, Dict]] = None
|
|
detected_user_stage: str = ""
|
|
user_profile_context: str = ""
|
|
persona: str = "default"
|
|
memory_evidence_text: str = ""
|
|
background_voice: str = "default"
|
|
occupation: str = ""
|
|
profile_birth_year: int | None = None
|
|
profile_era_place: str = ""
|
|
known_facts: List[KnownFact] | None = None
|
|
persona_threads: List[PersonaThread] | None = None
|
|
recent_questions: List[str] | None = None
|
|
turn_plan: InterviewTurnPlan | None = None
|
|
|
|
def guided_system_prompt(self) -> str:
|
|
"""用户原话仅以对话历史 + HumanMessage 注入模型。"""
|
|
from app.agents.chat.interview_turn_plan import (
|
|
format_interview_turn_directive_block,
|
|
)
|
|
from app.agents.chat.prompts_conversation import get_guided_conversation_prompt
|
|
|
|
directive = (
|
|
format_interview_turn_directive_block(self.turn_plan)
|
|
if self.turn_plan is not None
|
|
else ""
|
|
)
|
|
return get_guided_conversation_prompt(
|
|
current_stage=self.current_stage,
|
|
empty_slots=self.empty_slots,
|
|
filled_slots=self.filled_slots,
|
|
all_stages_coverage=self.all_stages_coverage,
|
|
detected_user_stage=self.detected_user_stage,
|
|
user_profile_context=self.user_profile_context,
|
|
persona=self.persona,
|
|
memory_evidence_text=self.memory_evidence_text,
|
|
background_voice=self.background_voice,
|
|
occupation=self.occupation,
|
|
profile_birth_year=self.profile_birth_year,
|
|
profile_era_place=self.profile_era_place,
|
|
known_facts=self.known_facts or [],
|
|
persona_threads=self.persona_threads or [],
|
|
recent_questions=self.recent_questions or [],
|
|
turn_directive_block=directive,
|
|
)
|