@musnows/scriverse 0.6.2 → 0.6.4

package/dist/store.js

@@ -1,6 +1,7 @@
 import { DRAFT_SETTING_MODULES } from "./domain.js";
 import { createHash } from "node:crypto";
 import { PLATFORM_AI_WORK_ID } from "./database.js";
+import { exportWorkDocx } from "./docx-export.js";
 import { AppError, notFound } from "./errors.js";
 import { accountReference, logger } from "./logger.js";
 import { paginated, paginationSql } from "./pagination.js";
@@ -9,6 +10,31 @@ import { classifyWorkModulePermissions, emptyWorkModulePermissions, fullWorkModu
 import { countWords, documentShortSearchTerms, id, json, normalizeDocumentSearchText, normalizeParagraphSpacing, now, splitDocumentParagraphs } from "./utils.js";
 import { buildWritingCalendar, writingDateKey } from "./writing-progress-time.js";
 export const attachmentPermissionModules = ["prose", "drafts", "settings", "characters", "races", "organizations"];
+export const WORK_AGENT_TOOL_IDS = [
+    "story_index",
+    "read_chapters",
+    "grep",
+    "search_story_entities",
+    "read_character_sections",
+    "search_drafts"
+];
+const DEFAULT_WORK_AGENT_TOOLS = [...WORK_AGENT_TOOL_IDS];
+export function normalizeWorkAgentTools(value) {
+    const source = Array.isArray(value)
+        ? value
+        : typeof value === "string"
+            ? json(value, DEFAULT_WORK_AGENT_TOOLS)
+            : DEFAULT_WORK_AGENT_TOOLS;
+    const enabled = new Set();
+    for (const item of source) {
+        if (typeof item !== "string")
+            continue;
+        const toolId = item === "query_story_knowledge" ? "search_story_entities" : item;
+        if (WORK_AGENT_TOOL_IDS.includes(toolId))
+            enabled.add(toolId);
+    }
+    return WORK_AGENT_TOOL_IDS.filter((toolId) => enabled.has(toolId));
+}
 const defaultPlatformPageSizes = {
     drafts: 30,
     settings: 30,
@@ -127,6 +153,7 @@ export const versionedEntityTypes = [
     "chapter-outline",
     "foreshadow"
 ];
+export const aiConversationTaskTypes = ["chat", "roleplay", "continue", "polish"];
 export function defaultAiConversationTitle(prompt) {
     const normalized = prompt.replace(/\s+/gu, " ").trim();
     return Array.from(normalized).slice(0, 15).join("") || "新对话";
@@ -205,6 +232,29 @@ function booleanValue(row, key) {
 export function normalizeCharacterName(value) {
     return value.normalize("NFKC").trim().replace(/\s+/gu, " ").toLocaleLowerCase("zh-CN");
 }
+const EMPTY_AI_INJECTED_ENTITIES = {
+    characters: [],
+    races: [],
+    organizations: []
+};
+function parseAiInjectedEntities(value) {
+    const parsed = typeof value === "string" ? json(value, {}) : isRecord(value) ? value : {};
+    const uniqueIds = (items) => [...new Set((Array.isArray(items) ? items : [])
+            .filter((item) => typeof item === "string" && item.trim().length > 0)
+            .map((item) => item.trim()))];
+    return {
+        characters: uniqueIds(parsed.characters),
+        races: uniqueIds(parsed.races),
+        organizations: uniqueIds(parsed.organizations)
+    };
+}
+function mergeAiInjectedEntities(base, extra) {
+    return {
+        characters: [...new Set([...base.characters, ...(extra.characters ?? [])])],
+        races: [...new Set([...base.races, ...(extra.races ?? [])])],
+        organizations: [...new Set([...base.organizations, ...(extra.organizations ?? [])])]
+    };
+}
 export class Store {
     db;
     constructor(db) {
@@ -685,6 +735,9 @@ export class Store {
         return {
             workId,
             systemPrompt: String(row?.system_prompt ?? ""),
+            dailyTokenQuota: row?.daily_token_quota === null || row?.daily_token_quota === undefined
+                ? null
+                : Math.max(10_000, Number(row.daily_token_quota)),
             autoRunEnabled: Number(row?.auto_run_enabled ?? 0) === 1,
             autoRunConcurrency: Math.min(8, Math.max(1, Number(row?.auto_run_concurrency ?? 2) || 2)),
             autoRunBatchLimit: Math.min(200, Math.max(1, Number(row?.auto_run_batch_limit ?? 20) || 20)),
@@ -696,9 +749,9 @@ export class Store {
             autoRunConsecutiveFailures: Math.max(0, Number(row?.auto_run_consecutive_failures ?? 0) || 0),
             bookSummaryContextPercent: Math.min(90, Math.max(1, Number(row?.book_summary_context_percent ?? 50) || 50)),
             contextCompactThreshold: Math.min(90, Math.max(50, Number(row?.context_compact_threshold ?? 85) || 85)),
-            agentTools: json(String(row?.agent_tools_json ?? '["story_index","read_chapters","search_story_entities","grep","read_character_sections","search_drafts"]'), ["story_index", "read_chapters", "search_story_entities", "grep", "read_character_sections", "search_drafts"])
-                .map((tool) => tool === "query_story_knowledge" ? "search_story_entities" : tool)
-                .filter((tool, index, tools) => tools.indexOf(tool) === index),
+            agentToolCallLimit: Math.min(48, Math.max(5, Number(row?.agent_tool_call_limit ?? 12) || 12)),
+            agentToolCallGlobalMultiplier: Math.min(6, Math.max(1, Number(row?.agent_tool_call_global_multiplier ?? 3) || 3)),
+            agentTools: normalizeWorkAgentTools(row?.agent_tools_json),
             titleGenerationModelId: row?.title_generation_model_id === null || row?.title_generation_model_id === undefined
                 ? null
                 : String(row.title_generation_model_id),
@@ -710,6 +763,9 @@ export class Store {
         const current = this.getWorkAiSettings(workId);
         const timestamp = now();
         const nextPrompt = input.systemPrompt ?? String(current.systemPrompt);
+        const nextDailyTokenQuota = input.dailyTokenQuota === undefined
+            ? (current.dailyTokenQuota === null ? null : Number(current.dailyTokenQuota))
+            : input.dailyTokenQuota;
         const nextEnabled = input.autoRunEnabled ?? Boolean(current.autoRunEnabled);
         const nextConcurrency = input.autoRunConcurrency ?? Number(current.autoRunConcurrency);
         const nextBatchLimit = input.autoRunBatchLimit ?? Number(current.autoRunBatchLimit);
@@ -717,18 +773,22 @@ export class Store {
         const nextFailureThreshold = input.autoRunFailureThreshold ?? Number(current.autoRunFailureThreshold);
         const nextBookSummaryContextPercent = input.bookSummaryContextPercent ?? Number(current.bookSummaryContextPercent);
         const nextContextCompactThreshold = input.contextCompactThreshold ?? Number(current.contextCompactThreshold);
-        const nextAgentTools = input.agentTools ?? current.agentTools;
+        const nextAgentToolCallLimit = input.agentToolCallLimit ?? Number(current.agentToolCallLimit);
+        const nextAgentToolCallGlobalMultiplier = input.agentToolCallGlobalMultiplier ?? Number(current.agentToolCallGlobalMultiplier);
+        const nextAgentTools = normalizeWorkAgentTools(input.agentTools ?? current.agentTools);
         const nextTitleGenerationModelId = input.titleGenerationModelId === undefined
             ? (current.titleGenerationModelId ? String(current.titleGenerationModelId) : null)
             : input.titleGenerationModelId?.trim() || null;
         this.db.run(`INSERT INTO work_ai_settings (
-         work_id, system_prompt, auto_run_enabled, auto_run_concurrency, auto_run_batch_limit,
+         work_id, system_prompt, daily_token_quota, auto_run_enabled, auto_run_concurrency, auto_run_batch_limit,
          auto_run_daily_task_limit, auto_run_failure_threshold, auto_run_paused, auto_run_pause_reason,
          auto_run_resume_at, auto_run_consecutive_failures, book_summary_context_percent,
-         context_compact_threshold, agent_tools_json, title_generation_model_id, updated_at
-       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+         context_compact_threshold, agent_tool_call_limit, agent_tool_call_global_multiplier,
+         agent_tools_json, title_generation_model_id, updated_at
+       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
        ON CONFLICT(work_id) DO UPDATE SET
          system_prompt = excluded.system_prompt,
+         daily_token_quota = excluded.daily_token_quota,
          auto_run_enabled = excluded.auto_run_enabled,
          auto_run_concurrency = excluded.auto_run_concurrency,
          auto_run_batch_limit = excluded.auto_run_batch_limit,
@@ -740,11 +800,14 @@ export class Store {
          auto_run_consecutive_failures = excluded.auto_run_consecutive_failures,
          book_summary_context_percent = excluded.book_summary_context_percent,
          context_compact_threshold = excluded.context_compact_threshold,
+         agent_tool_call_limit = excluded.agent_tool_call_limit,
+         agent_tool_call_global_multiplier = excluded.agent_tool_call_global_multiplier,
          agent_tools_json = excluded.agent_tools_json,
          title_generation_model_id = excluded.title_generation_model_id,
-         updated_at = excluded.updated_at`, workId, nextPrompt, nextEnabled ? 1 : 0, Math.min(8, Math.max(1, nextConcurrency)), Math.min(200, Math.max(1, nextBatchLimit)), Math.min(10_000, Math.max(0, nextDailyTaskLimit)), Math.min(10, Math.max(1, nextFailureThreshold)), current.autoRunPaused ? 1 : 0, String(current.autoRunPauseReason ?? ""), current.autoRunResumeAt === null ? null : String(current.autoRunResumeAt), Math.max(0, Number(current.autoRunConsecutiveFailures) || 0), Math.min(90, Math.max(1, nextBookSummaryContextPercent)), Math.min(90, Math.max(50, nextContextCompactThreshold)), JSON.stringify(nextAgentTools), nextTitleGenerationModelId, timestamp);
+         updated_at = excluded.updated_at`, workId, nextPrompt, nextDailyTokenQuota, nextEnabled ? 1 : 0, Math.min(8, Math.max(1, nextConcurrency)), Math.min(200, Math.max(1, nextBatchLimit)), Math.min(10_000, Math.max(0, nextDailyTaskLimit)), Math.min(10, Math.max(1, nextFailureThreshold)), current.autoRunPaused ? 1 : 0, String(current.autoRunPauseReason ?? ""), current.autoRunResumeAt === null ? null : String(current.autoRunResumeAt), Math.max(0, Number(current.autoRunConsecutiveFailures) || 0), Math.min(90, Math.max(1, nextBookSummaryContextPercent)), Math.min(90, Math.max(50, nextContextCompactThreshold)), Math.min(48, Math.max(5, nextAgentToolCallLimit)), Math.min(6, Math.max(1, nextAgentToolCallGlobalMultiplier)), JSON.stringify(nextAgentTools), nextTitleGenerationModelId, timestamp);
         this.audit(workId, "work.ai-settings.updated", "work-ai-settings", workId, {
             systemPromptChanged: input.systemPrompt !== undefined,
+            dailyTokenQuota: nextDailyTokenQuota,
             autoRunEnabled: nextEnabled,
             autoRunConcurrency: Math.min(8, Math.max(1, nextConcurrency)),
             autoRunBatchLimit: Math.min(200, Math.max(1, nextBatchLimit)),
@@ -752,6 +815,8 @@ export class Store {
             autoRunFailureThreshold: Math.min(10, Math.max(1, nextFailureThreshold)),
             bookSummaryContextPercent: Math.min(90, Math.max(1, nextBookSummaryContextPercent)),
             contextCompactThreshold: Math.min(90, Math.max(50, nextContextCompactThreshold)),
+            agentToolCallLimit: Math.min(48, Math.max(5, nextAgentToolCallLimit)),
+            agentToolCallGlobalMultiplier: Math.min(6, Math.max(1, nextAgentToolCallGlobalMultiplier)),
             agentTools: nextAgentTools,
             titleGenerationModelId: nextTitleGenerationModelId
         });
@@ -856,12 +921,22 @@ export class Store {
         return this.getWork(workId);
     }
     getWorkCover(workId) {
+        const cover = this.findWorkCover(workId);
+        if (!cover)
+            throw notFound("作品封面");
+        return cover;
+    }
+    findWorkCover(workId) {
         this.getWork(workId);
         const row = this.db.get("SELECT * FROM work_covers WHERE work_id = ?", workId);
         if (!row)
-            throw notFound("作品封面");
+            return null;
+        const mimeType = requiredString(row, "mime_type");
+        if (mimeType !== "image/jpeg" && mimeType !== "image/png" && mimeType !== "image/webp") {
+            throw new AppError(500, "INVALID_COVER_MIME", "作品封面类型无效");
+        }
         return {
-            mimeType: requiredString(row, "mime_type"),
+            mimeType,
             content: Buffer.from(row.content),
             byteLength: numberValue(row, "byte_length"),
             sha256: requiredString(row, "sha256"),
@@ -1185,14 +1260,16 @@ export class Store {
         });
     }
     createChapter(workId, input) {
-        this.getWork(workId);
-        const volume = this.getVolume(input.volumeId);
-        if (volume.workId !== workId)
-            throw new AppError(400, "VOLUME_WORK_MISMATCH", "卷不属于当前作品");
-        const last = this.db.get("SELECT COALESCE(MAX(sort_order), -1) AS value FROM chapters WHERE volume_id = ? AND deleted_at IS NULL", input.volumeId);
-        const chapterId = this.insertChapter(workId, input.volumeId, input.title, input.content ?? "", numberValue(last ?? {}, "value") + 1, "manual", null, input.chapterType ?? "正文");
-        this.audit(workId, "chapter.created", "chapter", chapterId);
-        return this.getChapter(chapterId);
+        return this.db.transaction(() => {
+            this.getWork(workId);
+            const volume = this.getVolume(input.volumeId);
+            if (volume.workId !== workId)
+                throw new AppError(400, "VOLUME_WORK_MISMATCH", "卷不属于当前作品");
+            const last = this.db.get("SELECT COALESCE(MAX(sort_order), -1) AS value FROM chapters WHERE volume_id = ? AND deleted_at IS NULL", input.volumeId);
+            const chapterId = this.insertChapter(workId, input.volumeId, input.title, input.content ?? "", numberValue(last ?? {}, "value") + 1, "manual", null, input.chapterType ?? "正文");
+            this.audit(workId, "chapter.created", "chapter", chapterId);
+            return this.getChapter(chapterId);
+        });
     }
     getChapter(chapterId) {
         const row = this.db.get("SELECT * FROM chapters WHERE id = ? AND deleted_at IS NULL", chapterId);
@@ -4283,11 +4360,12 @@ export class Store {
         const row = this.db.get("SELECT * FROM continuation_guard_runs WHERE suggestion_id = ? ORDER BY created_at DESC LIMIT 1", suggestionId);
         return row ? this.mapContinuationGuard(row) : null;
     }
-    createAiConversation(workId, title = "新对话") {
+    createAiConversation(workId, title = "新对话", taskType = null) {
         this.getWork(workId);
         const conversationId = id("conversation");
         const timestamp = now();
-        this.db.run("INSERT INTO ai_conversations (id, work_id, title, created_at, updated_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?)", conversationId, workId, title.trim() || "新对话", timestamp, timestamp, currentRequestActor()?.userId ?? null);
+        const agentTools = normalizeWorkAgentTools(this.getWorkAiSettings(workId).agentTools);
+        this.db.run("INSERT INTO ai_conversations (id, work_id, task_type, title, agent_tools_json, created_at, updated_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", conversationId, workId, taskType, title.trim() || "新对话", JSON.stringify(agentTools), timestamp, timestamp, currentRequestActor()?.userId ?? null);
         return this.getAiConversation(conversationId);
     }
     listAiConversations(workId) {
@@ -4354,10 +4432,12 @@ export class Store {
         const compactedMessageCount = Math.min(rows.length, Math.max(0, numberValue(conversation, "compacted_message_count")));
         return {
             workId,
+            roleplayCharacterId: optionalString(conversation, "roleplay_character_id"),
             summary: requiredString(conversation, "compacted_summary"),
             compactedMessageCount,
             totalMessageCount: rows.length,
             warningPending: Boolean(optionalString(conversation, "context_warning_at")),
+            injectedEntities: parseAiInjectedEntities(optionalString(conversation, "injected_entities_json") ?? EMPTY_AI_INJECTED_ENTITIES),
             messages: rows.slice(compactedMessageCount)
                 .filter((message) => requiredString(message, "id") !== excludeMessageId)
                 .map((message) => ({
@@ -4368,6 +4448,53 @@ export class Store {
             }))
         };
     }
+    getAiConversationInjectedEntities(conversationId, workId) {
+        const conversation = this.db.get("SELECT work_id, injected_entities_json FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        if (requiredString(conversation, "work_id") !== workId)
+            throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
+        return parseAiInjectedEntities(optionalString(conversation, "injected_entities_json") ?? EMPTY_AI_INJECTED_ENTITIES);
+    }
+    mergeAiConversationInjectedEntities(conversationId, workId, extra) {
+        const conversation = this.db.get("SELECT work_id, injected_entities_json FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        if (requiredString(conversation, "work_id") !== workId)
+            throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
+        const merged = mergeAiInjectedEntities(parseAiInjectedEntities(optionalString(conversation, "injected_entities_json") ?? EMPTY_AI_INJECTED_ENTITIES), extra);
+        this.db.run("UPDATE ai_conversations SET injected_entities_json = ?, updated_at = ? WHERE id = ?", JSON.stringify(merged), now(), conversationId);
+        return merged;
+    }
+    /** 对话首轮写入 system 时钟文案；已有值则原样返回，禁止后续覆盖。 */
+    ensureAiConversationSystemClock(conversationId, workId, candidate) {
+        const conversation = this.db.get("SELECT work_id, system_clock_text FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        if (requiredString(conversation, "work_id") !== workId)
+            throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
+        const existing = (optionalString(conversation, "system_clock_text") ?? "").trim();
+        if (existing)
+            return existing;
+        const clock = candidate.trim();
+        if (!clock)
+            return "";
+        this.db.run("UPDATE ai_conversations SET system_clock_text = ? WHERE id = ? AND TRIM(system_clock_text) = ''", clock, conversationId);
+        const refreshed = this.db.get("SELECT system_clock_text FROM ai_conversations WHERE id = ?", conversationId);
+        return (optionalString(refreshed ?? {}, "system_clock_text") ?? clock).trim() || clock;
+    }
+    listCharacterNameEntries(workId) {
+        this.getWork(workId);
+        return this.db.all(`SELECT character_id, normalized_name, display_name, kind FROM character_names
+       WHERE work_id = ?
+         AND character_id NOT IN (SELECT id FROM characters WHERE work_id = ? AND merged_into_character_id IS NOT NULL)
+       ORDER BY LENGTH(normalized_name) DESC, sort_order ASC`, workId, workId).map((row) => ({
+            characterId: requiredString(row, "character_id"),
+            normalizedName: requiredString(row, "normalized_name"),
+            displayName: requiredString(row, "display_name"),
+            kind: requiredString(row, "kind") === "alias" ? "alias" : "primary"
+        }));
+    }
     getAiConversationTitleContext(conversationId, workId) {
         const conversation = this.db.get("SELECT title, work_id FROM ai_conversations WHERE id = ?", conversationId);
         if (!conversation)
@@ -4401,6 +4528,94 @@ export class Store {
         this.db.run("UPDATE ai_conversations SET title = ?, updated_at = ? WHERE id = ?", normalizedTitle, now(), conversationId);
         return this.getAiConversation(conversationId);
     }
+    setAiConversationRoleplayCharacter(conversationId, characterId) {
+        const conversation = this.db.get("SELECT * FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        const workId = requiredString(conversation, "work_id");
+        const previousCharacterId = optionalString(conversation, "roleplay_character_id");
+        if (previousCharacterId === characterId)
+            return this.getAiConversationSummary(conversationId);
+        const messageCount = Number(this.db.get("SELECT COUNT(*) AS count FROM ai_conversation_messages WHERE conversation_id = ?", conversationId)?.count ?? 0);
+        if (messageCount > 0) {
+            throw new AppError(409, previousCharacterId ? "ROLEPLAY_CHARACTER_LOCKED" : "ROLEPLAY_CONVERSATION_STARTED", previousCharacterId ? "角色扮演对话开始后不能退出模式或更换角色卡" : "当前对话已经开始，不能中途切换为角色扮演");
+        }
+        if (characterId) {
+            const character = this.getCharacter(characterId);
+            if (String(character.workId) !== workId) {
+                throw new AppError(400, "ROLEPLAY_CHARACTER_WORK_MISMATCH", "角色卡不属于当前作品");
+            }
+            if (character.mergedIntoCharacterId) {
+                throw new AppError(409, "ROLEPLAY_CHARACTER_MERGED", "已合并角色不能用于角色扮演");
+            }
+        }
+        this.db.transaction(() => {
+            this.db.run("UPDATE ai_conversations SET roleplay_character_id = ?, task_type = CASE WHEN ? IS NOT NULL THEN 'roleplay' ELSE task_type END, updated_at = ? WHERE id = ?", characterId, characterId, now(), conversationId);
+            this.audit(workId, "ai-conversation.roleplay-updated", "ai-conversation", conversationId, {
+                previousCharacterId,
+                characterId
+            });
+        });
+        return this.getAiConversationSummary(conversationId);
+    }
+    setAiConversationTaskType(conversationId, taskType) {
+        const conversation = this.db.get("SELECT * FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        const workId = requiredString(conversation, "work_id");
+        const previousCharacterId = optionalString(conversation, "roleplay_character_id");
+        const previousTaskType = optionalString(conversation, "task_type") ?? (previousCharacterId ? "roleplay" : "chat");
+        if (previousTaskType === taskType)
+            return this.getAiConversationSummary(conversationId);
+        const messageCount = Number(this.db.get("SELECT COUNT(*) AS count FROM ai_conversation_messages WHERE conversation_id = ?", conversationId)?.count ?? 0);
+        if (messageCount > 0) {
+            throw new AppError(409, "AI_CONVERSATION_TASK_LOCKED", "对话开始后不能切换任务类型");
+        }
+        this.db.transaction(() => {
+            this.db.run("UPDATE ai_conversations SET task_type = ?, roleplay_character_id = CASE WHEN ? = 'roleplay' THEN roleplay_character_id ELSE NULL END, updated_at = ? WHERE id = ?", taskType, taskType, now(), conversationId);
+            this.audit(workId, "ai-conversation.task-type-updated", "ai-conversation", conversationId, {
+                previousTaskType,
+                taskType
+            });
+        });
+        return this.getAiConversationSummary(conversationId);
+    }
+    setAiConversationContextScope(conversationId, scope) {
+        const conversation = this.db.get("SELECT * FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        const workId = requiredString(conversation, "work_id");
+        const assertWork = (record, code, label) => {
+            if (String(record.workId) !== workId)
+                throw new AppError(400, code, `${label}不属于当前作品`);
+        };
+        if (scope.chapterId)
+            assertWork(this.getChapter(scope.chapterId), "CHAPTER_WORK_MISMATCH", "章节");
+        if (scope.volumeId)
+            assertWork(this.getVolume(scope.volumeId), "VOLUME_WORK_MISMATCH", "卷");
+        for (const chapterId of scope.chapterIds ?? [])
+            assertWork(this.getChapter(chapterId), "CHAPTER_WORK_MISMATCH", "章节");
+        for (const characterId of scope.characterIds ?? [])
+            assertWork(this.getCharacter(characterId), "CHARACTER_WORK_MISMATCH", "角色");
+        for (const settingId of scope.settingIds ?? [])
+            assertWork(this.getSetting(settingId), "SETTING_WORK_MISMATCH", "设定");
+        const previousScope = json(optionalString(conversation, "context_scope_json") ?? "", { type: "none" });
+        const serializedScope = JSON.stringify(scope);
+        if (JSON.stringify(previousScope) === serializedScope)
+            return this.getAiConversationSummary(conversationId);
+        const messageCount = Number(this.db.get("SELECT COUNT(*) AS count FROM ai_conversation_messages WHERE conversation_id = ?", conversationId)?.count ?? 0);
+        if (messageCount > 0) {
+            throw new AppError(409, "AI_CONVERSATION_CONTEXT_LOCKED", "对话开始后不能切换上下文引用");
+        }
+        this.db.transaction(() => {
+            this.db.run("UPDATE ai_conversations SET context_scope_json = ?, updated_at = ? WHERE id = ?", serializedScope, now(), conversationId);
+            this.audit(workId, "ai-conversation.context-scope-updated", "ai-conversation", conversationId, {
+                previousScope,
+                scope
+            });
+        });
+        return this.getAiConversationSummary(conversationId);
+    }
     addAiConversationMessage(conversationId, input) {
         const conversation = this.db.get("SELECT * FROM ai_conversations WHERE id = ?", conversationId);
         if (!conversation)
@@ -4444,8 +4659,13 @@ export class Store {
         const sourceCompactedCount = Math.max(0, numberValue(conversation, "compacted_message_count"));
         const forkCompactedCount = targetIndex + 1 >= sourceCompactedCount ? Math.min(sourceCompactedCount, targetIndex + 1) : 0;
         const forkSummary = forkCompactedCount ? requiredString(conversation, "compacted_summary") : "";
+        const injectedEntitiesJson = optionalString(conversation, "injected_entities_json")
+            ?? JSON.stringify(EMPTY_AI_INJECTED_ENTITIES);
+        const systemClockText = optionalString(conversation, "system_clock_text") ?? "";
         this.db.transaction(() => {
-            this.db.run("INSERT INTO ai_conversations (id, work_id, title, compacted_summary, compacted_message_count, created_at, updated_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", forkId, requiredString(conversation, "work_id"), title.slice(0, 200), forkSummary, forkCompactedCount, timestamp, timestamp, currentRequestActor()?.userId ?? null);
+            this.db.run("INSERT INTO ai_conversations (id, work_id, roleplay_character_id, task_type, context_scope_json, title, compacted_summary, compacted_message_count, agent_tools_json, injected_entities_json, system_clock_text, created_at, updated_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", forkId, requiredString(conversation, "work_id"), optionalString(conversation, "roleplay_character_id"), optionalString(conversation, "task_type"), optionalString(conversation, "context_scope_json"), title.slice(0, 200), forkSummary, forkCompactedCount, conversation.agent_tools_json == null
+                ? JSON.stringify(normalizeWorkAgentTools(this.getWorkAiSettings(requiredString(conversation, "work_id")).agentTools))
+                : String(conversation.agent_tools_json), injectedEntitiesJson, systemClockText, timestamp, timestamp, currentRequestActor()?.userId ?? null);
             for (const message of messages.slice(0, targetIndex + 1)) {
                 this.db.run("INSERT INTO ai_conversation_messages (id, conversation_id, role, content, citations_json, metadata_json, request_id, created_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", id("message"), forkId, requiredString(message, "role"), requiredString(message, "content"), requiredString(message, "citations_json"), requiredString(message, "metadata_json"), optionalString(message, "request_id"), requiredString(message, "created_at"), currentRequestActor()?.userId ?? null);
             }
@@ -4453,6 +4673,10 @@ export class Store {
         return this.getAiConversation(forkId);
     }
     mapAiConversation(row) {
+        const roleplayCharacterId = optionalString(row, "roleplay_character_id");
+        const roleplayCharacter = roleplayCharacterId
+            ? this.db.get("SELECT id, name, code FROM characters WHERE id = ? AND work_id = ?", roleplayCharacterId, requiredString(row, "work_id"))
+            : undefined;
         return {
             id: requiredString(row, "id"),
             workId: requiredString(row, "work_id"),
@@ -4462,10 +4686,36 @@ export class Store {
             compactedMessageCount: numberValue(row, "compacted_message_count"),
             hasCompactedSummary: Boolean(requiredString(row, "compacted_summary")),
             contextWarningPending: Boolean(optionalString(row, "context_warning_at")),
+            taskType: optionalString(row, "task_type") ?? (roleplayCharacterId ? "roleplay" : "chat"),
+            contextScope: json(optionalString(row, "context_scope_json") ?? "", { type: "none" }),
+            roleplayCharacter: roleplayCharacter ? {
+                id: requiredString(roleplayCharacter, "id"),
+                name: requiredString(roleplayCharacter, "name"),
+                code: requiredString(roleplayCharacter, "code")
+            } : null,
+            agentTools: row.agent_tools_json == null || row.agent_tools_json === undefined
+                ? null
+                : normalizeWorkAgentTools(row.agent_tools_json),
             createdAt: requiredString(row, "created_at"),
             updatedAt: requiredString(row, "updated_at")
         };
     }
+    /** 锁定本对话可用工具集；已锁定则保持不变，避免中途改作品设置破坏 prompt cache。 */
+    ensureAiConversationAgentTools(conversationId, workId) {
+        const conversation = this.db.get("SELECT id, work_id, agent_tools_json FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        if (requiredString(conversation, "work_id") !== workId) {
+            throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
+        }
+        if (conversation.agent_tools_json != null && conversation.agent_tools_json !== undefined) {
+            return normalizeWorkAgentTools(conversation.agent_tools_json);
+        }
+        const agentTools = normalizeWorkAgentTools(this.getWorkAiSettings(workId).agentTools);
+        this.db.run("UPDATE ai_conversations SET agent_tools_json = ?, updated_at = ? WHERE id = ? AND agent_tools_json IS NULL", JSON.stringify(agentTools), now(), conversationId);
+        const locked = this.db.get("SELECT agent_tools_json FROM ai_conversations WHERE id = ?", conversationId);
+        return normalizeWorkAgentTools(locked?.agent_tools_json ?? agentTools);
+    }
     mapAiConversationMessage(row) {
         return {
             id: requiredString(row, "id"),
@@ -5757,6 +6007,9 @@ export class Store {
         if (scope.type === "settings") {
             return [{ type: "settings", title: "仅设定集" }];
         }
+        if (scope.type === "settings-catalog") {
+            return [{ type: "settings-catalog", title: "设定库" }];
+        }
         if (scope.type === "selection" && typeof scope.selection === "string") {
             return [{ type: "selection", selection: scope.selection }];
         }
@@ -5869,6 +6122,22 @@ export class Store {
         }
         return lines.join("\n").trimEnd() + "\n";
     }
+    async exportDocx(workId) {
+        const tree = this.getWorkTree(workId);
+        const cover = this.findWorkCover(workId);
+        const volumes = tree.volumes.map((volume) => ({
+            title: String(volume.title),
+            chapters: volume.chapters.map((chapter) => ({
+                title: String(chapter.title),
+                content: String(chapter.content ?? "")
+            }))
+        }));
+        return exportWorkDocx({
+            title: String(tree.title),
+            volumes,
+            cover: cover ? { mimeType: cover.mimeType, content: cover.content } : null
+        });
+    }
     listAuditLogs(workId) {
         this.getWork(workId);
         return this.db.all(`SELECT log.*, user.display_name AS actor_display_name, user.username AS actor_username