@waterwx/dsh-novel-forge 1.6.0 → 1.7.0

package/src/index.ts

@@ -15,7 +15,7 @@ import z from 'schemastery'
 import type {} from '@deepseek-ai/dsh-host-webserver'
 import type {} from '@deepseek-ai/dsh-llm'
 import type {} from '@deepseek-ai/dsh-system-prompt'
-import type { ConfigPatch, NovelConfig } from './protocol.ts'
+import type { ConfigPatch, ImageModelConfig, NovelConfig } from './protocol.ts'
 import { makeRoutes } from './routes.ts'
 import { activeBookOutputDir } from './bookshelf.ts'
 
@@ -62,11 +62,13 @@ export interface Config {
   autoAuthorReview?: boolean
   /** 修订/润色产出草稿后自动附带一次 AI 审查（默认开，可在设置页关闭省 token）。 */
   autoReviewAfterRevise?: boolean
-  /** 豆包/Seedream 生图 API Key。 */
+  /** 生图模型库（多套并存，启用一条生效）。 */
+  imageModels?: ImageModelConfig[]
+  /** 豆包/Seedream 生图 API Key（旧字段，兼容迁移用）。 */
   imageApiKey?: string
-  /** 豆包/Seedream 生图模型 ID。 */
+  /** 豆包/Seedream 生图模型 ID（旧字段，兼容迁移用）。 */
   imageApiModel?: string
-  /** 是否启用豆包生图（默认关）。 */
+  /** 是否启用豆包生图（旧字段，兼容迁移用）。 */
   imageApiEnabled?: boolean
 }
 
@@ -85,6 +87,14 @@ export const Config: z<Config> = z.object({
   autoReview: z.boolean().default(true),
   autoAuthorReview: z.boolean().default(true),
   autoReviewAfterRevise: z.boolean().default(true),
+  imageModels: z.array(z.object({
+    id: z.string().default(''),
+    name: z.string().default(''),
+    baseURL: z.string().default(''),
+    apiKey: z.string().default(''),
+    model: z.string().default(''),
+    enabled: z.boolean().default(false),
+  })).default([]),
   imageApiKey: z.string().default(''),
   imageApiModel: z.string().default(''),
   imageApiEnabled: z.boolean().default(false),
@@ -126,12 +136,33 @@ export function resolveConfig(value: Partial<Config> | undefined): NovelConfig {
     autoReview: value?.autoReview ?? DEFAULT_AUTO_REVIEW,
     autoAuthorReview: value?.autoAuthorReview ?? DEFAULT_AUTO_AUTHOR_REVIEW,
     autoReviewAfterRevise: value?.autoReviewAfterRevise ?? DEFAULT_AUTO_REVIEW_AFTER_REVISE,
-    imageApiKey: value?.imageApiKey,
-    imageApiModel: value?.imageApiModel,
+    // 生图模型库：只有「从未保存过 imageModels」时才用旧配置迁移一条（豆包默认地址）；
+    // 已保存为空数组 = 用户主动删光，保持空，不再复活。
+    imageModels: value?.imageModels !== undefined
+      ? value.imageModels
+      : (value?.imageApiKey !== undefined && value.imageApiKey !== ''
+          ? [{ id: 'img-legacy', name: '豆包（旧配置）', baseURL: DEFAULT_IMAGE_BASE, apiKey: value.imageApiKey, model: value.imageApiModel ?? '', enabled: true }]
+          : []),
+    // 运行时生效值 = 启用条目（兼容旧代码读取 imageApiKey/imageApiModel/imageApiEnabled）。
+    imageBaseUrl: (() => {
+      const active = (value?.imageModels ?? []).find(m => m.enabled) ?? (value?.imageModels ?? [])[0]
+      return active?.baseURL !== undefined && active.baseURL !== '' ? active.baseURL : (value?.imageApiKey !== undefined && value.imageApiKey !== '' ? DEFAULT_IMAGE_BASE : undefined)
+    })(),
+    imageApiKey: (() => {
+      const active = (value?.imageModels ?? []).find(m => m.enabled) ?? (value?.imageModels ?? [])[0]
+      return active?.apiKey ?? value?.imageApiKey
+    })(),
+    imageApiModel: (() => {
+      const active = (value?.imageModels ?? []).find(m => m.enabled) ?? (value?.imageModels ?? [])[0]
+      return active?.model ?? value?.imageApiModel
+    })(),
     imageApiEnabled: value?.imageApiEnabled ?? false,
   }
 }
 
+/** 生图默认接口地址（豆包 ark，OpenAI 兼容）。 */
+const DEFAULT_IMAGE_BASE = 'https://ark.cn-beijing.volces.com/api/v3'
+
 /**
  * Mount the routes and announcement.
  * @param ctx - host plugin context carrying webServer/llm/systemPrompt.

package/src/protocol.ts

@@ -50,6 +50,8 @@ export const NOVEL_API = {
   storyboardTable: '/api/dsh-novel-forge/storyboard/table',
   /** 分镜·提示词级：分镜表 → 即梦可粘贴视频提示词。 */
   storyboardPrompts: '/api/dsh-novel-forge/storyboard/prompts',
+  /** 生图接口连通性测试（设置页每个模型条目用）。 */
+  imageTest: '/api/dsh-novel-forge/image-test',
   /** 重置项目（可选携带新大纲）：清空设定/卷/章节/伏笔/资产/事实库。 */
   reset: '/api/dsh-novel-forge/reset',
   /** 全书一致性质检：LLM 扫描已生成章节，输出矛盾问题清单。 */
@@ -875,12 +877,35 @@ export interface RolesRequest {
   label?: string
   /** op='imageGenerate' 时的漫画风格预设 id（用于统一角色图与漫画页风格）。 */
   style?: string
+  /** op='imageGenerate' 时指定生图模型库条目 id（缺省用启用条目）。 */
+  modelId?: string
   /** op='visual' / op='promptKit' 时的漫剧基底风格 id（角色图按方案风格重出）。 */
   styleId?: string
   /** op='visual' / op='promptKit' 时的可选滤镜风格 id。 */
   filterId?: string
 }
 
+/** POST /image-test 请求：测一个生图接口的连通性与延迟（不落盘）。 */
+export interface ImageTestRequest {
+  /** OpenAI 兼容生图接口地址。 */
+  baseURL: string
+  /** API Key。 */
+  apiKey: string
+  /** 模型 id（顺带校验是否在接口模型列表里，可选）。 */
+  model?: string
+}
+
+/** POST /image-test 响应。 */
+export interface ImageTestResponse {
+  ok: boolean
+  /** 连通延迟（毫秒）。 */
+  ms?: number
+  /** 失败原因。 */
+  message?: string
+  /** 模型 id 是否出现在接口的模型列表（探测成功时才有）。 */
+  modelFound?: boolean
+}
+
 /** POST /roles 响应。 */
 export interface RolesResponse {
   roles: RoleRecord[]
@@ -1038,6 +1063,22 @@ export interface ProjectState {
   updatedAt: string
 }
 
+/** 生图模型配置（模型库条目，可多套并存，启用一条生效）。 */
+export interface ImageModelConfig {
+  /** 条目 id（本地唯一）。 */
+  id: string
+  /** 展示名（如「豆包 Seedream」「即梦」）。 */
+  name: string
+  /** OpenAI 兼容生图接口地址（如 https://ark.cn-beijing.volces.com/api/v3）。 */
+  baseURL: string
+  /** API Key。 */
+  apiKey: string
+  /** 生图模型 id（如 doubao-seedream-5-0-pro-260628）。 */
+  model: string
+  /** 是否启用（同时只启用一条）。 */
+  enabled: boolean
+}
+
 /** Runtime config surface exposed to the panel (subset of plugin Config). */
 export interface NovelConfig {
   /** Absolute path of the default docx outline to load. */
@@ -1064,11 +1105,15 @@ export interface NovelConfig {
   autoAuthorReview: boolean
   /** 修订/润色产出草稿后是否自动附带一次 AI 审查（工作区显示新稿评分与剩余问题）。 */
   autoReviewAfterRevise: boolean
-  /** 豆包/Seedream 生图 API Key（保存在服务端配置，不暴露给前端明文回显）。 */
+  /** 生图模型库（多套并存，启用一条生效；旧 imageApiKey/imageApiModel 自动迁移为一条）。 */
+  imageModels?: ImageModelConfig[]
+  /** 运行时生效的生图接口地址（= 启用条目，兼容旧代码读取）。 */
+  imageBaseUrl?: string
+  /** 运行时生效的生图 API Key（= 启用条目，兼容旧代码读取）。 */
   imageApiKey?: string
-  /** 豆包/Seedream 生图模型 ID（如 doubao-seedream-5-0-pro-260628）。 */
+  /** 运行时生效的生图模型 id（= 启用条目，兼容旧代码读取）。 */
   imageApiModel?: string
-  /** 是否启用豆包生图（默认关，避免误调用付费 API）。 */
+  /** 是否启用生图（= 存在启用条目）。 */
   imageApiEnabled?: boolean
 }
 
@@ -1286,6 +1331,8 @@ export interface ConfigPatch {
   imageApiKey?: string
   imageApiModel?: string
   imageApiEnabled?: boolean
+  /** 生图模型库（完整替换保存）。 */
+  imageModels?: ImageModelConfig[]
 }
 
 // ------------------------------------------------------------ assistant

package/src/routes.ts

@@ -95,6 +95,8 @@ import {
   type StoryboardTableRequest,
   type StoryboardTableResponse,
   type StoryboardPromptsRequest,
+  type ImageTestRequest,
+  type ImageTestResponse,
   type StoryboardPromptsResponse,
   type StyleEngineRequest,
   type SummaryRequest,
@@ -146,6 +148,7 @@ import {
   reverseOutlineFromChapters,
   breakdownBook,
   generateRoleReferenceImage,
+  testImageEndpoint,
   generateStoryboardSkeleton,
   generateStoryboardTable,
   generateStoryboardPrompts,
@@ -1854,7 +1857,7 @@ export function makeRoutes(deps: NovelRoutesDeps): WebRoute[] {
           return
         }
         try {
-          const imageUrl = await generateRoleReferenceImage(ctx, config, project, config.outputDir, name, body?.style ?? '')
+          const imageUrl = await generateRoleReferenceImage(ctx, config, project, config.outputDir, name, body?.style ?? '', body?.modelId)
           project.updatedAt = new Date().toISOString()
           saveProject(config.outputDir, project)
           writeJson(res, 200, { roles: project.roles, imageUrl } satisfies RolesResponse)
@@ -2192,6 +2195,27 @@ export function makeRoutes(deps: NovelRoutesDeps): WebRoute[] {
     },
   }
 
+  // --------------------------------------------------------- image-test
+  /** 生图接口连通性测试：GET {baseURL}/models + 计时（设置页模型条目用）。 */
+  const imageTestRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.imageTest,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      const body = await readJsonBody<ImageTestRequest>(req)
+      if (body?.baseURL === undefined || body?.apiKey === undefined) {
+        writeJson(res, 400, { error: 'baseURL 与 apiKey 必填' })
+        return
+      }
+      try {
+        const result = await testImageEndpoint(body.baseURL, body.apiKey, body.model)
+        writeJson(res, 200, result satisfies ImageTestResponse)
+      } catch (error) {
+        writeJson(res, 500, { error: `测试失败：${(error as Error).message}` })
+      }
+    },
+  }
+
   // ---------------------------------------------------- storyboard table
   /** 分镜·导演级：骨架 → 分镜表（镜头级）。 */
   const storyboardTableRoute: WebRoute = {
@@ -2576,6 +2600,7 @@ export function makeRoutes(deps: NovelRoutesDeps): WebRoute[] {
     manhuaPlansRoute,
     styleImageRoute,
     storyboardSkeletonRoute,
+    imageTestRoute,
     storyboardTableRoute,
     storyboardPromptsRoute,
     breakdownRoute,