5a7875ba · permissionBRICK/SillyTavern

public/index.html+12 -9 Viewed

                             </select>
                         </div>
                         <div id="pollinations_form" data-source="pollinations">
+                            <h4>
+                                <a href="https://enter.pollinations.ai/" target="_blank" rel="noopener noreferrer" data-i18n="Pollinations API Key">
+                                    Pollinations API Key
+                                </a>
+                            </h4>
+                            <div class="flex-container">
+                                <input id="api_key_pollinations" name="api_key_pollinations" class="text_pole flex1" value="" type="text" autocomplete="off">
+                                <div title="Manage API keys" data-i18n="[title]Manage API keys" class="menu_button fa-solid fa-key fa-fw manage-api-keys" data-key="api_key_pollinations"></div>
+                            </div>
+                            <div data-for="api_key_pollinations" class="neutral_warning" data-i18n="For privacy reasons, your API key will be hidden after you click 'Connect'.">
+                                For privacy reasons, your API key will be hidden after you click 'Connect'.
+                            </div>
                             <h4 data-i18n="Pollinations Model">Pollinations Model</h4>
                             <select id="model_pollinations_select">
                                 <!-- Populated by JavaScript -->
                             </select>
-                            <div class="info-block hint">
-                                <a href="https://pollinations.ai/" target="_blank" rel="noopener noreferrer" data-i18n="Provided free of charge by Pollinations.AI">
-                                    Provided free of charge by Pollinations.AI
-                                </a>
-                                <br>
-                                <span data-i18n="Avoid sending sensitive information. Provider's outputs may include ads.">
-                                    Avoid sending sensitive information. Provider's outputs may include ads.
-                                </span>
-                            </div>
                         </div>
                         <div id="moonshot_form" data-source="moonshot">
                             <h4>

public/scripts/RossAscends-mods.js+1 -1 Viewed

                     || (secret_state[SECRET_KEYS.FIREWORKS] && oai_settings.chat_completion_source == chat_completion_sources.FIREWORKS)
                     || (secret_state[SECRET_KEYS.COMETAPI] && oai_settings.chat_completion_source == chat_completion_sources.COMETAPI)
                     || (secret_state[SECRET_KEYS.ZAI] && oai_settings.chat_completion_source == chat_completion_sources.ZAI)
-                    || (oai_settings.chat_completion_source === chat_completion_sources.POLLINATIONS)
+                    || (secret_state[SECRET_KEYS.POLLINATIONS] && oai_settings.chat_completion_source === chat_completion_sources.POLLINATIONS)
                     || (isValidUrl(oai_settings.custom_url) && oai_settings.chat_completion_source == chat_completion_sources.CUSTOM)
                     || (secret_state[SECRET_KEYS.AZURE_OPENAI] && oai_settings.chat_completion_source == chat_completion_sources.AZURE_OPENAI)
                 ) {

public/scripts/extensions/caption/index.js+2 -1 Viewed

                         'nanogpt': SECRET_KEYS.NANOGPT,
                         'chutes': SECRET_KEYS.CHUTES,
                         'electronhub': SECRET_KEYS.ELECTRONHUB,
+                        'pollinations': SECRET_KEYS.POLLINATIONS,
                     };
                     if (chatCompletionApis[api] && secret_state[chatCompletionApis[api]]) {

@@ -530,7 +531,7 @@ jQuery(async function () {
530	}	531	}
531		532
532	// Custom API doesn't need additional checks	533	// Custom API doesn't need additional checks
533	if (api === 'custom' \|\| api === 'pollinations') {	534	if (api === 'custom') {
534	return true;	535	return true;
535	}	536	}
536	}	537	}

public/scripts/extensions/shared.js+4 -0 Viewed

@@ -279,6 +279,10 @@ function throwIfInvalidModel(useReverseProxy) {
279	if (multimodalApi === 'zai' && !secret_state[SECRET_KEYS.ZAI]) {	279	if (multimodalApi === 'zai' && !secret_state[SECRET_KEYS.ZAI]) {
280	throw new Error('Z.AI API key is not set.');	280	throw new Error('Z.AI API key is not set.');
281	}	281	}
		282
		283	if (multimodalApi === 'pollinations' && !secret_state[SECRET_KEYS.POLLINATIONS]) {
		284	throw new Error('Pollinations API key is not set.');
		285	}
282	}	286	}
283		287
284	/**	288	/**

public/scripts/extensions/stable-diffusion/index.js+16 -10 Viewed

+}
 async function loadPollinationsModels() {
+    $('#sd_pollinations_key').toggleClass('success', !!secret_state[SECRET_KEYS.POLLINATIONS]);
     const result = await fetch('/api/sd/pollinations/models', {
         method: 'POST',
         headers: getRequestHeaders({ omitContentType: true }),

     if (result.ok) {
         const data = await result.json();
-        return { format: 'jpg', data: data?.image };
+        return { format: data?.format, data: data?.image };
     } else {
         const text = await result.text();
         throw new Error(text);

         case sources.togetherai:
             return secret_state[SECRET_KEYS.TOGETHERAI];
         case sources.pollinations:
-            return true;
+            return secret_state[SECRET_KEYS.POLLINATIONS];
         case sources.stability:
             return secret_state[SECRET_KEYS.STABILITY];
         case sources.huggingface:

     [event_types.SECRET_WRITTEN, event_types.SECRET_DELETED, event_types.SECRET_ROTATED].forEach(event => {
         eventSource.on(event, async (/** @type {string} */ key) => {
-            switch (key) {
+            const keySourceMap = {
-                case SECRET_KEYS.BFL:
+                [sources.bfl]: SECRET_KEYS.BFL,
-                case SECRET_KEYS.FALAI:
+                [sources.falai]: SECRET_KEYS.FALAI,
-                case SECRET_KEYS.STABILITY:
+                [sources.stability]: SECRET_KEYS.STABILITY,
-                case SECRET_KEYS.AIMLAPI:
+                [sources.aimlapi]: SECRET_KEYS.AIMLAPI,
-                case SECRET_KEYS.COMFY_RUNPOD:
+                [sources.comfy]: SECRET_KEYS.COMFY_RUNPOD,
-                    await loadSettingOptions();
+                [sources.pollinations]: SECRET_KEYS.POLLINATIONS,
-                    break;
+            };
+            const shouldReloadOptions = Object.entries(keySourceMap).some(([k, v]) => k === extension_settings.sd.source && v === key);
+            if (!shouldReloadOptions) {
+                return;
+            }
+            await loadSettingOptions();
         });
     });

public/scripts/extensions/stable-diffusion/settings.html+8 -3 Viewed

                 </div>
             </div>
             <div data-sd-source="pollinations">
-                <p>
+                <a href="https://enter.pollinations.ai">Pollinations.ai</a>
-                    <a href="https://pollinations.ai">Pollinations.ai</a>
+                <div class="flex-container flexnowrap alignItemsBaseline marginBot5">
-                </p>
+                    <strong class="flex1" data-i18n="API Key">API Key</strong>
+                    <div id="sd_pollinations_key" class="menu_button menu_button_icon manage-api-keys" data-key="api_key_pollinations">
+                        <i class="fa-fw fa-solid fa-key"></i>
+                        <span data-i18n="Click to set">Click to set</span>
+                    </div>
+                </div>
                 <div class="flex-container">
                     <label class="flex1 checkbox_label" for="sd_pollinations_enhance" data-i18n="[title]Enables prompt enhancing (passes prompts through an LLM to add detail)." title="Enables prompt enhancing (passes prompts through an LLM to add detail).">
                         <input id="sd_pollinations_enhance" type="checkbox" />

public/scripts/openai.js+2 -5 Viewed

@@ -2707,10 +2707,6 @@ export async function createGenerationParameters(settings, model, type, messages
2707	}	2707	}
2708	}	2708	}
2709		2709
2710	if (settings.chat_completion_source === chat_completion_sources.POLLINATIONS) {
2711	delete generate_data.max_tokens;
2712	}
2713
2714	// https://docs.electronhub.ai/api-reference/chat/completions	2710	// https://docs.electronhub.ai/api-reference/chat/completions
2715	if (settings.chat_completion_source === chat_completion_sources.ELECTRONHUB) {	2711	if (settings.chat_completion_source === chat_completion_sources.ELECTRONHUB) {
2716	generate_data.top_k = Number(settings.top_k_openai);	2712	generate_data.top_k = Number(settings.top_k_openai);

         [chat_completion_sources.AZURE_OPENAI]: { key: SECRET_KEYS.AZURE_OPENAI, selector: '#api_key_azure_openai', proxy: false },
         [chat_completion_sources.ZAI]: { key: SECRET_KEYS.ZAI, selector: '#api_key_zai', proxy: true },
         [chat_completion_sources.CHUTES]: { key: SECRET_KEYS.CHUTES, selector: '#api_key_chutes', proxy: false },
+        [chat_completion_sources.POLLINATIONS]: { key: SECRET_KEYS.POLLINATIONS, selector: '#api_key_pollinations', proxy: false },
     };
     // Vertex AI Express version - use API key

         case chat_completion_sources.ELECTRONHUB:
             return (Array.isArray(model_list) && model_list.find(m => m.id === oai_settings.electronhub_model)?.metadata?.vision);
         case chat_completion_sources.POLLINATIONS:
-            return (Array.isArray(model_list) && model_list.find(m => m.id === oai_settings.pollinations_model)?.vision);
+            return (Array.isArray(model_list) && model_list.find(m => m.id === oai_settings.pollinations_model)?.input_modalities?.includes('image'));
         case chat_completion_sources.COMETAPI:
             return true;
         case chat_completion_sources.MOONSHOT:

public/scripts/secrets.js+3 -0 Viewed

     ZAI: 'api_key_zai',
     SILICONFLOW: 'api_key_siliconflow',
     ELEVENLABS: 'api_key_elevenlabs',
+    POLLINATIONS: 'api_key_pollinations',
 };
 const FRIENDLY_NAMES = {

     [SECRET_KEYS.ZAI]: 'Z.AI',
     [SECRET_KEYS.SILICONFLOW]: 'SiliconFlow',
     [SECRET_KEYS.ELEVENLABS]: 'ElevenLabs TTS',
+    [SECRET_KEYS.POLLINATIONS]: 'Pollinations',
 };
 const INPUT_MAP = {

     [SECRET_KEYS.ZAI]: '#api_key_zai',
     [SECRET_KEYS.SILICONFLOW]: '#api_key_siliconflow',
     [SECRET_KEYS.COMFY_RUNPOD]: '#api_key_comfy_runpod',
+    [SECRET_KEYS.POLLINATIONS]: '#api_key_pollinations',
 };
 const getLabel = () => moment().format('L LT');

src/endpoints/backends/chat-completions.js+16 -12 Viewed

 const API_DEEPSEEK = 'https://api.deepseek.com/beta';
 const API_XAI = 'https://api.x.ai/v1';
 const API_AIMLAPI = 'https://api.aimlapi.com/v1';
-const API_POLLINATIONS = 'https://text.pollinations.ai/openai';
+const API_POLLINATIONS = 'https://gen.pollinations.ai/v1';
 const API_MOONSHOT = 'https://api.moonshot.ai/v1';
 const API_FIREWORKS = 'https://api.fireworks.ai/inference/v1';
 const API_COMETAPI = 'https://api.cometapi.com/v1';

             apiKey = readSecret(request.user.directories, SECRET_KEYS.AIMLAPI);
             headers = { ...AIMLAPI_HEADERS };
         } else if (request.body.chat_completion_source === CHAT_COMPLETION_SOURCES.POLLINATIONS) {
-            apiUrl = 'https://text.pollinations.ai';
+            apiUrl = 'https://gen.pollinations.ai/text';
-            apiKey = 'NONE';
+            apiKey = readSecret(request.user.directories, SECRET_KEYS.POLLINATIONS);
             headers = {};
         } else if (request.body.chat_completion_source === CHAT_COMPLETION_SOURCES.GROQ) {
             apiUrl = API_GROQ;

+            }
         } else if (request.body.chat_completion_source === CHAT_COMPLETION_SOURCES.POLLINATIONS) {
             apiUrl = API_POLLINATIONS;
-            apiKey = 'NONE';
+            apiKey = readSecret(request.user.directories, SECRET_KEYS.POLLINATIONS);
-            headers = {
+            headers = {};
-                'Authorization': '',
-            };
             bodyParams = {
                 reasoning_effort: request.body.reasoning_effort,
-                private: true,
-                referrer: 'sillytavern',
                 seed: request.body.seed ?? Math.floor(Math.random() * 99999999),
             };
             if (request.body.json_schema) {
-                setJsonObjectFormat(bodyParams, request.body.messages, request.body.json_schema);
+                bodyParams['response_format'] = {
+                    type: 'json_schema',
+                    json_schema: {
+                        schema: request.body.json_schema.value,
+                    },
+                };
+            }
         } else if (request.body.chat_completion_source === CHAT_COMPLETION_SOURCES.MOONSHOT) {
             apiUrl = new URL(request.body.reverse_proxy || API_MOONSHOT).toString();

 multimodalModels.post('/pollinations', async (_req, res) => {
     try {
-        const response = await fetch('https://text.pollinations.ai/models');
+        const response = await fetch('https://gen.pollinations.ai/models');
         if (!response.ok) {
             return res.json([]);

             return res.json([]);
+        }
-        const multimodalModels = data.filter(m => m?.vision).map(m => m.name);
+        const multimodalModels = data
+            .filter(m => Array.isArray(m?.input_modalities))
+            .filter(m => m.input_modalities.includes('image'))
+            .map(m => m.name);
         return res.json(multimodalModels);
     } catch (error) {
         console.error(error);

src/endpoints/openai.js+7 -3 Viewed

             bodyParams.max_tokens = 4096; // default is 1024
+        }
-        const noKeyTypes = ['custom', 'ooba', 'koboldcpp', 'vllm', 'llamacpp', 'pollinations'];
+        if (request.body.api === 'pollinations') {
+            key = readSecret(request.user.directories, SECRET_KEYS.POLLINATIONS);
+            bodyParams.seed = Math.floor(Math.random() * Math.pow(2, 32));
+        }
+        const noKeyTypes = ['custom', 'ooba', 'koboldcpp', 'vllm', 'llamacpp'];
         if (!key && !request.body.reverse_proxy && !noKeyTypes.includes(request.body.api)) {
             console.warn('No key found for API', request.body.api);
             return response.sendStatus(400);

@@ -173,8 +178,7 @@ router.post('/caption-image', async (request, response) => {
173	}	178	}
174		179
175	if (request.body.api === 'pollinations') {	180	if (request.body.api === 'pollinations') {
176	headers = { Authorization: '' };	181	apiUrl = 'https://gen.pollinations.ai/v1/chat/completions';
177	apiUrl = 'https://text.pollinations.ai/openai/chat/completions';
178	}	182	}
179		183
180	if (request.body.api === 'moonshot' && !request.body.reverse_proxy) {	184	if (request.body.api === 'moonshot' && !request.body.reverse_proxy) {

src/endpoints/secrets.js+1 -0 Viewed

     ZAI: 'api_key_zai',
     SILICONFLOW: 'api_key_siliconflow',
     ELEVENLABS: 'api_key_elevenlabs',
+    POLLINATIONS: 'api_key_pollinations',
 };
 /**

src/endpoints/speech.js+39 -8 Viewed

     try {
         const model = req.body.model || 'openai-audio';
-        const response = await fetch('https://text.pollinations.ai/models');
+        const response = await fetch('https://gen.pollinations.ai/text/models');
         if (!response.ok) {
             throw new Error('Failed to fetch Pollinations models');

 pollinations.post('/generate', async (req, res) => {
     try {
+        const key = readSecret(req.user.directories, SECRET_KEYS.POLLINATIONS);
+        if (!key) {
+            console.warn('No API key saved for Pollinations TTS.');
+            return res.sendStatus(400);
+        }
         const text = req.body.text;
         const model = req.body.model || 'openai-audio';
         const voice = req.body.voice || 'alloy';
-        const url = new URL(`https://text.pollinations.ai/generate/${encodeURIComponent(text)}`);
+        console.debug('Pollinations TTS request', { text, model, voice });
-        url.searchParams.append('model', model);
-        url.searchParams.append('voice', voice);
-        url.searchParams.append('referrer', 'sillytavern');
-        console.info('Pollinations request URL:', url.toString());
-        const response = await fetch(url);
+        const response = await fetch('https://gen.pollinations.ai/v1/chat/completions', {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${key}`,
+                'Content-Type': 'application/json',
+            },
+            body: JSON.stringify({
+                model: model,
+                stream: false,
+                modalities: ['text', 'audio'],
+                seed: Math.floor(Math.random() * Math.pow(2, 32)),
+                audio: {
+                    format: 'mp3',
+                    voice: voice,
+                },
+                messages: [{
+                    role: 'user',
+                    content: text,
+                }],
+            }),
+        });
         if (!response.ok) {
             const text = await response.text();
             throw new Error(`Failed to generate audio from Pollinations: ${text}`);
+        }
+        /** @type {any} */
+        const data = await response.json();
+        const audioData = data?.choices?.[0]?.message?.audio?.data;
+        if (!audioData) {
+            console.warn('Pollinations TTS audio data is missing from the response');
+            return res.sendStatus(500);
+        }
         res.set('Content-Type', 'audio/mpeg');
-        forwardFetchResponse(response, res);
+        return res.send(Buffer.from(audioData, 'base64'));
     } catch (error) {
         console.error(error);
         return res.sendStatus(500);

src/endpoints/stable-diffusion.js+18 -11 Viewed

 import FormData from 'form-data';
 import urlJoin from 'url-join';
 import _ from 'lodash';
+import mime from 'mime-types';
 import { delay, getBasicAuthHeader, isValidUrl, tryParse } from '../util.js';
 import { readSecret, SECRET_KEYS } from './secrets.js';

 pollinations.post('/models', async (_request, response) => {
     try {
-        const modelsUrl = new URL('https://image.pollinations.ai/models');
+        const modelsUrl = new URL('https://gen.pollinations.ai/image/models');
         const result = await fetch(modelsUrl);
         if (!result.ok) {

             throw new Error('Pollinations request failed.');
+        }
-        const models = data.map(x => ({ value: x, text: x }));
+        const models = data.map(x => ({ value: x.name, text: x.name }));
         return response.send(models);
     } catch (error) {
         console.error(error);

 pollinations.post('/generate', async (request, response) => {
     try {
-        const promptUrl = new URL(`https://image.pollinations.ai/prompt/${encodeURIComponent(request.body.prompt)}`);
+        const key = readSecret(request.user.directories, SECRET_KEYS.POLLINATIONS);
+        if (!key) {
+            console.warn('Pollinations API key not found.');
+            return response.sendStatus(400);
+        }
+        const promptUrl = new URL(`https://gen.pollinations.ai/image/${encodeURIComponent(request.body.prompt)}`);
         const params = new URLSearchParams({
             model: String(request.body.model),
             negative_prompt: String(request.body.negative_prompt),
             seed: String(request.body.seed >= 0 ? request.body.seed : Math.floor(Math.random() * 10_000_000)),
             width: String(request.body.width ?? 1024),
             height: String(request.body.height ?? 1024),
-            nologo: String(true),
-            nofeed: String(true),
-            private: String(true),
-            referrer: 'sillytavern',
         });
         if (request.body.enhance) {
             params.set('enhance', String(true));

         console.info('Pollinations request URL:', promptUrl.toString());
-        const result = await fetch(promptUrl);
+        const result = await fetch(promptUrl, {
+            method: 'GET',
+            headers: {
+                'Authorization': `Bearer ${key}`,
+            },
+        });
         if (!result.ok) {
             const text = await result.text();

             throw new Error('Pollinations request failed.');
+        }
+        const format = result.headers.get('Content-Type')?.toString() || 'image/jpeg';
         const buffer = await result.arrayBuffer();
-        const base64 = Buffer.from(buffer).toString('base64');
+        return response.send({ image: Buffer.from(buffer).toString('base64'), format: mime.extension(format) || 'jpg' });
-        return response.send({ image: base64 });
     } catch (error) {
         console.error(error);
         return response.sendStatus(500);

Update Pollinations API (#5060) * Upgrade Pollinations API Done: text, caption To do: TTS, image Fixes #5020 * Update Pollinations TTS to new API * Update Pollinations API for images