7f57f96c25
本次 squash merge 将 codex-story-first-image-intent 的整体改动合入 development，核心内容包括： 1. 后端数据与迁移：新增 stories、story_versions、story_image_intents、chapter_cover_intents、assets 等模型与 Alembic 迁移，建立 story-first、markdown-first、asset-first 的主数据链路。 2. 生成与任务链：引入 StoryBuilderOrchestrator、ChapterComposerOrchestrator、story_image_tasks、chapter_cover_tasks，图片生成从正文占位符改为结构化 intent -> asset -> markdown 回填。 3. 并发与一致性：为 story/chapter intent 增加 claim_token、claimed_at、attempt_count，采用数据库原子 claim 为主、Redis 锁为辅，避免重复生成、锁误删和 processing 卡死。 4. Memoir 读写路径：章节 canonical_markdown 成为正文真源，列表/详情接口补齐 markdown、cover_asset、word_count 等字段，PDF 与 asset 解析链路同步升级。 5. Memory / Retrieval：扩展 transcript ingest、chunking、evidence 检索与 story 聚合基础设施，为后续 story-first RAG 与多 agent 编排提供底座。 6. App 端体验：章节页继续走 MarkdownRenderer 阅读链，同时吸收 fix3-19 的跨平台 UI glitch 修复；更新对话页、首页、文案资源与章节列表映射逻辑。 7. 测试与文档：补充 asset resolver、story image task、章节封面派发、markdown 映射等回归测试，并加入图片占位符退役设计文档。
87 lines
3.6 KiB
Python
87 lines
3.6 KiB
Python
import unittest
|
||
|
||
from app.features.memoir.memoir_images.parser import (
|
||
build_initial_image_assets,
|
||
parse_image_placeholders,
|
||
parse_narrative_json,
|
||
parse_narrative_to_sections,
|
||
)
|
||
|
||
|
||
class MemoirImageParserTest(unittest.TestCase):
|
||
def test_parse_image_placeholders_preserves_order_and_offsets(self):
|
||
content = (
|
||
"那条路我一直记得。\n\n"
|
||
"{{{{IMAGE:南方小镇的青石板路}}}}\n\n"
|
||
"奶奶总坐在门口。\n\n"
|
||
"{{{{IMAGE:奶奶坐在院子里的藤椅上}}}}"
|
||
)
|
||
|
||
items = parse_image_placeholders(content, max_images=3)
|
||
|
||
self.assertEqual([item["index"] for item in items], [0, 1])
|
||
self.assertEqual(items[0]["description"], "南方小镇的青石板路")
|
||
self.assertEqual(
|
||
items[1]["placeholder"], "{{{{IMAGE:奶奶坐在院子里的藤椅上}}}}"
|
||
)
|
||
self.assertLess(items[0]["start_offset"], items[1]["start_offset"])
|
||
|
||
def test_build_initial_image_assets_marks_every_item_pending(self):
|
||
placeholders = [
|
||
{
|
||
"index": 0,
|
||
"description": "南方小镇的青石板路",
|
||
"placeholder": "{{{{IMAGE:南方小镇的青石板路}}}}",
|
||
"start_offset": 10,
|
||
}
|
||
]
|
||
|
||
assets = build_initial_image_assets(
|
||
placeholders=placeholders,
|
||
provider="liblib",
|
||
style="watercolor",
|
||
size="1024x1024",
|
||
now_iso="2026-03-10T10:00:00Z",
|
||
)
|
||
|
||
self.assertEqual(assets[0]["status"], "pending")
|
||
self.assertEqual(assets[0]["provider"], "liblib")
|
||
self.assertEqual(assets[0]["url"], None)
|
||
|
||
def test_parse_image_placeholders_accepts_double_brace_variant(self):
|
||
content = "开头。\n\n{{IMAGE:1938年初的上海弄堂口，冬日萧瑟}}\n\n结尾。"
|
||
|
||
items = parse_image_placeholders(content, max_images=2)
|
||
|
||
self.assertEqual(len(items), 1)
|
||
self.assertEqual(
|
||
items[0]["placeholder"], "{{IMAGE:1938年初的上海弄堂口，冬日萧瑟}}"
|
||
)
|
||
self.assertEqual(items[0]["description"], "1938年初的上海弄堂口，冬日萧瑟")
|
||
|
||
def test_parse_narrative_json_ignores_image_description(self):
|
||
raw = '{"paragraphs": [{"content": "那年春天。", "image_description": "南方小镇的青石板路"}, {"content": "奶奶坐在藤椅上。", "image_description": "奶奶的藤椅"}]}'
|
||
segments = parse_narrative_json(raw)
|
||
self.assertEqual(len(segments), 2)
|
||
self.assertEqual(segments[0]["content"], "那年春天。")
|
||
self.assertIsNone(segments[0]["placeholder_info"])
|
||
self.assertEqual(segments[1]["content"], "奶奶坐在藤椅上。")
|
||
self.assertIsNone(segments[1]["placeholder_info"])
|
||
|
||
def test_parse_narrative_to_sections_json_then_plain_strips_placeholders(self):
|
||
json_raw = (
|
||
'{"paragraphs": [{"content": "段落一", "image_description": "图一"}]}'
|
||
)
|
||
segments = parse_narrative_to_sections(json_raw)
|
||
self.assertEqual(len(segments), 1)
|
||
self.assertEqual(segments[0]["content"], "段落一")
|
||
self.assertIsNone(segments[0]["placeholder_info"])
|
||
|
||
placeholder_raw = "正文。\n\n{{{{IMAGE:描述}}}}\n\n结尾。"
|
||
segments2 = parse_narrative_to_sections(placeholder_raw)
|
||
self.assertEqual(len(segments2), 2)
|
||
self.assertEqual(segments2[0]["content"], "正文。")
|
||
self.assertEqual(segments2[1]["content"], "结尾。")
|
||
self.assertIsNone(segments2[0]["placeholder_info"])
|
||
self.assertIsNone(segments2[1]["placeholder_info"])
|