From c87731a501a1ec622ad32fc60933b86ed7de416d Mon Sep 17 00:00:00 2001
From: Josephasafg <ajgard7@gmail.com>
Date: Tue, 28 May 2024 15:16:27 +0300
Subject: [PATCH] feat: Added beta to client

---
 ai21/clients/studio/ai21_client.py             | 2 ++
 ai21/clients/studio/resources/beta/__init__.py | 0
 ai21/clients/studio/resources/beta/beta.py     | 5 +++++
 3 files changed, 7 insertions(+)
 create mode 100644 ai21/clients/studio/resources/beta/__init__.py
 create mode 100644 ai21/clients/studio/resources/beta/beta.py

diff --git a/ai21/clients/studio/ai21_client.py b/ai21/clients/studio/ai21_client.py
index 5a47e7ce..3d29575a 100644
--- a/ai21/clients/studio/ai21_client.py
+++ b/ai21/clients/studio/ai21_client.py
@@ -5,6 +5,7 @@
 
 from ai21.ai21_env_config import _AI21EnvConfig, AI21EnvConfig
 from ai21.ai21_http_client import AI21HTTPClient
+from ai21.clients.studio.resources.beta.beta import Beta
 from ai21.clients.studio.resources.studio_answer import StudioAnswer
 from ai21.clients.studio.resources.studio_chat import StudioChat
 from ai21.clients.studio.resources.studio_completion import StudioCompletion
@@ -65,6 +66,7 @@ def __init__(
         self.answer = StudioAnswer(self._http_client)
         self.library = StudioLibrary(self._http_client)
         self.segmentation = StudioSegmentation(self._http_client)
+        self.beta = Beta(self._http_client)
 
     def count_tokens(self, text: str, tokenizer_name: str = PreTrainedTokenizers.J2_TOKENIZER) -> int:
         warnings.warn(
diff --git a/ai21/clients/studio/resources/beta/__init__.py b/ai21/clients/studio/resources/beta/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/ai21/clients/studio/resources/beta/beta.py b/ai21/clients/studio/resources/beta/beta.py
new file mode 100644
index 00000000..7baed087
--- /dev/null
+++ b/ai21/clients/studio/resources/beta/beta.py
@@ -0,0 +1,5 @@
+from ai21.clients.studio.resources.studio_resource import StudioResource
+
+
+class Beta(StudioResource):
+    pass