phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #4724 from BerriAI/litellm_Set_max_file_size_transc

Browse source 
[Feat] - set max file size on /audio/transcriptions
This commit is contained in:
Ishaan Jaff 2024-07-15 20:42:24 -07:00 • committed by GitHub
commit 254ac37f65
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 144 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/proxy_server.py
View file

@ -143,7 +143,10 @@ from litellm.proxy.common_utils.encrypt_decrypt_utils import (
decrypt_value_helper,
encrypt_value_helper,
)
from litellm.proxy.common_utils.http_parsing_utils import _read_request_body
from litellm.proxy.common_utils.http_parsing_utils import (
_read_request_body,
check_file_size_under_limit,
)
from litellm.proxy.common_utils.init_callbacks import initialize_callbacks_on_proxy
from litellm.proxy.common_utils.openai_endpoint_utils import (
remove_sensitive_info_from_deployment,
@ -3796,7 +3799,13 @@ async def audio_transcriptions(
param="file",
)
# Instead of writing to a file
# Check if File can be read in memory before reading
check_file_size_under_limit(
request_data=data,
file=file,
router_model_names=router_model_names,
)
file_content = await file.read()
file_object = io.BytesIO(file_content)
file_object.name = file.filename