From 0a6926454b454f4308a140dbe88ccc42c3428c66 Mon Sep 17 00:00:00 2001
From: Asaf Gardin <asafg@ai21.com>
Date: Tue, 2 Jan 2024 16:31:44 +0200
Subject: [PATCH] fix: Added envconfig class to init

---
 ai21/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

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",