From 67110df3e13ed1b1a0e1d5d825aa1ad5262dd448 Mon Sep 17 00:00:00 2001 From: Tim Xia Date: Sat, 2 Mar 2024 00:07:21 -0500 Subject: [PATCH] change to snake case 'cause of aws docs --- litellm/llms/bedrock.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/litellm/llms/bedrock.py b/litellm/llms/bedrock.py index e0938ded8ee..1a25e167a9d 100644 --- a/litellm/llms/bedrock.py +++ b/litellm/llms/bedrock.py @@ -287,25 +287,25 @@ class AmazonMistralConfig: Reference: https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters-mistral.html Supported Params for the Amazon / Mistral models: - - `maxTokens` (integer) max tokens, + - `max_tokens` (integer) max tokens, - `temperature` (float) temperature for model, - - `topP` (float) top p for model + - `top_p` (float) top p for model - `stop` [string] A list of stop sequences that if generated by the model, stops the model from generating further output. - - `topK` (float) top k for model + - `top_k` (float) top k for model """ - maxTokens: Optional[int] = None + max_tokens: Optional[int] = None temperature: Optional[float] = None - topP: Optional[float] = None - topK: Optional[float] = None + top_p: Optional[float] = None + top_k: Optional[float] = None stop: Optional[list[str]] = None def __init__( self, - maxTokens: Optional[int] = None, + max_tokens: Optional[int] = None, temperature: Optional[float] = None, - topP: Optional[int] = None, - topK: Optional[float] = None, + top_p: Optional[int] = None, + top_k: Optional[float] = None, stop: Optional[list[str]] = None, ) -> None: locals_ = locals()