@fre4x/comfyui 1.1.0-beta.4 → 1.1.0

package/dist/index.js

@@ -3223,8 +3223,8 @@ var require_utils = __commonJS({
       }
       return ind;
     }
-    function removeDotSegments(path2) {
-      let input = path2;
+    function removeDotSegments(path4) {
+      let input = path4;
       const output = [];
       let nextSlash = -1;
       let len = 0;
@@ -3423,8 +3423,8 @@ var require_schemes = __commonJS({
         wsComponent.secure = void 0;
       }
       if (wsComponent.resourceName) {
-        const [path2, query] = wsComponent.resourceName.split("?");
-        wsComponent.path = path2 && path2 !== "/" ? path2 : void 0;
+        const [path4, query] = wsComponent.resourceName.split("?");
+        wsComponent.path = path4 && path4 !== "/" ? path4 : void 0;
         wsComponent.query = query;
         wsComponent.resourceName = void 0;
       }
@@ -6786,12 +6786,12 @@ var require_dist = __commonJS({
         throw new Error(`Unknown format "${name}"`);
       return f;
     };
-    function addFormats(ajv, list, fs2, exportName) {
+    function addFormats(ajv, list, fs4, exportName) {
       var _a2;
       var _b;
       (_a2 = (_b = ajv.opts.code).formats) !== null && _a2 !== void 0 ? _a2 : _b.formats = (0, codegen_1._)`require("ajv-formats/dist/formats").${exportName}`;
       for (const f of list)
-        ajv.addFormat(f, fs2[f]);
+        ajv.addFormat(f, fs4[f]);
     }
     module.exports = exports = formatsPlugin;
     Object.defineProperty(exports, "__esModule", { value: true });
@@ -6800,11 +6800,9 @@ var require_dist = __commonJS({
 });
 
 // src/index.ts
-import * as fs from "node:fs/promises";
-import * as crypto from "node:crypto";
+import * as fs3 from "node:fs/promises";
 import { realpathSync } from "node:fs";
-import { homedir } from "node:os";
-import path from "node:path";
+import path3 from "node:path";
 import { fileURLToPath } from "node:url";
 
 // ../packages/shared/dist/errors.js
@@ -7873,10 +7871,10 @@ function mergeDefs(...defs) {
 function cloneDef(schema) {
   return mergeDefs(schema._zod.def);
 }
-function getElementAtPath(obj, path2) {
-  if (!path2)
+function getElementAtPath(obj, path4) {
+  if (!path4)
     return obj;
-  return path2.reduce((acc, key) => acc?.[key], obj);
+  return path4.reduce((acc, key) => acc?.[key], obj);
 }
 function promiseAllObject(promisesObj) {
   const keys = Object.keys(promisesObj);
@@ -8259,11 +8257,11 @@ function aborted(x, startIndex = 0) {
   }
   return false;
 }
-function prefixIssues(path2, issues) {
+function prefixIssues(path4, issues) {
   return issues.map((iss) => {
     var _a2;
     (_a2 = iss).path ?? (_a2.path = []);
-    iss.path.unshift(path2);
+    iss.path.unshift(path4);
     return iss;
   });
 }
@@ -8446,7 +8444,7 @@ function formatError(error48, mapper = (issue2) => issue2.message) {
 }
 function treeifyError(error48, mapper = (issue2) => issue2.message) {
   const result = { errors: [] };
-  const processError = (error49, path2 = []) => {
+  const processError = (error49, path4 = []) => {
     var _a2, _b;
     for (const issue2 of error49.issues) {
       if (issue2.code === "invalid_union" && issue2.errors.length) {
@@ -8456,7 +8454,7 @@ function treeifyError(error48, mapper = (issue2) => issue2.message) {
       } else if (issue2.code === "invalid_element") {
         processError({ issues: issue2.issues }, issue2.path);
       } else {
-        const fullpath = [...path2, ...issue2.path];
+        const fullpath = [...path4, ...issue2.path];
         if (fullpath.length === 0) {
           result.errors.push(mapper(issue2));
           continue;
@@ -8488,8 +8486,8 @@ function treeifyError(error48, mapper = (issue2) => issue2.message) {
 }
 function toDotPath(_path) {
   const segs = [];
-  const path2 = _path.map((seg) => typeof seg === "object" ? seg.key : seg);
-  for (const seg of path2) {
+  const path4 = _path.map((seg) => typeof seg === "object" ? seg.key : seg);
+  for (const seg of path4) {
     if (typeof seg === "number")
       segs.push(`[${seg}]`);
     else if (typeof seg === "symbol")
@@ -20466,13 +20464,13 @@ function resolveRef(ref, ctx) {
   if (!ref.startsWith("#")) {
     throw new Error("External $ref is not supported, only local refs (#/...) are allowed");
   }
-  const path2 = ref.slice(1).split("/").filter(Boolean);
-  if (path2.length === 0) {
+  const path4 = ref.slice(1).split("/").filter(Boolean);
+  if (path4.length === 0) {
     return ctx.rootSchema;
   }
   const defsKey = ctx.version === "draft-2020-12" ? "$defs" : "definitions";
-  if (path2[0] === defsKey) {
-    const key = path2[1];
+  if (path4[0] === defsKey) {
+    const key = path4[1];
     if (!key || !ctx.defs[key]) {
       throw new Error(`Reference not found: ${ref}`);
     }
@@ -21264,8 +21262,8 @@ function getErrorMap2() {
 
 // ../node_modules/zod/v3/helpers/parseUtil.js
 var makeIssue = (params) => {
-  const { data, path: path2, errorMaps, issueData } = params;
-  const fullPath = [...path2, ...issueData.path || []];
+  const { data, path: path4, errorMaps, issueData } = params;
+  const fullPath = [...path4, ...issueData.path || []];
   const fullIssue = {
     ...issueData,
     path: fullPath
@@ -21380,11 +21378,11 @@ var errorUtil;
 
 // ../node_modules/zod/v3/types.js
 var ParseInputLazyPath = class {
-  constructor(parent, value, path2, key) {
+  constructor(parent, value, path4, key) {
     this._cachedPath = [];
     this.parent = parent;
     this.data = value;
-    this._path = path2;
+    this._path = path4;
     this._key = key;
   }
   get path() {
@@ -30427,6 +30425,37 @@ var StdioServerTransport = class {
   }
 };
 
+// src/errors.ts
+var ComfyApiError = class extends Error {
+  constructor(message, statusCode, details) {
+    super(message);
+    this.statusCode = statusCode;
+    this.details = details;
+    this.name = "ComfyApiError";
+  }
+};
+var WorkflowInputError = class extends Error {
+  constructor(message, field, kind = "validation") {
+    super(message);
+    this.field = field;
+    this.kind = kind;
+    this.name = "WorkflowInputError";
+  }
+};
+var WorkflowTimeoutError = class extends Error {
+  constructor(promptId, timeoutSeconds, queueSnapshot) {
+    super(`Timeout waiting for workflow ${promptId}`);
+    this.promptId = promptId;
+    this.timeoutSeconds = timeoutSeconds;
+    this.queueSnapshot = queueSnapshot;
+    this.name = "WorkflowTimeoutError";
+  }
+};
+
+// src/api.ts
+import * as fs from "node:fs/promises";
+import path from "node:path";
+
 // src/mock.ts
 var IS_MOCK = process.env.MOCK === "true";
 var MOCK_FIXTURES = {
@@ -30577,233 +30606,181 @@ var MOCK_FIXTURES = {
   }
 };
 
-// src/converter.ts
-var WIDGET_TYPES = /* @__PURE__ */ new Set(["INT", "FLOAT", "STRING", "BOOLEAN"]);
-var UI_CONTROL_VALUES = /* @__PURE__ */ new Set([
-  "fixed",
-  "increment",
-  "decrement",
-  "randomize"
-]);
-function isRecord(value) {
-  return typeof value === "object" && value !== null;
+// src/api.ts
+var COMFYUI_SERVER_URL = process.env.COMFYUI_SERVER_URL || "http://localhost:8188";
+async function fetchComfyJson(apiPath, options) {
+  const url2 = new URL(apiPath, COMFYUI_SERVER_URL).toString();
+  let response;
+  try {
+    response = await fetch(url2, {
+      ...options,
+      headers: {
+        Accept: "application/json",
+        ...options?.headers || {}
+      }
+    });
+  } catch (error48) {
+    throw new Error(
+      `Failed to connect to ComfyUI server at ${COMFYUI_SERVER_URL}: ${error48 instanceof Error ? error48.message : String(error48)}`
+    );
+  }
+  if (response.status === 404) {
+    throw new ComfyApiError(`Not found: ${apiPath}`, 404, null);
+  }
+  const text = await response.text();
+  let body;
+  try {
+    body = text ? JSON.parse(text) : null;
+  } catch {
+    throw new ComfyApiError(
+      `Invalid JSON response from ComfyUI: ${text.slice(0, 100)}`,
+      response.status,
+      text
+    );
+  }
+  if (!response.ok) {
+    throw new ComfyApiError(
+      "The ComfyUI service returned an error.",
+      response.status,
+      body
+    );
+  }
+  return body;
 }
-function isWebUIFormat(json2) {
-  return isRecord(json2) && Array.isArray(json2.nodes) && Array.isArray(json2.links);
+async function fetchImageAsset(filename, type, subfolder) {
+  const params = new URLSearchParams({ filename, type });
+  if (subfolder) params.set("subfolder", subfolder);
+  const url2 = new URL(
+    `/view?${params.toString()}`,
+    COMFYUI_SERVER_URL
+  ).toString();
+  const response = await fetch(url2);
+  if (!response.ok)
+    throw new Error(`Failed to fetch image: ${response.statusText}`);
+  const arrayBuffer = await response.arrayBuffer();
+  const mimeType = response.headers.get("content-type") || "image/png";
+  return {
+    buffer: Buffer.from(arrayBuffer),
+    mimeType
+  };
 }
-function isLinkData(link) {
-  return Array.isArray(link) && link.length >= 6 && typeof link[0] === "number" && typeof link[1] === "number" && typeof link[2] === "number" && typeof link[3] === "number" && typeof link[4] === "number" && typeof link[5] === "string";
+function createMockImageAsset() {
+  return {
+    buffer: Buffer.from("mock-image-data"),
+    mimeType: "image/png"
+  };
 }
-function getOrderedInputEntries(group, orderedNames) {
-  if (!group) {
-    return [];
-  }
-  const seen = /* @__PURE__ */ new Set();
-  const orderedEntries = [];
-  for (const inputName of orderedNames ?? []) {
-    const inputDefinition = group[inputName];
-    if (!inputDefinition) {
-      continue;
-    }
-    orderedEntries.push([inputName, inputDefinition]);
-    seen.add(inputName);
-  }
-  for (const [inputName, inputDefinition] of Object.entries(group)) {
-    if (seen.has(inputName)) {
-      continue;
-    }
-    orderedEntries.push([inputName, inputDefinition]);
+function getSessionWorkspaceDir() {
+  return process.env.MCP_WORKSPACE_DIR;
+}
+function getSafePathSegments(subfolder) {
+  if (!subfolder) return [];
+  return subfolder.split(/[/\\]/).filter((s) => s.length > 0 && s !== ".." && s !== ".");
+}
+function sanitizePathSegment(value, fallback) {
+  const normalized = value.trim().replace(/[\\/]+/g, "_").replace(/\.\.+/g, "_").replace(/_+/g, "_").replace(/[^A-Za-z0-9._-]/g, "_");
+  return normalized.length > 0 ? normalized : fallback;
+}
+async function saveToWorkspace(promptId, image, asset) {
+  const workspaceDir = getSessionWorkspaceDir();
+  if (!workspaceDir) {
+    return void 0;
   }
-  return orderedEntries;
+  const safePromptId = sanitizePathSegment(promptId, "unknown-prompt");
+  const safeFileName = path.basename(image.filename);
+  const safeSubfolderSegments = getSafePathSegments(image.subfolder);
+  const relativePath = path.join(
+    "comfyui",
+    safePromptId,
+    ...safeSubfolderSegments,
+    safeFileName
+  );
+  const workspacePath = path.join(workspaceDir, relativePath);
+  await fs.mkdir(path.dirname(workspacePath), { recursive: true });
+  await fs.writeFile(workspacePath, asset.buffer);
+  return {
+    workspace_path: workspacePath,
+    relative_path: relativePath,
+    filename: safeFileName,
+    mime_type: asset.mimeType,
+    subfolder: image.subfolder
+  };
 }
-function getWidgetTypeName(inputTypeData) {
-  return Array.isArray(inputTypeData) ? "COMBO" : String(inputTypeData).toUpperCase();
+async function getObjectInfo() {
+  return IS_MOCK ? MOCK_FIXTURES.object_info : await fetchComfyJson("/object_info");
 }
-function isWidgetInput(inputTypeData) {
-  const typeName = getWidgetTypeName(inputTypeData);
-  return typeName === "COMBO" || WIDGET_TYPES.has(typeName);
+function summarizeErrorMessage(error48) {
+  return error48 instanceof Error ? error48.message : String(error48);
 }
-function isSkippableUiArtifact(value, expectedTypeName) {
-  if (expectedTypeName !== "STRING" && typeof value === "string" && UI_CONTROL_VALUES.has(value)) {
-    return true;
+async function tryGetObjectInfo() {
+  try {
+    return {
+      objectInfo: await getObjectInfo()
+    };
+  } catch (error48) {
+    return {
+      warning: `Node schema details unavailable from ComfyUI: ${summarizeErrorMessage(
+        error48
+      )}`
+    };
   }
-  return typeof value === "object" && value !== null && !Array.isArray(value);
 }
-function normalizeWidgetValue(value, typeName, extraInfo) {
-  const defaultValue = extraInfo?.default;
-  const shouldUseDefaultForNumeric = (typeName === "INT" || typeName === "FLOAT") && (value === "" || value === void 0 || value === null);
-  if (shouldUseDefaultForNumeric && defaultValue !== void 0) {
-    return defaultValue;
+async function getQueueSnapshot() {
+  try {
+    return IS_MOCK ? MOCK_FIXTURES.queue : await fetchComfyJson("/queue");
+  } catch {
+    return void 0;
   }
-  return value;
 }
-function convertWebUIToAPIWithMetadata(webUIJson, objectInfo) {
-  const prompt = {};
-  const widgetPathMap = {};
-  const links = webUIJson.links || [];
-  const nodes = webUIJson.nodes || [];
-  const serverNodeIds = /* @__PURE__ */ new Set();
-  for (const node of nodes) {
-    if (!node.id || !node.type || !objectInfo[node.type]) {
-      continue;
-    }
-    serverNodeIds.add(String(node.id));
+function formatQueueSnapshotText(queueSnapshot) {
+  if (!queueSnapshot) {
+    return "";
   }
-  const linkMap = /* @__PURE__ */ new Map();
-  for (const link of links) {
-    if (isLinkData(link) && serverNodeIds.has(String(link[3]))) {
-      linkMap.set(link[0], link);
-    }
+  return `Queue snapshot: ${queueSnapshot.queue_running.length} running, ${queueSnapshot.queue_pending.length} pending.`;
+}
+function getPromptQueueVisibility(promptId, queueSnapshot) {
+  if (!queueSnapshot) {
+    return null;
   }
-  for (const node of nodes) {
-    if (!node.id || !node.type) continue;
-    const objInfo = objectInfo[node.type];
-    if (!objInfo) continue;
-    const nodeIdStr = String(node.id);
-    const promptNode = {
-      class_type: node.type,
-      inputs: {}
-    };
-    const connectedInputNames = /* @__PURE__ */ new Set();
-    if (Array.isArray(node.inputs)) {
-      for (const input of node.inputs) {
-        if (input.name && input.link !== null && input.link !== void 0) {
-          const linkData = linkMap.get(input.link);
-          if (Array.isArray(linkData) && linkData.length >= 6 && String(linkData[3]) === nodeIdStr) {
-            const originNodeId = String(linkData[1]);
-            const originSlot = linkData[2];
-            promptNode.inputs[input.name] = [
-              originNodeId,
-              originSlot
-            ];
-            connectedInputNames.add(input.name);
-          }
-        }
+  return JSON.stringify(queueSnapshot).includes(promptId);
+}
+async function waitForWorkflowResult(prompt_id, timeout_seconds) {
+  const start = Date.now();
+  const end = start + timeout_seconds * 1e3;
+  while (Date.now() < end) {
+    let history = MOCK_FIXTURES.history;
+    if (!IS_MOCK) {
+      try {
+        history = await fetchComfyJson(
+          `/history/${prompt_id}`
+        );
+      } catch {
+        history = {};
       }
     }
-    if (objInfo) {
-      const widgets = Array.isArray(node.widgets_values) ? node.widgets_values : [];
-      let widgetIndex = 0;
-      const processInputGroup = (group, orderedNames) => {
-        for (const [inputName, inputDataRaw] of getOrderedInputEntries(
-          group,
-          orderedNames
-        )) {
-          const [inputTypeData, inputExtraInfo] = inputDataRaw;
-          if (!isWidgetInput(inputTypeData)) {
-            continue;
-          }
-          const typeName = getWidgetTypeName(inputTypeData);
-          while (widgetIndex < widgets.length && isSkippableUiArtifact(widgets[widgetIndex], typeName)) {
-            widgetIndex++;
-          }
-          const isConnectedWidget = connectedInputNames.has(inputName);
-          if (widgetIndex >= widgets.length) {
-            if (!isConnectedWidget && inputExtraInfo?.default !== void 0) {
-              promptNode.inputs[inputName] = inputExtraInfo.default;
-            }
-            continue;
-          }
-          const widgetSourceIndex = widgetIndex;
-          const value = normalizeWidgetValue(
-            widgets[widgetSourceIndex],
-            typeName,
-            inputExtraInfo
-          );
-          widgetIndex++;
-          if (!isConnectedWidget) {
-            promptNode.inputs[inputName] = value;
-            widgetPathMap[`${nodeIdStr}.widgets[${widgetSourceIndex}]`] = `${nodeIdStr}.inputs.${inputName}`;
-          }
-        }
-      };
-      processInputGroup(
-        objInfo.input?.required,
-        objInfo.input_order?.required
-      );
-      processInputGroup(
-        objInfo.input?.optional,
-        objInfo.input_order?.optional
-      );
+    if (history?.[prompt_id]) {
+      return history[prompt_id];
     }
-    prompt[nodeIdStr] = promptNode;
+    await new Promise((r) => setTimeout(r, 1e3));
   }
-  return {
-    prompt,
-    widgetPathMap
-  };
+  const queueSnapshot = await getQueueSnapshot();
+  throw new WorkflowTimeoutError(prompt_id, timeout_seconds, queueSnapshot);
 }
 
-// src/index.ts
-var zNamespace = zod_exports;
-var z3 = zNamespace.z ?? zNamespace.default ?? zNamespace;
-var COMFYUI_SERVER_URL = process.env.COMFYUI_SERVER_URL || "http://localhost:8188";
-var PACKAGE_VERSION = getPackageVersion(import.meta.url);
-var ComfyApiError = class extends Error {
-  constructor(message, statusCode, details) {
-    super(message);
-    this.statusCode = statusCode;
-    this.details = details;
-    this.name = "ComfyApiError";
-  }
-};
-var WorkflowInputError = class extends Error {
-  constructor(message, field, kind = "validation") {
-    super(message);
-    this.field = field;
-    this.kind = kind;
-    this.name = "WorkflowInputError";
-  }
-};
-var INPUT_SEGMENT = "inputs";
-var OverrideRecordSchema = z3.record(z3.string(), z3.unknown());
-var OverridesSchema = z3.union([OverrideRecordSchema, z3.string()]).describe(
-  'Dot-notation overrides as an object or JSON string. e.g. {"6.seed": 7}'
-);
-var WorkflowTimeoutError = class extends Error {
-  constructor(promptId, timeoutSeconds, queueSnapshot) {
-    super(`Timeout waiting for workflow ${promptId}`);
-    this.promptId = promptId;
-    this.timeoutSeconds = timeoutSeconds;
-    this.queueSnapshot = queueSnapshot;
-    this.name = "WorkflowTimeoutError";
-  }
-};
-function isRecord2(value) {
+// src/workflow/common.ts
+function isRecord(value) {
   return typeof value === "object" && value !== null;
 }
-function isComfyPromptNode(value) {
-  return isRecord2(value) && typeof value.class_type === "string" && isRecord2(value.inputs);
-}
-function isUiOnlyPromptNode(nodeId, node) {
-  return nodeId === "meta" || ["Note", "MarkdownNote", "PrimitiveNode"].includes(node.class_type);
-}
-function extractRunnablePromptNodes(prompt) {
-  const runnablePrompt = {};
-  for (const [nodeId, rawNode] of Object.entries(prompt)) {
-    if (!isComfyPromptNode(rawNode) || isUiOnlyPromptNode(nodeId, rawNode)) {
-      continue;
-    }
-    runnablePrompt[nodeId] = rawNode;
-  }
-  if (Object.keys(runnablePrompt).length === 0) {
-    throw new WorkflowInputError(
-      "Workflow prompt does not contain any runnable ComfyUI nodes.",
-      "workflow"
-    );
-  }
-  return runnablePrompt;
-}
 function toStructuredContent(value) {
-  return isRecord2(value) ? value : {};
+  return isRecord(value) ? value : {};
 }
 function summarizeNodeErrors(nodeErrors) {
-  if (!isRecord2(nodeErrors)) {
+  if (!isRecord(nodeErrors)) {
     return [];
   }
   const lines = [];
   for (const [nodeId, rawNodeError] of Object.entries(nodeErrors)) {
-    if (!isRecord2(rawNodeError)) {
+    if (!isRecord(rawNodeError)) {
       continue;
     }
     const classType = typeof rawNodeError.class_type === "string" ? ` (${rawNodeError.class_type})` : "";
@@ -30811,7 +30788,7 @@ function summarizeNodeErrors(nodeErrors) {
     const missingInputs = [];
     const otherMessages = [];
     for (const rawError of rawErrors) {
-      if (!isRecord2(rawError)) {
+      if (!isRecord(rawError)) {
         continue;
       }
       const errorType = rawError.type;
@@ -30841,11 +30818,11 @@ function summarizeNodeErrors(nodeErrors) {
   return lines;
 }
 function summarizeComfyErrorDetails(details) {
-  if (!isRecord2(details)) {
+  if (!isRecord(details)) {
     return "";
   }
   const lines = [];
-  const errorDetails = isRecord2(details.error) ? details.error : void 0;
+  const errorDetails = isRecord(details.error) ? details.error : void 0;
   if (typeof errorDetails?.message === "string") {
     lines.push(errorDetails.message);
   } else if (typeof details.message === "string") {
@@ -30864,159 +30841,260 @@ function formatWorkflowNodeErrors(nodeErrors) {
   }
   return ["Workflow submission failed validation.", ...nodeLines].join("\n");
 }
-async function fetchComfyJson(apiPath, options) {
-  const url2 = new URL(apiPath, COMFYUI_SERVER_URL).toString();
-  let response;
-  try {
-    response = await fetch(url2, {
-      ...options,
-      headers: {
-        Accept: "application/json",
-        ...options?.headers || {}
-      }
-    });
-  } catch (error48) {
-    throw new Error(
-      `Failed to connect to ComfyUI server at ${COMFYUI_SERVER_URL}: ${error48 instanceof Error ? error48.message : String(error48)}`
-    );
-  }
-  if (response.status === 404) {
-    throw new ComfyApiError(`Not found: ${apiPath}`, 404, null);
+function formatValuePreview(value) {
+  if (typeof value === "string") {
+    return value.length > 100 ? `${value.slice(0, 97)}...` : value;
   }
-  const text = await response.text();
-  let body;
   try {
-    body = text ? JSON.parse(text) : null;
+    const serialized = JSON.stringify(value);
+    if (!serialized) {
+      return String(value);
+    }
+    return serialized.length > 100 ? `${serialized.slice(0, 97)}...` : serialized;
   } catch {
-    throw new ComfyApiError(
-      `Invalid JSON response from ComfyUI: ${text.slice(0, 100)}`,
-      response.status,
-      text
-    );
+    return String(value);
   }
-  if (!response.ok) {
-    throw new ComfyApiError(
-      "The ComfyUI service returned an error.",
-      response.status,
-      body
-    );
+}
+function getInputDefinition(objectInfo, prompt, nodeId, inputName) {
+  if (!objectInfo) {
+    return void 0;
   }
-  return body;
+  const node = prompt[nodeId];
+  if (!node) {
+    return void 0;
+  }
+  const nodeDefinition = objectInfo[node.class_type];
+  if (!nodeDefinition) {
+    return void 0;
+  }
+  return nodeDefinition.input?.required?.[inputName] ?? nodeDefinition.input?.optional?.[inputName];
 }
-async function fetchImageAsset(filename, type, subfolder) {
-  const params = new URLSearchParams({ filename, type });
-  if (subfolder) params.set("subfolder", subfolder);
-  const url2 = new URL(
-    `/view?${params.toString()}`,
-    COMFYUI_SERVER_URL
-  ).toString();
-  const response = await fetch(url2);
-  if (!response.ok)
-    throw new Error(`Failed to fetch image: ${response.statusText}`);
-  const arrayBuffer = await response.arrayBuffer();
-  const mimeType = response.headers.get("content-type") || "image/png";
-  return {
-    buffer: Buffer.from(arrayBuffer),
-    mimeType
-  };
+function getExpectedTypeName(inputDefinition) {
+  if (!inputDefinition) {
+    return void 0;
+  }
+  const [typeData] = inputDefinition;
+  return Array.isArray(typeData) ? "COMBO" : String(typeData).toUpperCase();
 }
-function createMockImageAsset() {
+function getInputOptionsSummary(inputDefinition, previewLimit = 8) {
+  if (!inputDefinition) {
+    return void 0;
+  }
+  const [typeData] = inputDefinition;
+  if (!Array.isArray(typeData)) {
+    return void 0;
+  }
   return {
-    buffer: Buffer.from("mock-image-data"),
-    mimeType: "image/png"
+    options_count: typeData.length,
+    options_preview: typeData.slice(0, previewLimit),
+    options_truncated: typeData.length > previewLimit
   };
 }
-function getSessionWorkspaceDir() {
-  return process.env.MCP_WORKSPACE_DIR;
+function isPlainObject3(value) {
+  return isRecord(value) && !Array.isArray(value);
 }
-function getSafePathSegments(subfolder) {
-  if (!subfolder) return [];
-  return subfolder.split(/[/\\]/).filter((s) => s.length > 0 && s !== ".." && s !== ".");
+function getValueTypeName(value) {
+  if (Array.isArray(value)) {
+    return "array";
+  }
+  if (value === null) {
+    return "null";
+  }
+  return typeof value;
 }
-async function saveToWorkspace(promptId, image, asset) {
-  const workspaceDir = getSessionWorkspaceDir();
-  if (!workspaceDir) {
-    return void 0;
+
+// src/workflow/load.ts
+import * as fs2 from "node:fs/promises";
+import * as crypto from "node:crypto";
+import { homedir } from "node:os";
+import path2 from "node:path";
+
+// src/converter.ts
+var WIDGET_TYPES = /* @__PURE__ */ new Set(["INT", "FLOAT", "STRING", "BOOLEAN"]);
+var UI_CONTROL_VALUES = /* @__PURE__ */ new Set([
+  "fixed",
+  "increment",
+  "decrement",
+  "randomize"
+]);
+function isRecord2(value) {
+  return typeof value === "object" && value !== null;
+}
+function isWebUIFormat(json2) {
+  return isRecord2(json2) && Array.isArray(json2.nodes) && Array.isArray(json2.links);
+}
+function isLinkData(link) {
+  return Array.isArray(link) && link.length >= 6 && typeof link[0] === "number" && typeof link[1] === "number" && typeof link[2] === "number" && typeof link[3] === "number" && typeof link[4] === "number" && typeof link[5] === "string";
+}
+function getOrderedInputEntries(group, orderedNames) {
+  if (!group) {
+    return [];
   }
-  const safeFileName = path.basename(image.filename);
-  const safeSubfolderSegments = getSafePathSegments(image.subfolder);
-  const relativePath = path.join(
-    "comfyui",
-    promptId,
-    ...safeSubfolderSegments,
-    safeFileName
-  );
-  const workspacePath = path.join(workspaceDir, relativePath);
-  await fs.mkdir(path.dirname(workspacePath), { recursive: true });
-  await fs.writeFile(workspacePath, asset.buffer);
-  return {
-    workspace_path: workspacePath,
-    relative_path: relativePath,
-    filename: safeFileName,
-    mime_type: asset.mimeType,
-    subfolder: image.subfolder
-  };
+  const seen = /* @__PURE__ */ new Set();
+  const orderedEntries = [];
+  for (const inputName of orderedNames ?? []) {
+    const inputDefinition = group[inputName];
+    if (!inputDefinition) {
+      continue;
+    }
+    orderedEntries.push([inputName, inputDefinition]);
+    seen.add(inputName);
+  }
+  for (const [inputName, inputDefinition] of Object.entries(group)) {
+    if (seen.has(inputName)) {
+      continue;
+    }
+    orderedEntries.push([inputName, inputDefinition]);
+  }
+  return orderedEntries;
 }
-function handleComfyError(error48) {
-  if (error48 instanceof ZodError) {
-    return createValidationError(
-      "arguments",
-      error48.issues[0]?.message || "Invalid input."
-    );
+function getWidgetTypeName(inputTypeData) {
+  return Array.isArray(inputTypeData) ? "COMBO" : String(inputTypeData).toUpperCase();
+}
+function isWidgetInput(inputTypeData) {
+  const typeName = getWidgetTypeName(inputTypeData);
+  return typeName === "COMBO" || WIDGET_TYPES.has(typeName);
+}
+function isSkippableUiArtifact(value, expectedTypeName) {
+  if (expectedTypeName !== "STRING" && typeof value === "string" && UI_CONTROL_VALUES.has(value)) {
+    return true;
   }
-  if (error48 instanceof WorkflowInputError) {
-    return error48.kind === "not_found" ? createNotFoundError(error48.message) : createValidationError(error48.field, error48.message);
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function normalizeWidgetValue(value, typeName, extraInfo) {
+  const defaultValue = extraInfo?.default;
+  const shouldUseDefaultForNumeric = (typeName === "INT" || typeName === "FLOAT") && (value === "" || value === void 0 || value === null);
+  if (shouldUseDefaultForNumeric && defaultValue !== void 0) {
+    return defaultValue;
+  }
+  return value;
+}
+function convertWebUIToAPIWithMetadata(webUIJson, objectInfo) {
+  const prompt = {};
+  const widgetPathMap = {};
+  const links = webUIJson.links || [];
+  const nodes = webUIJson.nodes || [];
+  const serverNodeIds = /* @__PURE__ */ new Set();
+  for (const node of nodes) {
+    if (!node.id || !node.type || !objectInfo[node.type]) {
+      continue;
+    }
+    serverNodeIds.add(String(node.id));
+  }
+  const linkMap = /* @__PURE__ */ new Map();
+  for (const link of links) {
+    if (isLinkData(link) && serverNodeIds.has(String(link[3]))) {
+      linkMap.set(link[0], link);
+    }
+  }
+  for (const node of nodes) {
+    if (!node.id || !node.type) continue;
+    const objInfo = objectInfo[node.type];
+    if (!objInfo) continue;
+    const nodeIdStr = String(node.id);
+    const promptNode = {
+      class_type: node.type,
+      inputs: {}
+    };
+    const connectedInputNames = /* @__PURE__ */ new Set();
+    if (Array.isArray(node.inputs)) {
+      for (const input of node.inputs) {
+        if (input.name && input.link !== null && input.link !== void 0) {
+          const linkData = linkMap.get(input.link);
+          if (Array.isArray(linkData) && linkData.length >= 6 && String(linkData[3]) === nodeIdStr) {
+            const originNodeId = String(linkData[1]);
+            const originSlot = linkData[2];
+            promptNode.inputs[input.name] = [
+              originNodeId,
+              originSlot
+            ];
+            connectedInputNames.add(input.name);
+          }
+        }
+      }
+    }
+    if (objInfo) {
+      const widgets = Array.isArray(node.widgets_values) ? node.widgets_values : [];
+      let widgetIndex = 0;
+      const processInputGroup = (group, orderedNames) => {
+        for (const [inputName, inputDataRaw] of getOrderedInputEntries(
+          group,
+          orderedNames
+        )) {
+          const [inputTypeData, inputExtraInfo] = inputDataRaw;
+          if (!isWidgetInput(inputTypeData)) {
+            continue;
+          }
+          const typeName = getWidgetTypeName(inputTypeData);
+          while (widgetIndex < widgets.length && isSkippableUiArtifact(widgets[widgetIndex], typeName)) {
+            widgetIndex++;
+          }
+          const isConnectedWidget = connectedInputNames.has(inputName);
+          if (widgetIndex >= widgets.length) {
+            if (!isConnectedWidget && inputExtraInfo?.default !== void 0) {
+              promptNode.inputs[inputName] = inputExtraInfo.default;
+            }
+            continue;
+          }
+          const widgetSourceIndex = widgetIndex;
+          const value = normalizeWidgetValue(
+            widgets[widgetSourceIndex],
+            typeName,
+            inputExtraInfo
+          );
+          widgetIndex++;
+          if (!isConnectedWidget) {
+            promptNode.inputs[inputName] = value;
+            widgetPathMap[`${nodeIdStr}.widgets[${widgetSourceIndex}]`] = `${nodeIdStr}.inputs.${inputName}`;
+          }
+        }
+      };
+      processInputGroup(
+        objInfo.input?.required,
+        objInfo.input_order?.required
+      );
+      processInputGroup(
+        objInfo.input?.optional,
+        objInfo.input_order?.optional
+      );
+    }
+    prompt[nodeIdStr] = promptNode;
   }
-  if (error48 instanceof ComfyApiError) {
-    const detailSummary = summarizeComfyErrorDetails(error48.details);
-    return {
-      content: [
-        {
-          type: "text",
-          text: detailSummary.length > 0 ? `${error48.message}
-${detailSummary}` : error48.message
-        }
-      ],
-      structuredContent: toStructuredContent(error48.details),
-      isError: true
-    };
+  return {
+    prompt,
+    widgetPathMap
+  };
+}
+
+// src/workflow/load.ts
+function isComfyPromptNode(value) {
+  return isRecord(value) && typeof value.class_type === "string" && isRecord(value.inputs);
+}
+function isUiOnlyPromptNode(nodeId, node) {
+  return nodeId === "meta" || ["Note", "MarkdownNote", "PrimitiveNode"].includes(node.class_type);
+}
+function extractRunnablePromptNodes(prompt) {
+  const runnablePrompt = {};
+  for (const [nodeId, rawNode] of Object.entries(prompt)) {
+    if (!isComfyPromptNode(rawNode) || isUiOnlyPromptNode(nodeId, rawNode)) {
+      continue;
+    }
+    runnablePrompt[nodeId] = rawNode;
   }
-  if (error48 instanceof WorkflowTimeoutError) {
-    const queueSummary = formatQueueSnapshotText(error48.queueSnapshot);
-    const promptVisibility = getPromptQueueVisibility(
-      error48.promptId,
-      error48.queueSnapshot
+  if (Object.keys(runnablePrompt).length === 0) {
+    throw new WorkflowInputError(
+      "Workflow prompt does not contain any runnable ComfyUI nodes.",
+      "workflow"
     );
-    const promptVisibilityText = promptVisibility === true ? "The prompt is still visible in the ComfyUI queue snapshot." : promptVisibility === false ? "The prompt is not visible in the ComfyUI queue snapshot; inspect ComfyUI history or server logs if it never completes." : "ComfyUI queue state was unavailable at timeout.";
-    return {
-      content: [
-        {
-          type: "text",
-          text: [
-            `${error48.message}.`,
-            queueSummary,
-            promptVisibilityText,
-            `You can re-attach by calling 'comfyui_wait_for_workflow' again with prompt_id '${error48.promptId}'.`
-          ].filter((line) => line && line.length > 0).join("\n")
-        }
-      ],
-      structuredContent: {
-        prompt_id: error48.promptId,
-        timeout_seconds: error48.timeoutSeconds,
-        queue: error48.queueSnapshot ?? null,
-        prompt_visible_in_queue: promptVisibility
-      },
-      isError: true
-    };
   }
-  return createInternalError(error48);
+  return runnablePrompt;
 }
 function getWorkflowDirectories(extraPath) {
   const directories = [
     process.cwd(),
-    path.join(process.cwd(), "workflows"),
-    path.join(homedir(), ".fre4x-comfyui", "workflows"),
-    path.join(
+    path2.join(process.cwd(), "workflows"),
+    path2.join(homedir(), ".fre4x-comfyui", "workflows"),
+    path2.join(
       homedir(),
       "comfy",
       "ComfyUI",
@@ -31029,7 +31107,7 @@ function getWorkflowDirectories(extraPath) {
     directories.push(extraPath);
   }
   return [
-    ...new Set(directories.map((directory) => path.resolve(directory)))
+    ...new Set(directories.map((directory) => path2.resolve(directory)))
   ];
 }
 function getWorkflowNameCandidates(name) {
@@ -31050,48 +31128,32 @@ function normalizeWorkflowFileName(fileName) {
       "output_file_name"
     );
   }
-  return trimmed.endsWith(".json") ? trimmed : `${trimmed}.json`;
-}
-function isConnectionValue(value) {
-  return Array.isArray(value) && value.length >= 2 && (typeof value[0] === "string" || typeof value[0] === "number") && typeof value[1] === "number";
-}
-async function getObjectInfo() {
-  return IS_MOCK ? MOCK_FIXTURES.object_info : await fetchComfyJson("/object_info");
-}
-function summarizeErrorMessage(error48) {
-  return error48 instanceof Error ? error48.message : String(error48);
-}
-async function tryGetObjectInfo() {
-  try {
-    return {
-      objectInfo: await getObjectInfo()
-    };
-  } catch (error48) {
-    return {
-      warning: `Node schema details unavailable from ComfyUI: ${summarizeErrorMessage(
-        error48
-      )}`
-    };
+  if (path2.isAbsolute(trimmed)) {
+    throw new WorkflowInputError(
+      "Absolute paths are not allowed.",
+      "output_file_name"
+    );
   }
+  return trimmed.endsWith(".json") ? trimmed : `${trimmed}.json`;
 }
 async function resolveWorkflowReference(args) {
   if (args.workflow_file_path) {
     const targetPath = args.workflow_file_path;
     try {
-      const resolvedPath = path.resolve(targetPath);
-      const content = await fs.readFile(resolvedPath, "utf8");
+      const resolvedPath = path2.resolve(targetPath);
+      const content = await fs2.readFile(resolvedPath, "utf8");
       return {
         targetPath: resolvedPath,
         content
       };
     } catch {
-      if (path.basename(targetPath) === targetPath) {
+      if (path2.basename(targetPath) === targetPath) {
         const candidates = getWorkflowNameCandidates(targetPath);
         for (const dir of getWorkflowDirectories()) {
           for (const candidate of candidates) {
-            const candidatePath = path.join(dir, candidate);
+            const candidatePath = path2.join(dir, candidate);
             try {
-              const content = await fs.readFile(
+              const content = await fs2.readFile(
                 candidatePath,
                 "utf8"
               );
@@ -31122,12 +31184,12 @@ async function resolveWorkflowReference(args) {
     const objectInfoAvailability = await tryGetObjectInfo();
     for (const dir of getWorkflowDirectories()) {
       try {
-        const entries = await fs.readdir(dir, { withFileTypes: true });
+        const entries = await fs2.readdir(dir, { withFileTypes: true });
         for (const entry of entries) {
           if (!entry.isFile() || !entry.name.endsWith(".json")) {
             continue;
           }
-          const fullPath = path.join(dir, entry.name);
+          const fullPath = path2.join(dir, entry.name);
           const inspected = await inspectWorkflowFile(
             fullPath,
             objectInfoAvailability.objectInfo
@@ -31157,7 +31219,7 @@ async function resolveWorkflowReference(args) {
   );
 }
 async function normalizeWorkflowPrompt(workflow, objectInfo) {
-  if (!isRecord2(workflow)) {
+  if (!isRecord(workflow)) {
     throw new WorkflowInputError(
       "Workflow must be a JSON object.",
       "workflow"
@@ -31181,8 +31243,8 @@ async function normalizeWorkflowPrompt(workflow, objectInfo) {
     normalizedWorkflow = converted.prompt;
     widgetPathMap = converted.widgetPathMap;
   }
-  const promptCandidate = isRecord2(normalizedWorkflow.prompt) && normalizedWorkflow.prompt ? normalizedWorkflow.prompt : normalizedWorkflow;
-  if (!isRecord2(promptCandidate)) {
+  const promptCandidate = isRecord(normalizedWorkflow.prompt) && normalizedWorkflow.prompt ? normalizedWorkflow.prompt : normalizedWorkflow;
+  if (!isRecord(promptCandidate)) {
     throw new WorkflowInputError(
       "Workflow prompt must be a JSON object.",
       "workflow"
@@ -31214,8 +31276,8 @@ async function loadWorkflow(args, objectInfo) {
   };
 }
 async function inspectWorkflowFile(fullPath, objectInfo) {
-  const content = await fs.readFile(fullPath, "utf8");
-  const defaultName = path.basename(fullPath, ".json");
+  const content = await fs2.readFile(fullPath, "utf8");
+  const defaultName = path2.basename(fullPath, ".json");
   const defaultId = getWorkflowId(fullPath);
   let parsed;
   try {
@@ -31224,10 +31286,10 @@ async function inspectWorkflowFile(fullPath, objectInfo) {
     return void 0;
   }
   let name = defaultName;
-  if (isRecord2(parsed)) {
+  if (isRecord(parsed)) {
     if (typeof parsed.name === "string" && parsed.name.trim().length > 0) {
       name = parsed.name;
-    } else if (isRecord2(parsed.workflow) && typeof parsed.workflow.name === "string" && parsed.workflow.name.trim().length > 0) {
+    } else if (isRecord(parsed.workflow) && typeof parsed.workflow.name === "string" && parsed.workflow.name.trim().length > 0) {
       name = parsed.workflow.name;
     }
   }
@@ -31282,6 +31344,23 @@ async function inspectWorkflowFile(fullPath, objectInfo) {
     return void 0;
   }
 }
+function deriveSavedWorkflowFileName(sourcePath, requestedName) {
+  if (requestedName) {
+    return normalizeWorkflowFileName(requestedName);
+  }
+  if (sourcePath) {
+    return normalizeWorkflowFileName(path2.basename(sourcePath));
+  }
+  return `workflow-${Date.now()}.json`;
+}
+function getWorkflowId(absolutePath) {
+  return crypto.createHash("sha256").update(absolutePath).digest("hex").substring(0, 8);
+}
+
+// src/workflow/editable.ts
+function isConnectionValue(value) {
+  return Array.isArray(value) && value.length >= 2 && (typeof value[0] === "string" || typeof value[0] === "number") && typeof value[1] === "number";
+}
 function inferEditableRole(nodeId, classType, inputName, consumerMap) {
   if (classType === "CLIPTextEncode" && inputName === "text") {
     const consumers = consumerMap.get(nodeId) ?? [];
@@ -31302,167 +31381,46 @@ function inferEditableRole(nodeId, classType, inputName, consumerMap) {
       for (const consumer of textConsumers) {
         const clipConsumers = consumerMap.get(consumer.nodeId) ?? [];
         if (clipConsumers.some(
-          (clipConsumer) => clipConsumer.inputName === "positive"
-        )) {
-          return "positive_prompt";
-        }
-        if (clipConsumers.some(
-          (clipConsumer) => clipConsumer.inputName === "negative"
-        )) {
-          return "negative_prompt";
-        }
-      }
-      return "prompt_text";
-    }
-  }
-  switch (inputName) {
-    case "seed":
-      return "seed";
-    case "steps":
-      return "steps";
-    case "cfg":
-      return "cfg";
-    case "sampler_name":
-      return "sampler";
-    case "scheduler":
-      return "scheduler";
-    case "denoise":
-      return "denoise";
-    case "width":
-      return "width";
-    case "height":
-      return "height";
-    case "batch_size":
-      return "batch_size";
-    case "ckpt_name":
-      return "checkpoint";
-    case "filename_prefix":
-      return "filename_prefix";
-    case "text":
-      return "text";
-    default:
-      return void 0;
-  }
-}
-function formatValuePreview(value) {
-  if (typeof value === "string") {
-    return value.length > 100 ? `${value.slice(0, 97)}...` : value;
-  }
-  try {
-    const serialized = JSON.stringify(value);
-    if (!serialized) {
-      return String(value);
-    }
-    return serialized.length > 100 ? `${serialized.slice(0, 97)}...` : serialized;
-  } catch {
-    return String(value);
-  }
-}
-function getInputDefinition(objectInfo, prompt, nodeId, inputName) {
-  if (!objectInfo) {
-    return void 0;
-  }
-  const node = prompt[nodeId];
-  if (!node) {
-    return void 0;
-  }
-  const nodeDefinition = objectInfo[node.class_type];
-  if (!nodeDefinition) {
-    return void 0;
-  }
-  return nodeDefinition.input?.required?.[inputName] ?? nodeDefinition.input?.optional?.[inputName];
-}
-function getExpectedTypeName(inputDefinition) {
-  if (!inputDefinition) {
-    return void 0;
-  }
-  const [typeData] = inputDefinition;
-  return Array.isArray(typeData) ? "COMBO" : String(typeData).toUpperCase();
-}
-function getInputOptionsSummary(inputDefinition, previewLimit = 8) {
-  if (!inputDefinition) {
-    return void 0;
-  }
-  const [typeData] = inputDefinition;
-  if (!Array.isArray(typeData)) {
-    return void 0;
-  }
-  return {
-    options_count: typeData.length,
-    options_preview: typeData.slice(0, previewLimit),
-    options_truncated: typeData.length > previewLimit
-  };
-}
-function isPlainObject3(value) {
-  return isRecord2(value) && !Array.isArray(value);
-}
-function getValueTypeName(value) {
-  if (Array.isArray(value)) {
-    return "array";
-  }
-  if (value === null) {
-    return "null";
-  }
-  return typeof value;
-}
-function validateOverrideValueType(path2, value, inputDefinition) {
-  if (!inputDefinition) {
-    return;
-  }
-  const [typeData] = inputDefinition;
-  if (Array.isArray(typeData)) {
-    if (!typeData.some((option) => option === value)) {
-      throw new WorkflowInputError(
-        `Invalid override for '${path2}': expected one of ${typeData.map((option) => JSON.stringify(option)).join(", ")}, received ${JSON.stringify(value)}.`,
-        "overrides"
-      );
-    }
-    return;
-  }
-  const typeName = String(typeData).toUpperCase();
-  switch (typeName) {
-    case "INT":
-      if (typeof value !== "number" || !Number.isInteger(value)) {
-        throw new WorkflowInputError(
-          `Invalid override for '${path2}': expected INT, received ${getValueTypeName(
-            value
-          )}.`,
-          "overrides"
-        );
-      }
-      return;
-    case "FLOAT":
-      if (typeof value !== "number") {
-        throw new WorkflowInputError(
-          `Invalid override for '${path2}': expected FLOAT, received ${getValueTypeName(
-            value
-          )}.`,
-          "overrides"
-        );
-      }
-      return;
-    case "STRING":
-      if (typeof value !== "string") {
-        throw new WorkflowInputError(
-          `Invalid override for '${path2}': expected STRING, received ${getValueTypeName(
-            value
-          )}.`,
-          "overrides"
-        );
-      }
-      return;
-    case "BOOLEAN":
-      if (typeof value !== "boolean") {
-        throw new WorkflowInputError(
-          `Invalid override for '${path2}': expected BOOLEAN, received ${getValueTypeName(
-            value
-          )}.`,
-          "overrides"
-        );
+          (clipConsumer) => clipConsumer.inputName === "positive"
+        )) {
+          return "positive_prompt";
+        }
+        if (clipConsumers.some(
+          (clipConsumer) => clipConsumer.inputName === "negative"
+        )) {
+          return "negative_prompt";
+        }
       }
-      return;
+      return "prompt_text";
+    }
+  }
+  switch (inputName) {
+    case "seed":
+      return "seed";
+    case "steps":
+      return "steps";
+    case "cfg":
+      return "cfg";
+    case "sampler_name":
+      return "sampler";
+    case "scheduler":
+      return "scheduler";
+    case "denoise":
+      return "denoise";
+    case "width":
+      return "width";
+    case "height":
+      return "height";
+    case "batch_size":
+      return "batch_size";
+    case "ckpt_name":
+      return "checkpoint";
+    case "filename_prefix":
+      return "filename_prefix";
+    case "text":
+      return "text";
     default:
-      return;
+      return void 0;
   }
 }
 function getEditableAliases(canonicalPath, nodeId, inputName, role, widgetPathMap) {
@@ -31512,15 +31470,15 @@ function buildEditableInputIndex(editableInputs) {
     pathsByNode
   };
 }
-function getPathSegmentHelp(prompt, path2) {
-  const parts = path2.split(".");
+function getPathSegmentHelp(prompt, path4) {
+  const parts = path4.split(".");
   if (parts.length === 0 || parts[0].length === 0) {
     return void 0;
   }
   const nodeId = parts[0];
   return {
     nodeId,
-    nodeExists: isRecord2(prompt[nodeId])
+    nodeExists: isRecord(prompt[nodeId])
   };
 }
 function describeAvailableNodePaths(editableIndex, nodeId) {
@@ -31702,6 +31660,221 @@ function buildOverrideExamples(editableInputs, limit = 5) {
 function formatOverrideExamplesText(examples) {
   return JSON.stringify(examples, null, 2);
 }
+function analyzeEditableInputs(prompt, objectInfo, widgetPathMap = {}) {
+  const consumerMap = /* @__PURE__ */ new Map();
+  for (const [nodeId, node] of Object.entries(prompt)) {
+    for (const [inputName, value] of Object.entries(node.inputs)) {
+      if (!isConnectionValue(value)) {
+        continue;
+      }
+      const sourceNodeId = String(value[0]);
+      const consumers = consumerMap.get(sourceNodeId) ?? [];
+      consumers.push({
+        inputName,
+        classType: node.class_type,
+        nodeId
+      });
+      consumerMap.set(sourceNodeId, consumers);
+    }
+  }
+  const editableInputs = [];
+  for (const [nodeId, node] of Object.entries(prompt)) {
+    for (const [inputName, value] of Object.entries(node.inputs)) {
+      if (isConnectionValue(value)) {
+        continue;
+      }
+      editableInputs.push({
+        path: `${nodeId}.inputs.${inputName}`,
+        node_id: nodeId,
+        class_type: node.class_type,
+        input_name: inputName,
+        value,
+        value_preview: formatValuePreview(value),
+        role: inferEditableRole(
+          nodeId,
+          node.class_type,
+          inputName,
+          consumerMap
+        ),
+        aliases: [],
+        expected_type: getExpectedTypeName(
+          getInputDefinition(objectInfo, prompt, nodeId, inputName)
+        ),
+        ...getInputOptionsSummary(
+          getInputDefinition(objectInfo, prompt, nodeId, inputName)
+        )
+      });
+    }
+  }
+  return editableInputs.map((entry) => ({
+    ...entry,
+    aliases: getEditableAliases(
+      entry.path,
+      entry.node_id,
+      entry.input_name,
+      entry.role,
+      widgetPathMap
+    )
+  })).sort(
+    (left, right) => left.path.localeCompare(right.path, void 0, { numeric: true })
+  );
+}
+function formatEditableInputsText(editableInputs, options = {}) {
+  if (editableInputs.length === 0) {
+    return "No literal editable inputs were detected.";
+  }
+  return formatListItems(editableInputs, (entry) => {
+    const role = entry.role ? ` [${entry.role}]` : "";
+    const classType = entry.class_type ? ` (${entry.class_type})` : "";
+    const expectedType = entry.expected_type ? ` <${entry.expected_type}>` : "";
+    const aliasSuffix = entry.aliases.length > 0 ? ` | aliases: ${entry.aliases.join(", ")}` : "";
+    const optionsSuffix = options.includeOptions ? formatEditableInputOptionsText(entry) : "";
+    return `${entry.path}${role}${classType}${expectedType}: ${entry.value_preview}${aliasSuffix}${optionsSuffix}`;
+  });
+}
+function formatInputDefinitionSummary(inputName, inputDefinition) {
+  const [typeData, extraInfo] = inputDefinition;
+  const typeLabel = Array.isArray(typeData) ? `COMBO (${typeData.length} options)` : String(typeData).toUpperCase();
+  const details = [];
+  if (Array.isArray(typeData)) {
+    details.push(
+      `options: ${typeData.slice(0, 5).map((option) => JSON.stringify(option)).join(", ")}${typeData.length > 5 ? ", ..." : ""}`
+    );
+  }
+  if (isRecord(extraInfo)) {
+    if (extraInfo.default !== void 0) {
+      details.push(`default: ${formatValuePreview(extraInfo.default)}`);
+    }
+    if (typeof extraInfo.min === "number") {
+      details.push(`min: ${extraInfo.min}`);
+    }
+    if (typeof extraInfo.max === "number") {
+      details.push(`max: ${extraInfo.max}`);
+    }
+  }
+  return `${inputName} <${typeLabel}>${details.length > 0 ? ` | ${details.join(" | ")}` : ""}`;
+}
+function formatNodeDefinitionText(nodeClass, node) {
+  const requiredInputs = Object.entries(node.input.required ?? {});
+  const optionalInputs = Object.entries(node.input.optional ?? {});
+  const lines = [
+    `Node: ${nodeClass}`,
+    `Display name: ${node.display_name || node.name || nodeClass}`,
+    `Category: ${node.category || "uncategorized"}`
+  ];
+  if (typeof node.description === "string" && node.description.trim().length > 0) {
+    lines.push(`Description: ${node.description.trim()}`);
+  }
+  lines.push(
+    `Required inputs (${requiredInputs.length}): ${requiredInputs.length > 0 ? requiredInputs.map(
+      ([inputName, inputDefinition]) => formatInputDefinitionSummary(
+        inputName,
+        inputDefinition
+      )
+    ).join("\n") : "None."}`
+  );
+  lines.push(
+    `Optional inputs (${optionalInputs.length}): ${optionalInputs.length > 0 ? optionalInputs.map(
+      ([inputName, inputDefinition]) => formatInputDefinitionSummary(
+        inputName,
+        inputDefinition
+      )
+    ).join("\n") : "None."}`
+  );
+  lines.push(`Outputs: ${node.output.join(", ") || "None."}`);
+  return lines.join("\n");
+}
+function formatNodeListText(classes, rawInfo, pagination) {
+  const text = `Available nodes (${pagination.total} total):
+` + formatListItems(classes, (nodeClass) => {
+    const node = rawInfo[nodeClass];
+    const displayName = node.display_name && node.display_name !== nodeClass ? ` \u2014 ${node.display_name}` : "";
+    const category = node.category ? ` | category: ${node.category}` : "";
+    const description = typeof node.description === "string" && node.description.trim().length > 0 ? ` | ${node.description.trim()}` : "";
+    return `${nodeClass}${displayName}${category}${description}`;
+  });
+  if (!pagination.hasMore) {
+    return text;
+  }
+  return `${text}
+
+${formatPaginationFooter(
+    pagination.offset,
+    pagination.limit,
+    pagination.total
+  )}`;
+}
+
+// src/zod.ts
+var zNamespace = zod_exports;
+var z3 = zNamespace.z ?? zNamespace.default ?? zNamespace;
+
+// src/workflow/overrides.ts
+var INPUT_SEGMENT = "inputs";
+var OverrideRecordSchema = z3.record(z3.string(), z3.unknown());
+var OverridesSchema = z3.union([OverrideRecordSchema, z3.string()]).describe(
+  'Dot-notation overrides as an object or JSON string. e.g. {"6.seed": 7}'
+);
+function validateOverrideValueType(path4, value, inputDefinition) {
+  if (!inputDefinition) {
+    return;
+  }
+  const [typeData] = inputDefinition;
+  if (Array.isArray(typeData)) {
+    if (!typeData.some((option) => option === value)) {
+      throw new WorkflowInputError(
+        `Invalid override for '${path4}': expected one of ${typeData.map((option) => JSON.stringify(option)).join(", ")}, received ${JSON.stringify(value)}.`,
+        "overrides"
+      );
+    }
+    return;
+  }
+  const typeName = String(typeData).toUpperCase();
+  switch (typeName) {
+    case "INT":
+      if (typeof value !== "number" || !Number.isInteger(value)) {
+        throw new WorkflowInputError(
+          `Invalid override for '${path4}': expected INT, received ${getValueTypeName(
+            value
+          )}.`,
+          "overrides"
+        );
+      }
+      return;
+    case "FLOAT":
+      if (typeof value !== "number") {
+        throw new WorkflowInputError(
+          `Invalid override for '${path4}': expected FLOAT, received ${getValueTypeName(
+            value
+          )}.`,
+          "overrides"
+        );
+      }
+      return;
+    case "STRING":
+      if (typeof value !== "string") {
+        throw new WorkflowInputError(
+          `Invalid override for '${path4}': expected STRING, received ${getValueTypeName(
+            value
+          )}.`,
+          "overrides"
+        );
+      }
+      return;
+    case "BOOLEAN":
+      if (typeof value !== "boolean") {
+        throw new WorkflowInputError(
+          `Invalid override for '${path4}': expected BOOLEAN, received ${getValueTypeName(
+            value
+          )}.`,
+          "overrides"
+        );
+      }
+      return;
+    default:
+      return;
+  }
+}
 function resolveOverridePath(prompt, rawPath, editableIndex) {
   if (!editableIndex) {
     return rawPath;
@@ -31720,25 +31893,9 @@ function resolveOverridePath(prompt, rawPath, editableIndex) {
   }
   const resolved = editableIndex.aliasToPath.get(rawPath);
   if (resolved) {
-    return resolved;
-  }
-  const rawParts = rawPath.split(".");
-  if (rawParts.length >= 3 && rawParts[1] === INPUT_SEGMENT) {
-    const pathHelp2 = getPathSegmentHelp(prompt, rawPath);
-    if (pathHelp2?.nodeId && pathHelp2.nodeExists) {
-      const validPaths = describeAvailableNodePaths(
-        editableIndex,
-        pathHelp2.nodeId
-      );
-      if (validPaths.length > 0) {
-        throw new WorkflowInputError(
-          `Unsupported override path '${rawPath}'. Only detected editable inputs can be overridden. Valid editable paths for node ${pathHelp2.nodeId}:
-${validPaths}`,
-          "overrides"
-        );
-      }
-    }
+    return resolved;
   }
+  const rawParts = rawPath.split(".");
   const pathHelp = getPathSegmentHelp(prompt, rawPath);
   if (pathHelp?.nodeId && pathHelp.nodeExists) {
     const validPaths = describeAvailableNodePaths(
@@ -31746,11 +31903,10 @@ ${validPaths}`,
       pathHelp.nodeId
     );
     if (validPaths.length > 0) {
-      throw new WorkflowInputError(
-        `Unknown override path '${rawPath}'. Valid editable paths for node ${pathHelp.nodeId}:
-${validPaths}`,
-        "overrides"
-      );
+      const message = rawParts.length >= 3 && rawParts[1] === INPUT_SEGMENT ? `Unsupported override path '${rawPath}'. Only detected editable inputs can be overridden. Valid editable paths for node ${pathHelp.nodeId}:
+${validPaths}` : `Unknown override path '${rawPath}'. Valid editable paths for node ${pathHelp.nodeId}:
+${validPaths}`;
+      throw new WorkflowInputError(message, "overrides");
     }
   }
   return rawPath;
@@ -31779,78 +31935,6 @@ function parseOverrides(rawOverrides) {
   }
   return parsed;
 }
-function analyzeEditableInputs(prompt, objectInfo, widgetPathMap = {}) {
-  const consumerMap = /* @__PURE__ */ new Map();
-  for (const [nodeId, node] of Object.entries(prompt)) {
-    for (const [inputName, value] of Object.entries(node.inputs)) {
-      if (!isConnectionValue(value)) {
-        continue;
-      }
-      const sourceNodeId = String(value[0]);
-      const consumers = consumerMap.get(sourceNodeId) ?? [];
-      consumers.push({
-        inputName,
-        classType: node.class_type,
-        nodeId
-      });
-      consumerMap.set(sourceNodeId, consumers);
-    }
-  }
-  const editableInputs = [];
-  for (const [nodeId, node] of Object.entries(prompt)) {
-    for (const [inputName, value] of Object.entries(node.inputs)) {
-      if (isConnectionValue(value)) {
-        continue;
-      }
-      editableInputs.push({
-        path: `${nodeId}.inputs.${inputName}`,
-        node_id: nodeId,
-        class_type: node.class_type,
-        input_name: inputName,
-        value,
-        value_preview: formatValuePreview(value),
-        role: inferEditableRole(
-          nodeId,
-          node.class_type,
-          inputName,
-          consumerMap
-        ),
-        aliases: [],
-        expected_type: getExpectedTypeName(
-          getInputDefinition(objectInfo, prompt, nodeId, inputName)
-        ),
-        ...getInputOptionsSummary(
-          getInputDefinition(objectInfo, prompt, nodeId, inputName)
-        )
-      });
-    }
-  }
-  return editableInputs.map((entry) => ({
-    ...entry,
-    aliases: getEditableAliases(
-      entry.path,
-      entry.node_id,
-      entry.input_name,
-      entry.role,
-      widgetPathMap
-    )
-  })).sort(
-    (left, right) => left.path.localeCompare(right.path, void 0, { numeric: true })
-  );
-}
-function formatEditableInputsText(editableInputs, options = {}) {
-  if (editableInputs.length === 0) {
-    return "No literal editable inputs were detected.";
-  }
-  return formatListItems(editableInputs, (entry) => {
-    const role = entry.role ? ` [${entry.role}]` : "";
-    const classType = entry.class_type ? ` (${entry.class_type})` : "";
-    const expectedType = entry.expected_type ? ` <${entry.expected_type}>` : "";
-    const aliasSuffix = entry.aliases.length > 0 ? ` | aliases: ${entry.aliases.join(", ")}` : "";
-    const optionsSuffix = options.includeOptions ? formatEditableInputOptionsText(entry) : "";
-    return `${entry.path}${role}${classType}${expectedType}: ${entry.value_preview}${aliasSuffix}${optionsSuffix}`;
-  });
-}
 function validatePromptPreflight(prompt, objectInfo, widgetPathMap = {}) {
   if (!objectInfo) {
     return {
@@ -31904,106 +31988,6 @@ ${issues.join("\n")}`,
     combo_inputs: comboInputs
   };
 }
-function formatInputDefinitionSummary(inputName, inputDefinition) {
-  const [typeData, extraInfo] = inputDefinition;
-  const typeLabel = Array.isArray(typeData) ? `COMBO (${typeData.length} options)` : String(typeData).toUpperCase();
-  const details = [];
-  if (Array.isArray(typeData)) {
-    details.push(
-      `options: ${typeData.slice(0, 5).map((option) => JSON.stringify(option)).join(", ")}${typeData.length > 5 ? ", ..." : ""}`
-    );
-  }
-  if (isRecord2(extraInfo)) {
-    if (extraInfo.default !== void 0) {
-      details.push(`default: ${formatValuePreview(extraInfo.default)}`);
-    }
-    if (typeof extraInfo.min === "number") {
-      details.push(`min: ${extraInfo.min}`);
-    }
-    if (typeof extraInfo.max === "number") {
-      details.push(`max: ${extraInfo.max}`);
-    }
-  }
-  return `${inputName} <${typeLabel}>${details.length > 0 ? ` | ${details.join(" | ")}` : ""}`;
-}
-function formatNodeDefinitionText(nodeClass, node) {
-  const requiredInputs = Object.entries(node.input.required ?? {});
-  const optionalInputs = Object.entries(node.input.optional ?? {});
-  const lines = [
-    `Node: ${nodeClass}`,
-    `Display name: ${node.display_name || node.name || nodeClass}`,
-    `Category: ${node.category || "uncategorized"}`
-  ];
-  if (typeof node.description === "string" && node.description.trim().length > 0) {
-    lines.push(`Description: ${node.description.trim()}`);
-  }
-  lines.push(
-    `Required inputs (${requiredInputs.length}): ${requiredInputs.length > 0 ? requiredInputs.map(
-      ([inputName, inputDefinition]) => formatInputDefinitionSummary(
-        inputName,
-        inputDefinition
-      )
-    ).join("\n") : "None."}`
-  );
-  lines.push(
-    `Optional inputs (${optionalInputs.length}): ${optionalInputs.length > 0 ? optionalInputs.map(
-      ([inputName, inputDefinition]) => formatInputDefinitionSummary(
-        inputName,
-        inputDefinition
-      )
-    ).join("\n") : "None."}`
-  );
-  lines.push(`Outputs: ${node.output.join(", ") || "None."}`);
-  return lines.join("\n");
-}
-function formatNodeListText(classes, rawInfo, pagination) {
-  const text = `Available nodes (${pagination.total} total):
-` + formatListItems(classes, (nodeClass) => {
-    const node = rawInfo[nodeClass];
-    const displayName = node.display_name && node.display_name !== nodeClass ? ` \u2014 ${node.display_name}` : "";
-    const category = node.category ? ` | category: ${node.category}` : "";
-    const description = typeof node.description === "string" && node.description.trim().length > 0 ? ` | ${node.description.trim()}` : "";
-    return `${nodeClass}${displayName}${category}${description}`;
-  });
-  if (!pagination.hasMore) {
-    return text;
-  }
-  return `${text}
-
-${formatPaginationFooter(
-    pagination.offset,
-    pagination.limit,
-    pagination.total
-  )}`;
-}
-async function getQueueSnapshot() {
-  try {
-    return IS_MOCK ? MOCK_FIXTURES.queue : await fetchComfyJson("/queue");
-  } catch {
-    return void 0;
-  }
-}
-function formatQueueSnapshotText(queueSnapshot) {
-  if (!queueSnapshot) {
-    return "";
-  }
-  return `Queue snapshot: ${queueSnapshot.queue_running.length} running, ${queueSnapshot.queue_pending.length} pending.`;
-}
-function getPromptQueueVisibility(promptId, queueSnapshot) {
-  if (!queueSnapshot) {
-    return null;
-  }
-  return JSON.stringify(queueSnapshot).includes(promptId);
-}
-function deriveSavedWorkflowFileName(sourcePath, requestedName) {
-  if (requestedName) {
-    return normalizeWorkflowFileName(requestedName);
-  }
-  if (sourcePath) {
-    return normalizeWorkflowFileName(path.basename(sourcePath));
-  }
-  return `workflow-${Date.now()}.json`;
-}
 function applyOverrides(prompt, overrides, options = {}) {
   for (const [rawKey, value] of Object.entries(overrides)) {
     const key = resolveOverridePath(prompt, rawKey, options.editableIndex);
@@ -32012,7 +31996,7 @@ function applyOverrides(prompt, overrides, options = {}) {
     for (let i = 0; i < parts.length - 1; i++) {
       const part = parts[i];
       const next = current[part];
-      if (!isRecord2(next)) {
+      if (!isRecord(next) || Array.isArray(next)) {
         const pathHelp = getPathSegmentHelp(prompt, key);
         const validPaths = pathHelp?.nodeId && pathHelp.nodeExists ? describeAvailableNodePaths(
           options.editableIndex,
@@ -32040,6 +32024,63 @@ ${validPaths}` : "";
     current[parts[parts.length - 1]] = value;
   }
 }
+
+// src/index.ts
+var PACKAGE_VERSION = getPackageVersion(import.meta.url);
+function handleComfyError(error48) {
+  if (error48 instanceof zod_exports.ZodError) {
+    return createValidationError(
+      "arguments",
+      error48.issues[0]?.message || "Invalid input."
+    );
+  }
+  if (error48 instanceof WorkflowInputError) {
+    return error48.kind === "not_found" ? createNotFoundError(error48.message) : createValidationError(error48.field, error48.message);
+  }
+  if (error48 instanceof ComfyApiError) {
+    const detailSummary = summarizeComfyErrorDetails(error48.details);
+    return {
+      content: [
+        {
+          type: "text",
+          text: detailSummary.length > 0 ? `${error48.message}
+${detailSummary}` : error48.message
+        }
+      ],
+      structuredContent: toStructuredContent(error48.details),
+      isError: true
+    };
+  }
+  if (error48 instanceof WorkflowTimeoutError) {
+    const queueSummary = formatQueueSnapshotText(error48.queueSnapshot);
+    const promptVisibility = getPromptQueueVisibility(
+      error48.promptId,
+      error48.queueSnapshot
+    );
+    const promptVisibilityText = promptVisibility === true ? "The prompt is still visible in the ComfyUI queue snapshot." : promptVisibility === false ? "The prompt is not visible in the ComfyUI queue snapshot; inspect ComfyUI history or server logs if it never completes." : "ComfyUI queue state was unavailable at timeout.";
+    return {
+      content: [
+        {
+          type: "text",
+          text: [
+            `${error48.message}.`,
+            queueSummary,
+            promptVisibilityText,
+            `You can re-attach by calling 'comfyui_wait_for_workflow' again with prompt_id '${error48.promptId}'.`
+          ].filter((line) => line && line.length > 0).join("\n")
+        }
+      ],
+      structuredContent: {
+        prompt_id: error48.promptId,
+        timeout_seconds: error48.timeoutSeconds,
+        queue: error48.queueSnapshot ?? null,
+        prompt_visible_in_queue: promptVisibility
+      },
+      isError: true
+    };
+  }
+  return createInternalError(error48);
+}
 var InspectNodeSchema = paginationSchema.extend({
   node_class: z3.string().optional().describe(
     "Specific node class to inspect. If omitted, lists all available nodes."
@@ -32139,9 +32180,6 @@ async function handleInspectNode(args) {
     return handleComfyError(error48);
   }
 }
-function getWorkflowId(absolutePath) {
-  return crypto.createHash("sha256").update(absolutePath).digest("hex").substring(0, 8);
-}
 async function handleDiscoverWorkflows(args) {
   const pathsToScan = getWorkflowDirectories(args.directory_path);
   const objectInfoAvailability = await tryGetObjectInfo();
@@ -32149,10 +32187,10 @@ async function handleDiscoverWorkflows(args) {
   let skippedNonWorkflowFiles = 0;
   for (const dir of pathsToScan) {
     try {
-      const entries = await fs.readdir(dir, { withFileTypes: true });
+      const entries = await fs3.readdir(dir, { withFileTypes: true });
       for (const entry of entries) {
         if (entry.isFile() && entry.name.endsWith(".json")) {
-          const fullPath = path.join(dir, entry.name);
+          const fullPath = path3.join(dir, entry.name);
           try {
             const inspected = await inspectWorkflowFile(
               fullPath,
@@ -32237,28 +32275,6 @@ Hint: If your workflows are stored elsewhere, provide a custom path using the 'd
     }
   };
 }
-async function waitForWorkflowResult(prompt_id, timeout_seconds) {
-  const start = Date.now();
-  const end = start + timeout_seconds * 1e3;
-  while (Date.now() < end) {
-    let history = MOCK_FIXTURES.history;
-    if (!IS_MOCK) {
-      try {
-        history = await fetchComfyJson(
-          `/history/${prompt_id}`
-        );
-      } catch {
-        history = {};
-      }
-    }
-    if (history?.[prompt_id]) {
-      return history[prompt_id];
-    }
-    await new Promise((r) => setTimeout(r, 1e3));
-  }
-  const queueSnapshot = await getQueueSnapshot();
-  throw new WorkflowTimeoutError(prompt_id, timeout_seconds, queueSnapshot);
-}
 async function handleWaitForWorkflow(args) {
   try {
     const historyEntry = await waitForWorkflowResult(
@@ -32438,11 +32454,11 @@ async function handleSaveWorkflow(args) {
         "Directory traversal sequences (..) are not allowed."
       );
     }
-    const outputPath = path.resolve(process.cwd(), outputFileName);
-    await fs.mkdir(path.dirname(outputPath), { recursive: true });
+    const outputPath = path3.resolve(process.cwd(), outputFileName);
+    await fs3.mkdir(path3.dirname(outputPath), { recursive: true });
     let existedBeforeSave = false;
     try {
-      await fs.access(outputPath);
+      await fs3.access(outputPath);
       existedBeforeSave = true;
     } catch {
       existedBeforeSave = false;
@@ -32453,7 +32469,7 @@ async function handleSaveWorkflow(args) {
         `File already exists: ${outputFileName}. Set overwrite=true to replace it.`
       );
     }
-    await fs.writeFile(
+    await fs3.writeFile(
       outputPath,
       `${JSON.stringify(prompt, null, 2)}
 `,