diff --git a/llms/providers/anthropic.py b/llms/providers/anthropic.py index 8b70565..3731ade 100644 --- a/llms/providers/anthropic.py +++ b/llms/providers/anthropic.py @@ -147,7 +147,7 @@ def _prepare_message_inputs( "messages": messages, "system": system_message, "temperature": temperature, - "max_tokens_to_sample": max_tokens, + "max_tokens": max_tokens, "stop_sequences": stop_sequences, } return model_inputs