a3f61fcc0f
- DB: segments 用户输入文本（Alembic 0002） - Chat: 阶段检测/阶段提示/回复限制，编排与访谈/画像 prompts 调整 - Memoir: 忠实度检查 agent，叙事与分类等链路更新 - Core: agent 日志、Alembic 启动、LangChain/日志/配置等 - Story: time_hints；Memory 检索与相关测试 - Expo: 助手头像、会话页与消息拆分、实时会话与文案/i18n - Docs/scripts/tests: 迁移脚本、LLM JSON/记忆检索文档、新增单测
68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
"""叙事分区、口述过短回退、配图字数门闸（纯函数/无 DB）。"""
|
||
|
||
import pytest
|
||
|
||
from app.agents.memoir.prompts import format_narrative_user_content
|
||
from app.features.memoir import story_pipeline_sync as sps
|
||
|
||
|
||
def test_format_narrative_user_content_oral_only() -> None:
|
||
assert format_narrative_user_content("hello", "") == "【本段用户口述】\nhello"
|
||
|
||
|
||
def test_format_narrative_user_content_with_evidence() -> None:
|
||
out = format_narrative_user_content("口述A", "摘录B")
|
||
assert "【本段用户口述】" in out
|
||
assert "口述A" in out
|
||
assert "摘录B" in out
|
||
assert "非本段口述" in out
|
||
|
||
|
||
def test_should_fallback_to_transcript_short_md(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
) -> None:
|
||
monkeypatch.setattr(sps.settings, "memoir_narrative_fallback_body_ratio", 0.5)
|
||
monkeypatch.setattr(sps.settings, "memoir_narrative_fallback_min_chars", 20)
|
||
oral = "我一九九九年出生在上海，后来全家搬到苏州生活了好几年。"
|
||
assert sps._should_fallback_to_transcript("1999", oral) is True
|
||
assert sps._should_fallback_to_transcript(oral, oral) is False
|
||
|
||
|
||
def test_apply_narrative_fallbacks_merge_shrink_appends_oral() -> None:
|
||
"""整篇合并 JSON 输出过短：保留旧文并拼本段口述。"""
|
||
long_existing = "x" * 500
|
||
raw = '{"paragraphs": [{"content": "短"}]}'
|
||
out = sps._apply_narrative_fallbacks(
|
||
raw,
|
||
"新口述补充",
|
||
existing_for_narrative=long_existing,
|
||
existing_chapter_md="",
|
||
chapter_category="childhood",
|
||
)
|
||
assert long_existing in out
|
||
assert "新口述补充" in out
|
||
|
||
|
||
def test_apply_narrative_fallbacks_json_too_short_returns_oral(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
) -> None:
|
||
monkeypatch.setattr(sps.settings, "memoir_narrative_fallback_body_ratio", 0.5)
|
||
monkeypatch.setattr(sps.settings, "memoir_narrative_fallback_min_chars", 20)
|
||
oral = "我1999年出生在上海，小学时爷爷常带我去河边散步。"
|
||
raw = '{"paragraphs": [{"content": "1999"}]}'
|
||
out = sps._apply_narrative_fallbacks(
|
||
raw,
|
||
oral,
|
||
existing_for_narrative="",
|
||
existing_chapter_md="",
|
||
chapter_category="childhood",
|
||
)
|
||
assert out.strip() == oral
|
||
|
||
|
||
def test_memoir_image_settings_min_body_field() -> None:
|
||
from app.features.memoir.memoir_images.settings import MemoirImageSettings
|
||
|
||
cfg = MemoirImageSettings(story_image_min_body_chars=799)
|
||
assert cfg.story_image_min_body_chars == 799
|