From 0bd747ef7e0d17e5d3ffd26f2816fce5353bcf05 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 15 Jul 2024 19:58:41 -0700 Subject: [PATCH] max_file_size_mb in float --- litellm/types/router.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/types/router.py b/litellm/types/router.py index 6da80d5065..e7b8971bcb 100644 --- a/litellm/types/router.py +++ b/litellm/types/router.py @@ -154,7 +154,7 @@ class GenericLiteLLMParams(BaseModel): input_cost_per_second: Optional[float] = None output_cost_per_second: Optional[float] = None - max_file_size_mb: Optional[int] = None + max_file_size_mb: Optional[float] = None model_config = ConfigDict(extra="allow", arbitrary_types_allowed=True) @@ -187,7 +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, + max_file_size_mb: Optional[float] = None, **params, ): args = locals() @@ -248,7 +248,7 @@ class LiteLLM_Params(GenericLiteLLMParams): 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, + max_file_size_mb: Optional[float] = None, **params, ): args = locals()