feat(ai-writing): 研究模板独立实体 + 配置页 + 内置范例 + 点击直接进入写作（0616 #5）

后端： - 新持久化包 research_templates（表 ai_writing_research_templates，create_all 建表，注册 models/__init__），deps 接 research_template_store - 路由 /api/ai-writing/research-templates CRUD（在 ai_writing.py） - 内置范例网摘/周报/要讯（DEFAULT_RESEARCH_TEMPLATES + seed_defaults，表空时由 list 接口幂等播种） - tests/test_research_templates.py（9 passed）前端： - api/ai-writing-research-templates.ts + hooks/useResearchTemplates.ts - 配置页 ResearchTemplatesConfigPage（/page/canvas/research-templates）CRUD - ResearchSidebar 研究模板改用研究模板实体；点击回填整张表单并直接 startWriting 进入写作；顶部齿轮进配置页 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-06-16 17:59:05 +08:00
1 parent 0e4caa5a66
commit ba5aeefe98
@@ -42,6 +42,7 @@ from deerflow.persistence.models.run_event import RunEventRow
 from deerflow.persistence.notifications.model import NotificationRow
 from deerflow.persistence.parallel_agent_tasks.model import ParallelAgentTaskRow
 from deerflow.persistence.recommended_questions.model import RecommendedQuestionRow
+from deerflow.persistence.research_templates.model import ResearchTemplateRow
 from deerflow.persistence.roundtable_chains.model import RoundtableChainRow
 from deerflow.persistence.roundtable_draft_shares.model import RoundtableDraftShareRow
 from deerflow.persistence.roundtable_drafts.model import (
@@ -89,6 +90,7 @@ __all__ = [
    "NotificationRow",
    "ParallelAgentTaskRow",
    "RecommendedQuestionRow",
+    "ResearchTemplateRow",
    "RoundtableChainRow",
    "RoundtableDraftRow",
    "RoundtableDraftShareRow",
@@ -0,0 +1,15 @@
+from deerflow.persistence.research_templates.model import ResearchTemplateRow
+from deerflow.persistence.research_templates.sql import ResearchTemplateRepository
+
+
+def make_research_template_store(session_factory) -> ResearchTemplateRepository | None:
+    if session_factory is None:
+        return None
+    return ResearchTemplateRepository(session_factory)
+
+
+__all__ = [
+    "ResearchTemplateRow",
+    "ResearchTemplateRepository",
+    "make_research_template_store",
+]
@@ -0,0 +1,35 @@
+"""ORM model for AI writing research templates (课题研究模板).
+
+A research template bundles a set of writing-form presets — article type,
+outline, material source, word count, audience, strict mode and a topic/intent —
+so a user can one-click pre-fill the whole 课题研究 form. Maintained from the
+research-template config page (需求 0616 #5).
+"""
+
+from __future__ import annotations
+
+from datetime import UTC, datetime
+
+from sqlalchemy import Boolean, DateTime, Integer, String, Text
+from sqlalchemy.orm import Mapped, mapped_column
+
+from deerflow.persistence.base import Base
+
+
+class ResearchTemplateRow(Base):
+    __tablename__ = "ai_writing_research_templates"
+
+    id: Mapped[str] = mapped_column(String(64), primary_key=True)
+    name: Mapped[str] = mapped_column(String(128), nullable=False)
+    description: Mapped[str | None] = mapped_column(Text, nullable=True)
+    # 预设字段（点击模板时回填表单）。
+    article_type_key: Mapped[str | None] = mapped_column(String(64), nullable=True)
+    outline: Mapped[str | None] = mapped_column(Text, nullable=True)
+    material_source: Mapped[str] = mapped_column(String(32), nullable=False, default="general")
+    word_count: Mapped[int] = mapped_column(Integer, nullable=False, default=800)
+    audience: Mapped[str | None] = mapped_column(String(128), nullable=True)
+    strict_mode: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
+    topic: Mapped[str | None] = mapped_column(Text, nullable=True)
+    sort_order: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC))
+    updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=lambda: datetime.now(UTC))
@@ -0,0 +1,221 @@
+"""SQLAlchemy-backed research template storage."""
+
+from __future__ import annotations
+
+import uuid
+from datetime import UTC, datetime
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
+
+from deerflow.persistence.research_templates.model import ResearchTemplateRow
+
+_UNSET = object()
+
+
+# 新机器内置范例模板（表为空时由 seed_defaults() 写入；作为给用户的范例，内容务求规范好用）。
+# article_type_key 取自默认文章类型（news/analysis/feature/report，见 20260517_02 迁移）。
+DEFAULT_RESEARCH_TEMPLATES: list[dict] = [
+    {
+        "name": "网摘",
+        "description": "汇编近期网络热点信息，分条摘录核心要点并标注来源，快速掌握全貌",
+        "article_type_key": "news",
+        "material_source": "general",
+        "word_count": 1000,
+        "audience": "关注热点动态的一般读者",
+        "strict_mode": False,
+        "topic": (
+            "围绕【主题】检索并汇编近一周网络上的重要信息与热点动态，"
+            "逐条摘录核心事实、关键数据与原文出处，帮助读者快速掌握全貌。"
+        ),
+        "outline": (
+            "一、热点速览（用一句话清单列出本期要点）\n"
+            "二、重点摘录\n"
+            "　1.【事件/话题标题】——核心事实、关键数据、信息来源\n"
+            "　2.【事件/话题标题】——……\n"
+            "　3.（按重要性依次罗列）\n"
+            "三、关联动态与背景补充（串联相关信息、补足来龙去脉）\n"
+            "四、小结与观察（提炼趋势或值得关注的看点）"
+        ),
+        "sort_order": 0,
+    },
+    {
+        "name": "周报",
+        "description": "按周梳理工作进展、数据指标与问题风险，输出结构化周度汇报",
+        "article_type_key": "report",
+        "material_source": "general",
+        "word_count": 1200,
+        "audience": "团队成员与上级主管",
+        "strict_mode": True,
+        "topic": (
+            "汇总本周【项目/团队】的工作进展，撰写周报："
+            "覆盖关键成果、数据指标、问题风险与下周计划，简明、客观、可量化。"
+        ),
+        "outline": (
+            "一、本周概览（一句话总结 + 关键指标对比上周）\n"
+            "二、重点进展\n"
+            "　1. 已完成事项与成果（结论先行，附支撑数据）\n"
+            "　2. 关键数据 / 里程碑达成情况\n"
+            "三、问题与风险（逐条列出问题、影响面及应对措施）\n"
+            "四、下周计划与资源需求（目标、负责人、所需支持）"
+        ),
+        "sort_order": 1,
+    },
+    {
+        "name": "要讯",
+        "description": "提炼最重要的资讯，精炼成简报式要点，突出核心事实与潜在影响",
+        "article_type_key": "analysis",
+        "material_source": "general",
+        "word_count": 600,
+        "audience": "需快速掌握要点的决策者",
+        "strict_mode": True,
+        "topic": (
+            "围绕【主题/领域】提炼近期最重要的资讯，编写要讯："
+            "每条聚焦核心事实、关键背景与潜在影响，文字精炼、判断审慎。"
+        ),
+        "outline": (
+            "一、头条要讯（1–3 条，每条用一句话点明核心）\n"
+            "二、要点详述（逐条展开）\n"
+            "　· 事实：发生了什么、何时何地、涉及谁\n"
+            "　· 背景：为何重要、与既有态势的关系\n"
+            "　· 影响：可能带来的后果与连锁反应\n"
+            "三、研判与建议（给出倾向性判断与可行动建议）"
+        ),
+        "sort_order": 2,
+    },
+]
+
+
+class ResearchTemplateRepository:
+    def __init__(self, session_factory: async_sessionmaker[AsyncSession]) -> None:
+        self._sf = session_factory
+
+    @staticmethod
+    def _row_to_dict(row: ResearchTemplateRow) -> dict:
+        d = row.to_dict()
+        for k in ("created_at", "updated_at"):
+            if isinstance(d.get(k), datetime):
+                d[k] = d[k].isoformat()
+        return d
+
+    async def create(
+        self,
+        *,
+        name: str,
+        description: str | None = None,
+        article_type_key: str | None = None,
+        outline: str | None = None,
+        material_source: str = "general",
+        word_count: int = 800,
+        audience: str | None = None,
+        strict_mode: bool = False,
+        topic: str | None = None,
+        sort_order: int = 0,
+    ) -> dict:
+        now = datetime.now(UTC)
+        row = ResearchTemplateRow(
+            id=str(uuid.uuid4()),
+            name=name,
+            description=description,
+            article_type_key=article_type_key,
+            outline=outline,
+            material_source=material_source,
+            word_count=word_count,
+            audience=audience,
+            strict_mode=strict_mode,
+            topic=topic,
+            sort_order=sort_order,
+            created_at=now,
+            updated_at=now,
+        )
+        async with self._sf() as session:
+            session.add(row)
+            await session.commit()
+            await session.refresh(row)
+            return self._row_to_dict(row)
+
+    async def update(
+        self,
+        template_id: str,
+        *,
+        name=_UNSET,
+        description=_UNSET,
+        article_type_key=_UNSET,
+        outline=_UNSET,
+        material_source=_UNSET,
+        word_count=_UNSET,
+        audience=_UNSET,
+        strict_mode=_UNSET,
+        topic=_UNSET,
+        sort_order=_UNSET,
+    ) -> dict | None:
+        async with self._sf() as session:
+            row = await session.get(ResearchTemplateRow, template_id)
+            if row is None:
+                return None
+            if name is not _UNSET:
+                row.name = name
+            if description is not _UNSET:
+                row.description = description
+            if article_type_key is not _UNSET:
+                row.article_type_key = article_type_key
+            if outline is not _UNSET:
+                row.outline = outline
+            if material_source is not _UNSET:
+                row.material_source = material_source
+            if word_count is not _UNSET:
+                row.word_count = word_count
+            if audience is not _UNSET:
+                row.audience = audience
+            if strict_mode is not _UNSET:
+                row.strict_mode = strict_mode
+            if topic is not _UNSET:
+                row.topic = topic
+            if sort_order is not _UNSET:
+                row.sort_order = sort_order
+            row.updated_at = datetime.now(UTC)
+            await session.commit()
+            await session.refresh(row)
+            return self._row_to_dict(row)
+
+    async def get(self, template_id: str) -> dict | None:
+        async with self._sf() as session:
+            row = await session.get(ResearchTemplateRow, template_id)
+            if row is None:
+                return None
+            return self._row_to_dict(row)
+
+    async def list(self) -> list[dict]:
+        stmt = select(ResearchTemplateRow).order_by(
+            ResearchTemplateRow.sort_order.asc(), ResearchTemplateRow.created_at.asc()
+        )
+        async with self._sf() as session:
+            result = await session.execute(stmt)
+            return [self._row_to_dict(r) for r in result.scalars()]
+
+    async def delete(self, template_id: str) -> bool:
+        async with self._sf() as session:
+            row = await session.get(ResearchTemplateRow, template_id)
+            if row is None:
+                return False
+            await session.delete(row)
+            await session.commit()
+            return True
+
+    async def seed_defaults(self) -> int:
+        """新机器/空表时写入内置范例模板（网摘/周报/要讯）。
+
+        幂等：仅当表为空时插入；已有任何模板则跳过（避免覆盖用户配置或重复播种）。
+        返回插入条数。
+        """
+        async with self._sf() as session:
+            existing = await session.execute(select(ResearchTemplateRow.id).limit(1))
+            if existing.first() is not None:
+                return 0
+            now = datetime.now(UTC)
+            for t in DEFAULT_RESEARCH_TEMPLATES:
+                session.add(
+                    ResearchTemplateRow(id=str(uuid.uuid4()), created_at=now, updated_at=now, **t)
+                )
+            await session.commit()
+            return len(DEFAULT_RESEARCH_TEMPLATES)