scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

change to snake case 'cause of aws docs

Browse source
This commit is contained in:
Tim Xia 2024-03-02 00:07:21 -05:00
parent 2c97b952ea
commit 67110df3e1
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/llms/bedrock.py
View file

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