image model needs to passed into tool definition

paulbatum · paulbatum · commit acfde07c8c8e · 2025-12-04T11:40:59.000-08:00
diff --git a/sdk/ai/azure-ai-projects/tests/agents/tools/test_agent_image_generation.py b/sdk/ai/azure-ai-projects/tests/agents/tools/test_agent_image_generation.py
@@ -73,7 +73,7 @@ def test_agent_image_generation(self, **kwargs):
                 definition=PromptAgentDefinition(
                     model=model,
                     instructions="Generate images based on user prompts",
-                    tools=[ImageGenTool(quality="low", size="1024x1024")],
+                    tools=[ImageGenTool(quality="low", size="1024x1024", model=image_model_deployment)],
                 ),
                 description="Agent for testing image generation.",
             )

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ def test_agent_image_generation(self, **kwargs):`
`73`	`73`	`definition=PromptAgentDefinition(`
`74`	`74`	`model=model,`
`75`	`75`	`instructions="Generate images based on user prompts",`
`76`		`- tools=[ImageGenTool(quality="low", size="1024x1024")],`
	`76`	`+ tools=[ImageGenTool(quality="low", size="1024x1024", model=image_model_deployment)],`
`77`	`77`	`),`
`78`	`78`	`description="Agent for testing image generation.",`
`79`	`79`	`)`