scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

allow setting max_file_size_mb

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-15 19:25:24 -07:00
parent a900f352b5
commit 865469e43f
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/types/router.py
View file

@ -154,6 +154,8 @@ class GenericLiteLLMParams(BaseModel):
input_cost_per_second: Optional[float] = None
output_cost_per_second: Optional[float] = None
max_file_size_mb: Optional[int] = None
model_config = ConfigDict(extra="allow", arbitrary_types_allowed=True)
def __init__(
@ -185,6 +187,7 @@ class GenericLiteLLMParams(BaseModel):
output_cost_per_token: Optional[float] = None,
input_cost_per_second: Optional[float] = None,
output_cost_per_second: Optional[float] = None,
max_file_size_mb: Optional[int] = None,
**params,
):
args = locals()
@ -243,6 +246,9 @@ class LiteLLM_Params(GenericLiteLLMParams):
aws_access_key_id: Optional[str] = None,
aws_secret_access_key: Optional[str] = None,
aws_region_name: Optional[str] = None,
# OpenAI / Azure Whisper
# set a max-size of file that can be passed to litellm proxy
max_file_size_mb: Optional[int] = None,
**params,
):
args = locals()