@goodandready/dsh-image-gen 0.10.24 → 0.10.26

package/lib/providers.js

@@ -26,6 +26,42 @@ import {
   normalizeCount,
 } from './provider-utils.js'
 
+// Per-backend factories (#218)
+import { createFalGenerator } from './providers/backends/fal.js'
+import { createCustomGenerator } from './providers/backends/custom.js'
+import { createSubscriptionGenerator } from './providers/backends/subscription.js'
+import { createLocalGenerator } from './providers/backends/local.js'
+import { createSeedreamGenerator } from './providers/backends/seedream.js'
+import { createGeminiGenerator } from './providers/backends/gemini.js'
+import { createReplicateGenerator } from './providers/backends/replicate.js'
+export {
+  falAuthHeader,
+  normalizeMediaType,
+  submitJob,
+  pollStatus,
+  buildEditForm,
+  estimateCost,
+  pxSize,
+  ASPECT_RATIOS,
+  sizeToPixels,
+  snapToMultipleOf64,
+  snapDimensions,
+  extractComfyNodeErrors,
+} from './providers/shared-helpers.js'
+
+import {
+  falAuthHeader,
+  normalizeMediaType,
+  submitJob,
+  pollStatus,
+  buildEditForm,
+  estimateCost,
+  pxSize,
+  snapToMultipleOf64,
+  snapDimensions,
+  extractComfyNodeErrors,
+} from './providers/shared-helpers.js'
+
 // Re-export utilities so existing imports from './providers.js' continue to work
 export {
   resolveApiKeyCandidates,
@@ -74,25 +110,7 @@ export async function tryGenerate(generators, order, seed, promptArg) {
  * Supports named sizes (landscape_16_9, etc.), aspect ratios (16:9, 3:2, 9:16),
  * and pixel dimensions [width, height], preserving requested proportions.
  */
-export function falAuthHeader(key) {
-  const trimmed = String(key ?? '').trim()
-  if (!trimmed) return ''
-  return trimmed.startsWith('Key ') || trimmed.startsWith('key ')
-    ? trimmed
-    : `Key ${trimmed}`
-}
-
 /** Map a content type onto the attachment service's supported set. */
-export function normalizeMediaType(contentType, fallbackFormat) {
-  const raw = String(contentType ?? '').toLowerCase()
-  if (raw.includes('jpeg') || raw.includes('jpg')) return 'image/jpeg'
-  if (raw.includes('webp')) return 'image/webp'
-  if (raw.includes('png')) return 'image/png'
-  if (fallbackFormat === 'jpeg') return 'image/jpeg'
-  if (fallbackFormat === 'webp') return 'image/webp'
-  return 'image/png'
-}
-
 /** Generation metadata written to companion sidecar file. */
 export function buildSidecar({
   prompt, size, format, seed, provider, deliverAs,
@@ -102,58 +120,7 @@ export function buildSidecar({
 }
 
 /** Submit a generation job to the FAL queue. */
-export async function submitJob(fetchImpl, baseURL, model, key, body, signal) {
-  const res = await fetchImpl(`${baseURL}/${model}`, {
-    method: 'POST',
-    headers: {
-      Authorization: falAuthHeader(key),
-      'Content-Type': 'application/json',
-    },
-    body: JSON.stringify(body),
-    signal,
-  })
-  const data = await res.json().catch(() => ({}))
-  if (!res.ok || !data.request_id) {
-    const detail = typeof data.detail === 'string' ? data.detail : JSON.stringify(data).slice(0, 600)
-    throw new Error(`FAL submit failed (HTTP ${res.status}): ${detail}`)
-  }
-  return data
-}
-
 /** Poll the FAL status endpoint until completion, failure, timeout, or abort. */
-export async function pollStatus(fetchImpl, statusUrl, key, signal, pollIntervalMs = 500, timeoutMs = 180000) {
-  const deadline = Date.now() + timeoutMs
-  let attempt = 0
-  for (;;) {
-    if (signal?.aborted) throw new Error('FAL generation cancelled')
-    if (Date.now() > deadline) throw new Error(`FAL generation timed out after ${timeoutMs} ms`)
-    if (attempt > 0) {
-      const delay = calculateBackoff(attempt - 1, pollIntervalMs, 5000)
-      await new Promise((resolve) => {
-        const timer = setTimeout(resolve, delay)
-        if (signal) {
-          signal.addEventListener('abort', () => {
-            clearTimeout(timer)
-            resolve()
-          }, { once: true })
-        }
-      })
-    }
-    attempt++
-    if (signal?.aborted) throw new Error('FAL generation cancelled')
-    const res = await fetchImpl(statusUrl, { headers: { Authorization: falAuthHeader(key) }, signal })
-    const data = await res.json().catch(() => ({}))
-    const status = data.status
-    if (status === 'COMPLETED') return data
-    if (status === 'ERROR' || data.error || data.detail) {
-      throw new Error(`FAL generation failed: ${JSON.stringify(data).slice(0, 600)}`)
-    }
-    if (status !== 'IN_QUEUE' && status !== 'IN_PROGRESS') {
-      throw new Error(`Unexpected FAL status "${status}": ${JSON.stringify(data).slice(0, 300)}`)
-    }
-  }
-}
-
 /**
  * @param deps {{fetchImpl: Function, resolveKey: (ref: string) => Promise<string>, cfg: object}}
  * @param job {{prompt: string, size: string, format: string, seed: number|undefined, signal: AbortSignal}}
@@ -164,51 +131,11 @@ export async function pollStatus(fetchImpl, statusUrl, key, signal, pollInterval
 
 /** Named size -> [width, height] for local APIs. */
 /** Aspect ratio -> [width, height] (pixels, normalized to 1024 base). */
-export const ASPECT_RATIOS = {
-  '1:1': [1024, 1024],
-  '16:9': [1344, 768],
-  '9:16': [768, 1344],
-  '4:3': [1152, 896],
-  '3:4': [896, 1152],
-  '3:2': [1152, 768],
-  '2:3': [768, 1152],
-}
-
 /** Pixel dimensions: aspectPixels (if specified) or mapped from named size. */
 /** Snap a pixel dimension to nearest multiple of 64. */
-export function snapToMultipleOf64(dim, minVal = 256, maxVal = 2048) {
-  const n = Math.round(Number(dim) / 64) * 64
-  return Math.max(minVal, Math.min(maxVal, n))
-}
-
-export function snapDimensions(width, height) {
-  return [snapToMultipleOf64(width), snapToMultipleOf64(height)]
-}
 
-export function pxSize(aspectPixels, size) {
-  if (aspectPixels) return snapDimensions(aspectPixels[0], aspectPixels[1])
-  const [w, h] = sizeToPixels(size)
-  return snapDimensions(w, h)
-}
-
-export function sizeToPixels(size) {
-  const px = SIZE_PIXELS[size]
-  if (!px) return [1024, 1024]
-  const [w, h] = px.split('x').map(Number)
-  return [w, h]
-}
 
 /** Assemble multipart body for /images/edits (OpenAI-compatible edit). */
-export function buildEditForm({ source, mask, prompt, size, strength }) {
-  const form = new FormData()
-  form.append('image', new Blob([source.bytes], { type: source.mediaType || 'image/png' }), 'source.png')
-  if (mask) form.append('mask', new Blob([mask.bytes], { type: mask.mediaType || 'image/png' }), 'mask.png')
-  form.append('prompt', prompt)
-  if (size) form.append('size', size)
-  if (strength !== undefined) form.append('strength', String(strength))
-  return form
-}
-
 /** Compare two images: ratio of differing pixels (0..1). */
 export async function pixelDiff(a, b) {
   if (!a || !b) return { error: 'missing image' }
@@ -256,438 +183,19 @@ export async function saveAttachmentSafe(ctx, { bytes, mediaType, name }) {
   }
 }
 
-export function extractComfyNodeErrors(entry) {
-  if (!entry || typeof entry !== 'object') return ''
-  const status = entry.status
-  if (status && status.status_str === 'error') {
-    const msgs = status.messages || []
-    const errList = []
-    for (const m of msgs) {
-      if (Array.isArray(m) && m[0] === 'execution_error') {
-        const d = m[1] || {}
-        errList.push(`node ${d.node_id || 'unknown'} (${d.node_type || ''}): ${d.exception_message || d.exception_type || 'execution failed'}`)
-      }
-    }
-    if (errList.length) return errList.join('; ')
-    return status.status_str || 'execution error'
-  }
-  return ''
-}
 
 export function makeProviders(deps, job) {
-  const { fetchImpl, resolveKey, cfg } = deps
-  const { prompt, size, format, seed, signal, negativePrompt, guidanceScale, source, mask, strength, quality, style, aspectPixels, aspectRatio } = job
-
-  async function fal(seedArg = seed, promptArg = prompt) {
-    const key = await resolveKey(cfg.apiKeyEnv)
-    const isEdit = Boolean(source && source.bytes)
-    const targetModel = isEdit
-      ? (mask && mask.bytes ? 'fal-ai/flux-pro/v1/inpaint' : 'fal-ai/flux/dev/image-to-image')
-      : cfg.model
-    const body = { prompt: promptArg, image_size: size, num_images: 1 }
-    if (seedArg !== undefined) body.seed = seedArg
-    if (negativePrompt !== undefined) body.negative_prompt = negativePrompt
-    if (guidanceScale !== undefined) body.guidance_scale = guidanceScale
-    if (format !== 'png') body.output_format = format
-    if (isEdit) {
-      body.image_url = `data:${source.mediaType || 'image/png'};base64,${Buffer.from(source.bytes).toString('base64')}`
-      if (mask && mask.bytes) {
-        body.mask_url = `data:${mask.mediaType || 'image/png'};base64,${Buffer.from(mask.bytes).toString('base64')}`
-      }
-      if (strength !== undefined) {
-        body.strength = strength
-      }
-    }
-
-    const submit = await submitJob(fetchImpl, cfg.baseURL, targetModel, key, body, signal)
-    const statusUrl = submit.status_url || `${cfg.baseURL}/${cfg.model}/requests/${submit.request_id}/status`
-    const statusBody = await pollStatus(fetchImpl, statusUrl, key, signal, cfg.pollIntervalMs, cfg.timeoutMs)
-
-    const resultRes = await fetchImpl(statusBody.response_url, {
-      headers: { Authorization: falAuthHeader(key) },
-      signal,
-    })
-    const result = await resultRes.json().catch(() => ({}))
-    const image = result.images && result.images[0]
-    if (!image || !image.url) {
-      throw new Error(`FAL returned no images: ${JSON.stringify(result).slice(0, 600)}`)
-    }
-    const download = await fetchImpl(image.url, { signal })
-    if (!download.ok) {
-      throw new Error(`Failed to download generated image (HTTP ${download.status})`)
-    }
-    return {
-      bytes: Buffer.from(await download.arrayBuffer()),
-      mediaType: normalizeMediaType(image.content_type, format),
-      width: image.width ?? 0,
-      height: image.height ?? 0,
-      seed: result.seed ?? seedArg ?? 0,
-      sourceUrl: image.url,
-      cost: result.cost,
-    }
-  }
-
-  // OpenAI-compatible image API. Single HTTP request instead of queue polling.
-  async function custom(seedArg = seed, promptArg = prompt) {
-    const base = String(cfg.customBaseURL || '').replace(/\/+$/, '')
-    if (!base) throw new Error('Custom image provider: base URL is not configured (Settings → Image generation)')
-    if (!cfg.customModel) throw new Error('Custom image provider: model is not configured')
-
-    // Empty key reference indicates unauthenticated gateway.
-    const key = cfg.customKeyEnv ? await resolveKey(cfg.customKeyEnv) : ''
-    const headers = { 'Content-Type': 'application/json' }
-    if (key) headers.Authorization = `Bearer ${key}`
-
-    // Omit response_format; modern models return either base64 or URL.
-    const endpoint = buildEndpointUrl(base, source ? 'images/edits' : 'images/generations')
-    const res = await fetchImpl(endpoint, {
-      method: 'POST',
-      headers,
-      body: source
-        ? buildEditForm({ source, mask, prompt: promptArg, size: cfg.customSize || (aspectPixels ? aspectPixels.join('x') : SIZE_PIXELS[size]) || size, strength })
-        : JSON.stringify({
-            model: cfg.customModel,
-            prompt: promptArg,
-            n: 1,
-            size: cfg.customSize || (aspectPixels ? aspectPixels.join('x') : SIZE_PIXELS[size]) || size,
-            ...(negativePrompt !== undefined ? { negative_prompt: negativePrompt } : {}),
-            ...(guidanceScale !== undefined ? { guidance_scale: guidanceScale } : {}),
-            ...(quality && quality !== 'auto' ? { quality } : {}),
-          }),
-      signal,
-    })
-    const data = await res.json().catch(() => ({}))
-    if (!res.ok) {
-      if (res.status === 503) {
-        const msg = (data?.error?.message || JSON.stringify(data)).toLowerCase()
-        if (msg.includes('no available channel') || msg.includes('model_not_found')) {
-          throw new Error(`Custom image API has no provisioned channel for "${cfg.customModel}" (HTTP 503); check gateway channel and routing configuration`)
-        }
-      }
-      const detail = formatErrorMessage(data?.error || data)
-      throw new Error(`Image API failed (HTTP ${res.status}): ${detail}`)
-    }
-    const item = data?.data?.[0]
-    if (!item) {
-      throw new Error(`Image API returned no images: ${JSON.stringify(data).slice(0, 600)}`)
-    }
-
-    if (item.b64_json) {
-      return {
-        bytes: Buffer.from(item.b64_json, 'base64'),
-        mediaType: normalizeMediaType(data.output_format || '', format),
-        width: 0,
-        height: 0,
-        seed: seedArg ?? 0,
-        sourceUrl: '',
-      }
-    }
-    if (!item.url) {
-      throw new Error(`Image API returned neither b64_json nor url: ${JSON.stringify(item).slice(0, 300)}`)
-    }
-    const download = await fetchImpl(item.url, { signal })
-    if (!download.ok) {
-      throw new Error(`Failed to download generated image (HTTP ${download.status})`)
-    }
-    const contentType = download.headers && typeof download.headers.get === 'function'
-      ? download.headers.get('content-type')
-      : ''
-    return {
-      bytes: Buffer.from(await download.arrayBuffer()),
-      mediaType: normalizeMediaType(contentType, format),
-      width: 0,
-      height: 0,
-      seed: seedArg ?? 0,
-      sourceUrl: item.url,
-    }
-  }
-
-  // Subscription generation: dsh-subscriptions handles session authentication.
-  function subscription(provider) {
-    return async function generate(seedArg = seed, promptArg = prompt) {
-      if (source) {
-        return { ok: false, provider, reason: `${provider}: does not support image editing — use fal, custom, or local` }
-      }
-      const images = deps.subscriptionImages
-      if (!images || typeof images.generate !== 'function') {
-        return {
-          ok: false,
-          provider,
-          reason: `${provider}: requires dsh-subscriptions plugin to manage session authentication`,
-        }
-      }
-      let produced
-      try {
-        produced = await images.generate({
-          provider,
-          prompt: promptArg,
-          size: resolveSubscriptionSize(size, aspectPixels, aspectRatio),
-          quality: cfg.subscriptionQuality || undefined,
-          signal,
-        })
-      } catch (e) {
-        return { ok: false, provider, reason: formatErrorMessage(e, provider) }
-      }
-      const first = Array.isArray(produced) ? produced[0] : null
-      if (!first || !first.b64_json) {
-        return { ok: false, provider, reason: `${provider}: no image returned in response` }
-      }
-      return {
-        bytes: Buffer.from(first.b64_json, 'base64'),
-        // Subscription outputs default to PNG media type.
-        mediaType: 'image/png',
-        width: 0,
-        height: 0,
-        seed: seedArg ?? 0,
-        sourceUrl: '',
-        revisedPrompt: first.revisedPrompt || '',
-      }
-    }
-  }
-
-  // Local generation: ComfyUI (queue + poll) or Automatic1111 (txt2img).
-  async function local(seedArg = seed, promptArg = prompt) {
-    const base = String(cfg.localBaseURL || '').replace(/\/+$/, '')
-    if (!base) throw new Error('Local image provider: server address is not configured (Settings → Image generation)')
-    const [width, height] = pxSize(aspectPixels, size)
-    const kind = cfg.localKind === 'a1111' ? 'a1111' : 'comfyui'
-
-    if (kind === 'a1111') {
-      const isImg2Img = Boolean(source && source.bytes)
-      const endpoint = `${base}${isImg2Img ? '/sdapi/v1/img2img' : '/sdapi/v1/txt2img'}`
-      const body = {
-        prompt: promptArg,
-        negative_prompt: negativePrompt,
-        width,
-        height,
-        steps: cfg.localSteps ?? 20,
-        cfg_scale: cfg.localCfg ?? 7,
-        seed: seedArg ?? -1,
-      }
-      if (isImg2Img) {
-        body.init_images = [Buffer.from(source.bytes).toString('base64')]
-        body.denoising_strength = strength ?? (mask ? 0.75 : 0.35)
-        if (mask && mask.bytes) {
-          body.mask = Buffer.from(mask.bytes).toString('base64')
-        }
-      }
-      if (cfg.localModel) body.override_settings = { sd_model_checkpoint: cfg.localModel }
-      const res = await fetchImpl(endpoint, {
-        method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify(body),
-        signal,
-      })
-      if (!res.ok) {
-        throw new Error(`Local A1111 failed (HTTP ${res.status}): ${String(await res.text().catch(() => '')).slice(0, 300)}`)
-      }
-      const data = await res.json().catch(() => ({}))
-      const b64 = data.images && data.images[0]
-      if (!b64) throw new Error('Local A1111 returned no images')
-      return {
-        bytes: Buffer.from(b64, 'base64'),
-        mediaType: normalizeMediaType('image/png', format),
-        width,
-        height,
-        seed: seedArg ?? 0,
-        sourceUrl: '',
-      }
-    }
-
-    // ComfyUI: submit via /prompt, poll /history/{prompt_id} until completed.
-    const promptId = `dsh-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`
-    const workflow = {
-      prompt: {
-        '3': { class_type: 'KSampler', inputs: { seed: seedArg ?? 0, steps: cfg.localSteps ?? 20, cfg: cfg.localCfg ?? 7, sampler_name: 'euler', scheduler: 'normal', denoise: 1, model: ['4', 0], positive: ['6', 0], negative: ['7', 0], latent_image: ['5', 0] } },
-        '4': { class_type: 'CheckpointLoaderSimple', inputs: { ckpt_name: cfg.localModel || 'v1-5-pruned-emaonly.safetensors' } },
-        '5': { class_type: 'EmptyLatentImage', inputs: { width, height, batch_size: 1 } },
-        '6': { class_type: 'CLIPTextEncode', inputs: { text: promptArg, clip: ['4', 1] } },
-        '7': { class_type: 'CLIPTextEncode', inputs: { text: negativePrompt || '', clip: ['4', 1] } },
-        '8': { class_type: 'VAEDecode', inputs: { samples: ['3', 0], vae: ['4', 2] } },
-        '9': { class_type: 'SaveImage', inputs: { filename_prefix: 'dsh', images: ['8', 0] } },
-      },
-    }
-    const submit = await fetchImpl(`${base}/prompt`, {
-      method: 'POST',
-      headers: { 'Content-Type': 'application/json' },
-      body: JSON.stringify({ prompt: workflow, client_id: promptId }),
-      signal,
-    })
-    if (!submit.ok) {
-      throw new Error(`Local ComfyUI submit failed (HTTP ${submit.status}): ${String(await submit.text().catch(() => '')).slice(0, 300)}`)
-    }
-    const submitData = await submit.json().catch(() => ({}))
-    const pid = submitData.prompt_id
-    if (!pid) throw new Error('Local ComfyUI did not return a prompt_id')
-
-    const deadline = Date.now() + cfg.timeoutMs
-    let attempt = 0
-    for (;;) {
-      if (signal?.aborted) throw new Error('Local ComfyUI generation cancelled')
-      if (Date.now() > deadline) throw new Error(`Local ComfyUI timed out after ${cfg.timeoutMs} ms`)
-      if (attempt > 0) {
-        const cDelay = calculateBackoff(attempt - 1, cfg.pollIntervalMs || 1000, 5000)
-        await new Promise((resolve) => {
-          const timer = setTimeout(resolve, cDelay)
-          if (signal) {
-            signal.addEventListener('abort', () => { clearTimeout(timer); resolve() }, { once: true })
-          }
-        })
-      }
-      attempt++
-      const hist = await fetchImpl(`${base}/history/${pid}`, { signal })
-      if (!hist.ok) continue
-      const histData = await hist.json().catch(() => ({}))
-      const entry = histData[pid]
-      if (entry) {
-        const comfyErr = extractComfyNodeErrors(entry)
-        if (comfyErr) {
-          throw new Error(`Local ComfyUI execution failed: ${comfyErr}`)
-        }
-      }
-      if (entry && entry.outputs) {
-        const outputs = entry.outputs
-        const img = Object.values(outputs).flatMap((o) => o.images || []).find((i) => i && i.filename)
-        if (img) {
-          const dl = await fetchImpl(`${base}/view?filename=${encodeURIComponent(img.filename)}&subfolder=${encodeURIComponent(img.subfolder || '')}&type=${encodeURIComponent(img.type || 'output')}`, { signal })
-          if (!dl.ok) throw new Error(`Local ComfyUI download failed (HTTP ${dl.status})`)
-          return {
-            bytes: Buffer.from(await dl.arrayBuffer()),
-            mediaType: normalizeMediaType('image/png', format),
-            width,
-            height,
-            seed: seedArg ?? 0,
-            sourceUrl: '',
-          }
-        }
-      }
-    }
-  }
-
-  // Seedream (ByteDance): OpenAI-compatible images API.
-  async function seedream(seedArg = seed, promptArg = prompt) {
-    const key = await resolveKey(cfg.seedreamKeyEnv)
-    const base = (cfg.seedreamBaseURL || 'https://api.bytedanceapi.com/v1').replace(/\/+$/, '')
-    const res = await fetchImpl(`${base}/images/generations`, {
-      method: 'POST',
-      headers: { 'Content-Type': 'application/json', Authorization: `Bearer ${key}` },
-      body: JSON.stringify({
-        model: cfg.seedreamModel || 'seedream-4.0',
-        prompt: promptArg,
-        n: 1,
-        size: (aspectPixels ? aspectPixels.join('x') : SIZE_PIXELS[size]) || size,
-        ...(negativePrompt !== undefined ? { negative_prompt: negativePrompt } : {}),
-        ...(quality !== undefined ? { quality } : {}),
-        ...(style !== undefined ? { style } : {}),
-      }),
-      signal,
-    })
-    const data = await res.json().catch(() => ({}))
-    if (!res.ok) {
-      const detail = data?.error?.message || JSON.stringify(data).slice(0, 600)
-      throw new Error(`Seedream failed (HTTP ${res.status}): ${detail}`)
-    }
-    const item = data?.data?.[0]
-    if (!item) throw new Error('Seedream returned no images')
-    if (item.b64_json) {
-      return { bytes: Buffer.from(item.b64_json, 'base64'), mediaType: normalizeMediaType('image/png', format), width: 0, height: 0, seed: seedArg ?? 0, sourceUrl: '' }
-    }
-    if (!item.url) throw new Error('Seedream returned neither b64_json nor url')
-    const dl = await fetchImpl(item.url, { signal })
-    if (!dl.ok) throw new Error(`Seedream download failed (HTTP ${dl.status})`)
-    return { bytes: Buffer.from(await dl.arrayBuffer()), mediaType: normalizeMediaType('image/png', format), width: 0, height: 0, seed: seedArg ?? 0, sourceUrl: item.url }
-  }
-
-  // Gemini (Google): generateContent / imagen via GenAI API.
-  async function gemini(seedArg = seed, promptArg = prompt) {
-    const key = await resolveKey(cfg.geminiKeyEnv)
-    const model = cfg.geminiModel || 'gemini-2.0-flash-exp-image-generation'
-    const res = await fetchImpl(`https://generativelanguage.googleapis.com/v1beta/models/${model}:generateContent?key=${encodeURIComponent(key)}`, {
-      method: 'POST',
-      headers: { 'Content-Type': 'application/json' },
-      body: JSON.stringify({
-        contents: [{ parts: [{ text: promptArg }] }],
-        generationConfig: { responseModalities: ['IMAGE'], ...(quality !== undefined ? { imageConfig: { imageQuality: quality } } : {}), ...(style !== undefined ? { imageConfig: { imageStyle: style } } : {}) },
-      }),
-      signal,
-    })
-    const data = await res.json().catch(() => ({}))
-    if (!res.ok) {
-      const detail = data?.error?.message || JSON.stringify(data).slice(0, 600)
-      throw new Error(`Gemini failed (HTTP ${res.status}): ${detail}`)
-    }
-    const part = data?.candidates?.[0]?.content?.parts?.find((p) => p.inlineData?.data)
-    if (!part) throw new Error('Gemini returned no image')
-    return { bytes: Buffer.from(part.inlineData.data, 'base64'), mediaType: normalizeMediaType(part.inlineData.mimeType || 'image/png', format), width: 0, height: 0, seed: seedArg ?? 0, sourceUrl: '' }
-  }
-
-
-  // Replicate provider: API submit & polling
-  async function replicate(seedArg = seed, promptArg = prompt) {
-    const key = await resolveKey(cfg.replicateKeyEnv || 'REPLICATE_API_TOKEN')
-    if (!key) throw new Error('Replicate API token is not configured (Settings → Image generation → replicateKeyEnv)')
-    const model = cfg.replicateModel || 'black-forest-labs/flux-schnell'
-    const [width, height] = pxSize(aspectPixels, size)
-    const body = {
-      input: {
-        prompt: promptArg,
-        aspect_ratio: aspectPixels ? `${aspectPixels[0]}:${aspectPixels[1]}` : (ASPECT_RATIOS[size] || '1:1'),
-        seed: seedArg,
-        ...(negativePrompt ? { negative_prompt: negativePrompt } : {}),
-        ...(source && source.bytes ? {
-          image: `data:${source.mediaType || 'image/png'};base64,${Buffer.from(source.bytes).toString('base64')}`,
-          ...(mask && mask.bytes ? { mask: `data:${mask.mediaType || 'image/png'};base64,${Buffer.from(mask.bytes).toString('base64')}` } : {}),
-          ...(strength !== undefined ? { prompt_strength: strength } : {}),
-        } : {}),
-      },
-    }
-    const res = await fetchImpl(`https://api.replicate.com/v1/models/${model}/predictions`, {
-      method: 'POST',
-      headers: {
-        Authorization: `Bearer ${key}`,
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify(body),
-      signal,
-    })
-    const data = await res.json().catch(() => ({}))
-    if (!res.ok) {
-      throw new Error(`Replicate submit failed (HTTP ${res.status}): ${data.detail || JSON.stringify(data).slice(0, 300)}`)
-    }
-    let pred = data
-    if (pred.status !== 'succeeded') {
-      const pollUrl = pred.urls?.get || `https://api.replicate.com/v1/predictions/${pred.id}`
-      const deadline = Date.now() + (cfg.timeoutMs || 180000)
-      let attempt = 0
-    while (pred.status !== 'succeeded') {
-        if (signal?.aborted) throw new Error('Replicate generation cancelled')
-        if (Date.now() > deadline) throw new Error('Replicate generation timed out')
-        if (pred.status === 'failed' || pred.status === 'canceled') {
-          throw new Error(`Replicate failed: ${pred.error || 'unknown error'}`)
-        }
-        const repDelay = calculateBackoff(attempt++, cfg.pollIntervalMs || 1000, 5000)
-        await new Promise((r) => setTimeout(r, repDelay))
-        const pRes = await fetchImpl(pollUrl, { headers: { Authorization: `Bearer ${key}` }, signal })
-        pred = await pRes.json().catch(() => ({}))
-      }
-    }
-    const output = Array.isArray(pred.output) ? pred.output[0] : pred.output
-    if (!output) throw new Error('Replicate returned no image output')
-    const dl = await fetchImpl(output, { signal })
-    if (!dl.ok) throw new Error(`Replicate download failed (HTTP ${dl.status})`)
-    const bytes = Buffer.from(await dl.arrayBuffer())
-    return {
-      bytes,
-      mediaType: normalizeMediaType(dl.headers?.get?.('content-type') || 'image/png', format),
-      width,
-      height,
-      seed: seedArg ?? 0,
-      cost: estimateCost('replicate', model),
-      sourceUrl: output,
-    }
+  const subscription = createSubscriptionGenerator(deps, job)
+  return {
+    fal: createFalGenerator(deps, job),
+    custom: createCustomGenerator(deps, job),
+    codex: subscription('codex'),
+    grok: subscription('grok'),
+    local: createLocalGenerator(deps, job),
+    seedream: createSeedreamGenerator(deps, job),
+    gemini: createGeminiGenerator(deps, job),
+    replicate: createReplicateGenerator(deps, job),
   }
-
-  return { fal, custom, codex: subscription('codex'), grok: subscription('grok'), local, seedream, gemini, replicate }
 }
 
 
@@ -792,21 +300,6 @@ export function traceToSvg(imageBytes, { colorMode = 'color', paletteSize = 16,
 }
 
 
-export function estimateCost(provider, model, { count = 1 } = {}) {
-  const p = String(provider).toLowerCase()
-  if (p === 'fal') {
-    if (String(model).includes('schnell') || String(model).includes('klein')) return 0.003 * count
-    if (String(model).includes('dev')) return 0.025 * count
-    if (String(model).includes('clarity') || String(model).includes('upscale')) return 0.01 * count
-    return 0.005 * count
-  }
-  if (p === 'replicate') return 0.003 * count
-  if (p === 'seedream') return 0.004 * count
-  if (p === 'gemini') return 0.03 * count
-  if (p === 'codex' || p === 'grok' || p === 'local') return 0.0
-  return 0.01 * count
-}
-
 
 export const STYLE_PRESETS = {
   cinematic: {

package/lib/register-tools.js

@@ -7,6 +7,10 @@ import { registerProcessingAdvancedTools } from './tools/processing-advanced.js'
 import { registerEditingTools } from './tools/editing.js'
 import { registerInspectTools } from './tools/inspect.js'
 import { registerFrontendTools } from './tools/frontend.js'
+import { registerSketchTools } from './tools/sketch.js'
+import { registerSpritesheetTools } from './tools/spritesheet.js'
+import { registerPatternTools } from './tools/pattern.js'
+import { registerResponsiveTools } from './tools/responsive.js'
 
 /**
  * Registers every image-gen tool on the host tools service.
@@ -21,4 +25,8 @@ export function registerAllTools(ctx, deps) {
   registerEditingTools(ctx, deps)
   registerInspectTools(ctx, deps)
   registerFrontendTools(ctx, deps)
+  registerSketchTools(ctx, deps)
+  registerSpritesheetTools(ctx, deps)
+  registerPatternTools(ctx, deps)
+  registerResponsiveTools(ctx, deps)
 }