life-echo/api/tests/test_interview_prompts.py

"""访谈提示词：追问触发与性格（Persona）拼接回归。"""

from app.agents.chat.personas import normalize_interview_persona
from app.agents.chat.prompts_conversation import (
    get_guided_conversation_prompt,
    get_opening_prompt,
)


def test_guided_prompt_contains_mandatory_followup_when_heuristic_matches():
    p = get_guided_conversation_prompt(
        current_stage="childhood",
        empty_slots=["place", "people"],
        filled_slots={},
        user_message="厉害吧 那个女生叫娟娟",
        conversation_turn=1,
        same_topic_turns=1,
        all_stages_coverage=None,
        detected_user_stage="childhood",
        user_profile_context="",
        persona="default",
    )
    assert "什么时候追问" in p
    assert "本轮判定" in p


def test_guided_prompt_persona_curious_guide():
    p = get_guided_conversation_prompt(
        current_stage="education",
        empty_slots=["school"],
        filled_slots={},
        user_message="还行吧",
        conversation_turn=0,
        same_topic_turns=0,
        all_stages_coverage=None,
        detected_user_stage="education",
        user_profile_context="",
        persona="curious_guide",
    )
    assert "好奇引导" in p


def test_normalize_interview_persona_unknown_falls_back():
    assert normalize_interview_persona("not_a_real_persona") == "default"
    assert normalize_interview_persona("") == "default"


def test_guided_prompt_contains_memoir_orientation():
    p = get_guided_conversation_prompt(
        current_stage="childhood",
        empty_slots=["place"],
        filled_slots={},
        user_message="后来我就去上班了",
        conversation_turn=0,
        same_topic_turns=0,
        all_stages_coverage=None,
        detected_user_stage="childhood",
        user_profile_context="",
        persona="default",
    )
    assert "对话方向" in p
    assert "人生故事" in p


def test_guided_prompt_contains_memory_section_when_evidence():
    p = get_guided_conversation_prompt(
        current_stage="childhood",
        empty_slots=["place"],
        filled_slots={},
        user_message="后来我就去上班了",
        conversation_turn=0,
        same_topic_turns=0,
        all_stages_coverage=None,
        detected_user_stage="childhood",
        user_profile_context="",
        persona="default",
        memory_evidence_text="[摘要:rolling] 1990年生于上海。",
    )
    assert "相关记忆摘录" in p
    assert "过往口述" in p
    assert "1990年生于上海" in p


def test_guided_prompt_chit_chat_hint():
    p = get_guided_conversation_prompt(
        current_stage="childhood",
        empty_slots=["place"],
        filled_slots={},
        user_message="今天天气真好哈哈",
        conversation_turn=0,
        same_topic_turns=0,
        all_stages_coverage=None,
        detected_user_stage="childhood",
        user_profile_context="",
        persona="default",
    )
    assert "偏闲聊" in p


def test_guided_prompt_reply_length_section_explicit_expanded():
    p = get_guided_conversation_prompt(
        current_stage="childhood",
        empty_slots=["place"],
        filled_slots={},
        user_message="还行吧",
        conversation_turn=0,
        same_topic_turns=0,
        all_stages_coverage=None,
        detected_user_stage="childhood",
        user_profile_context="",
        persona="default",
        reply_length_mode="expanded",
    )
    assert "本轮回复长度" in p
    assert "当前档位：expanded" in p
    assert "expanded" in p


def test_guided_prompt_reply_length_explicit_brief():
    """档位由 Agent 的 ReplyPlan 传入，prompt 不再自行推导。"""
    p = get_guided_conversation_prompt(
        current_stage="childhood",
        empty_slots=["place"],
        filled_slots={},
        user_message="嗯",
        conversation_turn=0,
        same_topic_turns=0,
        all_stages_coverage=None,
        detected_user_stage="childhood",
        user_profile_context="",
        persona="default",
        reply_length_mode="brief",
    )
    assert "当前档位：brief" in p


def test_guided_prompt_background_voice_military() -> None:
    p = get_guided_conversation_prompt(
        current_stage="childhood",
        empty_slots=["place"],
        filled_slots={},
        user_message="后来我就去上班了",
        conversation_turn=0,
        same_topic_turns=0,
        all_stages_coverage=None,
        detected_user_stage="childhood",
        user_profile_context="",
        persona="default",
        background_voice="military",
    )
    assert "背景语气：军队语境" in p
    assert "真诚承接" in p


def test_opening_prompt_military_has_examples_note() -> None:
    p = get_opening_prompt(
        current_stage="childhood",
        empty_slots_readable=["成长的地方"],
        user_profile_context="",
        persona="default",
        background_voice="military",
    )
    assert "军队语境" in p
    assert "（军队语境：简洁" in p or "军队语境" in p