diff --git a/sdk/ai/azure-ai-projects/.env.template b/sdk/ai/azure-ai-projects/.env.template
index 9e4c3ca99a21..647e445d9d90 100644
--- 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
index 1a9a0cd8e5f0..a7a8b6d318d1 100644
--- 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 = {