scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

fix download_python_file_from_s3

Browse source
This commit is contained in:
Ishaan Jaff 2025-07-15 15:39:42 -07:00
parent 6a52204946
commit a078e43d44
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
litellm/proxy/common_utils/load_config_utils.py
View file

@ -65,7 +65,11 @@ async def get_config_file_contents_from_gcs(bucket_name, object_key):
return None
def download_python_file_from_s3(bucket_name, object_key, local_file_path):
def download_python_file_from_s3(
bucket_name: str,
object_key: str,
local_file_path: str,
) -> bool:
"""
Download a Python file from S3 and save it to local filesystem.
@ -99,6 +103,7 @@ def download_python_file_from_s3(bucket_name, object_key, local_file_path):
# Read the file contents
file_contents = response["Body"].read().decode("utf-8")
verbose_proxy_logger.debug(f"File contents: {file_contents}")
# Ensure directory exists
os.makedirs(os.path.dirname(local_file_path), exist_ok=True)
@ -114,11 +119,15 @@ def download_python_file_from_s3(bucket_name, object_key, local_file_path):
verbose_proxy_logger.error(f"ImportError: {str(e)}")
return False
except Exception as e:
verbose_proxy_logger.error(f"Error downloading Python file: {str(e)}")
verbose_proxy_logger.exception(f"Error downloading Python file: {str(e)}")
return False
async def download_python_file_from_gcs(bucket_name, object_key, local_file_path):
async def download_python_file_from_gcs(
bucket_name: str,
object_key: str,
local_file_path: str,
) -> bool:
"""
Download a Python file from GCS and save it to local filesystem.
@ -154,7 +163,7 @@ async def download_python_file_from_gcs(bucket_name, object_key, local_file_path
return True
except Exception as e:
verbose_proxy_logger.error(f"Error downloading Python file from GCS: {str(e)}")
verbose_proxy_logger.exception(f"Error downloading Python file from GCS: {str(e)}")
return False