updated code to fix int tests

Author: eavanvalkenburg

.github/workflows/python-integration-tests.yml

@@ -99,7 +99,7 @@ jobs:
       AZURE_OPENAI_CHAT_DEPLOYMENT_NAME: ${{ vars.AZUREOPENAI__CHATDEPLOYMENTNAME }}
       AZURE_OPENAI_RESPONSES_DEPLOYMENT_NAME: ${{ vars.AZUREOPENAI__RESPONSESDEPLOYMENTNAME }}
       AZURE_OPENAI_DEPLOYMENT_NAME: ${{ vars.AZUREOPENAI__RESPONSESDEPLOYMENTNAME }}
-      AZURE_OPENAI_EMBEDDING_DEPLOYMENT_NAME: ${{ vars.AZUREOPENAI__EMBEDDINGDEPLOYMENTNAME }}
+      AZURE_OPENAI_EMBEDDING_DEPLOYMENT_NAME: ${{ vars.AZURE_OPENAI_EMBEDDING_DEPLOYMENT_NAME }}
       AZURE_OPENAI_ENDPOINT: ${{ vars.AZUREOPENAI__ENDPOINT }}
       AZURE_OPENAI_API_VERSION: ${{ vars.AZUREOPENAI__APIVERSION || '2024-12-01-preview' }}
     defaults:
@@ -251,8 +251,8 @@ jobs:
           --retries 2 --retry-delay 5
 
   # Azure AI integration tests
-  python-tests-azure-ai:
-    name: Python Integration Tests - Azure AI
+  python-tests-foundry:
+    name: Python Integration Tests - Foundry
     runs-on: ubuntu-latest
     environment: integration
     timeout-minutes: 60

.github/workflows/python-merge-tests.yml

@@ -198,7 +198,7 @@ jobs:
       AZURE_OPENAI_CHAT_DEPLOYMENT_NAME: ${{ vars.AZUREOPENAI__CHATDEPLOYMENTNAME }}
       AZURE_OPENAI_RESPONSES_DEPLOYMENT_NAME: ${{ vars.AZUREOPENAI__RESPONSESDEPLOYMENTNAME }}
       AZURE_OPENAI_DEPLOYMENT_NAME: ${{ vars.AZUREOPENAI__RESPONSESDEPLOYMENTNAME }}
-      AZURE_OPENAI_EMBEDDING_DEPLOYMENT_NAME: ${{ vars.AZUREOPENAI__EMBEDDINGDEPLOYMENTNAME }}
+      AZURE_OPENAI_EMBEDDING_DEPLOYMENT_NAME: ${{ vars.AZURE_OPENAI_EMBEDDING_DEPLOYMENT_NAME }}
       AZURE_OPENAI_ENDPOINT: ${{ vars.AZUREOPENAI__ENDPOINT }}
       AZURE_OPENAI_API_VERSION: ${{ vars.AZUREOPENAI__APIVERSION || '2024-12-01-preview' }}
     defaults:
@@ -389,8 +389,8 @@ jobs:
           fail-on-empty: false
           title: Functions integration test results
 
-  python-tests-azure-ai:
-    name: Python Tests - Azure AI
+  python-tests-foundry:
+    name: Python Integration Tests - Foundry
     needs: paths-filter
     if: >
       github.event_name != 'pull_request' &&

python/packages/openai/agent_framework_openai/_chat_client.py

@@ -431,6 +431,7 @@ def __init__(
             env_file_encoding=env_file_encoding,
             openai_model_fields=("responses_model", "model"),
             azure_deployment_fields=("responses_deployment_name", "deployment_name"),
+            responses_mode=True,
         )
 
         self.client = client

python/packages/openai/agent_framework_openai/_chat_completion_client.py

@@ -84,7 +84,7 @@
 
 logger = logging.getLogger("agent_framework.openai")
 
-DEFAULT_AZURE_OPENAI_CHAT_COMPLETION_API_VERSION = "2024-10-21"
+DEFAULT_AZURE_OPENAI_CHAT_COMPLETION_API_VERSION = "2024-12-01-preview"
 
 ResponseModelBoundT = TypeVar("ResponseModelBoundT", bound=BaseModel)
 ResponseModelT = TypeVar("ResponseModelT", bound=BaseModel | None, default=None)

python/packages/openai/agent_framework_openai/_shared.py

@@ -194,6 +194,7 @@ def load_openai_service_settings(
     env_file_encoding: str | None,
     openai_model_fields: Sequence[OpenAIModelSettingName] = ("model",),
     azure_deployment_fields: Sequence[AzureDeploymentSettingName] = ("deployment_name",),
+    responses_mode: bool = False,
 ) -> tuple[dict[str, Any], AsyncOpenAI, bool]:
     """Load OpenAI settings, including Azure OpenAI aliases.
 
@@ -298,7 +299,10 @@ def load_openai_service_settings(
         return azure_settings, client, True  # type: ignore[return-value]
     client_args["default_headers"] = merged_headers
     if endpoint := azure_settings.get("endpoint"):
-        client_args["azure_endpoint"] = endpoint
+        if responses_mode:
+            client_args["base_url"] = f"{endpoint.rstrip('/')}/openai/v1/"
+        else:
+            client_args["azure_endpoint"] = endpoint
     if base_url := azure_settings.get("base_url"):
         client_args["base_url"] = base_url
     if api_key := azure_settings.get("api_key"):

python/packages/openai/tests/openai/test_openai_chat_client_azure.py

@@ -31,10 +31,6 @@
 )
 
 
-def _get_azure_responses_deployment_name() -> str:
-    return os.getenv("AZURE_OPENAI_RESPONSES_DEPLOYMENT_NAME") or os.environ["AZURE_OPENAI_DEPLOYMENT_NAME"]
-
-
 class OutputStruct(BaseModel):
     """A structured output for testing purposes."""
 
@@ -51,10 +47,7 @@ def _create_azure_openai_chat_client(
         api_key if api_key is not None else None if credential is not None else os.environ["AZURE_OPENAI_API_KEY"]
     )
     return OpenAIChatClient(
-        model=_get_azure_responses_deployment_name(),
         api_key=resolved_api_key,
-        azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
-        api_version=os.getenv("AZURE_OPENAI_API_VERSION"),
         credential=credential,
     )