Switch from OpenAI to Azure OpenAI

Author: AnatoliB

samples-v2/openai_agents/.gitignore

@@ -126,7 +126,7 @@ dmypy.json
 bin
 obj
 appsettings.json
-# local.settings.json
+local.settings.json
 
 # Azurite artifacts
 __blobstorage__

samples-v2/openai_agents/function_app.py

@@ -1,7 +1,34 @@
+import os
 import azure.functions as func
-from agents import Agent, Runner
+from openai import AsyncAzureOpenAI
+from agents import Agent, Runner, set_default_openai_client
 from durable_decorators import durable_openai_agent_orchestrator
 from model_activity import create_invoke_model_activity
+from azure.identity import DefaultAzureCredential
+
+
+#region Regular Azure OpenAI setup
+
+# Initialize Azure credential
+credential = DefaultAzureCredential()
+
+# Token provider function that returns the token
+def get_azure_token():
+    return credential.get_token("https://cognitiveservices.azure.com/.default").token
+
+# Initialize Azure OpenAI client with DefaultAzureCredential
+openai_client = AsyncAzureOpenAI(
+    azure_ad_token_provider=get_azure_token,
+    api_version=os.getenv("AZURE_OPENAI_API_VERSION", "2024-02-01"),
+    azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
+    azure_deployment=os.getenv("AZURE_OPENAI_DEPLOYMENT", "gpt-4")
+)
+
+# Set the default OpenAI client for the Agents SDK
+set_default_openai_client(openai_client)
+
+# endregion
+
 
 app = func.FunctionApp(http_auth_level=func.AuthLevel.FUNCTION)
 

samples-v2/openai_agents/local.settings.json

@@ -0,0 +1,10 @@
+{
+  "IsEncrypted": false,
+  "Values": {
+    "AzureWebJobsStorage": "UseDevelopmentStorage=true",
+    "FUNCTIONS_WORKER_RUNTIME": "python",
+    "AZURE_OPENAI_ENDPOINT": "https://your-openai-service.openai.azure.com/",
+    "AZURE_OPENAI_DEPLOYMENT": "your-gpt-deployment-name",
+    "AZURE_OPENAI_API_VERSION": "2025-03-01-preview"
+  }
+}

samples-v2/openai_agents/local.settings.json.template

@@ -204,9 +204,7 @@ def from_json(cls, json_str: str) -> 'ActivityModelInput':
 class ModelInvoker:
     def __init__(self, model_provider: Optional[ModelProvider] = None):
         """Initialize the activity with a model provider."""
-        self._model_provider = model_provider or OpenAIProvider(
-            openai_client=AsyncOpenAI(max_retries=0)
-        )
+        self._model_provider = model_provider or OpenAIProvider()
 
     async def invoke_model_activity(self, input: ActivityModelInput) -> ModelResponse:
         """Activity that invokes a model with the given input."""

samples-v2/openai_agents/model_invoker.py

@@ -4,4 +4,6 @@
 
 azure-functions
 azure-functions-durable
+azure-identity
+openai
 openai-agents