add explicit config exports

microsoft · Nov 22, 2024 · 48fb6e8 · 48fb6e8
1 parent 2a3404b
commit 48fb6e8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/python/packages/autogen-ext/src/autogen_ext/models/__init__.py b/python/packages/autogen-ext/src/autogen_ext/models/__init__.py
@@ -1,9 +1,8 @@
 from ._openai._openai_client import (
     AzureOpenAIChatCompletionClient,
-    AzureOpenAIClientConfiguration,
     OpenAIChatCompletionClient,
-    OpenAIClientConfiguration,
 )
+from ._openai.config import AzureOpenAIClientConfiguration, OpenAIClientConfiguration
 from ._reply_chat_completion_client import ReplayChatCompletionClient
 
 __all__ = [

diff --git a/python/packages/autogen-ext/src/autogen_ext/models/_openai/config/__init__.py b/python/packages/autogen-ext/src/autogen_ext/models/_openai/config/__init__.py
@@ -49,3 +49,6 @@ class AzureOpenAIClientConfiguration(BaseOpenAIClientConfiguration, total=False)
     azure_ad_token_provider: AsyncAzureADTokenProvider
     # Must be provided
     model_capabilities: Required[ModelCapabilities]
+
+
+__all__ = ["AzureOpenAIClientConfiguration", "OpenAIClientConfiguration"]