Files
life-echo/api/app/agents/chat/prompt_context.py
Kevin 3121d1384d WIP: memory system improvements (in progress)
Interview/chat prompt layers, reply planner, style profiles, memory
injection, interview meta store, and related tests. Work not finished.

Made-with: Cursor
2026-04-22 16:56:28 +08:00

64 lines
2.4 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
"""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 注入模型。
本轮模式硬指令统一由 `InterviewTurnPlan.render_system_directive()` 产出,
主 prompt 不得重复立法(见 interview_turn_plan 模块 docstring
"""
from app.agents.chat.prompts_conversation import get_guided_conversation_prompt
directive = (
self.turn_plan.render_system_directive()
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,
)