Python: Azure AI mapping HostedImageGenerationTool to ImageGenTool

python/packages/azure-ai/agent_framework_azure_ai/_shared.py

@@ -10,6 +10,7 @@
     HostedCodeInterpreterTool,
     HostedFileContent,
     HostedFileSearchTool,
+    HostedImageGenerationTool,
     HostedMCPTool,
     HostedVectorStoreContent,
     HostedWebSearchTool,
@@ -31,6 +32,7 @@
     CodeInterpreterTool,
     CodeInterpreterToolAuto,
     FunctionTool,
+    ImageGenTool,
     MCPTool,
     ResponseTextFormatConfigurationJsonObject,
     ResponseTextFormatConfigurationJsonSchema,
@@ -442,6 +444,20 @@ def to_azure_ai_tools(
                                 timezone=location.get("timezone"),
                             )
                     azure_tools.append(ws_tool)
+                case HostedImageGenerationTool():
+                    opts = tool.options or {}
+                    # Azure ImageGenTool requires the constant model "gpt-image-1"
+                    # Cast optional fields to expected Literal unions for mypy compliance
+                    azure_tools.append(
+                        ImageGenTool(
+                            model="gpt-image-1",
+                            size=cast(
+                                Literal["1024x1024", "1024x1536", "1536x1024", "auto"] | None, opts.get("image_size")
+                            ),
+                            output_format=cast(Literal["png", "webp", "jpeg"] | None, opts.get("media_type")),
+                            partial_images=opts.get("streaming_count"),
+                        )
+                    )
                 case _:
                     logger.debug("Unsupported tool passed (type: %s)", type(tool))
         else:

python/samples/getting_started/agents/azure_ai/azure_ai_with_image_generation.py

@@ -1,9 +1,10 @@
 # Copyright (c) Microsoft. All rights reserved.
 import asyncio
+import tempfile
 from pathlib import Path
 
 import aiofiles
-from agent_framework import DataContent, HostedImageGenerationTool
+from agent_framework import HostedImageGenerationTool, ImageGenerationToolResultContent
 from agent_framework.azure import AzureAIProjectAgentProvider
 from azure.identity.aio import AzureCliCredential
 
@@ -32,8 +33,7 @@ async def main() -> None:
             tools=[
                 HostedImageGenerationTool(
                     options={
-                        "model": "gpt-image-1-mini",
-                        "quality": "low",
+                        "model": "gpt-image-1",
                         "size": "1024x1024",
                     }
                 )
@@ -54,15 +54,14 @@ async def main() -> None:
         # Save the image to a file
         print("Downloading generated image...")
         image_data = [
-            content
+            content.outputs
             for content in result.messages[0].contents
-            if isinstance(content, DataContent) and content.media_type == "image/png"
+            if isinstance(content, ImageGenerationToolResultContent) and content.outputs is not None
         ]
         if image_data and image_data[0]:
-            # Save to the same directory as this script
+            # Save to the OS temporary directory
             filename = "microsoft.png"
-            current_dir = Path(__file__).parent.resolve()
-            file_path = current_dir / filename
+            file_path = Path(tempfile.gettempdir()) / filename
             async with aiofiles.open(file_path, "wb") as f:
                 await f.write(image_data[0].get_data_bytes())