diff --git a/ai21/clients/studio/resources/studio_completion.py b/ai21/clients/studio/resources/studio_completion.py index 48f85fa7..10c1890f 100644 --- a/ai21/clients/studio/resources/studio_completion.py +++ b/ai21/clients/studio/resources/studio_completion.py @@ -18,7 +18,6 @@ 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, @@ -26,9 +25,6 @@ def create( 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 index cb286df2..f549306a 100644 --- 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,