fix: change signatures for the stream function

anthropics · Jul 29, 2024 · d83a4d6 · d83a4d6
1 parent 70f3f5c
commit d83a4d6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 26 deletions.
diff --git a/src/anthropic/_utils/_reflection.py b/src/anthropic/_utils/_reflection.py
@@ -34,7 +34,7 @@ def assert_signatures_in_sync(
 
         if custom_param.annotation != source_param.annotation:
             errors.append(
-                f"types for the `{name}` param are do not match; source={repr(source_param.annotation)} checking={repr(source_param.annotation)}"
+                f"types for the `{name}` param are do not match; source={repr(source_param.annotation)} checking={repr(custom_param.annotation)}"
             )
             continue
 

diff --git a/src/anthropic/resources/messages.py b/src/anthropic/resources/messages.py
@@ -885,18 +885,7 @@ def stream(
         *,
         max_tokens: int,
         messages: Iterable[MessageParam],
-        model: Union[
-            str,
-            Literal[
-                "claude-3-5-sonnet-20240620",
-                "claude-3-opus-20240229",
-                "claude-3-sonnet-20240229",
-                "claude-3-haiku-20240307",
-                "claude-2.1",
-                "claude-2.0",
-                "claude-instant-1.2",
-            ],
-        ],
+        model: ModelParam,
         metadata: message_create_params.Metadata | NotGiven = NOT_GIVEN,
         stop_sequences: List[str] | NotGiven = NOT_GIVEN,
         system: Union[str, Iterable[TextBlockParam]] | NotGiven = NOT_GIVEN,
@@ -1800,18 +1789,7 @@ def stream(
         *,
         max_tokens: int,
         messages: Iterable[MessageParam],
-        model: Union[
-            str,
-            Literal[
-                "claude-3-5-sonnet-20240620",
-                "claude-3-opus-20240229",
-                "claude-3-sonnet-20240229",
-                "claude-3-haiku-20240307",
-                "claude-2.1",
-                "claude-2.0",
-                "claude-instant-1.2",
-            ],
-        ],
+        model: ModelParam,
         metadata: message_create_params.Metadata | NotGiven = NOT_GIVEN,
         stop_sequences: List[str] | NotGiven = NOT_GIVEN,
         system: Union[str, Iterable[TextBlockParam]] | NotGiven = NOT_GIVEN,

diff --git a/tests/lib/streaming/test_messages.py b/tests/lib/streaming/test_messages.py
@@ -192,7 +192,7 @@ def test_stream_method_definition_in_sync(sync: bool) -> None:
 
         if custom_param.annotation != generated_param.annotation:
             errors.append(
-                f"types for the `{name}` param are do not match; generated={repr(generated_param.annotation)} custom={repr(generated_param.annotation)}"
+                f"types for the `{name}` param are do not match; generated={repr(generated_param.annotation)} custom={repr(custom_param.annotation)}"
             )
             continue