a3f61fcc0f
- DB: segments 用户输入文本（Alembic 0002） - Chat: 阶段检测/阶段提示/回复限制，编排与访谈/画像 prompts 调整 - Memoir: 忠实度检查 agent，叙事与分类等链路更新 - Core: agent 日志、Alembic 启动、LangChain/日志/配置等 - Story: time_hints；Memory 检索与相关测试 - Expo: 助手头像、会话页与消息拆分、实时会话与文案/i18n - Docs/scripts/tests: 迁移脚本、LLM JSON/记忆检索文档、新增单测
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
"""Celery：story 变更后重组关联章节的 canonical_markdown（物化视图）。"""
|
||
|
||
from celery import shared_task
|
||
from sqlalchemy import select
|
||
|
||
from app.core.db import get_sync_db
|
||
from app.core.logging import get_logger
|
||
from app.features.memoir import repo as memoir_repo
|
||
from app.features.memoir.models import Chapter, ChapterStoryLink
|
||
|
||
logger = get_logger(__name__)
|
||
|
||
|
||
@shared_task(bind=True, max_retries=3, default_retry_delay=30)
|
||
def recompose_chapters_for_story(self, story_id: str) -> dict:
|
||
try:
|
||
with get_sync_db() as session:
|
||
stmt = (
|
||
select(Chapter.id)
|
||
.join(
|
||
ChapterStoryLink,
|
||
ChapterStoryLink.chapter_id == Chapter.id,
|
||
)
|
||
.where(
|
||
ChapterStoryLink.story_id == story_id,
|
||
Chapter.markdown_compose_dirty.is_(True),
|
||
)
|
||
)
|
||
ids = list(session.scalars(stmt).all())
|
||
for cid in ids:
|
||
memoir_repo.compose_chapter_from_story_links_sync(session, cid)
|
||
session.commit()
|
||
logger.info(
|
||
"recompose_chapters_for_story: story={} recomposed_chapters={}",
|
||
story_id,
|
||
ids,
|
||
)
|
||
return {"story_id": story_id, "recomposed_chapter_ids": ids}
|
||
except Exception as exc:
|
||
logger.warning(
|
||
"recompose_chapters_for_story failed story={} err={}", story_id, exc
|
||
)
|
||
raise self.retry(exc=exc) from exc
|