bb16d3a5c9
访谈与阶段 - 新增 app/agents/stage_constants.py：集中 CHAT_STAGES、章节分类/顺序、阶段到默认 memoir 类别等，与 MemoirState 默认槽位顺序对齐；减少散落在 prompts 内的重复常量。 - 新增 app/agents/chat/prompt_context.py：以 ChatPromptContext 汇总 guided 系统提示所需字段（阶段、槽位、轮次、人设、记忆证据、回复长度模式、背景声线、职业等），统一走 get_guided_conversation_prompt。 - 大幅收敛 app/agents/chat/prompts_conversation.py；调整 prompts.py、stage_prompts.py、stage_detection.py；同步 interview_agent、profile_agent、helpers 与 state_schema，使对话侧构造提示的方式一致、可测。 回忆录流水线 - memoir/prompts.py 删除已迁至 stage_constants / 独立模板的大段常量与图片占位相关逻辑；classification / extraction / fidelity / narrative agents 与 orchest（全量历史仍可用于计数，注入模型时按轮次与字符上限截断）、image_prompt_fallback_disabled。 - dependencies 增加 get_llm_provider_fast（LRU 缓存，可与默认共用密钥与 base_url）。 任务与编排 - memoir_tasks：prepare_batches 注入 llm_fast；开启独立快档模型时打结构化日志。 - chapter_cover_tasks、story_image_tasks：与图片 prompt / JSON 工具路径或策略变更对齐（import 与行为一致）。 - story_pipeline_sync 等小处同步。 其它核心 - langchain_llm、text_normalize 随上述调用链微调。 开发者体验 - .cursor/settings.json：启用 redis-development、postman 插件。 测试 - 新增 test_image_prompt_policy：覆盖「禁止回退」等图片 prompt 策略。 - 更新 test_interview_prompts、test_interview_reply_length、test_experience_regressions、test_json_and_memory_utils，匹配新常量位置、json_utils 与对话/长度行为。
115 lines
3.4 KiB
Python
115 lines
3.4 KiB
Python
"""
|
||
共享状态 Schema（对话 Agent 与后台 Agent 共用）
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Dict, List, Optional
|
||
|
||
from pydantic import BaseModel, Field
|
||
|
||
from app.agents.stage_constants import CHAT_STAGES
|
||
|
||
|
||
class SlotData(BaseModel):
|
||
"""Slot 数据结构"""
|
||
|
||
snippet: Optional[str] = None
|
||
segment_ids: List[str] = Field(default_factory=list)
|
||
|
||
|
||
class MemoirStateSchema(BaseModel):
|
||
"""回忆录状态"""
|
||
|
||
stage_order: List[str]
|
||
current_stage: str
|
||
covered_stages: List[str]
|
||
slots: Dict[str, Dict[str, SlotData]]
|
||
|
||
def empty_slots_for_current_stage(self) -> List[str]:
|
||
stage_slots = self.slots.get(self.current_stage, {})
|
||
empty_keys: List[str] = []
|
||
for key, value in stage_slots.items():
|
||
if not value.snippet:
|
||
empty_keys.append(key)
|
||
return empty_keys
|
||
|
||
def empty_slots_for_stage(self, stage: str) -> List[str]:
|
||
"""获取指定阶段的空槽位"""
|
||
stage_slots = self.slots.get(stage, {})
|
||
return [key for key, value in stage_slots.items() if not value.snippet]
|
||
|
||
def filled_slots_for_stage(self, stage: str) -> Dict[str, str]:
|
||
"""获取指定阶段已填充的槽位及其内容"""
|
||
stage_slots = self.slots.get(stage, {})
|
||
return {
|
||
key: value.snippet for key, value in stage_slots.items() if value.snippet
|
||
}
|
||
|
||
def all_stages_coverage(self) -> Dict[str, Dict]:
|
||
"""获取所有阶段的覆盖情况摘要"""
|
||
coverage: Dict[str, Dict] = {}
|
||
for stage in self.stage_order:
|
||
stage_slots = self.slots.get(stage, {})
|
||
total = len(stage_slots)
|
||
filled = sum(1 for v in stage_slots.values() if v.snippet)
|
||
coverage[stage] = {
|
||
"total": total,
|
||
"filled": filled,
|
||
"empty": total - filled,
|
||
"ratio": filled / total if total > 0 else 0,
|
||
}
|
||
return coverage
|
||
|
||
|
||
# 与 stage_constants.CHAT_STAGES 同一顺序；list() 避免与元组共享可变别名
|
||
DEFAULT_STAGE_ORDER: list[str] = list(CHAT_STAGES)
|
||
|
||
|
||
def default_slots() -> Dict[str, Dict[str, SlotData]]:
|
||
return {
|
||
"childhood": {
|
||
"place": SlotData(),
|
||
"people": SlotData(),
|
||
"daily_life": SlotData(),
|
||
"emotion": SlotData(),
|
||
"turning_event": SlotData(),
|
||
},
|
||
"education": {
|
||
"school": SlotData(),
|
||
"city": SlotData(),
|
||
"motivation": SlotData(),
|
||
"challenge": SlotData(),
|
||
"change": SlotData(),
|
||
},
|
||
"career": {
|
||
"job": SlotData(),
|
||
"environment": SlotData(),
|
||
"decision": SlotData(),
|
||
"pressure": SlotData(),
|
||
"growth": SlotData(),
|
||
},
|
||
"family": {
|
||
"relationship": SlotData(),
|
||
"conflict": SlotData(),
|
||
"support": SlotData(),
|
||
"responsibility": SlotData(),
|
||
"change": SlotData(),
|
||
},
|
||
"belief": {
|
||
"value": SlotData(),
|
||
"regret": SlotData(),
|
||
"pride": SlotData(),
|
||
"lesson": SlotData(),
|
||
},
|
||
}
|
||
|
||
|
||
def default_state() -> MemoirStateSchema:
|
||
return MemoirStateSchema(
|
||
stage_order=DEFAULT_STAGE_ORDER,
|
||
current_stage=DEFAULT_STAGE_ORDER[0],
|
||
covered_stages=[],
|
||
slots=default_slots(),
|
||
)
|