feat: add flexible enum to model param (#75)

anthropics · Jul 18, 2023 · e29b8f9 · e29b8f9
1 parent d9b7e30
commit e29b8f9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/src/anthropic/resources/completions.py b/src/anthropic/resources/completions.py
@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import List, overload
+from typing import List, Union, overload
 from typing_extensions import Literal
 
 from ..types import Completion, completion_create_params
@@ -21,7 +21,7 @@ def create(
         self,
         *,
         max_tokens_to_sample: int,
-        model: str,
+        model: Union[str, Literal["claude-2", "claude-instant-1"]],
         prompt: str,
         metadata: completion_create_params.CompletionRequestNonStreamingMetadata | NotGiven = NOT_GIVEN,
         stop_sequences: List[str] | NotGiven = NOT_GIVEN,
@@ -113,7 +113,7 @@ def create(
         self,
         *,
         max_tokens_to_sample: int,
-        model: str,
+        model: Union[str, Literal["claude-2", "claude-instant-1"]],
         prompt: str,
         stream: Literal[True],
         metadata: completion_create_params.CompletionRequestStreamingMetadata | NotGiven = NOT_GIVEN,
@@ -205,7 +205,7 @@ def create(
         self,
         *,
         max_tokens_to_sample: int,
-        model: str,
+        model: Union[str, Literal["claude-2", "claude-instant-1"]],
         prompt: str,
         metadata: completion_create_params.CompletionRequestNonStreamingMetadata
         | completion_create_params.CompletionRequestStreamingMetadata
@@ -253,7 +253,7 @@ async def create(
         self,
         *,
         max_tokens_to_sample: int,
-        model: str,
+        model: Union[str, Literal["claude-2", "claude-instant-1"]],
         prompt: str,
         metadata: completion_create_params.CompletionRequestNonStreamingMetadata | NotGiven = NOT_GIVEN,
         stop_sequences: List[str] | NotGiven = NOT_GIVEN,
@@ -345,7 +345,7 @@ async def create(
         self,
         *,
         max_tokens_to_sample: int,
-        model: str,
+        model: Union[str, Literal["claude-2", "claude-instant-1"]],
         prompt: str,
         stream: Literal[True],
         metadata: completion_create_params.CompletionRequestStreamingMetadata | NotGiven = NOT_GIVEN,
@@ -437,7 +437,7 @@ async def create(
         self,
         *,
         max_tokens_to_sample: int,
-        model: str,
+        model: Union[str, Literal["claude-2", "claude-instant-1"]],
         prompt: str,
         metadata: completion_create_params.CompletionRequestNonStreamingMetadata
         | completion_create_params.CompletionRequestStreamingMetadata

diff --git a/src/anthropic/types/completion_create_params.py b/src/anthropic/types/completion_create_params.py
@@ -22,7 +22,7 @@ class CompletionRequestNonStreaming(TypedDict, total=False):
     only specifies the absolute maximum number of tokens to generate.
     """
 
-    model: Required[str]
+    model: Required[Union[str, Literal["claude-2", "claude-instant-1"]]]
     """The model that will complete your prompt.
 
     As we improve Claude, we develop new versions of it that you can query. This
@@ -109,7 +109,7 @@ class CompletionRequestStreaming(TypedDict, total=False):
     only specifies the absolute maximum number of tokens to generate.
     """
 
-    model: Required[str]
+    model: Required[Union[str, Literal["claude-2", "claude-instant-1"]]]
     """The model that will complete your prompt.
 
     As we improve Claude, we develop new versions of it that you can query. This