@exreve/exk 1.0.62 → 1.0.64

package/dist/cli/agentSession.js

@@ -273,16 +273,17 @@ export class AgentSessionManager {
         // Priority: explicit parameter > env var
         const requested = agentBackend || process.env.TTC_AGENT_BACKEND;
         if (requested === 'pi') {
+            let piBackend;
             try {
-                const { piBackend } = require('./piBackend.js');
-                if (piBackend.isAvailable()) {
-                    return piBackend;
-                }
-                console.warn('[AgentSessionManager] Pi backend requested but SDK not installed, falling back to Claude');
+                piBackend = require('./piBackend.js').piBackend;
+            }
+            catch (err) {
+                throw new Error(`Pi backend requested but module failed to load: ${err.message}. Install with: cd $(npm root -g)/@exreve/exk && npm install @mariozechner/pi-coding-agent`);
             }
-            catch {
-                console.warn('[AgentSessionManager] Pi backend not available, falling back to Claude');
+            if (!piBackend.isAvailable()) {
+                throw new Error('Pi backend requested but SDK (@mariozechner/pi-coding-agent) is not installed. Install with: cd $(npm root -g)/@exreve/exk && npm install @mariozechner/pi-coding-agent');
             }
+            return piBackend;
         }
         return claudeBackend;
     }

package/dist/cli/moduleMcpServer.js

@@ -7,7 +7,7 @@
  */
 import { createSdkMcpServer, tool } from '@anthropic-ai/claude-agent-sdk';
 import { z } from 'zod';
-import { executeAnalyzeImage, executeSendFile, executeBrowserQuery } from './sharedTools.js';
+import { executeAnalyzeImage, executeSendFile, executeBrowserQuery, executeGenImage } from './sharedTools.js';
 /**
  * Build the shared config from MCP server config.
  */
@@ -56,6 +56,15 @@ export function createModuleMcpServer(config) {
             country: args.country,
             mobile: args.mobile,
         }, sharedConfig)),
+        tool('gen_image', 'Generate an image from a text prompt using Flux.2-flex. The generated PNG is saved to a temp directory and the file path is returned. ' +
+            'Use this to create illustrations, icons, diagrams, or any visual content for the user. ' +
+            'The image is saved locally — use send_file to display it in chat after generation.', {
+            prompt: z.string().describe('Detailed text description of the image to generate. Be specific about style, colors, composition, etc.'),
+            filename: z.string().optional().describe('Output filename (without extension, or with .png). Defaults to gen_<timestamp>.png'),
+        }, async (args) => executeGenImage({
+            prompt: args.prompt,
+            filename: args.filename,
+        }, sharedConfig)),
     ];
     return createSdkMcpServer({
         name: 'claude-voice-modules',

package/dist/cli/piBackend.js

@@ -15,10 +15,8 @@
  */
 import { executeAnalyzeImage, executeSendFile, executeBrowserQuery } from './sharedTools.js';
 // Conditional import — will be undefined if package not installed
-// @ts-expect-error — Pi SDK is an optional dependency, may not be installed
 let piSdk;
 try {
-    // @ts-expect-error — optional dependency
     piSdk = await import('@mariozechner/pi-coding-agent');
 }
 catch {
@@ -62,20 +60,12 @@ export class PiBackend {
         // Set model if specified
         if (model) {
             try {
-                // Try to find a built-in model first
-                const builtInModel = piSdk.getModel?.('anthropic', model);
-                if (builtInModel) {
-                    sessionOpts.model = builtInModel;
-                }
-                else {
-                    // Try as a custom model — Pi supports custom providers via models.json
-                    // For now, we try the model string directly
-                    const found = modelRegistry.find(provider || 'anthropic', model);
-                    if (found) {
-                        sessionOpts.model = found;
-                    }
-                    // If not found, Pi will use its default model
+                // Try as a custom model via ModelRegistry
+                const found = modelRegistry.find(provider || 'anthropic', model);
+                if (found) {
+                    sessionOpts.model = found;
                 }
+                // If not found, Pi will use its default model
             }
             catch {
                 // Model not found — Pi will use default
@@ -330,7 +320,8 @@ export class PiBackend {
                 required: ['image_path', 'question'],
             },
             execute: async (_toolCallId, params) => {
-                return executeAnalyzeImage({ image_path: params.image_path, question: params.question }, sharedConfig);
+                const result = await executeAnalyzeImage({ image_path: params.image_path, question: params.question }, sharedConfig);
+                return { ...result, details: {} };
             },
         });
     }
@@ -352,7 +343,8 @@ export class PiBackend {
                 },
             },
             execute: async (_toolCallId, params) => {
-                return executeSendFile(params, sharedConfig);
+                const result = await executeSendFile(params, sharedConfig);
+                return { ...result, details: {} };
             },
         });
     }
@@ -373,7 +365,8 @@ export class PiBackend {
                 required: ['query'],
             },
             execute: async (_toolCallId, params) => {
-                return executeBrowserQuery({ query: params.query, maxSteps: params.maxSteps }, sharedConfig);
+                const result = await executeBrowserQuery({ query: params.query, maxSteps: params.maxSteps }, sharedConfig);
+                return { ...result, details: {} };
             },
         });
     }

package/dist/cli/sharedTools.js

@@ -257,3 +257,72 @@ export async function executeBrowserQuery(args, config) {
         };
     }
 }
+// ============ Image Generation ============
+const FLUX_MODEL = 'black-forest-labs/flux.2-flex';
+/**
+ * gen_image — generate an image using Flux.2-flex via OpenRouter.
+ * Saves the result as a PNG file in a temp directory and returns the file path.
+ */
+export async function executeGenImage(args, _config) {
+    const apiKey = getOpenrouterApiKey();
+    if (!apiKey) {
+        return { content: [{ type: 'text', text: 'Error: OPENROUTER_API_KEY not configured. Set it in ai-config.json or OPENROUTER_API_KEY env var.' }], isError: true };
+    }
+    try {
+        const res = await fetch('https://openrouter.ai/api/v1/chat/completions', {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${apiKey}`,
+                'Content-Type': 'application/json',
+                'HTTP-Referer': 'https://talk-to-code.com',
+                'X-Title': 'TalkToCode',
+            },
+            body: JSON.stringify({
+                model: FLUX_MODEL,
+                messages: [{ role: 'user', content: args.prompt }],
+                modalities: ['image'],
+            }),
+            signal: AbortSignal.timeout(120_000), // 2 min timeout for image gen
+        });
+        if (!res.ok) {
+            const errText = await res.text();
+            return { content: [{ type: 'text', text: `Error from Flux API (${res.status}): ${errText.slice(0, 500)}` }], isError: true };
+        }
+        const data = await res.json();
+        // Extract image from response
+        const images = data.choices?.[0]?.message?.images;
+        if (!images || images.length === 0) {
+            return { content: [{ type: 'text', text: 'Error: No image in response from Flux API.' }], isError: true };
+        }
+        const dataUrl = images[0]?.image_url?.url || images[0];
+        if (!dataUrl || !dataUrl.startsWith('data:')) {
+            return { content: [{ type: 'text', text: `Error: Unexpected image format from Flux API.` }], isError: true };
+        }
+        // Decode base64 and save to temp dir
+        const base64 = dataUrl.split(',')[1];
+        if (!base64) {
+            return { content: [{ type: 'text', text: 'Error: Empty image data from Flux API.' }], isError: true };
+        }
+        const imgBuf = Buffer.from(base64, 'base64');
+        const tmpDir = path.join(os.tmpdir(), 'talk-to-code', 'gen-images');
+        fs.mkdirSync(tmpDir, { recursive: true });
+        const safeName = (args.filename || `gen_${Date.now()}`).replace(/[^a-zA-Z0-9._-]/g, '_');
+        const fileName = safeName.endsWith('.png') ? safeName : `${safeName}.png`;
+        const filePath = path.join(tmpDir, fileName);
+        fs.writeFileSync(filePath, imgBuf);
+        const sizeKb = (imgBuf.length / 1024).toFixed(0);
+        return {
+            content: [{
+                    type: 'text',
+                    text: `Image generated successfully.\n\nFile: ${filePath}\nSize: ${sizeKb} KB\nPrompt: ${args.prompt}`,
+                }],
+            details: { filePath, sizeKb, prompt: args.prompt },
+        };
+    }
+    catch (error) {
+        if (error.name === 'TimeoutError') {
+            return { content: [{ type: 'text', text: 'Image generation timed out after 2 minutes. Try a simpler prompt.' }], isError: true };
+        }
+        return { content: [{ type: 'text', text: `Error generating image: ${error.message}` }], isError: true };
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exreve/exk",
-  "version": "1.0.62",
+  "version": "1.0.64",
   "description": "exk - Control Claude CLI with voice and programmable interfaces",
   "type": "module",
   "bin": {
@@ -36,6 +36,7 @@
     "@anthropic-ai/claude-agent-sdk": "^0.2.126",
     "@anthropic-ai/sdk": "^0.92.0",
     "@fastify/static": "^9.0.0",
+    "@mariozechner/pi-coding-agent": "^0.73.1",
     "@xenova/transformers": "^2.17.2",
     "anthropic-proxy": "^1.3.0",
     "chokidar": "^3.6.0",