agent/agent-framework
2
0
Fork 0
mirror of https://github.com/microsoft/agent-framework.git synced 2026-06-16 21:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Python: Add base_url support to OpenAI settings and improve initialization (#788)

Browse Source 
Co-authored-by: Chris <66376200+crickman@users.noreply.github.com>
This commit is contained in:
Yuge Zhang
2025-09-17 15:56:11 +08:00
committed by GitHub
Unverified
parent a1b0a28f9c
commit 3050c8c8bd
3 changed files with 24 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/main/tests/openai/test_openai_chat_client.py
+16
View File
@@ -80,6 +80,22 @@ def test_init_base_url(openai_unit_test_env: dict[str, str]) -> None:
assert str(open_ai_chat_completion.client.base_url) == "http://localhost:1234/v1/"
def test_init_base_url_from_settings_env() -> None:
"""Test that base_url from OpenAISettings environment variable is properly used."""
# Set environment variable for base_url
with patch.dict(
os.environ,
{
"OPENAI_API_KEY": "dummy",
"OPENAI_CHAT_MODEL_ID": "gpt-5",
"OPENAI_BASE_URL": "https://custom-openai-endpoint.com/v1",
},
):
client = OpenAIChatClient()
assert client.ai_model_id == "gpt-5"
assert str(client.client.base_url) == "https://custom-openai-endpoint.com/v1/"
@pytest.mark.parametrize("exclude_list", [["OPENAI_CHAT_MODEL_ID"]], indirect=True)
def test_init_with_empty_model_id(openai_unit_test_env: dict[str, str]) -> None:
with pytest.raises(ServiceInitializationError):