fix: removed experimental

AI21Labs · Dec 26, 2023 · e316760 · e316760
1 parent c55cbee
commit e316760
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/ai21/clients/studio/resources/studio_completion.py b/ai21/clients/studio/resources/studio_completion.py
@@ -18,17 +18,13 @@ def create(
         top_p: Optional[float] = 1,
         top_k_return: Optional[int] = 0,
         custom_model: Optional[str] = None,
-        experimental_mode: bool = False,
         stop_sequences: Optional[List[str]] = None,
         frequency_penalty: Optional[Dict[str, Any]] = None,
         presence_penalty: Optional[Dict[str, Any]] = None,
         count_penalty: Optional[Dict[str, Any]] = None,
         epoch: Optional[int] = None,
         **kwargs,
     ) -> CompletionsResponse:
-        if experimental_mode:
-            model = f"experimental/{model}"
-
         url = f"{self._client.get_base_url()}/{model}"
 
         if custom_model is not None:
@@ -45,7 +41,6 @@ def create(
             top_p=top_p,
             top_k_return=top_k_return,
             custom_model=custom_model,
-            experimental_mode=experimental_mode,
             stop_sequences=stop_sequences,
             frequency_penalty=frequency_penalty,
             presence_penalty=presence_penalty,

diff --git a/ai21/resources/bases/completion_base.py b/ai21/resources/bases/completion_base.py
@@ -20,7 +20,6 @@ def create(
         top_p=1,
         top_k_return=0,
         custom_model: Optional[str] = None,
-        experimental_mode: bool = False,
         stop_sequences: Optional[List[str]] = (),
         frequency_penalty: Optional[Dict[str, Any]] = {},
         presence_penalty: Optional[Dict[str, Any]] = {},
@@ -44,7 +43,6 @@ def _create_body(
         top_p: Optional[int],
         top_k_return: Optional[int],
         custom_model: Optional[str],
-        experimental_mode: bool,
         stop_sequences: Optional[List[str]],
         frequency_penalty: Optional[Dict[str, Any]],
         presence_penalty: Optional[Dict[str, Any]],
@@ -54,7 +52,6 @@ def _create_body(
         return {
             "model": model,
             "customModel": custom_model,
-            "experimentalModel": experimental_mode,
             "prompt": prompt,
             "maxTokens": max_tokens,
             "numResults": num_results,