Add deprecated alias for openai model clients (#4641)

Add alias for openai model clients
microsoft · Dec 10, 2024 · c234e77 · c234e77
1 parent bfb8aea
commit c234e77
Showing 1 changed file with 18 additions and 0 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
@@ -0,0 +1,18 @@
+from typing_extensions import deprecated
+
+from .openai import AzureOpenAIChatCompletionClient as AzureOpenAIChatCompletionClientAlias
+from .openai import OpenAIChatCompletionClient as OpenAIChatCompletionClientAlias
+
+
+@deprecated(
+    "autogen_ext.models.OpenAIChatCompletionClient moved to autogen_ext.models.openai.OpenAIChatCompletionClient. This alias will be removed in 0.4.0."
+)
+class OpenAIChatCompletionClient(OpenAIChatCompletionClientAlias):
+    pass
+
+
+@deprecated(
+    "autogen_ext.models.AzureOpenAIChatCompletionClient moved to autogen_ext.models.openai.AzureOpenAIChatCompletionClient. This alias will be removed in 0.4.0."
+)
+class AzureOpenAIChatCompletionClient(AzureOpenAIChatCompletionClientAlias):
+    pass