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,16 @@ def to_azure_ai_tools(
                                 timezone=location.get("timezone"),
                             )
                     azure_tools.append(ws_tool)
+                case HostedImageGenerationTool():
+                    opts = tool.options or {}
+                    azure_tools.append(
+                        ImageGenTool(
+                            model=opts.get("model_id", "gpt-image-1"),
+                            size=opts.get("image_size"),
+                            output_format=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,7 +33,7 @@ async def main() -> None:
             tools=[
                 HostedImageGenerationTool(
                     options={
-                        "model": "gpt-image-1-mini",
+                        "model": "gpt-image-1",
                         "quality": "low",
                         "size": "1024x1024",
                     }
@@ -54,15 +55,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())