From ba008cf4c00f9736ffdcb327c3a01c21db1fae5b Mon Sep 17 00:00:00 2001
From: Eduard van Valkenburg <eavanvalkenburg@users.noreply.github.com>
Date: Mon, 28 Jul 2025 11:12:28 +0200
Subject: [PATCH] Merge tests (#247)

* removed hardcoded model names from tests

* add debug line
---
 python/packages/azure/tests/test_azure_chat_client.py    | 9 +++++----
 .../main/tests/openai/test_openai_chat_client.py         | 8 ++++----
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/python/packages/azure/tests/test_azure_chat_client.py b/python/packages/azure/tests/test_azure_chat_client.py
index 614f7e8471..a2fe54604f 100644
--- a/python/packages/azure/tests/test_azure_chat_client.py
+++ b/python/packages/azure/tests/test_azure_chat_client.py
@@ -646,9 +646,10 @@ def get_story_text() -> str:
 @skip_if_azure_integration_tests_disabled
 async def test_azure_openai_chat_client_response() -> None:
     """Test Azure OpenAI chat completion responses."""
-    azure_chat_client = AzureChatClient(deployment_name="gpt-4o")
+    azure_chat_client = AzureChatClient()
 
     assert isinstance(azure_chat_client, ChatClient)
+    print(f"Using Azure OpenAI endpoint: {azure_chat_client.client._azure_endpoint}")  # noqa
 
     messages: list[ChatMessage] = []
     messages.append(
@@ -673,7 +674,7 @@ async def test_azure_openai_chat_client_response() -> None:
 @skip_if_azure_integration_tests_disabled
 async def test_azure_openai_chat_client_response_tools() -> None:
     """Test AzureOpenAI chat completion responses."""
-    azure_chat_client = AzureChatClient(deployment_name="gpt-4o")
+    azure_chat_client = AzureChatClient()
 
     assert isinstance(azure_chat_client, ChatClient)
 
@@ -695,7 +696,7 @@ async def test_azure_openai_chat_client_response_tools() -> None:
 @skip_if_azure_integration_tests_disabled
 async def test_azure_openai_chat_client_streaming() -> None:
     """Test Azure OpenAI chat completion responses."""
-    azure_chat_client = AzureChatClient(deployment_name="gpt-4o")
+    azure_chat_client = AzureChatClient()
 
     assert isinstance(azure_chat_client, ChatClient)
 
@@ -728,7 +729,7 @@ async def test_azure_openai_chat_client_streaming() -> None:
 @skip_if_azure_integration_tests_disabled
 async def test_azure_openai_chat_client_streaming_tools() -> None:
     """Test AzureOpenAI chat completion responses."""
-    azure_chat_client = AzureChatClient(deployment_name="gpt-4o")
+    azure_chat_client = AzureChatClient()
 
     assert isinstance(azure_chat_client, ChatClient)
 
diff --git a/python/packages/main/tests/openai/test_openai_chat_client.py b/python/packages/main/tests/openai/test_openai_chat_client.py
index 3f3f0c4ecb..8ba5efc440 100644
--- a/python/packages/main/tests/openai/test_openai_chat_client.py
+++ b/python/packages/main/tests/openai/test_openai_chat_client.py
@@ -127,7 +127,7 @@ def get_story_text() -> str:
 @skip_if_openai_integration_tests_disabled
 async def test_openai_chat_completion_response() -> None:
     """Test OpenAI chat completion responses."""
-    openai_chat_client = OpenAIChatClient(ai_model_id="gpt-4.1-mini")
+    openai_chat_client = OpenAIChatClient()
 
     assert isinstance(openai_chat_client, ChatClient)
 
@@ -154,7 +154,7 @@ async def test_openai_chat_completion_response() -> None:
 @skip_if_openai_integration_tests_disabled
 async def test_openai_chat_completion_response_tools() -> None:
     """Test OpenAI chat completion responses."""
-    openai_chat_client = OpenAIChatClient(ai_model_id="gpt-4.1-mini")
+    openai_chat_client = OpenAIChatClient()
 
     assert isinstance(openai_chat_client, ChatClient)
 
@@ -176,7 +176,7 @@ async def test_openai_chat_completion_response_tools() -> None:
 @skip_if_openai_integration_tests_disabled
 async def test_openai_chat_client_streaming() -> None:
     """Test Azure OpenAI chat completion responses."""
-    openai_chat_client = OpenAIChatClient(ai_model_id="gpt-4.1-mini")
+    openai_chat_client = OpenAIChatClient()
 
     assert isinstance(openai_chat_client, ChatClient)
 
@@ -209,7 +209,7 @@ async def test_openai_chat_client_streaming() -> None:
 @skip_if_openai_integration_tests_disabled
 async def test_openai_chat_client_streaming_tools() -> None:
     """Test AzureOpenAI chat completion responses."""
-    openai_chat_client = OpenAIChatClient(ai_model_id="gpt-4.1-mini")
+    openai_chat_client = OpenAIChatClient()
 
     assert isinstance(openai_chat_client, ChatClient)