Skip to content

Commit 6608db6

Browse files
committed
Make agent model deployment name configurable via environment variable
1 parent e549ce0 commit 6608db6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sdk/ai/azure-ai-projects/tests/test_base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
# Copyright (c) Microsoft Corporation.
44
# Licensed under the MIT License.
55
# ------------------------------------
6+
import os
67
import random
78
import re
89
import functools
@@ -100,7 +101,7 @@ class TestBase(AzureRecordedTestCase):
100101
}
101102

102103
test_agents_params = {
103-
"model_deployment_name": "gpt-4o",
104+
"model_deployment_name": os.environ.get("AZURE_AI_PROJECTS_TESTS_AGENTS_MODEL_DEPLOYMENT_NAME") or "gpt-4o",
104105
"agent_name": "agent-for-python-projects-sdk-testing",
105106
}
106107

0 commit comments

Comments
 (0)