max_file_size_mb in float

nidhishs · Jul 16, 2024 · 0bd747e · 0bd747e
1 parent 38cef1c
commit 0bd747e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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 @@ def __init__(
  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 @@ def __init__(
  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()