@kolbo/mcp 1.81.5 → 1.81.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kolbo/mcp",
-  "version": "1.81.5",
+  "version": "1.81.6",
   "description": "Kolbo AI MCP Server - Generate images, videos, music, speech, and sound effects from Claude Code",
   "main": "src/index.js",
   "bin": {

package/src/apps/widgets/generation.js

@@ -702,6 +702,11 @@ function handleBatchStatus(sc, st) {
       resolved.model_name = r.model_name;
       resolved.model_icon = r.model_icon;
     }
+    // Same for the references actually used (reference_details from the
+    // server) — every id in the batch shares one reference set.
+    if (!resolved.reference_images && r.reference_images && r.reference_images.length) {
+      resolved.reference_images = r.reference_images;
+    }
     scenes.push({
       scene_number: i + 1,
       title: (sc.prompts && sc.prompts[i]) || '',

package/src/tools/_shared.js

@@ -716,8 +716,11 @@ function mediaRefs(p) {
     ? p.reference_images.filter(Boolean)
     : (p.reference_image ? [p.reference_image] : []);
   return {
-    reference_images: images,
-    reference_image: p.reference_image || images[0],
+    // Omitted when empty: a live card MERGES status payloads over its own
+    // state, so a present-but-empty reference_images (the single-id status
+    // path has no input refs of its own) wiped the submit-time thumbnails
+    // off the finished card.
+    ...(images.length ? { reference_images: images, reference_image: p.reference_image || images[0] } : {}),
     ...(Array.isArray(p.reference_videos) && p.reference_videos.length
       ? { reference_videos: p.reference_videos.filter(Boolean) } : {}),
     ...(Array.isArray(p.reference_audio) && p.reference_audio.length

package/src/tools/generate.js

@@ -282,7 +282,10 @@ function registerGenerateTools(server, client, options = {}) {
       return uiCompleted({
         tool: 'generate_image', kind: 'image', gen, client, model, prompt,
         count: num_images, settings: imageSettings(shared),
-        reference_images,
+        // The status result reports what the server ACTUALLY conditioned on
+        // (reference_details: DNA stills, merged sources, auto-routing) — the
+        // input arg is only what the caller happened to name.
+        reference_images: result.result.reference_images || reference_images,
         urls: result.result.urls,
         credits_used: creditFields(result).credits_used,
       }, JSON.stringify({
@@ -371,7 +374,8 @@ function registerGenerateTools(server, client, options = {}) {
       return uiCompleted({
         tool: 'generate_image_edit', kind: 'image', gen, client, model, prompt,
         count: num_images, settings: imageSettings(shared),
-        reference_images: [...(source_images || []), ...(reference_images || [])],
+        reference_images: result.result.reference_images
+          || [...(source_images || []), ...(reference_images || [])],
         urls: result.result.urls,
         credits_used: creditFields(result).credits_used,
       }, JSON.stringify({
@@ -1096,6 +1100,11 @@ function registerGenerateTools(server, client, options = {}) {
           state: single.state,
           urls: done ? urls : undefined,
           thumbnail_url: res.thumbnail_url,
+          // The refs the server actually conditioned on (reference_details) —
+          // the live card merges this payload over its submit-time state, so
+          // the finished card shows every reference, including server-side
+          // additions (DNA stills, merged sources) the tool call never named.
+          reference_images: res.reference_images,
           prompt: res.prompt_used || res.prompt || undefined,
           credits_used: creditFields(single).credits_used,
           items: [{
@@ -1136,9 +1145,19 @@ function registerGenerateTools(server, client, options = {}) {
       // moodboards.js / visual_dna.js / listResult(). uiResult always embeds
       // both text and structuredContent, so hosts without a UI-app renderer
       // are unaffected — only the fallback quality changes.
+      // A live batch card (prompts[] fan-out) merges this payload over its
+      // submit-time state, so a hardcoded 'Generations' model here painted
+      // over the real model chip on every finished batch. When all results
+      // ran the same model (the batch case), name it; only a genuinely mixed
+      // multi-id check falls back to the generic label. Same for the
+      // references actually used — one shared set per batch.
+      const modelsRan = [...new Set(results.map(r => r.result && r.result.model).filter(Boolean))];
+      const actualRefs = results
+        .map(r => r.result && r.result.reference_images)
+        .find(refs => Array.isArray(refs) && refs.length);
       return uiCompleted({
         tool: 'get_generation_status', kind: 'status', client,
-        model: 'Generations',
+        model: modelsRan.length === 1 ? modelsRan[0] : 'Generations',
         gen: {
           generation_id: ids[0],
           session_id: results[0] && results[0].session_id,
@@ -1146,6 +1165,7 @@ function registerGenerateTools(server, client, options = {}) {
           type: results[0] && results[0].type,
         },
         settings: {},
+        reference_images: actualRefs,
         items: results.map(r => {
           const res = r.result || {};
           return {