@goodandready/dsh-image-gen 0.8.6 → 0.8.7

package/lib/client.js

@@ -419,11 +419,11 @@ window.__ModuleLoader__.load({
     }
     const OUTPUT_FORMATS = ['png', 'jpeg', 'webp']
     const DELIVERY_MODES = ['link', 'image']
-    const PROVIDERS = ['fal', 'custom', 'codex', 'grok']
+    const PROVIDERS = ['fal', 'custom', 'codex', 'grok', 'local']
 
     /** Field specs in display order. */
     const FIELDS = [
-      { field: 'provider', spec: selectField('provider', PROVIDERS), kind: 'select', options: PROVIDERS, labelKey: 'f.provider', hintKey: 'f.providerHint' },
+      { field: 'provider', spec: selectField('provider', PROVIDERS), kind: 'select', options: PROVIDERS, labelKey: 'f.provider', hintKey: 'f.providerHint', providerNeeds: { fal: 'ключ FAL', custom: 'адрес + ключ', codex: 'подписка ChatGPT', grok: 'подписка Grok', local: 'адрес сервера' } },
       { field: 'model', spec: textField('model'), kind: 'text', labelKey: 'f.model', hintKey: 'f.modelHint', placeholderKey: 'p.model' },
       { field: 'apiKeyEnv', spec: textField('apiKeyEnv'), kind: 'text', labelKey: 'f.apiKeyEnv', hintKey: 'f.apiKeyEnvHint', placeholderKey: 'p.apiKeyEnv' },
       { field: 'baseURL', spec: textField('baseURL'), kind: 'text', labelKey: 'f.baseURL', hintKey: 'f.baseURLHint', placeholderKey: 'p.baseURL' },
@@ -457,7 +457,8 @@ window.__ModuleLoader__.load({
       }
       if (entry.kind === 'select') {
         const px = entry.sizePixels && entry.sizePixels[state.text] ? ` — ${entry.sizePixels[state.text]}` : ''
-        const hint = base.hint + px
+        const need = entry.providerNeeds && entry.providerNeeds[state.text] ? ` (нужно: ${entry.providerNeeds[state.text]})` : ''
+        const hint = base.hint + px + need
         return react_jsx_runtime.jsx(SelectField, { ...base, hint, options: entry.options, inheritLabel: t('f.inherit') })
       }
       return react_jsx_runtime.jsx(ValueField, { ...base, numeric: entry.kind === 'number', placeholder: entry.placeholderKey ? t(entry.placeholderKey) : '' })
@@ -742,6 +743,9 @@ window.__ModuleLoader__.load({
 
       const body = []
       if (prompt) body.push(react.createElement('div', { className: 'fal-prompt', key: 'p' }, prompt))
+      if (args.originalPrompt && args.originalPrompt !== prompt) {
+        body.push(react.createElement('div', { className: 'fal-meta', key: 'orig' }, `Исходный запрос: ${args.originalPrompt}`))
+      }
       if (failed) {
         body.push(react.createElement('div', { className: 'fal-err', key: 'e' }, parsed.text || 'неизвестная ошибка'))
       } else if (!parsed.attachment && parsed.url) {
@@ -772,6 +776,15 @@ window.__ModuleLoader__.load({
         body.push(react.createElement('div', { className: 'fal-meta', key: 'm' }, parsed.text))
       }
 
+      // Ссылка для скачивания: та же, что у картинки (наш роут или провайдерский url).
+      const downloadUrl = parsed.attachment
+        ? '/dsh-image-gen/image?id=' + encodeURIComponent(parsed.attachment.attachmentId ?? parsed.attachment.id ?? '')
+          + '&mt=' + encodeURIComponent(parsed.attachment.mediaType || 'image/png')
+          + '&b=' + encodeURIComponent(String(parsed.attachment.bytes ?? 0))
+          + '&w=' + encodeURIComponent(String(parsed.attachment.width ?? 0))
+          + '&h=' + encodeURIComponent(String(parsed.attachment.height ?? 0))
+        : parsed.url
+
       // Диагностика: метаданные генерации (prompt/seed/size/format), чтобы было
       // понятно, по каким параметрам получилась именно эта картинка.
       const metaParts = []
@@ -792,6 +805,14 @@ window.__ModuleLoader__.load({
           onClick: repeat,
         }, 'Повторить генерацию'))
       }
+      if (!running && !failed && downloadUrl) {
+        body.push(react.createElement('a', {
+          key: 'download',
+          className: 'fal-repeat',
+          href: downloadUrl,
+          download: 'generated-image.png',
+        }, 'Скачать'))
+      }
 
       return react.createElement('div', { className: 'fal-card' }, head, body)
     }

package/lib/index.js

@@ -15,7 +15,7 @@
 import z from '@deepseek-ai/schemastery'
 import { defineTool } from '@deepseek-ai/dsh-tools'
 import { credentialRef } from '@deepseek-ai/dsh-credentials'
-import { mkdir, writeFile } from 'node:fs/promises'
+import { mkdir, readFile, writeFile } from 'node:fs/promises'
 import path from 'node:path'
 import {
   IMAGE_SIZES,
@@ -40,7 +40,7 @@ const NS = 'dsh-image-gen'
 // настройки лежат под старым именем. Оно читается как запасное и переносится
 // под новое имя один раз — молча, при первом запуске после обновления.
 const LEGACY_NS = 'dsh-fal-image-gen'
-export const inject = ['tools', 'attachments', 'credentials', 'webServer', 'settings']
+export const inject = ['tools', 'attachments', 'credentials', 'webServer', 'settings', 'llm']
 
 export const Config = z.object({
   provider: z
@@ -108,13 +108,62 @@ export const Config = z.object({
     .description('provider=custom: fixed size sent to the API, e.g. 1024x1024. '
       + 'Empty means the named size is translated automatically — set this only for an API picky about sizes.')
     .default(''),
+  localKind: z
+    .string()
+    .description('provider=local: which local API to use — "comfyui" or "a1111".')
+    .default('comfyui'),
+  localBaseURL: z
+    .string()
+    .description('provider=local: server address, e.g. http://127.0.0.1:8188 (ComfyUI) or http://127.0.0.1:7860 (A1111).')
+    .default(''),
+  localModel: z
+    .string()
+    .description('provider=local: model id (A1111) or workflow/schema name (ComfyUI). Empty means the server default.')
+    .default(''),
+  localSteps: z
+    .number()
+    .description('provider=local: sampling steps.')
+    .default(20),
+  localCfg: z
+    .number()
+    .description('provider=local: CFG scale.')
+    .default(7),
   outputDir: z
     .string()
     .description('Where generated images are saved. A relative path resolves against the session working directory; an absolute path is used as given.')
     .default('generated/images'),
+  historyLimit: z
+    .number()
+    .description('How many recent generations to keep in the in-memory history list.')
+    .default(50),
+  enhancePrompt: z
+    .boolean()
+    .description('Expand a short prompt into a detailed one through the chat model before generating. Off by default.')
+    .default(false),
+  enhanceModel: z
+    .string()
+    .description('Model used to enhance the prompt. Empty means the same model that leads the conversation.')
+    .default(''),
+  enhanceBelowChars: z
+    .number()
+    .description('Only enhance prompts shorter than this many characters.')
+    .default(200),
 })
 
 /** Keep a file stem safe for the filesystem. */
+/** Прочитать исходное изображение для правки: путь к файлу или attachment id. */
+export async function resolveSource(ctx, exec, ref) {
+  if (!ref) return undefined
+  const sessionCwd = exec.agent?.session?.header?.cwd
+  if (ref.startsWith('sha256:')) {
+    const stored = await ctx.attachments.readImage({ attachmentId: ref, mediaType: 'image/png', bytes: 0, width: 0, height: 0 })
+    return { bytes: Buffer.from(stored.data), mediaType: stored.ref?.mediaType || 'image/png' }
+  }
+  const p = path.resolve(sessionCwd || process.cwd(), ref)
+  const bytes = await readFile(p)
+  return { bytes, mediaType: 'image/png' }
+}
+
 export function slugify(input) {
   const stem = String(input ?? '')
     .toLowerCase()
@@ -165,6 +214,53 @@ function migrateLegacySettings(sctx, scope) {
   }
 }
 
+/** In-memory история генераций (пополняется в execute, где доступен cwd). */
+const history = []
+
+/** Отфильтровать записи, чьи файлы ещё существуют; новые первыми. */
+export function filterHistory(entries, exists) {
+  return entries.filter((e) => exists(e.path)).slice(0, 50)
+}
+
+/** Собрать текст из итератора llm.stream (text-delta / block-end). */
+export async function collectText(iterable) {
+  let out = ''
+  let sawDelta = false
+  for await (const chunk of iterable) {
+    if (chunk && chunk.type === 'text-delta' && typeof chunk.text === 'string') {
+      out += chunk.text
+      sawDelta = true
+    } else if (
+      !sawDelta && chunk && chunk.type === 'block-end'
+      && chunk.block && chunk.block.type === 'text' && typeof chunk.block.text === 'string'
+    ) {
+      out += chunk.block.text
+    }
+  }
+  return out.trim()
+}
+
+/** Развернуть короткий промпт через чат-модель; при ошибке вернуть исходный. */
+export async function enhancePrompt(ctx, cfg, prompt, signal) {
+  if (!cfg.enhancePrompt) return { prompt, enhanced: false }
+  if (String(prompt).length >= (cfg.enhanceBelowChars || 200)) return { prompt, enhanced: false }
+  try {
+    const chunks = ctx.llm.stream({
+      ...(signal ? { signal } : {}),
+      ...(cfg.enhanceModel ? { model: cfg.enhanceModel } : {}),
+      messages: [
+        { role: 'system', content: 'You expand a short image-generation prompt into a detailed one. Reply with only the expanded prompt, no commentary.' },
+        { role: 'user', content: prompt },
+      ],
+    })
+    const text = await collectText(chunks)
+    if (!text) return { prompt, enhanced: false }
+    return { prompt: text, enhanced: true }
+  } catch (e) {
+    return { prompt, enhanced: false }
+  }
+}
+
 export function apply(ctx, config) {
 
   // The Web card edits this namespace; without registering it the card binds to
@@ -242,6 +338,22 @@ export function apply(ctx, config) {
     }), `dsh-image-gen: image route ${path}`)
   }
 
+  // История генераций: in-memory список, отфильтрованный по существованию файлов.
+  ctx.effect(() => ctx.webServer.register({
+    kind: 'exact',
+    path: '/dsh-image-gen/history',
+    handler: async (req, res) => {
+      if (req.method !== 'GET') {
+        res.writeHead(405, { 'Content-Type': 'application/json' })
+        res.end(JSON.stringify({ error: 'GET only' }))
+        return
+      }
+      const exists = (p) => { try { return require('node:fs').existsSync(p) } catch { return false } }
+      res.writeHead(200, { 'Content-Type': 'application/json' })
+      res.end(JSON.stringify(filterHistory(history, exists)))
+    },
+  }), 'dsh-image-gen: history route')
+
   ctx.tools.register(
     defineTool({
       name: 'generate_image',
@@ -286,6 +398,18 @@ export function apply(ctx, config) {
           type: 'number',
           description: 'Optional prompt adherence (e.g. 1-20); sent to providers that support it (ignored otherwise).',
         },
+        source_image: {
+          type: 'string',
+          description: 'Optional path to an image file or an attachment id (sha256:...) to edit instead of drawing from scratch. Only providers that support image editing accept it; others refuse with a clear reason.',
+        },
+        mask: {
+          type: 'string',
+          description: 'Optional path or attachment id of a mask image (same size as source) to restrict the edit to a region. Only used when source_image is set and the provider supports masks.',
+        },
+        strength: {
+          type: 'number',
+          description: 'Optional edit strength (0-1) for providers that accept it; higher keeps more of the original.',
+        },
       },
       output: {
         schema: {
@@ -361,9 +485,13 @@ export function apply(ctx, config) {
         let subscriptionImages
         try { subscriptionImages = ctx.get && ctx.get('subscriptionImages') } catch (noService) { subscriptionImages = undefined }
 
+        const source = args.source_image ? await resolveSource(ctx, exec, args.source_image) : undefined
+        const mask = args.mask ? await resolveSource(ctx, exec, args.mask) : undefined
+        const enhanced = await enhancePrompt(ctx, cfg, args.prompt, exec.signal)
+        const effectivePrompt = enhanced.prompt
         const providers = makeProviders(
           { fetchImpl: fetch, resolveKey: (ref) => resolveApiKey(ctx, ref), cfg, subscriptionImages },
-          { prompt: args.prompt, size, format, seed: args.seed, signal: exec.signal, negativePrompt: args.negative_prompt, guidanceScale: args.guidance_scale },
+          { prompt: effectivePrompt, size, format, seed: args.seed, signal: exec.signal, negativePrompt: args.negative_prompt, guidanceScale: args.guidance_scale, source, mask, strength: args.strength },
         )
         // Подписочные провайдеры (codex/grok) отдают {ok:false, reason} вместо исключения:
         // отказ должен дойти до модели текстом. Без проверки execute шёл дальше с пустыми
@@ -375,7 +503,7 @@ export function apply(ctx, config) {
           const bytes = gen.bytes
           const mediaType = gen.mediaType
           const extension = mediaType === 'image/jpeg' ? 'jpg' : mediaType === 'image/webp' ? 'webp' : 'png'
-          const stem = `${slugify(args.output_name || args.prompt)}-${Date.now().toString(36)}-${jobSeed}`
+          const stem = `${slugify(args.output_name || effectivePrompt)}-${Date.now().toString(36)}-${jobSeed}`
           const name = `${stem}.${extension}`
 
           const attachment = await ctx.attachments.saveImage({
@@ -401,7 +529,7 @@ export function apply(ctx, config) {
           await writeFile(
             path.join(outDir, `${stem}.json`),
             JSON.stringify(buildSidecar({
-              prompt: args.prompt,
+              prompt: effectivePrompt,
               size,
               format,
               seed: gen.seed,
@@ -412,16 +540,31 @@ export function apply(ctx, config) {
               mediaType,
               attachmentId: attachment.attachmentId,
               url: deliverAs === 'image' && gen.sourceUrl ? gen.sourceUrl : localUrl,
+              cost: gen.cost,
             }), null, 2),
           )
 
+          history.unshift({
+            path: filePath,
+            prompt: effectivePrompt,
+            provider,
+            size,
+            format,
+            seed: gen.seed,
+            createdAt: new Date().toISOString(),
+            attachmentId: attachment.attachmentId,
+          })
+          if (history.length > (cfg.historyLimit || 50)) history.length = cfg.historyLimit || 50
+
           return {
             path: filePath,
             url: deliverAs === 'image' && gen.sourceUrl ? gen.sourceUrl : localUrl,
             width: gen.width || attachment.width,
             height: gen.height || attachment.height,
             seed: gen.seed,
-            prompt: args.prompt,
+            prompt: effectivePrompt,
+            originalPrompt: enhanced.enhanced ? args.prompt : undefined,
+            cost: gen.cost,
             format: mediaType.replace('image/', ''),
             attachment: {
               attachmentId: attachment.attachmentId,

package/lib/providers.js

@@ -7,7 +7,7 @@
 // Сеть приходит параметром (fetchImpl), ключ — через resolveKey, поэтому оба
 // провайдера проверяются юнит-тестами без единого реального запроса.
 
-export const PROVIDER_KEYS = ['fal', 'custom', 'codex', 'grok']
+export const PROVIDER_KEYS = ['fal', 'custom', 'codex', 'grok', 'local']
 
 /** Привести count из аргумента инструмента к диапазону 1..4. */
 /** Порядок провайдеров для fallback: основной первым, остальные по PROVIDER_KEYS. */
@@ -99,9 +99,9 @@ export function normalizeMediaType(contentType, fallbackFormat) {
 /** Метаданные одной генерации для sidecar-файла рядом с картинкой. */
 export function buildSidecar({
   prompt, size, format, seed, provider, deliverAs,
-  width, height, mediaType, attachmentId, url, createdAt = new Date().toISOString(),
+  width, height, mediaType, attachmentId, url, cost, createdAt = new Date().toISOString(),
 }) {
-  return { prompt, size, format, seed, provider, deliverAs, width, height, mediaType, attachmentId, url, createdAt }
+  return { prompt, size, format, seed, provider, deliverAs, width, height, mediaType, attachmentId, url, cost, createdAt }
 }
 
 /** Submit a generation job to the FAL queue. */
@@ -157,9 +157,29 @@ export async function pollStatus(fetchImpl, statusUrl, key, signal, pollInterval
  *   {bytes, mediaType, width, height, seed, sourceUrl}. Нулевые width/height
  *   означают «спроси у службы вложений»: она всё равно измеряет байты сама.
  */
+
+/** Именованный размер -> [width, height] для локальных API. */
+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]
+}
+
+/** Собрать multipart-тело для /images/edits (OpenAI-совместимый 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
+}
+
 export function makeProviders(deps, job) {
   const { fetchImpl, resolveKey, cfg } = deps
-  const { prompt, size, format, seed, signal, negativePrompt, guidanceScale } = job
+  const { prompt, size, format, seed, signal, negativePrompt, guidanceScale, source, mask, strength } = job
 
   async function fal(seedArg = seed) {
     const key = await resolveKey(cfg.apiKeyEnv)
@@ -193,6 +213,7 @@ export function makeProviders(deps, job) {
       height: image.height ?? 0,
       seed: result.seed ?? seedArg ?? 0,
       sourceUrl: image.url,
+      cost: result.cost,
     }
   }
 
@@ -210,17 +231,20 @@ export function makeProviders(deps, job) {
 
     // response_format не отправляем: новые модели OpenAI его отвергают, а ответ
     // всё равно приходит либо base64, либо ссылкой — принимаем оба.
-    const res = await fetchImpl(`${base}/images/generations`, {
+    const endpoint = source ? `${base}/images/edits` : `${base}/images/generations`
+    const res = await fetchImpl(endpoint, {
       method: 'POST',
       headers,
-      body: JSON.stringify({
-        model: cfg.customModel,
-        prompt,
-        n: 1,
-        size: cfg.customSize || SIZE_PIXELS[size] || size,
-        ...(negativePrompt !== undefined ? { negative_prompt: negativePrompt } : {}),
-        ...(guidanceScale !== undefined ? { guidance_scale: guidanceScale } : {}),
-      }),
+      body: source
+        ? buildEditForm({ source, mask, prompt, size: cfg.customSize || SIZE_PIXELS[size] || size, strength })
+        : JSON.stringify({
+            model: cfg.customModel,
+            prompt,
+            n: 1,
+            size: cfg.customSize || SIZE_PIXELS[size] || size,
+            ...(negativePrompt !== undefined ? { negative_prompt: negativePrompt } : {}),
+            ...(guidanceScale !== undefined ? { guidance_scale: guidanceScale } : {}),
+          }),
       signal,
     })
     const data = await res.json().catch(() => ({}))
@@ -268,6 +292,9 @@ export function makeProviders(deps, job) {
   // отдаёт готовую картинку, а не ключ доступа.
   function subscription(provider) {
     return async function generate(seedArg = seed) {
+      if (source) {
+        return { ok: false, provider, reason: `${provider}: не умеет править изображения — используйте fal, custom или local` }
+      }
       const images = deps.subscriptionImages
       if (!images || typeof images.generate !== 'function') {
         return {
@@ -305,5 +332,102 @@ export function makeProviders(deps, job) {
     }
   }
 
-  return { fal, custom, codex: subscription('codex'), grok: subscription('grok') }
+  // Локальная генерация: ComfyUI (очередь + опрос) или Automatic1111 (txt2img).
+  async function local(seedArg = seed) {
+    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] = sizeToPixels(size)
+    const kind = cfg.localKind === 'a1111' ? 'a1111' : 'comfyui'
+
+    if (kind === 'a1111') {
+      const body = {
+        prompt,
+        negative_prompt: negativePrompt,
+        width,
+        height,
+        steps: cfg.localSteps ?? 20,
+        cfg_scale: cfg.localCfg ?? 7,
+        seed: seedArg ?? -1,
+      }
+      if (cfg.localModel) body.override_settings = { sd_model_checkpoint: cfg.localModel }
+      const res = await fetchImpl(`${base}/sdapi/v1/txt2img`, {
+        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 через /prompt, опрос /history/{prompt_id} до готовности.
+    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: prompt, 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
+    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`)
+      await new Promise((resolve) => {
+        const timer = setTimeout(resolve, cfg.pollIntervalMs || 2000)
+        signal.addEventListener('abort', () => { clearTimeout(timer); resolve() }, { once: true })
+      })
+      const hist = await fetchImpl(`${base}/history/${pid}`, { signal })
+      if (!hist.ok) continue
+      const histData = await hist.json().catch(() => ({}))
+      const entry = histData[pid]
+      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: '',
+          }
+        }
+      }
+    }
+  }
+
+  return { fal, custom, codex: subscription('codex'), grok: subscription('grok'), local }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goodandready/dsh-image-gen",
-  "version": "0.8.6",
+  "version": "0.8.7",
   "description": "Image generation for DeepSeek Harness: a generate_image tool with pluggable providers — the FAL queue, any OpenAI-compatible images API, or a ChatGPT/Grok subscription with no API key at all. The picture is shown inline in the conversation; the model receives either a link (works with any chat model) or the image itself (needs dsh-vision-bridge or a vision-capable model).",
   "keywords": [
     "deepseek-harness",