log file_size_in_mb in metadata

This commit is contained in:
Ishaan Jaff 2024-07-29 08:00:28 -07:00
parent 0627468455
commit b2fcf65653

View file

@ -56,6 +56,9 @@ def check_file_size_under_limit(
file_contents_size = file.size or 0
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
if llm_router is not None and request_data["model"] in router_model_names: