@kolbo/mcp 1.87.5 → 1.87.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kolbo/mcp",
-  "version": "1.87.5",
+  "version": "1.87.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/index.js

@@ -71,7 +71,13 @@ const WIDGET_CSP = {
     // Public hosts owned by Kolbo.
     'https://api.kolbo.ai',
     'https://app.kolbo.ai',
-    'https://cdn.kolbo.ai',
+    'https://cdn.kolbo.ai',
+    // Preset thumbnails and library media live on the media-* hosts (prod
+    // presets still reference media-dev); without these every preset tile
+    // rendered as a black box. All three are on kolbo-api's download allowlist.
+    'https://media.kolbo.ai',
+    'https://media-staging.kolbo.ai',
+    'https://media-dev.kolbo.ai',
     ...KOLBO_MEDIA_DOMAINS,
     'https://kolbo-general-media.fra1.digitaloceanspaces.com',
     'https://kolbo-general-media.fra1.cdn.digitaloceanspaces.com',

package/src/apps/widgets/generation.js

@@ -195,6 +195,7 @@ function isListPayload(sc, toolName) {
 function boot(sc) {
   if (!sc) return;
   if (isListPayload(sc, sc.tool)) return renderList(sc);
+  sc = liveFromStatus(sc);
   state = sc;
   el('tool-title').textContent = TOOL_TITLES[sc.tool] || 'Generation';
   setPrompt(sc.prompt ? promptHTML(sc.prompt) : '', sc.prompt);
@@ -486,6 +487,51 @@ function capAt(sc, i) {
   return sc._caps[i] || sc.prompts[i];
 }
 
+// A completed TOOL CALL is not a completed GENERATION. When a generate_* call
+// outlives its poll window on a text host (Claude desktop over stdio), the
+// server answers with a kind:'status' grid whose items are still processing,
+// and that used to render as a dead card of "processing" badges that never
+// polled again: the user watched skeletons that would never fill while the
+// images landed in the library. Turn any such payload back into a LIVE card,
+// the same generating/polling path a fresh submit takes, cells filled per id.
+function isTerminal(s) { return s === 'completed' || s === 'failed' || s === 'cancelled'; }
+function liveFromStatus(sc) {
+  if (!sc || sc.phase !== 'completed' || !Array.isArray(sc.items) || !sc.items.length) return sc;
+  var ids = sc.items.map(function (it) { return it && it.id; });
+  if (ids.some(function (id) { return !id; })) return sc;
+  if (!sc.items.some(function (it) { return !isTerminal(it.state); })) return sc;
+  var kind = sc.kind === 'status' || !sc.kind ? kindFromTool(sc.tool, {}) : sc.kind;
+  var live = Object.assign({}, sc, {
+    phase: 'generating', kind: kind, items: undefined,
+    count: ids.length, poll_tool: 'get_generation_status', generation_id: ids[0]
+  });
+  if (ids.length > 1) {
+    live.generation_ids = ids;
+    live.prompts = sc.items.map(function (it) { return it.title || ''; });
+    live.status_args = { generation_ids: ids, wait: true };
+  } else {
+    live.status_args = { generation_id: ids[0], wait: true };
+  }
+  return live;
+}
+// Multi-id status responses carry their per-generation results as items[] in
+// structuredContent (the shape the status grid renders) and as generations[]
+// only in the text, and structured() reads structuredContent first, so a live
+// batch card never saw generations[] on a host that ships structuredContent.
+function itemsToGens(items) {
+  if (!Array.isArray(items)) return null;
+  return items.map(function (it) {
+    return {
+      generation_id: it.id, state: it.state, credits_used: it.credits_used,
+      result: {
+        urls: Array.isArray(it.urls) ? it.urls : (it.url ? [it.url] : []),
+        model: it.model, model_name: it.model_name, model_icon: it.model_icon,
+        reference_images: it.reference_images
+      }
+    };
+  });
+}
+
 function renderGenerating(sc) {
   setPhaseChip('Generating', true);
   var n = Math.min(sc.count || 1, isBatch(sc) ? 8 : 4);
@@ -670,9 +716,14 @@ function poll(sc) {
       if (++pollErrors >= MAX_POLL_ERRORS) return renderTrackingIssue(st.error || 'Tracking paused. The generation may still be running.');
       return schedulePoll(sc);
     }
-    // Batch (prompts[] fan-out): multi-id status shape { all_done, generations[] }.
-    if (isBatch(sc) && Array.isArray(st.generations)) {
-      return handleBatchStatus(sc, st);
+    // Batch (prompts[] fan-out): multi-id status shape { all_done, generations[] },
+    // or the same data as items[] when it came via structuredContent.
+    var gens = Array.isArray(st.generations) ? st.generations : itemsToGens(st.items);
+    if (isBatch(sc) && gens) {
+      return handleBatchStatus(sc, {
+        all_done: st.all_done != null ? st.all_done : gens.every(function (g) { return isTerminal(g.state); }),
+        generations: gens
+      });
     }
     if (stateName === 'completed') {
       pollErrors = 0;
@@ -724,8 +775,13 @@ function poll(sc) {
 function handleBatchStatus(sc, st) {
   pollErrors = 0;
   var gens = st.generations || [];
+  // Cells are laid out in submit order; match results by id, not position.
+  var cellIndex = function (g, i) {
+    var idx = sc.generation_ids ? sc.generation_ids.indexOf(g.generation_id) : -1;
+    return idx < 0 ? i : idx;
+  };
   gens.forEach(function (g, i) {
-    if (g.state === 'completed') fillBatchCell(sc, i, g);
+    if (g.state === 'completed') fillBatchCell(sc, cellIndex(g, i), g);
   });
   if (!st.all_done) return schedulePoll(sc);
 
@@ -753,8 +809,8 @@ function handleBatchStatus(sc, st) {
       resolved.reference_images = r.reference_images;
     }
     scenes.push({
-      scene_number: i + 1,
-      title: capAt(sc, i),
+      scene_number: cellIndex(g, i) + 1,
+      title: capAt(sc, cellIndex(g, i)),
       image_urls: sc.kind === 'video' ? [] : urls,
       video_urls: sc.kind === 'video' ? urls : []
     });
@@ -1305,6 +1361,41 @@ function kindFromTool(tool, sc) {
 
 // Recover a successful legacy/text result when a host mounted the iframe from
 // declaration metadata but the server did not recognize its Apps capability.
+// Every generate_* tool on a text host answers a poll-window timeout with the
+// same plain shape from pollOrTimedOut(): { state:'processing', generation_id,
+// _timed_out }. No phase, no widget, no urls, so nothing above recognised it
+// and the card stayed on its pre-render skeleton forever while the generation
+// finished in the library. Rebuild the live card from the tool INPUT (same
+// source completedFromPlain uses) and let it poll to completion.
+function liveFromTimedOut(sc) {
+  if (!sc || !sc.generation_id || isTerminal(sc.state) || Array.isArray(sc.urls)) return null;
+  var ids = Array.isArray(sc.generation_ids) && sc.generation_ids.length > 1 ? sc.generation_ids : null;
+  var live = {
+    widget: 'generation',
+    phase: 'generating',
+    tool: originTool,
+    kind: kindFromTool(originTool, sc),
+    prompt: originArgs.prompt || originArgs.text || '',
+    model: sc.model || originArgs.model,
+    settings: {
+      duration: originArgs.duration,
+      resolution: originArgs.resolution,
+      aspect_ratio: originArgs.aspect_ratio,
+      quality: originArgs.quality,
+      visual_dna_ids: originArgs.visual_dna_ids,
+      moodboard_id: originArgs.moodboard_id
+    },
+    count: ids ? ids.length : (originArgs.num_images || 1),
+    generation_id: sc.generation_id,
+    poll_tool: 'get_generation_status',
+    status_args: ids ? { generation_ids: ids, wait: true } : { generation_id: sc.generation_id, wait: true },
+    session_id: sc.session_id || originArgs.session_id,
+    project_id: sc.project_id || originArgs.project_id
+  };
+  if (ids) { live.generation_ids = ids; live.prompts = originArgs.prompts || []; }
+  return live;
+}
+
 function completedFromPlain(sc) {
   if (!sc || (!Array.isArray(sc.urls) && !Array.isArray(sc.scenes))) return null;
   return Object.assign({}, sc, {
@@ -1333,7 +1424,7 @@ window.kolbo.onToolResult(function (result) {
   var list = listPayload(sc);
   if (list) return renderList(list);
   if (sc && (sc.phase || sc.widget)) return boot(sc);
-  var recovered = completedFromPlain(sc);
+  var recovered = completedFromPlain(sc) || liveFromTimedOut(sc);
   if (recovered) return boot(recovered);
   // Tool errored (or returned plain text): show it instead of a dead blank card.
   var txt = '';

package/src/tools/generate.js

@@ -88,7 +88,7 @@ async function submitBatch(rawItems, submitOne) {
 // multi-id branch, same fix: always ship structuredContent via the shared
 // kind:'status' grid, which already renders any mix of completed/processing
 // items correctly.
-async function pollBatch(client, batch, { interval, timeout }, toolName) {
+async function pollBatch(client, batch, { interval, timeout }, toolName, submittedModel) {
   const polls = await Promise.all(batch.ids.map((id) => pollOrTimedOut(client, id, { interval, timeout })));
   const generations = polls.map((p, i) => p.timedOut
     ? { prompt: batch.ok[i].prompt, generation_id: batch.ids[i], status: 'processing', note: 'Still running — call get_generation_status with wait=true to collect it.' }
@@ -103,7 +103,7 @@ async function pollBatch(client, batch, { interval, timeout }, toolName) {
   const modelsRan = [...new Set(polls.filter((p) => !p.timedOut).map((p) => p.result.result && p.result.result.model).filter(Boolean))];
   return uiCompleted({
     tool: toolName, kind: 'status', client,
-    model: modelsRan.length === 1 ? modelsRan[0] : 'Generations',
+    model: modelsRan.length === 1 ? modelsRan[0] : (submittedModel || 'Generations'),
     gen: { generation_id: batch.ids[0], session_id: batch.ok[0].gen.session_id },
     settings: {},
     items: generations.map(g => ({
@@ -111,6 +111,8 @@ async function pollBatch(client, batch, { interval, timeout }, toolName) {
       state: g.status,
       title: g.prompt,
       url: Array.isArray(g.urls) ? g.urls[0] : undefined,
+      urls: Array.isArray(g.urls) ? g.urls : undefined,
+      credits_used: g.credits_used,
     })),
   }, text);
 }
@@ -292,7 +294,7 @@ function registerGenerateTools(server, client, options = {}) {
           status_args: { generation_ids: batch.ids, wait: true },
           reference_images
         });
-        return pollBatch(client, batch, { interval: (batch.ok[0].gen.poll_interval_hint || 3) * 1000, timeout: 150000 }, 'generate_image');
+        return pollBatch(client, batch, { interval: (batch.ok[0].gen.poll_interval_hint || 3) * 1000, timeout: 150000 }, 'generate_image', model);
       }
 
       const gen = await client.post('/v1/generate/image', { ...shared, prompt, num_images });
@@ -381,7 +383,7 @@ function registerGenerateTools(server, client, options = {}) {
           status_args: { generation_ids: batch.ids, wait: true },
           reference_images: [...(source_images || []), ...(reference_images || [])]
         });
-        return pollBatch(client, batch, { interval: (batch.ok[0].gen.poll_interval_hint || 3) * 1000, timeout: 150000 }, 'generate_image_edit');
+        return pollBatch(client, batch, { interval: (batch.ok[0].gen.poll_interval_hint || 3) * 1000, timeout: 150000 }, 'generate_image_edit', model);
       }
 
       const gen = await client.post('/v1/generate/image-edit', { ...shared, prompt, num_images });
@@ -634,7 +636,7 @@ function registerGenerateTools(server, client, options = {}) {
           status_args: { generation_ids: batch.ids, wait: true },
           reference_images
         });
-        return pollBatch(client, batch, { interval: (batch.ok[0].gen.poll_interval_hint || 8) * 1000, timeout: 150000 }, 'generate_video');
+        return pollBatch(client, batch, { interval: (batch.ok[0].gen.poll_interval_hint || 8) * 1000, timeout: 150000 }, 'generate_video', model);
       }
 
       const gen = await client.post('/v1/generate/video', { ...shared, prompt });
@@ -724,7 +726,7 @@ function registerGenerateTools(server, client, options = {}) {
           status_args: { generation_ids: batch.ids, wait: true },
           reference_images: items.map((item) => item.image_url)
         });
-        return pollBatch(client, batch, { interval: (batch.ok[0].gen.poll_interval_hint || 8) * 1000, timeout: 900000 }, 'generate_video_from_image');
+        return pollBatch(client, batch, { interval: (batch.ok[0].gen.poll_interval_hint || 8) * 1000, timeout: 900000 }, 'generate_video_from_image', model);
       }
 
       const gen = await client.post('/v1/generate/video/from-image', { ...shared, image_url, prompt });
@@ -1249,6 +1251,12 @@ function registerGenerateTools(server, client, options = {}) {
             state: r.state,
             title: res.prompt_used || res.prompt || undefined,
             url: Array.isArray(res.urls) ? res.urls[0] : undefined,
+            // A live batch card reads these from structuredContent (its only
+            // view of this response) to fill each cell and repaint its chips.
+            urls: Array.isArray(res.urls) ? res.urls : undefined,
+            credits_used: creditFields(r).credits_used,
+            model: res.model, model_name: res.model_name, model_icon: res.model_icon,
+            reference_images: res.reference_images,
           };
         }),
       }, text, extraContent);

package/src/tools/presets.js

@@ -46,10 +46,14 @@ function registerPresetTools(server, client, options = {}) {
 
       if (lookup) return { content: [{ type: 'text', text }] };
 
-      return uiResult(UI.list, text, {
-        widget: 'list',
+      // TOOL_WIDGETS declares mediaGrid for this tool, so hosts that mount from
+      // the declaration (Claude desktop) rendered a media grid fed a list-shaped
+      // 8-item page with no page_tool: a "Load more" that could never load. Ship
+      // the whole catalog (a few hundred small rows) to the grid it actually mounts.
+      return uiResult(UI.mediaGrid, text, {
+        widget: 'mediaGrid',
         title: 'Presets' + (type ? ' — ' + type : ''),
-        items: presets.slice(0, 8).map(p => ({
+        items: presets.slice(0, 300).map(p => ({
           id: p.id,
           title: p.name,
           subtitle: p.category,
@@ -60,7 +64,7 @@ function registerPresetTools(server, client, options = {}) {
           use_hint: 'Use preset "{TITLE}" (preset_id: {ID}) for my next generation — ask me for the prompt.'
         })),
         total: result.count || presets.length,
-        has_more: presets.length > 8
+        has_more: presets.length > 300
       });
     }
   );