phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

follow camelcase convention

Browse source
This commit is contained in:
Tim Xia 2024-03-01 23:58:03 -05:00
parent bf92f84b00
commit 2c97b952ea
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/llms/bedrock.py
View file

@ -287,14 +287,14 @@ class AmazonMistralConfig:
Reference: https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters-mistral.html
Supported Params for the Amazon / Mistral models:
- `max_tokens` (integer) max tokens,
- `maxTokens` (integer) max tokens,
- `temperature` (float) temperature for model,
- `top_p` (float) top p for model
- `topP` (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.
- `top_k` (float) top k for model
- `topK` (float) top k for model
"""
max_tokens: Optional[int] = None
maxTokens: Optional[int] = None
temperature: Optional[float] = None
topP: Optional[float] = None
topK: Optional[float] = None
@ -302,7 +302,7 @@ class AmazonMistralConfig:
def __init__(
self,
max_tokens: Optional[int] = None,
maxTokens: Optional[int] = None,
temperature: Optional[float] = None,
topP: Optional[int] = None,
topK: Optional[float] = None,
@ -1118,4 +1118,4 @@ def image_generation(
image_dict = {"url": artifact["base64"]}
model_response.data = image_dict
return model_response
return model_response