diff --git a/libs/partners/openai/langchain_openai/chat_models/__init__.py b/libs/partners/openai/langchain_openai/chat_models/__init__.py index 574128d2704..e1acbb4b87e 100644 --- a/libs/partners/openai/langchain_openai/chat_models/__init__.py +++ b/libs/partners/openai/langchain_openai/chat_models/__init__.py @@ -1,4 +1,6 @@ from langchain_openai.chat_models.azure import AzureChatOpenAI from langchain_openai.chat_models.base import ChatOpenAI +from langchain_openai.chat_models.batch import BatchError, BatchStatus + +__all__ = ["ChatOpenAI", "AzureChatOpenAI", "BatchError", "BatchStatus"] -__all__ = ["ChatOpenAI", "AzureChatOpenAI"]