koishi-plugin-aka-comfy-agent 0.2.0 → 0.2.2

package/dist/index.d.mts

@@ -11,6 +11,20 @@ interface LlmProviderConfig {
     /** 视觉模型请求正文（仅 vision 使用；留空则用默认） */
     userText: string;
 }
+/**
+ * 程序级锚点 / 背景模板的可配置覆盖（框架化 = 默认留空，回退 prompts.ts 框架默认）。
+ * 完整措辞只保留在私有文档，经此处 koishi.yml 配置还原，无需改源码。
+ */
+interface PromptOverrideConfig {
+    /** 身份锁定锚点（完整措辞；留空 = 用 prompts.ts 框架默认） */
+    identityAnchor: string;
+    /** 三点布光光场锚点（完整措辞；留空 = 用 prompts.ts 框架默认） */
+    lightfieldAnchor: string;
+    /** 背景替换模板，含 {color} 占位（留空 = 用 prompts.ts 框架默认） */
+    backgroundClauseTemplate: string;
+    /** 背景覆盖模板，含 {desc} 占位（留空 = 用 prompts.ts 框架默认） */
+    backgroundOverrideTemplate: string;
+}
 interface Config {
     mainLLM: LlmProviderConfig;
     visionLLM: LlmProviderConfig;
@@ -20,6 +34,8 @@ interface Config {
     };
     /** 背景色兜底：DS 未给或非法时用此值 */
     defaultBackgroundColor: string;
+    /** 程序级锚点/背景模板覆盖（框架化；完整措辞经此还原） */
+    prompt: PromptOverrideConfig;
     session: {
         maxTurns: number;
         idleTimeoutMs: number;

package/dist/index.d.ts

@@ -11,6 +11,20 @@ interface LlmProviderConfig {
     /** 视觉模型请求正文（仅 vision 使用；留空则用默认） */
     userText: string;
 }
+/**
+ * 程序级锚点 / 背景模板的可配置覆盖（框架化 = 默认留空，回退 prompts.ts 框架默认）。
+ * 完整措辞只保留在私有文档，经此处 koishi.yml 配置还原，无需改源码。
+ */
+interface PromptOverrideConfig {
+    /** 身份锁定锚点（完整措辞；留空 = 用 prompts.ts 框架默认） */
+    identityAnchor: string;
+    /** 三点布光光场锚点（完整措辞；留空 = 用 prompts.ts 框架默认） */
+    lightfieldAnchor: string;
+    /** 背景替换模板，含 {color} 占位（留空 = 用 prompts.ts 框架默认） */
+    backgroundClauseTemplate: string;
+    /** 背景覆盖模板，含 {desc} 占位（留空 = 用 prompts.ts 框架默认） */
+    backgroundOverrideTemplate: string;
+}
 interface Config {
     mainLLM: LlmProviderConfig;
     visionLLM: LlmProviderConfig;
@@ -20,6 +34,8 @@ interface Config {
     };
     /** 背景色兜底：DS 未给或非法时用此值 */
     defaultBackgroundColor: string;
+    /** 程序级锚点/背景模板覆盖（框架化；完整措辞经此还原） */
+    prompt: PromptOverrideConfig;
     session: {
         maxTurns: number;
         idleTimeoutMs: number;

package/dist/index.js

@@ -32,8 +32,8 @@ var import_koishi3 = require("koishi");
 var import_koishi = require("koishi");
 
 // src/llm/prompts.ts
-var MAIN_SYSTEM_PROMPT = `You are the edit-intent author for an image re-render pipeline. The pipeline prepends fixed identity, backdrop and lighting anchors before your output. Describe only the user's requested look (material or optical adjustments; a background change or light override when explicitly asked) as one short natural English clause. Placeholder default; the full prompt is restored via config override from the private design doc.`;
-var GLM_PROBE_PROMPT = `You are an asset extractor for a character image. From the screenshot, list every visual detail that a re-render could drift, and output a structured English JSON inventory (species, per-part colors, accessories, body surface, tail, pose, background). Do not invent details; use "none" when absent. Placeholder default; full prompt (with exact schema) is in the private design doc.`;
+var MAIN_SYSTEM_PROMPT = `You are the edit-intent author for an image re-render pipeline. The pipeline prepends fixed identity, backdrop and lighting anchors before your output. Describe only the user's requested look — material or optical adjustments, or a background change / light override when explicitly asked — as one short natural English clause.`;
+var GLM_PROBE_PROMPT = `You are an asset extractor for a character image. From the screenshot, list every visual detail that a re-render could drift, and output a structured English JSON inventory covering species, per-part colors, accessories, body surface, tail, pose, and background. Do not invent details; use "none" when absent.`;
 var VISION_USER_TEXT = "分析这张游戏角色截图，输出资产清单 JSON。";
 var VISION_CROSSCHECK_TEXT = `这是对同一张源图的【资产清单对抗复核】。
 
@@ -42,15 +42,19 @@ var VISION_CROSSCHECK_TEXT = `这是对同一张源图的【资产清单对抗
 
 请基于源图逐项对抗验证上一轮的每一项字段，明确指出错误/遗漏/脑补，修正会导致重绘跑偏的项，保持英文枚举与同一 schema 不变；不确定时写 "none"。只输出修正后的权威 JSON。`;
 var VISION_REPORT_LABEL = "资产清单(GLM 修正后)";
-var IDENTITY_ANCHOR = `Preserve the subject, its pose, outfit, accessories and colors of the source image exactly, keeping the same composition and framing, at high resolution, with no UI and no watermark. (Generic placeholder; the full locked wording is in the private design doc.)`;
-var LIGHTFIELD_ANCHOR = `Evenly light the character's front with gentle volume, keep the shadows soft and the face readable, and add a subtle rim so the silhouette lifts off the backdrop with a soft contact shadow. (Generic placeholder; the full locked wording is in the private design doc.)`;
-function buildBackdropClause(color) {
+var IDENTITY_ANCHOR = `Preserve the subject, its pose, outfit, accessories and colors of the source image exactly, keeping the same composition and framing, at high resolution, with no UI and no watermark.`;
+var LIGHTFIELD_ANCHOR = `Evenly light the character's front with gentle volume, keep the shadows soft and the face readable, and add a subtle rim so the silhouette lifts off the backdrop with a soft contact shadow.`;
+function buildBackdropClause(color, template) {
   const c = (color || "yellow").trim().toLowerCase();
-  return `Place the character on a plain solid ${c} seamless studio backdrop that fully replaces the source background. (Generic placeholder; full wording in private doc.)`;
+  const t = template?.trim();
+  if (t) return t.replace("{color}", c);
+  return `Place the character on a plain solid ${c} seamless studio backdrop that fully replaces the source background.`;
 }
-function buildBackdropFromOverride(desc) {
+function buildBackdropFromOverride(desc, template) {
   const d = (desc || "").trim().replace(/\s+/g, " ");
-  return `Place the character against ${d}, fully replacing the source background. (Generic placeholder; full wording in private doc.)`;
+  const t = template?.trim();
+  if (t) return t.replace("{desc}", d);
+  return `Place the character against ${d}, fully replacing the source background.`;
 }
 
 // src/config.ts
@@ -70,6 +74,12 @@ var Config = import_koishi.Schema.object({
     timeoutMs: import_koishi.Schema.number().min(3e4).max(6e5).default(18e4).description("超时(ms)")
   }).description("ComfyUI"),
   defaultBackgroundColor: import_koishi.Schema.string().default("yellow").description("背景色兜底（DS 未给或非法时使用；英文颜色词）"),
+  prompt: import_koishi.Schema.object({
+    identityAnchor: import_koishi.Schema.string().role("textarea").default("").description("身份锁定锚点完整措辞（留空=用 prompts.ts 框架默认；完整原文见私有文档）"),
+    lightfieldAnchor: import_koishi.Schema.string().role("textarea").default("").description("三点布光光场锚点完整措辞（留空=用 prompts.ts 框架默认；完整原文见私有文档）"),
+    backgroundClauseTemplate: import_koishi.Schema.string().role("textarea").default("").description("背景替换模板，含 {color} 占位（留空=用 prompts.ts 框架默认；完整原文见私有文档）"),
+    backgroundOverrideTemplate: import_koishi.Schema.string().role("textarea").default("").description("背景覆盖模板，含 {desc} 占位（留空=用 prompts.ts 框架默认；完整原文见私有文档）")
+  }).description("程序级锚点/背景模板覆盖（框架化：发布包默认空，完整措辞经此 koishi.yml 配置还原，无需改源码）"),
   session: import_koishi.Schema.object({
     maxTurns: import_koishi.Schema.number().min(1).max(50).default(10).description("最大轮次"),
     idleTimeoutMs: import_koishi.Schema.number().min(6e4).max(24 * 60 * 60 * 1e3).default(10 * 60 * 1e3).description("空闲销毁(ms)"),
@@ -420,9 +430,11 @@ async function runVisionAnalysis(vision, dataUrl, systemPrompt, userText) {
 async function runGenerateOnly(deps, session, userFeedback) {
   const { main, comfy, lock, logger } = deps;
   const decision = await runMainLlm(main, session, userFeedback, deps.mainSystemPrompt);
-  const backdrop = decision.background ? buildBackdropFromOverride(decision.background) : buildBackdropClause(deps.defaultBackgroundColor);
-  const promptText = `${IDENTITY_ANCHOR} ${backdrop} ${LIGHTFIELD_ANCHOR} ${decision.prompt}`;
-  logger.info("brain prompt len=%d ds=%d backdrop=%s lightfield=%d preview=%s", promptText.length, decision.prompt.length, decision.background?.slice(0, 40) || deps.defaultBackgroundColor || "yellow", LIGHTFIELD_ANCHOR.length, promptText.slice(0, 160).replace(/\s+/g, " "));
+  const identity = deps.identityAnchor?.trim() || IDENTITY_ANCHOR;
+  const lightfield = deps.lightfieldAnchor?.trim() || LIGHTFIELD_ANCHOR;
+  const backdrop = decision.background ? buildBackdropFromOverride(decision.background, deps.backgroundOverrideTemplate) : buildBackdropClause(deps.defaultBackgroundColor, deps.backgroundClauseTemplate);
+  const promptText = `${identity} ${backdrop} ${lightfield} ${decision.prompt}`;
+  logger.info("brain prompt len=%d ds=%d anchor=%d lightfield=%d preview=%s", promptText.length, decision.prompt.length, identity.length, lightfield.length, promptText.slice(0, 160).replace(/\s+/g, " "));
   logger.info("brain note=%s", decision.note ?? "");
   let seed = session.params.seed;
   if (!Number.isFinite(seed) || seed < 0) seed = randomSeed();
@@ -686,7 +698,11 @@ function createRenderFacade(deps) {
                 visionUserText: deps.visionUserText,
                 visionCrosscheckRounds: deps.visionCrosscheckRounds,
                 visionCrosscheckText: deps.visionCrosscheckText,
-                defaultBackgroundColor: deps.defaultBackgroundColor
+                defaultBackgroundColor: deps.defaultBackgroundColor,
+                identityAnchor: deps.identityAnchor,
+                lightfieldAnchor: deps.lightfieldAnchor,
+                backgroundClauseTemplate: deps.backgroundClauseTemplate,
+                backgroundOverrideTemplate: deps.backgroundOverrideTemplate
               },
               downloaded.dataUrl
             );
@@ -736,7 +752,11 @@ function createRenderFacade(deps) {
           visionUserText: deps.visionUserText,
           visionCrosscheckRounds: deps.visionCrosscheckRounds,
           visionCrosscheckText: deps.visionCrosscheckText,
-          defaultBackgroundColor: deps.defaultBackgroundColor
+          defaultBackgroundColor: deps.defaultBackgroundColor,
+          identityAnchor: deps.identityAnchor,
+          lightfieldAnchor: deps.lightfieldAnchor,
+          backgroundClauseTemplate: deps.backgroundClauseTemplate,
+          backgroundOverrideTemplate: deps.backgroundOverrideTemplate
         },
         session,
         input.instruction,
@@ -1268,6 +1288,10 @@ function apply(ctx, config) {
     visionCrosscheckRounds: config.visionCrosscheckRounds,
     visionCrosscheckText: VISION_CROSSCHECK_TEXT,
     defaultBackgroundColor: config.defaultBackgroundColor,
+    identityAnchor: config.prompt?.identityAnchor?.trim() || IDENTITY_ANCHOR,
+    lightfieldAnchor: config.prompt?.lightfieldAnchor?.trim() || LIGHTFIELD_ANCHOR,
+    backgroundClauseTemplate: config.prompt?.backgroundClauseTemplate?.trim() || void 0,
+    backgroundOverrideTemplate: config.prompt?.backgroundOverrideTemplate?.trim() || void 0,
     // Comfy 生成轮询期间编辑「正在生成」进度消息（约每 15s 一次，由 brain onTick 触发）
     onGenerateProgress: (elapsedMs) => {
       const g = genProgress;
@@ -1352,7 +1376,11 @@ function apply(ctx, config) {
     visionUserText: config.visionLLM.userText?.trim() || VISION_USER_TEXT,
     visionCrosscheckRounds: config.visionCrosscheckRounds,
     visionCrosscheckText: VISION_CROSSCHECK_TEXT,
-    defaultBackgroundColor: config.defaultBackgroundColor
+    defaultBackgroundColor: config.defaultBackgroundColor,
+    identityAnchor: config.prompt?.identityAnchor?.trim() || IDENTITY_ANCHOR,
+    lightfieldAnchor: config.prompt?.lightfieldAnchor?.trim() || LIGHTFIELD_ANCHOR,
+    backgroundClauseTemplate: config.prompt?.backgroundClauseTemplate?.trim() || void 0,
+    backgroundOverrideTemplate: config.prompt?.backgroundOverrideTemplate?.trim() || void 0
   });
   const lastGeneratedStore = {
     get: (id) => runtime.lastGenByConv.get(id),