From 03062f167555c47cb4fd38bcfc91d5f19f57851b Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Wed, 7 Jan 2026 17:50:25 +0530
Subject: [PATCH] add Image Editing

---
 .../my-website/docs/providers/azure_ai_img.md | 99 ++++++++++++++++++-
 1 file changed, 98 insertions(+), 1 deletion(-)

diff --git a/docs/my-website/docs/providers/azure_ai_img.md b/docs/my-website/docs/providers/azure_ai_img.md
index 6434856fd2a..513bbe858d0 100644
--- a/docs/my-website/docs/providers/azure_ai_img.md
+++ b/docs/my-website/docs/providers/azure_ai_img.md
@@ -12,7 +12,7 @@ Azure AI provides powerful image generation capabilities using FLUX models from
 | Description | Azure AI Image Generation uses FLUX models to generate high-quality images from text descriptions. |
 | Provider Route on LiteLLM | `azure_ai/` |
 | Provider Doc | [Azure AI FLUX Models ↗](https://techcommunity.microsoft.com/blog/azure-ai-foundry-blog/black-forest-labs-flux-1-kontext-pro-and-flux1-1-pro-now-available-in-azure-ai-f/4434659) |
-| Supported Operations | [`/images/generations`](#image-generation) |
+| Supported Operations | [`/images/generations`](#image-generation), [`/images/edits`](#image-editing) |
 
 ## Setup
 
@@ -275,6 +275,103 @@ curl --location 'http://localhost:4000/v1/images/generations' \
 </TabItem>
 </Tabs>
 
+## Image Editing
+
+FLUX 2 Pro supports image editing by passing an input image along with a prompt describing the desired modifications.
+
+### Usage - LiteLLM Python SDK
+
+<Tabs>
+<TabItem value="basic-edit" label="Basic Image Edit">
+
+```python showLineNumbers title="Basic Image Editing with FLUX 2 Pro"
+import litellm
+import os
+
+# Set your API credentials
+os.environ["AZURE_AI_API_KEY"] = "your-api-key-here"
+os.environ["AZURE_AI_API_BASE"] = "your-azure-ai-endpoint"  # e.g., https://litellm-ci-cd-prod.services.ai.azure.com
+
+# Edit an existing image
+response = litellm.image_edit(
+    model="azure_ai/flux.2-pro",
+    prompt="Add a red hat to the subject",
+    image=open("input_image.png", "rb"),
+    api_base=os.environ["AZURE_AI_API_BASE"],
+    api_key=os.environ["AZURE_AI_API_KEY"],
+    api_version="preview",
+)
+
+print(response.data[0].b64_json)  # FLUX 2 returns base64 encoded images
+```
+
+</TabItem>
+
+<TabItem value="async-edit" label="Async Image Edit">
+
+```python showLineNumbers title="Async Image Editing"
+import litellm
+import asyncio
+import os
+
+async def edit_image():
+    os.environ["AZURE_AI_API_KEY"] = "your-api-key-here"
+    os.environ["AZURE_AI_API_BASE"] = "your-azure-ai-endpoint"
+    
+    response = await litellm.aimage_edit(
+        model="azure_ai/flux.2-pro",
+        prompt="Change the background to a sunset beach",
+        image=open("input_image.png", "rb"),
+        api_base=os.environ["AZURE_AI_API_BASE"],
+        api_key=os.environ["AZURE_AI_API_KEY"],
+        api_version="preview",
+    )
+    
+    return response
+
+asyncio.run(edit_image())
+```
+
+</TabItem>
+</Tabs>
+
+### Usage - LiteLLM Proxy Server
+
+<Tabs>
+<TabItem value="curl-edit" label="cURL">
+
+```bash showLineNumbers title="Image Edit via Proxy - cURL"
+curl --location 'http://localhost:4000/v1/images/edits' \
+--header 'Authorization: Bearer sk-1234' \
+--form 'model="azure-flux-2-pro"' \
+--form 'prompt="Add sunglasses to the person"' \
+--form 'image=@"input_image.png"'
+```
+
+</TabItem>
+
+<TabItem value="openai-sdk-edit" label="OpenAI SDK">
+
+```python showLineNumbers title="Image Edit via Proxy - OpenAI SDK"
+from openai import OpenAI
+
+client = OpenAI(
+    base_url="http://localhost:4000",
+    api_key="sk-1234"
+)
+
+response = client.images.edit(
+    model="azure-flux-2-pro",
+    prompt="Make the sky more dramatic with storm clouds",
+    image=open("input_image.png", "rb"),
+)
+
+print(response.data[0].b64_json)
+```
+
+</TabItem>
+</Tabs>
+
 ## Supported Parameters
 
 Azure AI Image Generation supports the following OpenAI-compatible parameters: