fix: Fix provider string for AzureTokenProvider (#4992)

Fix provider string for AzureTokenProvider
microsoft · Jan 13, 2025 · 70f7e99 · 70f7e99
1 parent 1a6e976
commit 70f7e99
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/python/packages/autogen-ext/src/autogen_ext/auth/azure/__init__.py b/python/packages/autogen-ext/src/autogen_ext/auth/azure/__init__.py
@@ -16,7 +16,7 @@ class TokenProviderConfig(BaseModel):
 class AzureTokenProvider(Component[TokenProviderConfig]):
     component_type = "token_provider"
     component_config_schema = TokenProviderConfig
-    component_provider_override = "autogen_ext.models.openai.AzureTokenProvider"
+    component_provider_override = "autogen_ext.auth.azure.AzureTokenProvider"
 
     def __init__(self, credential: TokenProvider, *scopes: str):
         self.credential = credential

diff --git a/python/packages/autogen-ext/src/autogen_ext/models/openai/_openai_client.py b/python/packages/autogen-ext/src/autogen_ext/models/openai/_openai_client.py
@@ -1114,7 +1114,7 @@ class AzureOpenAIChatCompletionClient(
                 "azure_deployment": "{your-azure-deployment}",
                 "api_version": "2024-06-01",
                 "azure_ad_token_provider": {
-                    "provider": "autogen_ext.models.openai.AzureTokenProvider",
+                    "provider": "autogen_ext.auth.azure.AzureTokenProvider",
                     "config": {
                         "provider_kind": "DefaultAzureCredential",
                         "scopes": ["https://cognitiveservices.azure.com/.default"],

diff --git a/python/packages/autogen-magentic-one/README.md b/python/packages/autogen-magentic-one/README.md
@@ -143,7 +143,7 @@ To configure for Azure OpenAI service, use the following config:
         "azure_deployment": "{your-azure-deployment}",
         "api_version": "2024-06-01",
         "azure_ad_token_provider": {
-            "provider": "autogen_ext.models.openai.AzureTokenProvider",
+            "provider": "autogen_ext.auth.azure.AzureTokenProvider",
             "config": {
                 "provider_kind": "DefaultAzureCredential",
                 "scopes": [

diff --git a/python/samples/core_async_human_in_the_loop/model_config_template.json b/python/samples/core_async_human_in_the_loop/model_config_template.json
@@ -7,7 +7,7 @@
 //         "azure_deployment": "{your-azure-deployment}",
 //         "api_version": "2024-06-01",
 //         "azure_ad_token_provider": {
-//             "provider": "autogen_ext.models.openai.AzureTokenProvider",
+//             "provider": "autogen_ext.auth.azure.AzureTokenProvider",
 //             "config": {
 //                 "provider_kind": "DefaultAzureCredential",
 //                 "scopes": [

diff --git a/python/samples/core_chess_game/model_config_template.json b/python/samples/core_chess_game/model_config_template.json
@@ -7,7 +7,7 @@
 //         "azure_deployment": "{your-azure-deployment}",
 //         "api_version": "2024-06-01",
 //         "azure_ad_token_provider": {
-//             "provider": "autogen_ext.models.openai.AzureTokenProvider",
+//             "provider": "autogen_ext.auth.azure.AzureTokenProvider",
 //             "config": {
 //                 "provider_kind": "DefaultAzureCredential",
 //                 "scopes": [