phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

log file_size_in_mb in metadata

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-29 08:00:28 -07:00
parent b6b8bb4f02
commit cc0e790863
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/common_utils/http_parsing_utils.py
View file

@ -56,6 +56,9 @@ def check_file_size_under_limit(
file_contents_size = file.size or 0 file_contents_size = file.size or 0
file_content_size_in_mb = file_contents_size / (1024 * 1024) file_content_size_in_mb = file_contents_size / (1024 * 1024)
if "metadata" not in request_data:
request_data["metadata"] = {}
request_data["metadata"]["file_size_in_mb"] = file_content_size_in_mb
max_file_size_mb = None max_file_size_mb = None
if llm_router is not None and request_data["model"] in router_model_names: if llm_router is not None and request_data["model"] in router_model_names: