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()