image generation using workers AI

Author: siddharthsambharia-portkey

integrations/llms/workers-ai.mdx

@@ -91,6 +91,128 @@ Use the Portkey instance to send requests to Workers AI. You can also override t
 
 
 
+### **4\. Invoke Image Generation with** Workers AI
+
+Use the Portkey instance to send requests to Workers AI. You can also override the virtual key directly in the API call if needed.
+
+Portkey supports the OpenAI signature to make text-to-image requests.
+
+<Tabs>
+  <Tab title="NodeJS">
+```js
+import Portkey from 'portkey-ai';
+
+// Initialize the Portkey client
+const portkey = new Portkey({
+    apiKey: "PORTKEY_API_KEY",  // Replace with your Portkey API key
+    virtualKey: "VIRTUAL_KEY"   // Add your provider's virtual key
+});
+
+async function main() {
+  const image = await portkey.images.generate({
+    model: "image_model_name",
+    prompt: "Lucy in the sky with diamonds"
+  });
+
+  console.log(image.data);
+}
+
+main();
+```
+  </Tab>
+  <Tab title="Python">
+
+```py
+from portkey_ai import Portkey
+from IPython.display import display, Image
+
+# Initialize the Portkey client
+portkey = Portkey(
+    api_key="PORTKEY_API_KEY",  # Replace with your Portkey API key
+    virtual_key="VIRTUAL_KEY"   # Add your provider's virtual key
+)
+
+image = portkey.images.generate(
+  model="image_model_name",
+  prompt="Lucy in the sky with diamonds"
+)
+
+# Display the image
+display(Image(url=image.data[0].url))
+```
+  </Tab>
+  <Tab title="OpenAI NodeJS">
+
+```js
+import OpenAI from 'openai'; // We're using the v4 SDK
+import { PORTKEY_GATEWAY_URL, createHeaders } from 'portkey-ai'
+
+const openai = new OpenAI({
+  apiKey: 'WORKERS_AI_API_KEY', // defaults to process.env["WORKERS_AI_API_KEY"],
+  baseURL: PORTKEY_GATEWAY_URL,
+  defaultHeaders: createHeaders({
+    provider: "openai",
+    apiKey: "PORTKEY_API_KEY" // defaults to process.env["PORTKEY_API_KEY"]
+  })
+});
+
+async function main() {
+  const image = await openai.images.generate({
+    model: "image_model_name",
+    prompt: "Lucy in the sky with diamonds"
+  });
+
+  console.log(image.data);
+}
+
+main();
+```
+  </Tab>
+  <Tab title="OpenAI Python">
+
+```py
+from openai import OpenAI
+from portkey_ai import PORTKEY_GATEWAY_URL, createHeaders
+from IPython.display import display, Image
+
+client = OpenAI(
+    api_key='WORKERS_AI_API_KEY',
+    base_url=PORTKEY_GATEWAY_URL,
+    default_headers=createHeaders(
+        provider="openai",
+        api_key="PORTKEY_API_KEY"
+    )
+)
+
+image = client.images.generate(
+  model="image_model_name",
+  prompt="Lucy in the sky with diamonds",
+  n=1,
+  size="1024x1024"
+)
+
+# Display the image
+display(Image(url=image.data[0].url))
+```
+  </Tab>
+<Tab title="cURL">
+
+```sh
+curl "https://api.portkey.ai/v1/images/generations" \
+  -H "Content-Type: application/json" \
+  -H "x-portkey-api-key: $PORTKEY_API_KEY" \
+  -H "x-portkey-virtual-key: $WORKERS_AI_VIRTUAL_KEY" \
+  -d '{
+    "model": "image_model_name",
+    "prompt": "Lucy in the sky with diamonds"
+  }'
+```
+  </Tab>
+</Tabs>
+
+
+
+
 ## Managing Workers AI Prompts
 
 You can manage all prompts to Workers AI in the [Prompt Library](/product/prompt-library). All the current models of Workers AI are supported and you can easily start testing different prompts.