add new class to other samples

Author: eavanvalkenburg

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

@@ -3,7 +3,7 @@
 from pathlib import Path
 
 import aiofiles
-from agent_framework import DataContent
+from agent_framework import DataContent, HostedImageGenerationTool
 from agent_framework.azure import AzureAIClient
 from azure.identity.aio import AzureCliCredential
 
@@ -29,12 +29,13 @@ async def main() -> None:
             name="ImageGenAgent",
             instructions="Generate images based on user requirements.",
             tools=[
-                {
-                    "type": "image_generation",
-                    "model": "gpt-image-1-mini",
-                    "quality": "low",
-                    "size": "1024x1024",
-                }
+                HostedImageGenerationTool(
+                    options={
+                        "model": "gpt-image-1-mini",
+                        "quality": "low",
+                        "size": "1024x1024",
+                    }
+                )
             ],
         ) as agent,
     ):

python/samples/getting_started/agents/openai/openai_responses_client_streaming_image_generation.py

@@ -4,7 +4,7 @@
 import base64
 
 import anyio
-from agent_framework import DataContent
+from agent_framework import DataContent, HostedImageGenerationTool
 from agent_framework.openai import OpenAIResponsesClient
 
 """OpenAI Responses Client Streaming Image Generation Example
@@ -45,12 +45,13 @@ async def main():
     agent = OpenAIResponsesClient().create_agent(
         instructions="You are a helpful agent that can generate images.",
         tools=[
-            {
-                "type": "image_generation",
-                "size": "1024x1024",
-                "quality": "high",
-                "partial_images": 3,
-            }
+            HostedImageGenerationTool(
+                options={
+                    "size": "1024x1024",
+                    "quality": "high",
+                    "partial_images": 3,
+                }
+            )
         ],
     )