@musnows/scriverse 0.6.6 → 0.6.8

package/dist/store.js

@@ -16,7 +16,8 @@ export const WORK_AGENT_TOOL_IDS = [
     "grep",
     "search_story_entities",
     "read_character_sections",
-    "search_drafts"
+    "search_drafts",
+    "image"
 ];
 const DEFAULT_WORK_AGENT_TOOLS = [...WORK_AGENT_TOOL_IDS];
 export function normalizeWorkAgentTools(value) {
@@ -681,13 +682,20 @@ export class Store {
         const row = this.db.get("SELECT * FROM platform_ai_settings WHERE id = 1");
         return {
             systemPrompt: String(row?.system_prompt ?? ""),
+            imageToolModelId: row?.image_tool_model_id === null || row?.image_tool_model_id === undefined
+                ? null
+                : String(row.image_tool_model_id),
             updatedAt: String(row?.updated_at ?? "")
         };
     }
     updatePlatformAiSettings(input) {
         const timestamp = now();
-        this.db.run(`INSERT INTO platform_ai_settings (id, system_prompt, updated_at) VALUES (1, ?, ?)
-       ON CONFLICT(id) DO UPDATE SET system_prompt = excluded.system_prompt, updated_at = excluded.updated_at`, input.systemPrompt ?? String(this.getPlatformAiSettings().systemPrompt), timestamp);
+        const current = this.getPlatformAiSettings();
+        this.db.run(`INSERT INTO platform_ai_settings (id, system_prompt, image_tool_model_id, updated_at) VALUES (1, ?, ?, ?)
+       ON CONFLICT(id) DO UPDATE SET system_prompt = excluded.system_prompt,
+         image_tool_model_id = excluded.image_tool_model_id, updated_at = excluded.updated_at`, input.systemPrompt ?? String(current.systemPrompt), input.imageToolModelId === undefined
+            ? (current.imageToolModelId === null ? null : String(current.imageToolModelId))
+            : input.imageToolModelId, timestamp);
         return this.getPlatformAiSettings();
     }
     getPlatformUiSettings() {
@@ -754,6 +762,9 @@ export class Store {
             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),
+            imageToolModelId: row?.image_tool_model_id === null || row?.image_tool_model_id === undefined
+                ? null
+                : String(row.image_tool_model_id),
             alwaysIncludeSettingInfo: Number(row?.always_include_setting_info ?? 0) === 1,
             titleGenerationModelId: row?.title_generation_model_id === null || row?.title_generation_model_id === undefined
                 ? null
@@ -779,6 +790,9 @@ export class Store {
         const nextAgentToolCallLimit = input.agentToolCallLimit ?? Number(current.agentToolCallLimit);
         const nextAgentToolCallGlobalMultiplier = input.agentToolCallGlobalMultiplier ?? Number(current.agentToolCallGlobalMultiplier);
         const nextAgentTools = normalizeWorkAgentTools(input.agentTools ?? current.agentTools);
+        const nextImageToolModelId = input.imageToolModelId === undefined
+            ? (current.imageToolModelId ? String(current.imageToolModelId) : null)
+            : input.imageToolModelId?.trim() || null;
         const nextAlwaysIncludeSettingInfo = input.alwaysIncludeSettingInfo ?? Boolean(current.alwaysIncludeSettingInfo);
         const nextTitleGenerationModelId = input.titleGenerationModelId === undefined
             ? (current.titleGenerationModelId ? String(current.titleGenerationModelId) : null)
@@ -788,8 +802,8 @@ export class Store {
          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_tool_call_limit, agent_tool_call_global_multiplier,
-         agent_tools_json, title_generation_model_id, always_include_setting_info, updated_at
-       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+         agent_tools_json, title_generation_model_id, image_tool_model_id, always_include_setting_info, updated_at
+       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
        ON CONFLICT(work_id) DO UPDATE SET
          system_prompt = excluded.system_prompt,
          daily_token_quota = excluded.daily_token_quota,
@@ -808,8 +822,9 @@ export class Store {
          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,
+         image_tool_model_id = excluded.image_tool_model_id,
          always_include_setting_info = excluded.always_include_setting_info,
-         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, nextAlwaysIncludeSettingInfo ? 1 : 0, 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, nextImageToolModelId, nextAlwaysIncludeSettingInfo ? 1 : 0, timestamp);
         this.audit(workId, "work.ai-settings.updated", "work-ai-settings", workId, {
             systemPromptChanged: input.systemPrompt !== undefined,
             dailyTokenQuota: nextDailyTokenQuota,
@@ -823,6 +838,7 @@ export class Store {
             agentToolCallLimit: Math.min(48, Math.max(5, nextAgentToolCallLimit)),
             agentToolCallGlobalMultiplier: Math.min(6, Math.max(1, nextAgentToolCallGlobalMultiplier)),
             agentTools: nextAgentTools,
+            imageToolModelId: nextImageToolModelId,
             alwaysIncludeSettingInfo: nextAlwaysIncludeSettingInfo,
             titleGenerationModelId: nextTitleGenerationModelId
         });
@@ -3425,6 +3441,16 @@ export class Store {
             throw notFound("附件");
         return this.mapAttachment(row);
     }
+    getSettingAttachment(workId, attachmentId) {
+        const row = this.db.get(`SELECT attachment.*
+       FROM attachments attachment
+       JOIN attachment_references reference ON reference.attachment_id = attachment.id
+       WHERE attachment.id = ? AND attachment.work_id = ? AND reference.work_id = ? AND reference.entity_type = 'setting'
+       LIMIT 1`, attachmentId, workId, workId);
+        if (!row)
+            throw new AppError(404, "SETTING_IMAGE_ATTACHMENT_NOT_FOUND", "图片附件不存在或未被设定正文引用");
+        return this.mapAttachment(row);
+    }
     attachmentModules(attachmentId) {
         this.getAttachment(attachmentId);
         const modules = new Set();