@goodandready/dsh-image-gen 0.10.17 → 0.10.19

package/lib/generation-cache.js

@@ -7,6 +7,27 @@ import { existsSync, mkdirSync, readFileSync, writeFileSync, statSync, unlinkSyn
 
 const DEFAULT_CACHE_MAX_BYTES = 500 * 1024 * 1024 // 500 MB
 
+// In-Memory L1 LRU cache (up to 32 entries) for sub-millisecond retrieval
+const L1_CACHE = new Map()
+const L1_MAX_ENTRIES = 32
+
+function getFromL1(hash) {
+  if (!L1_CACHE.has(hash)) return null
+  const entry = L1_CACHE.get(hash)
+  L1_CACHE.delete(hash)
+  L1_CACHE.set(hash, entry) // refresh LRU order
+  return entry
+}
+
+function setToL1(hash, entry) {
+  if (L1_CACHE.has(hash)) L1_CACHE.delete(hash)
+  else if (L1_CACHE.size >= L1_MAX_ENTRIES) {
+    const oldestKey = L1_CACHE.keys().next().value
+    if (oldestKey) L1_CACHE.delete(oldestKey)
+  }
+  L1_CACHE.set(hash, entry)
+}
+
 function getCacheDir() {
   const base = process.env.DSH_HOME || join(homedir(), '.dsh')
   return join(base, 'cache', 'image-gen')
@@ -23,6 +44,9 @@ function getCacheDir() {
 export function getCachedGeneration(hash, { force = false } = {}) {
   if (force || !hash) return null
 
+  const l1Hit = getFromL1(hash)
+  if (l1Hit) return l1Hit
+
   const dir = getCacheDir()
   const metaPath = join(dir, `${hash}.json`)
   const dataPath = join(dir, `${hash}.bin`)
@@ -41,7 +65,7 @@ export function getCachedGeneration(hash, { force = false } = {}) {
       // Non-fatal touch failure: still return valid cache entry
     }
 
-    return {
+    const cachedEntry = {
       bytes,
       mediaType: meta.mediaType || 'image/png',
       width: meta.width,
@@ -49,6 +73,8 @@ export function getCachedGeneration(hash, { force = false } = {}) {
       seed: meta.seed,
       meta,
     }
+    setToL1(hash, cachedEntry)
+    return cachedEntry
   } catch {
     return null
   }
@@ -79,7 +105,17 @@ export function setCachedGeneration(hash, { bytes, mediaType = 'image/png', widt
     }
 
     writeFileSync(metaPath, JSON.stringify(cacheMeta, null, 2), 'utf8')
-    writeFileSync(dataPath, Buffer.isBuffer(bytes) ? bytes : Buffer.from(bytes))
+    const binBuf = Buffer.isBuffer(bytes) ? bytes : Buffer.from(bytes)
+    writeFileSync(dataPath, binBuf)
+
+    setToL1(hash, {
+      bytes: binBuf,
+      mediaType,
+      width,
+      height,
+      seed,
+      meta: cacheMeta,
+    })
 
     pruneCacheToLimit(maxBytes)
   } catch {

package/lib/index.js

@@ -27,6 +27,7 @@ import {
   resolveApiKeyCandidates,
   toLosslessJson,
   saveAttachmentSafe,
+  testProviderConnection,
 } from './providers.js'
 
 import { buildDualOutputMarkdown, resolveConversationImage, analyzeImageWithVision } from './resolve-image.js'
@@ -276,6 +277,14 @@ export const Config = z.object({
     .string()
     .description('Model used to enhance the prompt. Empty means the same model that leads the conversation.')
     .default(''),
+  autoEnhancePrompt: z
+    .boolean()
+    .description('Enrich prompts with photography, lighting, and composition quality tokens.')
+    .default(false),
+  defaultStylePreset: z
+    .string()
+    .description('Default style preset applied to generations (e.g. cinematic, anime, photorealistic).')
+    .default('none'),
   enhanceBelowChars: z
     .number()
     .description('Only enhance prompts shorter than this many characters.')
@@ -620,7 +629,36 @@ export function apply(ctx, config) {
     }), `dsh-image-gen: image route ${path}`)
   }
 
-  // История генераций: in-memory список, отфильтрованный по существованию файлов.
+  // Диагностика подключения к провайдерам
+  ctx.effect(() => ctx.webServer.register({
+    kind: 'exact',
+    path: '/dsh-image-gen/diagnostics/test',
+    handler: async (req, res) => {
+      if (req.method !== 'GET') {
+        res.writeHead(405, { 'Content-Type': 'application/json' })
+        res.end(JSON.stringify({ error: 'GET only' }))
+        return
+      }
+      try {
+        const u = new URL(req.url, 'http://localhost')
+        const provider = u.searchParams.get('provider') || config.provider || 'fal'
+        const deps = {
+          fetchImpl: fetch,
+          resolveKey: (ref) => resolveApiKey(ctx, ref),
+          cfg: (typeof getConfig === "function") ? getConfig() : config,
+          ctx,
+        }
+        const result = await testProviderConnection(deps, provider)
+        res.writeHead(200, { 'Content-Type': 'application/json' })
+        res.end(JSON.stringify({ provider, ...result }))
+      } catch (err) {
+        res.writeHead(500, { 'Content-Type': 'application/json' })
+        res.end(JSON.stringify({ ok: false, error: err.message }))
+      }
+    },
+  }), 'dsh-image-gen: diagnostics test route')
+
+    // История генераций: in-memory список, отфильтрованный по существованию файлов.
   ctx.effect(() => ctx.webServer.register({
     kind: 'exact',
     path: '/dsh-image-gen/history',

package/lib/prompt-polisher.js

@@ -0,0 +1,194 @@
+/**
+ * @file lib/prompt-polisher.js
+ * Curated artistic/photographic style presets, smart prompt enrichment, and palette harmonization.
+ */
+
+export const CURATED_STYLES = {
+  cinematic: {
+    id: 'cinematic',
+    label: 'Cinematic Film',
+    promptSuffix: 'cinematic 35mm film still, anamorphic lighting, shallow depth of field, color graded, photorealistic details',
+    negativePrompt: 'cartoon, illustration, 3d render, oversaturated, blurry, distorted, amateur',
+    guidanceScale: 6.5,
+  },
+  photorealistic: {
+    id: 'photorealistic',
+    label: 'Photorealistic Studio',
+    promptSuffix: 'photorealistic portraiture, 85mm lens, natural studio lighting, micro-textures, 8k resolution, award-winning photography',
+    negativePrompt: 'illustration, drawing, anime, 3d render, plastic skin, CGI, doll-like',
+    guidanceScale: 7.0,
+  },
+  anime: {
+    id: 'anime',
+    label: 'Anime Masterpiece',
+    promptSuffix: 'clean anime aesthetic, sharp clean lineart, vibrant cel shading, detailed background scenery, Makoto Shinkai style',
+    negativePrompt: 'photorealistic, realistic photo, noisy, deformed, 3d model, lowres',
+    guidanceScale: 7.0,
+  },
+  minimalist_vector: {
+    id: 'minimalist_vector',
+    label: 'Minimalist Vector',
+    promptSuffix: 'minimalist vector art, clean sharp outlines, flat color fills, modern graphic design, Bauhaus aesthetics, SVG vector style',
+    negativePrompt: 'photorealistic, heavy gradients, 3d render, noisy texture, clutter, photorealism',
+    guidanceScale: 7.5,
+  },
+  isometric_3d: {
+    id: 'isometric_3d',
+    label: 'Isometric 3D',
+    promptSuffix: 'isometric 3D render, ambient occlusion, miniature diorama, Octane render, smooth clay surface, soft global illumination',
+    negativePrompt: 'flat, 2d, hand-drawn sketch, chaotic background, realistic photo',
+    guidanceScale: 7.0,
+  },
+  analog_film: {
+    id: 'analog_film',
+    label: 'Analog 35mm Film',
+    promptSuffix: 'vintage 35mm photograph, Kodak Portra 400 color science, authentic film grain, subtle halation, nostalgic muted tones',
+    negativePrompt: 'digital CGI, harsh sharpening, plastic, 3d render, oversaturated digital look',
+    guidanceScale: 6.5,
+  },
+  cyberpunk: {
+    id: 'cyberpunk',
+    label: 'Cyberpunk Neon',
+    promptSuffix: 'cyberpunk aesthetic, high-tech dark city, neon volumetric lighting, rain-slicked reflective surfaces, moody atmosphere',
+    negativePrompt: 'daylight, sunshine, pastoral, cartoon, pastel colors, washed out',
+    guidanceScale: 7.5,
+  },
+  pixel_art: {
+    id: 'pixel_art',
+    label: '16-bit Retro Pixel Art',
+    promptSuffix: '16-bit retro pixel art, crisp pixel grid, authentic dithered shading, nostalgic arcade palette, master pixel artwork',
+    negativePrompt: 'smooth vector, 3d render, blurry, modern photography, anti-aliased gradients',
+    guidanceScale: 8.0,
+  },
+  oil_painting: {
+    id: 'oil_painting',
+    label: 'Classical Oil Painting',
+    promptSuffix: 'classical fine art oil painting, impasto brushstrokes, textured canvas, chiaroscuro lighting, museum masterpiece',
+    negativePrompt: 'photograph, digital vector, 3d CGI, plastic, flat anime',
+    guidanceScale: 7.0,
+  },
+  claymation: {
+    id: 'claymation',
+    label: 'Claymation Stop-Motion',
+    promptSuffix: 'handcrafted claymation stop-motion aesthetic, tactile plasticine clay texture, studio macro softbox lighting, Aardman inspired',
+    negativePrompt: 'digital CGI, flat vector, photographic humans, harsh synthetic gloss',
+    guidanceScale: 7.5,
+  },
+}
+
+/**
+ * Normalizes a style identifier or alias into a canonical style key.
+ */
+export function normalizeStyleKey(key) {
+  if (!key) return null
+  const cleaned = String(key).trim().toLowerCase().replace(/[-\s]/g, '_')
+  if (CURATED_STYLES[cleaned]) return cleaned
+  // Alias mapping
+  if (cleaned === 'isometric') return 'isometric_3d'
+  if (cleaned === 'vector' || cleaned === 'minimalist') return 'minimalist_vector'
+  if (cleaned === 'film' || cleaned === 'analog') return 'analog_film'
+  if (cleaned === 'photo' || cleaned === 'realism') return 'photorealistic'
+  return cleaned
+}
+
+/**
+ * Smart prompt enrichment: adds photographic & lighting cues if requested.
+ */
+export function enrichWithQualityCues(prompt, styleKey = null) {
+  if (!prompt || typeof prompt !== 'string') return prompt
+  let p = prompt.trim()
+  const lower = p.toLowerCase()
+  
+  // Only append if not already containing lighting/detail tokens
+  const hasLighting = lower.includes('lighting') || lower.includes('light') || lower.includes('illumination')
+  const hasDetail = lower.includes('detailed') || lower.includes('highres') || lower.includes('sharp') || lower.includes('resolution')
+
+  const additions = []
+  if (!hasLighting) additions.push('balanced natural lighting')
+  if (!hasDetail && styleKey !== 'minimalist_vector' && styleKey !== 'pixel_art') additions.push('high visual clarity')
+
+  if (additions.length > 0) {
+    p = p.endsWith('.') ? `${p} ${additions.join(', ')}.` : `${p}, ${additions.join(', ')}`
+  }
+  return p
+}
+
+/**
+ * Polish and enrich a prompt with styles, auto-enhancements, and optional palette constraints.
+ * 
+ * @param {string} basePrompt 
+ * @param {string|null} stylePreset 
+ * @param {object} [options]
+ * @param {boolean} [options.autoEnhance=false]
+ * @param {string[]} [options.paletteColors=null]
+ * @param {string} [options.existingNegative]
+ * @param {number} [options.existingGuidance]
+ * @returns {{ prompt: string, negativePrompt?: string, guidanceScale?: number, styleApplied?: string }}
+ */
+export function polishPrompt(basePrompt, stylePreset = null, {
+  autoEnhance = false,
+  paletteColors = null,
+  existingNegative = '',
+  existingGuidance = undefined,
+} = {}) {
+  let prompt = String(basePrompt || '').trim()
+  let negative = existingNegative ? String(existingNegative).trim() : ''
+  let guidance = existingGuidance
+  let styleApplied = null
+
+  const key = normalizeStyleKey(stylePreset)
+  const preset = key ? CURATED_STYLES[key] : null
+
+  if (preset) {
+    styleApplied = preset.id
+    if (preset.promptSuffix && !prompt.toLowerCase().includes(preset.promptSuffix.toLowerCase())) {
+      prompt = prompt.endsWith('.') ? `${prompt} ${preset.promptSuffix}.` : `${prompt}, ${preset.promptSuffix}`
+    }
+    if (preset.negativePrompt) {
+      negative = [negative, preset.negativePrompt].filter(Boolean).join(', ')
+    }
+    if (guidance === undefined && preset.guidanceScale !== undefined) {
+      guidance = preset.guidanceScale
+    }
+  } else if (stylePreset && typeof stylePreset === 'string' && stylePreset.trim() && stylePreset !== 'none') {
+    // Custom style suffix string
+    const customSuffix = stylePreset.trim()
+    if (!prompt.toLowerCase().includes(customSuffix.toLowerCase())) {
+      prompt = `${prompt}, in ${customSuffix} style`
+    }
+  }
+
+  // Auto-enhance prompt if toggled
+  if (autoEnhance) {
+    prompt = enrichWithQualityCues(prompt, styleApplied)
+  }
+
+  // Palette constraint injection
+  if (Array.isArray(paletteColors) && paletteColors.length > 0) {
+    const validHex = paletteColors
+      .map(c => String(c).trim().toUpperCase())
+      .filter(c => /^#([0-9A-F]{3}|[0-9A-F]{6})$/i.test(c))
+    
+    if (validHex.length > 0) {
+      const paletteDirective = `chromatic palette strictly dominated by ${validHex.join(', ')}, harmonious color grading`
+      prompt = `${prompt}, ${paletteDirective}`
+      const paletteNeg = 'clashing foreign colors, dissonant hues, out-of-palette tints'
+      negative = [negative, paletteNeg].filter(Boolean).join(', ')
+    }
+  }
+
+  return {
+    prompt,
+    negativePrompt: negative || undefined,
+    guidanceScale: guidance,
+    styleApplied: styleApplied || undefined,
+  }
+}
+
+export function listCuratedStyles() {
+  return Object.values(CURATED_STYLES).map(s => ({
+    id: s.id,
+    label: s.label,
+    guidanceScale: s.guidanceScale,
+  }))
+}

package/lib/providers.js

@@ -152,7 +152,12 @@ export function isFatalClientError(error) {
     msg.includes('bad request (http 400') ||
     msg.includes('invalid_prompt') ||
     msg.includes('prompt is required') ||
-    msg.includes('unsupported image format')
+    msg.includes('unsupported image format') ||
+    msg.includes('402 payment required') ||
+    msg.includes('insufficient_quota') ||
+    msg.includes('insufficient credits') ||
+    msg.includes('exceeded your current quota') ||
+    msg.includes('balance is insufficient')
   )
 }
 
@@ -371,16 +376,19 @@ export async function pollStatus(fetchImpl, statusUrl, key, signal, pollInterval
   for (;;) {
     if (signal?.aborted) throw new Error('FAL generation cancelled')
     if (Date.now() > deadline) throw new Error(`FAL generation timed out after ${timeoutMs} ms`)
-    const delay = calculateBackoff(attempt++, pollIntervalMs, 5000)
-    await new Promise((resolve) => {
-      const timer = setTimeout(resolve, delay)
-      if (signal) {
-        signal.addEventListener('abort', () => {
-          clearTimeout(timer)
-          resolve()
-        }, { once: true })
-      }
-    })
+    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(() => ({}))
@@ -770,13 +778,16 @@ export function makeProviders(deps, job) {
     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`)
-      const cDelay = calculateBackoff(attempt++, cfg.pollIntervalMs || 1000, 5000)
-      await new Promise((resolve) => {
-        const timer = setTimeout(resolve, cDelay)
-        if (signal) {
-          signal.addEventListener('abort', () => { clearTimeout(timer); resolve() }, { once: true })
-        }
-      })
+      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(() => ({}))
@@ -1180,3 +1191,87 @@ export function toLosslessJson(val) {
   }
   return null
 }
+
+
+/**
+ * Diagnostic ping/connection test for a given provider.
+ * Returns { ok: boolean, latencyMs: number, message: string }.
+ */
+export async function testProviderConnection(deps, providerId) {
+  const { fetchImpl = fetch, resolveKey, cfg } = deps
+  const start = Date.now()
+  const p = (providerId || cfg.defaultProvider || 'fal').toLowerCase()
+
+  try {
+    if (p === 'fal') {
+      const key = await resolveKey(cfg.apiKeyEnv || 'FAL_KEY')
+      if (!key) return { ok: false, latencyMs: 0, message: 'FAL API key not configured' }
+      const res = await fetchImpl('https://queue.fal.run/fal-ai/fast-sdxl/status', {
+        headers: { Authorization: falAuthHeader(key) },
+        signal: typeof AbortSignal !== 'undefined' && AbortSignal.timeout ? AbortSignal.timeout(6000) : undefined,
+      })
+      const latencyMs = Date.now() - start
+      if (res.status === 401 || res.status === 403) {
+        return { ok: false, latencyMs, message: `Fal auth failed (HTTP ${res.status})` }
+      }
+      return { ok: true, latencyMs, message: 'Fal.ai queue reachable' }
+    }
+
+    if (p === 'comfyui') {
+      const url = cfg.comfyuiUrl || cfg.localBaseURL || 'http://127.0.0.1:8188'
+      const res = await fetchImpl(`${url.replace(/\/+$/, '')}/system_stats`, {
+        signal: typeof AbortSignal !== 'undefined' && AbortSignal.timeout ? AbortSignal.timeout(5000) : undefined,
+      })
+      const latencyMs = Date.now() - start
+      if (res.ok) {
+        const data = await res.json().catch(() => ({}))
+        const vram = data?.devices?.[0]?.vram_free ? ` (${Math.round(data.devices[0].vram_free / 1048576)} MB free VRAM)` : ''
+        return { ok: true, latencyMs, message: `ComfyUI operational${vram}` }
+      }
+      return { ok: false, latencyMs, message: `ComfyUI returned HTTP ${res.status}` }
+    }
+
+    if (p === 'a1111') {
+      const url = cfg.a1111Url || cfg.localBaseURL || 'http://127.0.0.1:7860'
+      const res = await fetchImpl(`${url.replace(/\/+$/, '')}/sdapi/v1/samplers`, {
+        signal: typeof AbortSignal !== 'undefined' && AbortSignal.timeout ? AbortSignal.timeout(5000) : undefined,
+      })
+      const latencyMs = Date.now() - start
+      if (res.ok) {
+        return { ok: true, latencyMs, message: 'Automatic1111 operational' }
+      }
+      return { ok: false, latencyMs, message: `A1111 returned HTTP ${res.status}` }
+    }
+
+    if (p === 'custom') {
+      const url = cfg.customEndpoint || cfg.customBaseURL || 'https://api.openai.com/v1'
+      const key = await resolveKey(cfg.customApiKeyEnv || cfg.customKeyEnv || 'OPENAI_API_KEY')
+      const headers = key ? { Authorization: `Bearer ${key}` } : {}
+      const res = await fetchImpl(`${url.replace(/\/+$/, '')}/models`, {
+        headers,
+        signal: typeof AbortSignal !== 'undefined' && AbortSignal.timeout ? AbortSignal.timeout(6000) : undefined,
+      })
+      const latencyMs = Date.now() - start
+      if (res.status === 401 || res.status === 403) {
+        return { ok: false, latencyMs, message: `Custom endpoint auth error (HTTP ${res.status})` }
+      }
+      if (res.ok || res.status === 404 || res.status === 405) {
+        return { ok: true, latencyMs, message: `Custom gateway reachable (HTTP ${res.status})` }
+      }
+      return { ok: false, latencyMs, message: `Custom endpoint returned HTTP ${res.status}` }
+    }
+
+    if (p === 'subscription' || p === 'chatgpt' || p === 'grok' || p === 'codex') {
+      const hasSub = Boolean(deps.ctx?.subscriptions)
+      const latencyMs = Date.now() - start
+      if (hasSub) {
+        return { ok: true, latencyMs, message: 'Subscription bridge available' }
+      }
+      return { ok: false, latencyMs, message: 'dsh-subscriptions plugin not installed' }
+    }
+
+    return { ok: true, latencyMs: Date.now() - start, message: `Provider "${p}" configured` }
+  } catch (err) {
+    return { ok: false, latencyMs: Date.now() - start, message: err.message || 'Connection timeout' }
+  }
+}

package/lib/tools/editing.js

@@ -75,6 +75,10 @@ export function registerEditingTools(ctx, deps) {
           },
           mask: {
             type: 'string',
+          },
+          mask_image: {
+            type: 'string',
+            description: 'Optional path or attachment id of the inpaint mask image (alias for mask).',
             description: 'Optional path or attachment id of the inpaint mask image.',
           },
           strength: {
@@ -120,7 +124,8 @@ export function registerEditingTools(ctx, deps) {
             })
             assertBudgetAvailable(estimatedCostEdit, cfg.dailyBudgetUsd)
             const source = await resolveConversationImage(ctx, exec, args.image)
-          const mask = args.mask ? await resolveSource(ctx, exec, args.mask) : undefined
+          const maskRef = args.mask_image || args.mask
+          const mask = maskRef ? await resolveSource(ctx, exec, maskRef) : undefined
           const vision = await analyzeImageWithVision(ctx, exec, source)
 
           let effectivePrompt = args.prompt
@@ -130,7 +135,7 @@ export function registerEditingTools(ctx, deps) {
 
           const provider = cfg.provider || 'fal'
           const deps = {
-            fetchImpl: ctx.http ? (url, opt) => ctx.http(url, opt) : fetch,
+            fetchImpl: fetch,
             resolveKey: (ref) => resolveApiKey(ctx, ref),
             cfg,
           }
@@ -257,7 +262,7 @@ export function registerEditingTools(ctx, deps) {
 
           const provider = cfg.provider || 'fal'
           const deps = {
-            fetchImpl: ctx.http ? (url, opt) => ctx.http(url, opt) : fetch,
+            fetchImpl: fetch,
             resolveKey: (ref) => resolveApiKey(ctx, ref),
             cfg,
           }

package/lib/tools/generation.js

@@ -1,3 +1,20 @@
+import { polishPrompt } from '../prompt-polisher.js'
+
+/** Run an array of async task functions with a concurrency cap */
+async function asyncPool(tasks, concurrency = 3) {
+  const results = new Array(tasks.length)
+  let nextIdx = 0
+  async function worker() {
+    while (nextIdx < tasks.length) {
+      const idx = nextIdx++
+      results[idx] = await tasks[idx]()
+    }
+  }
+  const workers = Array.from({ length: Math.min(concurrency, tasks.length) }, () => worker())
+  await Promise.all(workers)
+  return results
+}
+
 // generation — image-gen tools (generate_image, generate_image_pack). Extracted from apply() (#216).
 
 import { defineTool } from '@deepseek-ai/dsh-tools'
@@ -95,6 +112,15 @@ export function registerGenerationTools(ctx, deps) {
           type: 'string',
           description: 'Optional custom directory to write images into (relative to session cwd or absolute). Defaults to outputDir from settings.',
         },
+        style_preset: {
+          type: 'string',
+          description: 'Optional style preset: cinematic, photorealistic, anime, minimalist_vector, isometric_3d, analog_film, cyberpunk, claymation.',
+        },
+        palette_colors: {
+          type: 'array',
+          items: { type: 'string' },
+          description: 'Optional array of hex color codes (#1e1b4b) to constrain and harmonize the color palette.',
+        },
         count: {
           type: 'integer',
           description: 'Number of variations to generate in one call, 1-4 (default 1). Cost scales with count.',
@@ -263,7 +289,7 @@ export function registerGenerationTools(ctx, deps) {
         assertBudgetAvailable(estimatedCost, cfg.dailyBudgetUsd)
         const providers = makeProviders(
           { fetchImpl: fetch, resolveKey: (ref) => resolveApiKey(ctx, ref), cfg, subscriptionImages },
-          { prompt: effectivePrompt, size, format, seed: args.seed, signal: exec.signal, negativePrompt: effectiveNegative, guidanceScale: effectiveGuidance, source, mask, strength: args.strength, quality: args.quality, style: args.style, aspectPixels, aspectRatio: args.aspect_ratio },
+          { prompt: effectivePrompt, size, format, seed: args.seed, signal: exec.signal, negativePrompt: effectivePolishedNegative, guidanceScale: effectiveGuidance, source, mask, strength: args.strength, quality: args.quality, style: args.style, aspectPixels, aspectRatio: args.aspect_ratio },
         )
         // Подписочные провайдеры (codex/grok) отдают {ok:false, reason} вместо исключения:
         // отказ должен дойти до модели текстом. Без проверки execute шёл дальше с пустыми
@@ -482,18 +508,21 @@ export function registerGenerationTools(ctx, deps) {
 
         const batchPrompts = Array.isArray(args.prompts) && args.prompts.length ? args.prompts : null
         if (batchPrompts) {
-          for (let i = 0; i < batchPrompts.length; i += 1) {
-            const item = batchPrompts[i]
+          const tasks = batchPrompts.map((item, i) => async () => {
             const text = typeof item === 'string' ? item : (item && item.text) || ''
             const cached = await checkCache(seedBase + i, text)
-            images.push(cached || await tryGenerate(generators, order, seedBase + i, text))
-          }
+            return cached || await tryGenerate(generators, order, seedBase + i, text)
+          })
+          const parallelResults = await asyncPool(tasks, 3)
+          images.push(...parallelResults)
         } else {
           const count = normalizeCount(args.count)
-          for (let i = 0; i < count; i += 1) {
+          const tasks = Array.from({ length: count }, (_, i) => async () => {
             const cached = await checkCache(seedBase + i, effectivePrompt)
-            images.push(cached || await tryGenerate(generators, order, seedBase + i))
-          }
+            return cached || await tryGenerate(generators, order, seedBase + i)
+          })
+          const parallelResults = await asyncPool(tasks, 3)
+          images.push(...parallelResults)
         }
         const first = images[0]
         const dualOutputSummary = buildDualOutputMarkdown({
@@ -542,8 +571,7 @@ export function registerGenerationTools(ctx, deps) {
           const warnings = []
           const seedBase = Math.floor(Math.random() * 100000)
 
-          for (let i = 0; i < ratios.length; i += 1) {
-            const ratio = ratios[i]
+          const packTasks = ratios.map((ratio) => async () => {
             const sizeName = ratio === '16:9' ? 'landscape_16_9' : ratio === '9:16' ? 'portrait_16_9' : ratio === '4:3' ? 'landscape_4_3' : ratio === '3:4' ? 'portrait_4_3' : 'square_hd'
             const name = `${slugify(args.output_name || 'pack')}-${ratio.replace(':', 'x')}`
             try {
@@ -553,9 +581,17 @@ export function registerGenerationTools(ctx, deps) {
                 seed: seedBase,
                 output_name: name,
               }, exec)
-              results.push({ ratio, ...genResult })
+              return { success: true, ratio, genResult }
             } catch (err) {
-              warnings.push(`Ratio ${ratio} failed: ${err?.message || String(err)}`)
+              return { success: false, ratio, error: err?.message || String(err) }
+            }
+          })
+          const settledPack = await Promise.all(packTasks.map((t) => t()))
+          for (const item of settledPack) {
+            if (item.success) {
+              results.push({ ratio: item.ratio, ...item.genResult })
+            } else {
+              warnings.push(`Ratio ${item.ratio} failed: ${item.error}`)
             }
           }