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

package/dist/tools/pollinations/shared.js

@@ -1,13 +1,14 @@
 /**
  * Shared utilities for Pollinations API tools
  *
- * Updated: 2026-02-12 - Verified API Reference
- * Tests: 18/18 passed
+ * Updated: 2026-02-18 - Sprint 2: Dynamic ModelRegistry integration
+ * Hardcoded model lists replaced by ModelRegistry lookups with static fallback.
  */
 import * as https from 'https';
 import * as fs from 'fs';
 import * as path from 'path';
 import { loadConfig } from '../../server/config.js';
+import { ModelRegistry } from '../../server/models/index.js';
 // ─── Configuration ───────────────────────────────────────────────────────
 const API_BASE = 'gen.pollinations.ai';
 const FREE_IMAGE_BASE = 'image.pollinations.ai';
@@ -19,22 +20,126 @@ export function hasApiKey() {
     const key = getApiKey();
     return !!(key && key.length > 5 && key !== 'dummy');
 }
-// ─── Verified Model Data (2026-02-12) ─────────────────────────────────────
+// ─── Model Data (Dynamic via ModelRegistry) ───────────────────────────────
 /**
- * FREE Image Models (image.pollinations.ai/models)
- * WARNING: flux removed from free, turbo broken (shows notice)
+ * FREE Image Models (DEPRECATED - image.pollinations.ai is dead)
  */
-export const FREE_IMAGE_MODELS = {
-    sana: { desc: 'Default free model', fileSize: '~60KB', reliable: true },
-    zimage: { desc: 'Alias sana/turbo low qual', fileSize: '~35KB', reliable: true },
-    turbo: { desc: 'DEPRECATED - shows notice', fileSize: '~4.1MB', reliable: false },
-};
+export const FREE_IMAGE_MODELS = {};
+/**
+ * Dynamic Paid Image Models accessor.
+ * Returns data from ModelRegistry if ready, otherwise falls back to static data.
+ *
+ * BACKWARD COMPATIBLE: Same shape as the old hardcoded PAID_IMAGE_MODELS
+ */
+export function getPaidImageModels() {
+    if (ModelRegistry.isReady()) {
+        const models = ModelRegistry.list('image');
+        const result = {};
+        for (const m of models) {
+            const costStr = formatPricingForDisplay(m);
+            result[m.name] = {
+                desc: m.description,
+                cost: costStr,
+                t2i: true, // All image models support T2I
+                i2i: m.supportsI2X,
+                params: m.supportsI2X
+                    ? ['width', 'height', 'image']
+                    : ['width', 'height'],
+                notes: m.paid_only ? 'Paid Only' : undefined,
+            };
+        }
+        return result;
+    }
+    return _STATIC_PAID_IMAGE_MODELS;
+}
 /**
- * Paid Image Models (gen.pollinations.ai)
- * I2I = Image-to-Image support
+ * Dynamic Video Models accessor.
+ * BACKWARD COMPATIBLE: Same shape as old VIDEO_MODELS
  */
-export const PAID_IMAGE_MODELS = {
+export function getVideoModels() {
+    if (ModelRegistry.isReady()) {
+        const models = ModelRegistry.list('video');
+        const result = {};
+        for (const m of models) {
+            const costStr = formatPricingForDisplay(m);
+            result[m.name] = {
+                desc: m.description,
+                cost: costStr,
+                t2v: !_STATIC_I2V_ONLY.has(m.name), // wan is I2V only
+                i2v: m.supportsI2X,
+                audio: !_STATIC_NO_AUDIO.has(m.name),
+                duration: m.durationRange || [1, 10],
+                aspectRatios: m.aspectRatios || ['16:9'],
+                costHeader: m.costHeader || 'x-usage-completion-video-seconds',
+                genTime: m.genTimeEstimate || '~30s',
+            };
+        }
+        return result;
+    }
+    return _STATIC_VIDEO_MODELS;
+}
+/**
+ * Dynamic Audio Models accessor.
+ * BACKWARD COMPATIBLE: Same shape as old AUDIO_MODELS
+ */
+export function getAudioModels() {
+    if (ModelRegistry.isReady()) {
+        const models = ModelRegistry.list('audio');
+        const result = {};
+        for (const m of models) {
+            const audioType = detectAudioType(m);
+            result[m.name] = {
+                desc: m.description,
+                type: audioType,
+                endpoint: _STATIC_AUDIO_ENDPOINTS[m.name] || (audioType === 'stt' ? '/v1/audio/transcriptions' : `/audio/{text}`),
+                params: audioType === 'stt' ? ['file'] : ['voice', 'format'],
+                voices: m.voices,
+                notes: m.paid_only ? 'Paid Only' : undefined,
+            };
+        }
+        return result;
+    }
+    return _STATIC_AUDIO_MODELS;
+}
+/**
+ * Music Model accessor (backward compatible)
+ */
+export function getMusicModel() {
+    // Check registry for elevenmusic
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('audio', 'elevenmusic');
+        if (m) {
+            return {
+                'elevenmusic': {
+                    desc: m.description,
+                    endpoint: '/audio/{text}',
+                    params: ['duration', 'instrumental'],
+                    duration: [3, 300],
+                }
+            };
+        }
+    }
+    return _STATIC_MUSIC_MODEL;
+}
+// ─── Backward Compatibility ──────────────────────────────────────────────
+// OLD const exports removed (caused TDZ error at module load).
+// Consumers must use the function forms:
+//   getPaidImageModels(), getVideoModels(), getAudioModels(), getMusicModel()
+// For direct model lookup: use ModelRegistry.getByNameOrAlias()
+// ─── Private Static Fallback Data ─────────────────────────────────────────
+// Used ONLY when ModelRegistry is not ready (startup race, offline).
+const _STATIC_I2V_ONLY = new Set(); // Models that are I2V only (no T2V)
+const _STATIC_NO_AUDIO = new Set(['seedance', 'seedance-pro']); // Video models without audio
+const _STATIC_AUDIO_ENDPOINTS = {
+    'openai-audio': '/v1/chat/completions',
+    'elevenlabs': '/audio/{text}',
+    'whisper': '/v1/audio/transcriptions',
+    'scribe': '/v1/audio/transcriptions',
+    'elevenmusic': '/audio/{text}',
+};
+const _STATIC_PAID_IMAGE_MODELS = {
     'flux': { desc: 'Flux Schnell', cost: '0.0002 🌻', t2i: true, i2i: false, params: ['width', 'height'] },
+    'sana': { desc: 'Sana (Efficient)', cost: '0.0002 🌻', t2i: true, i2i: false, params: ['width', 'height'] },
     'zimage': { desc: 'Z-Image Turbo (6B Flux 2x)', cost: '0.0002 🌻', t2i: true, i2i: false, params: ['width', 'height'] },
     'imagen-4': { desc: 'Imagen 4 (alpha)', cost: '0.0025 🌻', t2i: true, i2i: false, params: ['width', 'height'] },
     'klein': { desc: 'FLUX.2 Klein 4B', cost: '0.008 🌻', t2i: true, i2i: true, params: ['width', 'height', 'image'] },
@@ -47,17 +152,11 @@ export const PAID_IMAGE_MODELS = {
     'nanobanana': { desc: 'NanoBanana (Gemini 2.5 Flash)', cost: 'tokens', t2i: true, i2i: true, params: ['width', 'height', 'image'] },
     'nanobanana-pro': { desc: 'NanoBanana Pro (Gemini 3 Pro)', cost: 'tokens', t2i: true, i2i: true, params: ['width', 'height', 'image'], notes: 'Thinking Model' },
 };
-/**
- * Video Models (gen.pollinations.ai)
- * T2V = Text-to-Video, I2V = Image-to-Video
- */
-export const VIDEO_MODELS = {
+const _STATIC_VIDEO_MODELS = {
     'grok-video': {
         desc: 'Grok Video (alpha)',
         cost: '0.0025/sec',
-        t2v: true,
-        i2v: false,
-        audio: true,
+        t2v: true, i2v: false, audio: true,
         duration: [1, 15],
         aspectRatios: ['16:9', '9:16', '1:1', '4:3'],
         costHeader: 'x-usage-completion-video-seconds',
@@ -66,9 +165,7 @@ export const VIDEO_MODELS = {
     'ltx-2': {
         desc: 'LTX-2 (Lightricks)',
         cost: '0.01/sec',
-        t2v: true,
-        i2v: false,
-        audio: true,
+        t2v: true, i2v: false, audio: true,
         duration: [5, 20],
         aspectRatios: ['16:9'],
         costHeader: 'x-usage-completion-video-seconds',
@@ -77,9 +174,7 @@ export const VIDEO_MODELS = {
     'wan': {
         desc: 'Wan 2.6 (Alibaba)',
         cost: '0.025/sec',
-        t2v: false, // I2V ONLY!
-        i2v: true,
-        audio: true,
+        t2v: false, i2v: true, audio: true,
         duration: [5, 15],
         aspectRatios: ['16:9', '9:16', '1:1', '4:3'],
         costHeader: 'x-usage-completion-video-seconds',
@@ -88,10 +183,8 @@ export const VIDEO_MODELS = {
     'veo': {
         desc: 'Veo 3.1 Fast (Google)',
         cost: '0.15/sec 💎',
-        t2v: true,
-        i2v: true,
-        audio: true,
-        duration: [4, 8], // 4, 6, or 8 seconds
+        t2v: true, i2v: true, audio: true,
+        duration: [4, 8],
         aspectRatios: ['16:9', '9:16', '1:1'],
         costHeader: 'x-usage-completion-video-seconds',
         genTime: '~45-68s',
@@ -99,9 +192,7 @@ export const VIDEO_MODELS = {
     'seedance': {
         desc: 'Seedance Lite (BytePlus)',
         cost: 'tokens',
-        t2v: true,
-        i2v: true,
-        audio: false,
+        t2v: true, i2v: true, audio: false,
         duration: [4, 12],
         aspectRatios: ['16:9', '9:16', '1:1'],
         costHeader: 'x-usage-completion-video-tokens',
@@ -110,20 +201,14 @@ export const VIDEO_MODELS = {
     'seedance-pro': {
         desc: 'Seedance Pro-Fast (BytePlus)',
         cost: 'tokens',
-        t2v: true,
-        i2v: true,
-        audio: false,
+        t2v: true, i2v: true, audio: false,
         duration: [4, 12],
         aspectRatios: ['16:9', '9:16', '1:1'],
         costHeader: 'x-usage-completion-video-tokens',
         genTime: '~30s'
     },
 };
-/**
- * Audio Models
- * TTS = Text-to-Speech, STT = Speech-to-Text
- */
-export const AUDIO_MODELS = {
+const _STATIC_AUDIO_MODELS = {
     'openai-audio': {
         desc: 'GPT-4o Audio Preview',
         type: 'both',
@@ -147,17 +232,48 @@ export const AUDIO_MODELS = {
         notes: 'POST ONLY (multipart)'
     },
 };
-/**
- * Music Model (separate tool)
- */
-export const MUSIC_MODEL = {
+const _STATIC_MUSIC_MODEL = {
     'elevenmusic': {
         desc: 'ElevenLabs Music',
         endpoint: '/audio/{text}',
         params: ['duration', 'instrumental'],
-        duration: [3, 300], // 3-300 seconds
+        duration: [3, 300],
     }
 };
+// ─── Private Helpers ─────────────────────────────────────────────────────
+function formatPricingForDisplay(m) {
+    const p = m.pricing;
+    if (p.completionImageTokens) {
+        return p.completionImageTokens < 0.001
+            ? 'tokens'
+            : `${p.completionImageTokens} 🌻${m.paid_only ? ' 💎' : ''}`;
+    }
+    if (p.completionVideoSeconds) {
+        return `${p.completionVideoSeconds}/sec${m.paid_only ? ' 💎' : ''}`;
+    }
+    if (p.completionVideoTokens) {
+        return 'tokens';
+    }
+    if (p.completionAudioTokens) {
+        return `${p.completionAudioTokens} 🌻/tok`;
+    }
+    if (p.completionAudioSeconds) {
+        return `${p.completionAudioSeconds}/sec`;
+    }
+    if (p.promptAudioSeconds) {
+        return `${p.promptAudioSeconds}/sec`;
+    }
+    return 'tokens';
+}
+function detectAudioType(m) {
+    const hasAudioInput = m.input_modalities.includes('audio');
+    const hasAudioOutput = m.output_modalities.includes('audio');
+    if (hasAudioInput && hasAudioOutput)
+        return 'both';
+    if (hasAudioInput)
+        return 'stt';
+    return 'tts';
+}
 // ─── HTTP Helpers ─────────────────────────────────────────────────────────
 export function httpsGet(url, headers = {}) {
     return new Promise((resolve, reject) => {
@@ -299,38 +415,38 @@ export function httpsPostMultipart(url, fields, headers = {}) {
         req.end();
     });
 }
-// ─── Model Discovery ─────────────────────────────────────────────────────
-const MODEL_CACHE = {
-    image: [],
-    audio: [],
-    text: [],
-};
-let CACHE_TIME = 0;
-const CACHE_TTL = 5 * 60 * 1000; // 5 minutes
+// ─── Model Discovery (delegated to ModelRegistry) ─────────────────────────
+/**
+ * @deprecated Use ModelRegistry.list() directly
+ */
 export async function fetchModels(type) {
-    const now = Date.now();
-    if (MODEL_CACHE[type].length > 0 && now - CACHE_TIME < CACHE_TTL) {
-        return MODEL_CACHE[type];
-    }
-    const apiKey = getApiKey();
-    const headers = {};
-    if (apiKey) {
-        headers['Authorization'] = `Bearer ${apiKey}`;
-    }
-    try {
-        const { data } = await httpsGet(`https://${API_BASE}/${type}/models`, headers);
-        MODEL_CACHE[type] = JSON.parse(data.toString());
-        CACHE_TIME = now;
-        return MODEL_CACHE[type];
-    }
-    catch (err) {
-        console.error(`Failed to fetch ${type} models:`, err);
-        return [];
-    }
+    ModelRegistry.ensureFresh();
+    const models = ModelRegistry.list(type);
+    return models.map(m => ({
+        name: m.name,
+        pricing: m.pricing,
+        paid_only: m.paid_only,
+        input_modalities: m.input_modalities,
+        output_modalities: m.output_modalities,
+        description: m.description,
+    }));
 }
+/**
+ * @deprecated Use ModelRegistry.get() directly
+ */
 export async function getModelInfo(type, name) {
-    const models = await fetchModels(type);
-    return models.find(m => m.name === name);
+    ModelRegistry.ensureFresh();
+    const m = ModelRegistry.getByNameOrAlias(type, name);
+    if (!m)
+        return undefined;
+    return {
+        name: m.name,
+        pricing: m.pricing,
+        paid_only: m.paid_only,
+        input_modalities: m.input_modalities,
+        output_modalities: m.output_modalities,
+        description: m.description,
+    };
 }
 // ─── Cost Estimation & Tracking ───────────────────────────────────────────
 /**
@@ -347,10 +463,37 @@ export function extractCostFromHeaders(headers) {
         videoTokens: headers['x-usage-completion-video-tokens']
             ? parseFloat(headers['x-usage-completion-video-tokens'])
             : undefined,
+        costUsd: headers['x-usage-cost-usd']
+            ? parseFloat(headers['x-usage-cost-usd'])
+            : undefined,
         modelUsed: headers['x-model-used'],
         requestId: headers['x-request-id'],
     };
 }
+/**
+ * Fetch current Enter balance (`/account/balance`) for Real Cost calculation.
+ */
+export async function fetchEnterBalance() {
+    const apiKey = getApiKey();
+    if (!apiKey)
+        return null;
+    try {
+        const url = 'https://gen.pollinations.ai/account/balance';
+        // Using native fetch
+        const res = await fetch(url, {
+            headers: { 'Authorization': `Bearer ${apiKey}` },
+            signal: AbortSignal.timeout(5000)
+        });
+        if (!res.ok)
+            return null;
+        const data = await res.json();
+        // The endpoint usually returns just { "balance": 9.9... }
+        return data.balance !== undefined ? data.balance : null;
+    }
+    catch {
+        return null; // Silent catch
+    }
+}
 /**
  * Check if cost estimator is enabled in config
  */
@@ -358,33 +501,79 @@ export function isCostEstimatorEnabled() {
     const config = loadConfig();
     return config.costEstimator !== false; // Default true
 }
+// ─── COST ESTIMATION BENCHMARKS ──────────────────────────────────────────
+export function per1pollen(cost) {
+    if (!cost || cost <= 0)
+        return "—";
+    const x = 1 / cost;
+    if (x >= 1_000_000)
+        return `${(x / 1_000_000).toFixed(1)}M`;
+    if (x >= 100_000)
+        return `${Math.round(x / 1000)}K`;
+    if (x >= 10_000)
+        return `${(x / 1000).toFixed(1)}K`.replace(/\.0K$/, "K");
+    if (x >= 1_000)
+        return `${Math.round(x / 100) * 100}`.replace(/\B(?=(\d{3})+(?!\d))/g, ",");
+    if (x >= 100)
+        return `${Math.round(x)}`;
+    if (x >= 10)
+        return `${Math.round(x * 10) / 10}`;
+    return `${x.toFixed(1)}`;
+}
 export function estimateImageCost(model) {
-    const info = PAID_IMAGE_MODELS[model];
+    // Try ModelRegistry first
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('image', model);
+        if (m && m.averageCost !== undefined) {
+            return m.averageCost;
+        }
+    }
+    // Fallback to static
+    const info = _STATIC_PAID_IMAGE_MODELS[model];
     if (!info)
         return 0.0002;
     const costMatch = info.cost.match(/[\d.]+/);
     return costMatch ? parseFloat(costMatch[0]) : 0.0002;
 }
 export function estimateVideoCost(model, duration) {
-    const info = VIDEO_MODELS[model];
+    // Try ModelRegistry first
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('video', model);
+        if (m && m.averageCost !== undefined) {
+            return m.averageCost;
+        }
+    }
+    // Fallback to static
+    const info = _STATIC_VIDEO_MODELS[model];
     if (!info)
         return duration * 0.01;
     if (info.costHeader === 'x-usage-completion-video-tokens') {
-        // Token-based: 108900 tokens for 5s video
         const tokensPerSecond = 21780;
-        return (duration * tokensPerSecond) * 0.00001; // Approximate
+        return (duration * tokensPerSecond) * 0.00001;
     }
-    // Second-based
     const costMatch = info.cost.match(/[\d.]+/);
     const perSecond = costMatch ? parseFloat(costMatch[0]) : 0.01;
     return duration * perSecond;
 }
 export function estimateTtsCost(textLength) {
-    // Approximate: 1 char ≈ 1 token
+    // Try ModelRegistry first
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('audio', 'elevenlabs');
+        if (m && m.averageCost !== undefined) {
+            return m.averageCost;
+        }
+    }
     return (textLength / 1000) * 0.00018;
 }
 export function estimateMusicCost(duration) {
-    return duration * 0.005; // ~0.005/sec
+    // Try ModelRegistry first
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('audio', 'elevenmusic');
+        if (m && m.averageCost !== undefined) {
+            return m.averageCost;
+        }
+    }
+    return duration * 0.005;
 }
 // ─── File Utils ──────────────────────────────────────────────────────────
 export function ensureDir(dir) {
@@ -416,39 +605,61 @@ export function formatFileSize(bytes) {
         return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
     return `${(bytes / (1024 * 1024 * 1024)).toFixed(1)} GB`;
 }
-// ─── Validation Helpers ──────────────────────────────────────────────────
+// ─── Validation Helpers (Dynamic via ModelRegistry) ──────────────────────
 /**
  * Check if model supports Image-to-Image
  */
 export function supportsI2I(model) {
-    const info = PAID_IMAGE_MODELS[model];
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('image', model);
+        return m?.supportsI2X === true;
+    }
+    const info = _STATIC_PAID_IMAGE_MODELS[model];
     return info?.i2i === true;
 }
 /**
  * Check if video model supports Image-to-Video
  */
 export function supportsI2V(model) {
-    const info = VIDEO_MODELS[model];
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('video', model);
+        return m?.supportsI2X === true;
+    }
+    const info = _STATIC_VIDEO_MODELS[model];
     return info?.i2v === true;
 }
 /**
  * Check if video model requires Image-to-Video (no T2V)
  */
 export function requiresI2V(model) {
-    const info = VIDEO_MODELS[model];
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('video', model);
+        if (m) {
+            return _STATIC_I2V_ONLY.has(m.name); // Only wan is I2V-only for now
+        }
+    }
+    const info = _STATIC_VIDEO_MODELS[model];
     return info?.t2v === false && info?.i2v === true;
 }
 /**
  * Validate aspect ratio for video model
  */
 export function validateAspectRatio(model, ratio) {
-    const info = VIDEO_MODELS[model];
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('video', model);
+        return m?.aspectRatios?.includes(ratio) ?? false;
+    }
+    const info = _STATIC_VIDEO_MODELS[model];
     return info?.aspectRatios.includes(ratio) ?? false;
 }
 /**
  * Get valid duration range for video model
  */
 export function getDurationRange(model) {
-    const info = VIDEO_MODELS[model];
+    if (ModelRegistry.isReady()) {
+        const m = ModelRegistry.getByNameOrAlias('video', model);
+        return m?.durationRange ?? [1, 10];
+    }
+    const info = _STATIC_VIDEO_MODELS[model];
     return info?.duration ?? [1, 10];
 }

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

@@ -0,0 +1 @@
+export {};

package/dist/tools/pollinations/test_estimators.js

@@ -0,0 +1,22 @@
+import { estimateImageCost, estimateVideoCost, estimateTtsCost, per1pollen } from './shared.js';
+import { ModelRegistry } from '../../server/models/index.js';
+async function testEstimators() {
+    console.log("Loading models...");
+    await ModelRegistry.ensureFresh();
+    const imageTests = ['flux', 'flux-pro', 'turbo'];
+    console.log("\n=== IMAGE ESTIMATIONS ===");
+    for (const model of imageTests) {
+        const cost = estimateImageCost(model);
+        console.log(`[${model}]: Cost = ${cost}, 1 pollen ≈ ${per1pollen(cost)} images`);
+    }
+    const videoTests = ['ltx-2', 'wan', 'veo'];
+    console.log("\n=== VIDEO ESTIMATIONS (6s) ===");
+    for (const model of videoTests) {
+        const cost = estimateVideoCost(model, 6);
+        console.log(`[${model}]: Cost = ${cost}, 1 pollen ≈ ${per1pollen(cost)} vidéos`);
+    }
+    console.log("\n=== TTS ESTIMATIONS (200 chars) ===");
+    const ttsCost = estimateTtsCost(200);
+    console.log(`[elevenlabs]: Cost = ${ttsCost}, 1 pollen ≈ ${per1pollen(ttsCost)} generations`);
+}
+testEstimators();

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

@@ -3,15 +3,11 @@
  *
  * Updated: 2026-02-12 - Verified API Reference
  *
- * Two STT options:
- * 1. openai-audio (DEFAULT): GPT-4o Audio Preview - uses /v1/chat/completions with modalities
- *    - Least expensive option
- *    - Can handle both audio input and output
+ * 1. whisper-large-v3 (DEFAULT): High accuracy Whisper model
+ * 2. whisper-1: Standard Whisper model
+ * 3. scribe: ElevenLabs Scribe v2
  *
- * 2. whisper: OpenAI Whisper v3 - uses /v1/audio/transcriptions
- *    - POST ONLY with multipart/form-data
- *    - Specialized for transcription
- *    - Higher accuracy for long audio
+ * All models use /v1/audio/transcriptions (POST multipart)
  */
 import { type ToolDefinition } from '@opencode-ai/plugin/tool';
-export declare const transcribeAudioTool: ToolDefinition;
+export declare const polliSttTool: ToolDefinition;