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

use helper to check check_file_size_under_limit

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-15 19:40:05 -07:00
parent 6c060b1fdc
commit b5a2090720
1 changed files with 11 additions and 2 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, decrypt_value_helper,
encrypt_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.init_callbacks import initialize_callbacks_on_proxy
from litellm.proxy.common_utils.openai_endpoint_utils import ( from litellm.proxy.common_utils.openai_endpoint_utils import (
remove_sensitive_info_from_deployment, remove_sensitive_info_from_deployment,
@ -3796,7 +3799,13 @@ async def audio_transcriptions(
param="file", 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_content = await file.read()
file_object = io.BytesIO(file_content) file_object = io.BytesIO(file_content)
file_object.name = file.filename file_object.name = file.filename