agents tests - support model overrides

paulbatum · paulbatum · commit 9c0b240e21e4 · 2025-12-08T10:40:15.000-08:00
diff --git a/sdk/ai/azure-ai-projects/.env.template b/sdk/ai/azure-ai-projects/.env.template
@@ -50,6 +50,8 @@ AZURE_AI_PROJECTS_TESTS_PROJECT_ENDPOINT=
 
 # Used in Agent tests
 AZURE_AI_PROJECTS_TESTS_AGENTS_PROJECT_ENDPOINT=
+AZURE_AI_PROJECTS_TESTS_AGENTS_MODEL_DEPLOYMENT_NAME=
+AZURE_AI_PROJECTS_TESTS_AGENTS_IMAGE_MODEL_DEPLOYMENT_NAME=
 
 # Used in tracing tests
 AZURE_AI_PROJECTS_TESTS_TRACING_PROJECT_ENDPOINT=
diff --git a/sdk/ai/azure-ai-projects/tests/test_base.py b/sdk/ai/azure-ai-projects/tests/test_base.py
@@ -3,6 +3,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 # ------------------------------------
+import os
 import random
 import re
 import functools
@@ -84,12 +85,12 @@ class TestBase(AzureRecordedTestCase):
     }
 
     test_agents_params = {
-        "model_deployment_name": "gpt-4o",
+        "model_deployment_name": os.environ.get("AZURE_AI_PROJECTS_TESTS_AGENTS_MODEL_DEPLOYMENT_NAME") or "gpt-4o",
         "agent_name": "agent-for-python-projects-sdk-testing",
     }
 
     test_agents_tools_params = {
-        "image_generation_model_deployment_name": "gpt-image-1-mini",
+        "image_generation_model_deployment_name": os.environ.get("AZURE_AI_PROJECTS_TESTS_AGENTS_IMAGE_MODEL_DEPLOYMENT_NAME") or "gpt-image-1-mini",
     }
 
     test_inference_params = {

Original file line number	Diff line number	Diff line change
`@@ -3,6 +3,7 @@`
`3`	`3`	`# Copyright (c) Microsoft Corporation.`
`4`	`4`	`# Licensed under the MIT License.`
`5`	`5`	`# ------------------------------------`
	`6`	`+import os`
`6`	`7`	`import random`
`7`	`8`	`import re`
`8`	`9`	`import functools`
`@@ -84,12 +85,12 @@ class TestBase(AzureRecordedTestCase):`
`84`	`85`	`}`
`85`	`86`
`86`	`87`	`test_agents_params = {`
`87`		`- "model_deployment_name": "gpt-4o",`
	`88`	`+ "model_deployment_name": os.environ.get("AZURE_AI_PROJECTS_TESTS_AGENTS_MODEL_DEPLOYMENT_NAME") or "gpt-4o",`
`88`	`89`	`"agent_name": "agent-for-python-projects-sdk-testing",`
`89`	`90`	`}`
`90`	`91`
`91`	`92`	`test_agents_tools_params = {`
`92`		`- "image_generation_model_deployment_name": "gpt-image-1-mini",`
	`93`	`+ "image_generation_model_deployment_name": os.environ.get("AZURE_AI_PROJECTS_TESTS_AGENTS_IMAGE_MODEL_DEPLOYMENT_NAME") or "gpt-image-1-mini",`
`93`	`94`	`}`
`94`	`95`
`95`	`96`	`test_inference_params = {`