feat: add Pollinations.ai image plugin #308

TBXark · Nov 28, 2024 · 16d043a · 16d043a
1 parent 6836b5c
commit 16d043a
Show file tree

Hide file tree

Showing 5 changed files with 50 additions and 11 deletions.
diff --git a/dist/buildinfo.json b/dist/buildinfo.json
diff --git a/dist/index.js b/dist/index.js
diff --git a/packages/lib/core/src/config/version.ts b/packages/lib/core/src/config/version.ts
@@ -1,2 +1,2 @@
-export const BUILD_TIMESTAMP = 1732293307;
-export const BUILD_VERSION = '2f6b657';
+export const BUILD_TIMESTAMP = 1732772382;
+export const BUILD_VERSION = '6836b5c';
diff --git a/packages/lib/core/src/telegram/command/index.ts b/packages/lib/core/src/telegram/command/index.ts
@@ -80,10 +80,9 @@ async function handlePluginCommand(message: Telegram.Message, command: string, r
             DATA,
             ENV: ENV.PLUGINS_ENV,
         });
-        if (type === 'image') {
-            return sender.sendPhoto(content);
-        }
         switch (type) {
+            case 'image':
+                return sender.sendPhoto(content);
             case 'html':
                 return sender.sendRichText(content, 'HTML');
             case 'markdown':

diff --git a/plugins/pollinations.json b/plugins/pollinations.json
@@ -0,0 +1,30 @@
+{
+  "url": "https://image.pollinations.ai/prompt/{{DATA}}",
+  "method": "GET",
+  "headers": {
+    "accept": "application/dns-json"
+  },
+  "input": {
+    "type": "text",
+    "required": true
+  },
+  "query": {
+    "width": "{{ENV.POLLINATIONS_IMAGE_WIDTH}}",
+    "height": "{{ENV.POLLINATIONS_IMAGE_HEIGHT}}",
+    "model": "{{ENV.POLLINATIONS_MODEL}}",
+    "nologo": "{{ENV.POLLINATIONS_NOLOGO}}",
+    "private": "{{ENV.POLLINATIONS_PRIVATE}}",
+    "enhance": "{{ENV.POLLINATIONS_ENHANCE}}"
+  },
+  "response": {
+    "content": {
+      "input_type": "blob",
+      "output_type": "image"
+    },
+    "error": {
+      "input_type": "text",
+      "output_type": "text",
+      "output": "Error: {{.}}"
+    }
+  }
+}