opencode-pollinations-plugin 6.1.0-beta.2 → 6.1.0-beta.22

package/dist/tools/pollinations/gen_image.js

@@ -0,0 +1,225 @@
+/**
+ * gen_image Tool - Pollinations Image Generation
+ *
+ * Updated: 2026-02-19 - Dynamic ModelRegistry + Cost Guard + Toasts
+ *
+ * All models are dynamic from the Pollinations API.
+ * Unknown models are accepted as (beta) and passed through to the API.
+ * Cost Guard reads enablePaidTools, costConfirmationRequired, costThreshold.
+ */
+import { tool } from '@opencode-ai/plugin/tool';
+import * as fs from 'fs';
+import * as path from 'path';
+import { getApiKey, hasApiKey, httpsGet, ensureDir, generateFilename, getDefaultOutputDir, formatCost, formatFileSize, estimateImageCost, extractCostFromHeaders, isCostEstimatorEnabled, supportsI2I, getPaidImageModels, fetchEnterBalance, } from './shared.js';
+import { loadConfig } from '../../server/config.js';
+import { checkCostControl, isTokenBased } from './cost-guard.js';
+import { emitStatusToast } from '../../server/toast.js';
+// ─── Constants ─────────────────────────────────────────────────────────────
+const DEFAULT_MODEL = 'flux';
+// ─── Tool Definition ──────────────────────────────────────────────────────
+export const polliGenImageTool = tool({
+    description: `Generate an image from a text prompt using Pollinations AI.
+
+💡 **Modèles Image Dynamiques** :
+L'API Pollinations (Enter) possède une quantité importante de modèles (Flux, Midjourney, Seedream, etc.) et ils changent fréquemment. Le catalogue à jour est listé ci-dessous.
+
+**Exemples d'utilisation Optionnelle** :
+- **I2I (Image-to-Image)** : Utilisez le paramètre \`reference_image\` avec une URL ou chemin local si le modèle le supporte.
+- L'outil embarque un "costGuard" automatique gérant la confirmation des coûts.`,
+    args: {
+        prompt: tool.schema.string().describe('Description of the image to generate'),
+        model: tool.schema.string().optional().describe('Model to use (default: flux). Unknown models accepted as (beta).'),
+        width: tool.schema.number().min(256).max(4096).optional().describe('Image width (default: 1024)'),
+        height: tool.schema.number().min(256).max(4096).optional().describe('Image height (default: 1024)'),
+        reference_image: tool.schema.string().optional().describe('URL(s) for image-to-image editing (comma-separated for multi-image models)'),
+        seed: tool.schema.number().optional().describe('Seed for reproducibility (-1 for random)'),
+        quality: tool.schema.enum(['low', 'med', 'high']).optional().describe('Quality for gptimage models only'),
+        transparent: tool.schema.boolean().optional().describe('Transparent background for gptimage models only'),
+        save_to: tool.schema.string().optional().describe('Custom output directory'),
+        filename: tool.schema.string().optional().describe('Custom filename (without extension)'),
+    },
+    async execute(args, context) {
+        const apiKey = getApiKey();
+        const hasKey = hasApiKey();
+        // Determine model based on key presence
+        let model = args.model || DEFAULT_MODEL;
+        const width = args.width || 1024;
+        const height = args.height || 1024;
+        // Fetch known models from registry
+        const imageModels = getPaidImageModels();
+        const knownModel = !!imageModels[model];
+        const isBetaModel = !knownModel;
+        // Force Auth Check for ALL Image Generations
+        if (!hasKey) {
+            return `❌ **Clé API Requise** pour la génération d'images.
+💡 Utilisez \`/pollinations connect <clé>\` pour activer le service.
+💎 Modèles disponibles: ${Object.keys(imageModels).slice(0, 5).join(', ')}...`;
+        }
+        // Unknown model → beta passthrough (don't reject)
+        if (isBetaModel) {
+            emitStatusToast('warning', `Modèle "${model}" non référencé — mode (beta)`, '🎨 gen_image');
+        }
+        // Validate I2I support (for known models only; beta models get default behavior)
+        if (args.reference_image && knownModel && !supportsI2I(model)) {
+            return `⚠️ Le modèle "${model}" ne supporte pas l'Image-to-Image.
+💡 Modèles I2I supportés: ${Object.entries(imageModels)
+                .filter(([, info]) => info.i2i)
+                .map(([name]) => name)
+                .join(', ')}`;
+        }
+        // Estimate cost
+        const estimatedCost = estimateImageCost(model);
+        // Cost Guard check V2
+        const costCheck = checkCostControl('polli_gen_image', args, model, estimatedCost, 'image');
+        if (!costCheck.allowed) {
+            return costCheck.message || '❌ Opération bloquée par le Cost Guard.';
+        }
+        // Emit start toast
+        const config = loadConfig();
+        const argsStr = config.gui?.logs === 'verbose' ? `\nParameters: ${JSON.stringify(args)}` : '';
+        emitStatusToast('info', `Génération image: ${model} (${width}×${height})${argsStr}`, '🎨 polli_gen_image');
+        // Set metadata
+        context.metadata({ title: `🎨 Image: ${model}${isBetaModel ? ' (beta)' : ''}` });
+        try {
+            let imageData;
+            let responseHeaders = {};
+            let usedModel = model;
+            // === ENTER endpoint ONLY (gen.pollinations.ai) ===
+            const params = new URLSearchParams({
+                nologo: 'true',
+                private: 'true',
+                width: String(width),
+                height: String(height),
+            });
+            // Model parameter
+            params.set('model', model);
+            // Seed
+            if (args.seed !== undefined) {
+                params.set('seed', String(args.seed));
+            }
+            // I2I: reference image(s)
+            if (args.reference_image) {
+                // Check if it's a local file path
+                let imageUrl = args.reference_image;
+                if (!args.reference_image.startsWith('http')) {
+                    // For local files, we'd need to upload first
+                    // For now, require URL
+                    return `❌ Les fichiers locaux nécessitent d'être uploadés d'abord.
+💡 Utilisez l'outil \`file_to_url\` pour obtenir une URL publique.`;
+                }
+                params.set('image', imageUrl);
+            }
+            // Quality (gptimage only)
+            if (args.quality && model.startsWith('gptimage')) {
+                params.set('quality', args.quality);
+            }
+            // Transparent (gptimage only)
+            if (args.transparent !== undefined && model.startsWith('gptimage')) {
+                params.set('transparent', String(args.transparent));
+            }
+            const promptEncoded = encodeURIComponent(args.prompt);
+            const url = `https://gen.pollinations.ai/image/${promptEncoded}?${params}`;
+            const headers = {};
+            if (apiKey)
+                headers['Authorization'] = `Bearer ${apiKey}`;
+            // 1. Fetch balance avant génération
+            const balBefore = await fetchEnterBalance();
+            const result = await httpsGet(url, headers);
+            imageData = result.data;
+            responseHeaders = result.headers;
+            // Update used model from response if available
+            if (responseHeaders['x-model-used']) {
+                usedModel = responseHeaders['x-model-used'];
+            }
+            // Save the image
+            let outputDir = getDefaultOutputDir('images');
+            let filename = args.filename;
+            if (args.save_to) {
+                if (args.save_to.match(/\.(png|jpe?g|webp|gif)$/i)) {
+                    outputDir = path.dirname(args.save_to);
+                    filename = path.basename(args.save_to);
+                }
+                else {
+                    outputDir = args.save_to;
+                }
+            }
+            ensureDir(outputDir);
+            filename = filename || generateFilename('image', usedModel, 'png');
+            const filePath = path.join(outputDir, filename.includes('.') ? filename : `${filename}.png`);
+            fs.writeFileSync(filePath, imageData);
+            // 2. Fetch balance après génération (delay for API sync)
+            let balAfter = null;
+            let realCost;
+            if (balBefore !== null) {
+                await new Promise(r => setTimeout(r, 1000)); // Laisse le temps au ledger
+                balAfter = await fetchEnterBalance();
+                if (balAfter !== null) {
+                    realCost = Math.round((balBefore - balAfter) * 10000) / 10000;
+                }
+            }
+            // Extract cost from headers as fallback/info
+            const costTracking = extractCostFromHeaders(responseHeaders);
+            // Build result
+            const fileSize = fs.statSync(filePath).size;
+            const lines = [];
+            // Inject costWarning at top if present
+            if (costCheck.message && !costCheck.allowed) { // Assuming costWarning should come from costCheck if not allowed
+                lines.push(costCheck.message);
+                lines.push('');
+            }
+            lines.push(`🎨 Image Générée`);
+            lines.push(`━━━━━━━━━━━━━━━━━━`);
+            lines.push(`Prompt: ${args.prompt.substring(0, 100)}${args.prompt.length > 100 ? '...' : ''}`);
+            lines.push(`Modèle: ${usedModel}${isBetaModel ? ' (beta)' : ''}`);
+            lines.push(`Résolution: ${width}×${height}`);
+            // Add I2I info if used
+            if (args.reference_image) {
+                lines.push(`I2I Source: ${args.reference_image.substring(0, 50)}...`);
+            }
+            lines.push(`Fichier: ${filePath}`);
+            lines.push(`Taille: ${formatFileSize(fileSize)}`);
+            // Pricing details (Estimé vs Réel)
+            if (isCostEstimatorEnabled()) {
+                const maxCost = estimatedCost * 3;
+                lines.push(`\n💰 **Rapport Financier :**`);
+                if (isTokenBased('image', usedModel)) {
+                    lines.push(`- Coût Estimé   : ${formatCost(estimatedCost)} (Max théorique: ${formatCost(maxCost)})`);
+                }
+                else {
+                    lines.push(`- Coût Estimé   : ${formatCost(estimatedCost)}`);
+                }
+                if (realCost !== undefined) {
+                    lines.push(`- Coût Réel     : **${formatCost(realCost)}** (via Solde Wallet)`);
+                }
+                else if (costTracking.costUsd !== undefined) {
+                    lines.push(`- Coût Réel     : **${formatCost(costTracking.costUsd)}** (via Headers API)`);
+                }
+                else {
+                    lines.push(`- Coût Réel     : Inconnu (API injoignable)`);
+                }
+            }
+            if (responseHeaders['x-request-id']) {
+                lines.push(`Request ID: ${responseHeaders['x-request-id']}`);
+            }
+            // Emit success toast
+            emitStatusToast('success', `Image générée ✓ (${usedModel})`, '🎨 gen_image', { filePath: filePath });
+            return lines.join('\n');
+        }
+        catch (err) {
+            emitStatusToast('error', `Erreur: ${err.message?.substring(0, 60)}`, '🎨 gen_image');
+            if (err.message?.includes('402') || err.message?.includes('Payment')) {
+                return `❌ Crédits pollen insuffisants pour le modèle "${model}".
+💡 Vérifiez votre solde avec /pollinations usage`;
+            }
+            if (err.message?.includes('401') || err.message?.includes('403')) {
+                return `❌ Clé API invalide ou non autorisée.
+🔧 Vérifiez votre clé avec /pollinations connect`;
+            }
+            if (err.message?.includes('400')) {
+                return `❌ Paramètres invalides: ${err.message}
+💡 Vérifiez que le modèle supporte les paramètres fournis.`;
+            }
+            return `❌ Erreur génération image: ${err.message}`;
+        }
+    },
+});

package/dist/tools/pollinations/gen_music.d.ts

@@ -0,0 +1,14 @@
+/**
+ * gen_music Tool - Pollinations Music Generation
+ *
+ * Updated: 2026-02-12 - Verified API Reference
+ *
+ * Model: elevenmusic (ElevenLabs Music)
+ * Endpoint: gen.pollinations.ai/audio/{text}
+ *
+ * Parameters:
+ * - duration: 3-300 seconds
+ * - instrumental: boolean (vocals or instrumental only)
+ */
+import { type ToolDefinition } from '@opencode-ai/plugin/tool';
+export declare const polliGenMusicTool: ToolDefinition;

package/dist/tools/pollinations/gen_music.js

@@ -0,0 +1,180 @@
+/**
+ * gen_music Tool - Pollinations Music Generation
+ *
+ * Updated: 2026-02-12 - Verified API Reference
+ *
+ * Model: elevenmusic (ElevenLabs Music)
+ * Endpoint: gen.pollinations.ai/audio/{text}
+ *
+ * Parameters:
+ * - duration: 3-300 seconds
+ * - instrumental: boolean (vocals or instrumental only)
+ */
+import { tool } from '@opencode-ai/plugin/tool';
+import * as fs from 'fs';
+import * as path from 'path';
+import { getApiKey, httpsGet, ensureDir, generateFilename, getDefaultOutputDir, formatCost, formatFileSize, estimateMusicCost, extractCostFromHeaders, isCostEstimatorEnabled, } from './shared.js';
+import { loadConfig } from '../../server/config.js';
+import { checkCostControl, isTokenBased } from './cost-guard.js';
+import { emitStatusToast } from '../../server/toast.js';
+// ─── Constants ─────────────────────────────────────────────────────────────
+const MIN_DURATION = 3;
+const MAX_DURATION = 300; // 5 minutes
+const DEFAULT_DURATION = 10;
+const MODEL_NAME = 'elevenmusic';
+// ─── Tool Definition ──────────────────────────────────────────────────────
+export const polliGenMusicTool = tool({
+    description: `Generate music from a text description using Pollinations AI.
+
+**🎵 Model:** elevenmusic (ElevenLabs Music)
+
+**📝 Parameters:**
+- \`duration\`: 3-300 seconds (default: 10s)
+- \`instrumental\`: true = no vocals, false = vocals allowed
+
+**💡 Example Prompts:**
+- "upbeat jazz with saxophone solo"
+- "ambient electronic for meditation"
+- "epic orchestral film score with dramatic strings"
+- "lo-fi hip hop beats with piano"
+- "acoustic guitar ballad with soft vocals"
+- "electronic dance music with heavy bass drop"
+
+**💰 Cost:** ~0.005 🌻 per second
+- 10 seconds ≈ 0.05 🌻
+- 30 seconds ≈ 0.15 🌻
+- 60 seconds ≈ 0.30 🌻
+
+**⚠️ Notes:**
+- Generation time scales with duration (~1s per second of audio)
+- Longer tracks (60s+) may take 1-2 minutes
+- Instrumental mode produces cleaner results for background music`,
+    args: {
+        prompt: tool.schema.string().describe('Description of the music to generate'),
+        duration: tool.schema.number().min(MIN_DURATION).max(MAX_DURATION).optional()
+            .describe(`Duration in seconds (default: ${DEFAULT_DURATION}, max: ${MAX_DURATION})`),
+        instrumental: tool.schema.boolean().optional().describe('Instrumental only - no vocals (default: false)'),
+        seed: tool.schema.number().optional().describe('Seed for reproducibility (-1 for random)'),
+        save_to: tool.schema.string().optional().describe('Custom output directory'),
+        filename: tool.schema.string().optional().describe('Custom filename (without extension)'),
+    },
+    async execute(args, context) {
+        const apiKey = getApiKey();
+        if (!apiKey) {
+            return `❌ La génération musicale nécessite une clé API Pollinations.
+🔧 Connectez votre clé avec /pollinations connect`;
+        }
+        const duration = Math.min(args.duration || DEFAULT_DURATION, MAX_DURATION);
+        const instrumental = args.instrumental || false;
+        // Estimate cost
+        const estimatedCost = estimateMusicCost(duration);
+        // Cost Guard check V2
+        const costCheck = checkCostControl('polli_gen_music', args, MODEL_NAME, estimatedCost, 'audio');
+        if (!costCheck.allowed) {
+            return costCheck.message || '❌ Opération bloquée par le Cost Guard.';
+        }
+        // Estimate generation time
+        const genTimeSeconds = Math.ceil(duration * 1.2);
+        // Emit start toast
+        const config = loadConfig();
+        const argsStr = config.gui?.logs === 'verbose' ? `\nParameters: ${JSON.stringify(args)}` : '';
+        emitStatusToast('info', `Génération musique: ${duration}s (~${genTimeSeconds}s gen)${argsStr}`, '🎵 polli_gen_music');
+        // Metadata
+        context.metadata({ title: `🎵 Music: ${duration}s (~${genTimeSeconds}s gen time)` });
+        try {
+            // Build URL
+            const params = new URLSearchParams({
+                model: MODEL_NAME,
+                nologo: 'true',
+                private: 'true',
+                duration: String(duration),
+            });
+            if (instrumental) {
+                params.set('instrumental', 'true');
+            }
+            // Seed for reproducibility
+            if (args.seed !== undefined) {
+                params.set('seed', String(args.seed));
+            }
+            const promptEncoded = encodeURIComponent(args.prompt);
+            const url = `https://gen.pollinations.ai/audio/${promptEncoded}?${params}`;
+            const headers = {
+                'Authorization': `Bearer ${apiKey}`,
+            };
+            // Music generation takes time
+            const result = await httpsGet(url, headers);
+            const audioData = result.data;
+            const responseHeaders = result.headers;
+            // Save audio
+            let outputDir = getDefaultOutputDir('music');
+            let filename = args.filename;
+            if (args.save_to) {
+                if (args.save_to.match(/\.(mp3|wav|ogg|m4a)$/i)) {
+                    outputDir = path.dirname(args.save_to);
+                    filename = path.basename(args.save_to);
+                }
+                else {
+                    outputDir = args.save_to;
+                }
+            }
+            ensureDir(outputDir);
+            filename = filename || generateFilename('music', MODEL_NAME, 'mp3');
+            const filePath = path.join(outputDir, filename.endsWith('.mp3') ? filename : `${filename}.mp3`);
+            fs.writeFileSync(filePath, audioData);
+            const fileSize = fs.statSync(filePath).size;
+            let actualCost = estimatedCost;
+            if (responseHeaders) {
+                const costTracking = extractCostFromHeaders(responseHeaders);
+                if (costTracking.costUsd !== undefined)
+                    actualCost = costTracking.costUsd;
+            }
+            // Build result
+            const lines = [];
+            // Inject costWarning at top if present
+            if (costCheck.message && !costCheck.allowed) {
+                lines.push(costCheck.message);
+                lines.push('');
+            }
+            lines.push(`🎵 Musique Générée`);
+            lines.push(`━━━━━━━━━━━━━━━━━━`);
+            lines.push(`Prompt: ${args.prompt}`);
+            lines.push(`Durée: ~${duration}s`);
+            lines.push(`Mode: ${instrumental ? 'Instrumental (sans voix)' : 'Avec voix possible'}`);
+            lines.push(`Fichier: ${filePath}`);
+            lines.push(`Taille: ${formatFileSize(fileSize)}`);
+            // Cost info
+            if (isCostEstimatorEnabled()) {
+                if (isTokenBased('audio', MODEL_NAME)) {
+                    const maxCost = estimatedCost * 3;
+                    lines.push(`Coût: ${formatCost(actualCost)} (Max théorique: ${formatCost(maxCost)})`);
+                }
+                else {
+                    lines.push(`Coût: ${formatCost(actualCost)}`);
+                }
+            }
+            if (responseHeaders['x-model-used']) {
+                lines.push(`Modèle utilisé: ${responseHeaders['x-model-used']}`);
+            }
+            if (responseHeaders['x-request-id']) {
+                lines.push(`Request ID: ${responseHeaders['x-request-id']}`);
+            }
+            // Emit success toast
+            emitStatusToast('success', `Musique générée ✓ (${duration}s)`, '🎵 gen_music');
+            return lines.join('\n');
+        }
+        catch (err) {
+            emitStatusToast('error', `Erreur: ${err.message?.substring(0, 60)}`, '🎵 gen_music');
+            if (err.message?.includes('402') || err.message?.includes('Payment')) {
+                return `❌ Crédits pollen insuffisants.`;
+            }
+            if (err.message?.includes('401') || err.message?.includes('403')) {
+                return `❌ Clé API invalide ou non autorisée.`;
+            }
+            if (err.message?.includes('Timeout')) {
+                return `❌ Timeout - La génération musicale a pris trop de temps.
+💡 Essayez une durée plus courte.`;
+            }
+            return `❌ Erreur génération musicale: ${err.message}`;
+        }
+    },
+});

package/dist/tools/pollinations/gen_video.d.ts

@@ -0,0 +1,16 @@
+/**
+ * gen_video Tool - Pollinations Video Generation
+ *
+ * Updated: 2026-02-12 - Verified API Reference
+ *
+ * Video models with different capabilities:
+ * - T2V (Text-to-Video): grok-video, ltx-2, veo, seedance, seedance-pro
+ * - I2V (Image-to-Video): wan (I2V ONLY!), veo, seedance, seedance-pro
+ * - Veo Interpolation: Uses image=url1,url2 for transitions
+ *
+ * Response headers for cost tracking:
+ * - x-usage-completion-video-seconds (grok, ltx-2, veo, wan)
+ * - x-usage-completion-video-tokens (seedance, seedance-pro)
+ */
+import { type ToolDefinition } from '@opencode-ai/plugin/tool';
+export declare const polliGenVideoTool: ToolDefinition;