Revert "Correct anthropic param name"

This reverts commit 26676df.
kagisearch · May 14, 2024 · d4c7403 · d4c7403
1 parent 11093ad
commit d4c7403
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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