diff --git a/ai21/__init__.py b/ai21/__init__.py index bfc95fa8..afe7b643 100644 --- a/ai21/__init__.py +++ b/ai21/__init__.py @@ -1,5 +1,6 @@ from typing import Any +from ai21.ai21_env_config import AI21EnvConfig from ai21.clients.studio.ai21_client import AI21Client from ai21.errors import ( AI21APIError, @@ -50,6 +51,7 @@ def __getattr__(name: str) -> Any: __all__ = [ + "AI21EnvConfig", "AI21Client", "AI21APIError", "APITimeoutError",