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

use helper to get_config_file_contents_from_gcs

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-30 10:26:42 -07:00
parent 8d6a0bdc81
commit 051ac50fca
1 changed files with 20 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -42,7 +42,7 @@ def get_file_contents_from_s3(bucket_name, object_key):
config = yaml.safe_load(yaml_file)
return config
except ImportError:
except ImportError as e:
# this is most likely if a user is not using the litellm docker container
verbose_proxy_logger.error(f"ImportError: {str(e)}")
pass
@ -51,6 +51,25 @@ def get_file_contents_from_s3(bucket_name, object_key):
return None
async def get_config_file_contents_from_gcs(bucket_name, object_key):
try:
from litellm.integrations.gcs_bucket import GCSBucketLogger
gcs_bucket = GCSBucketLogger(
bucket_name=bucket_name,
)
file_contents = await gcs_bucket.download_gcs_object(object_key)
# file_contentis is a bytes object, so we need to convert it to yaml
file_contents = file_contents.decode("utf-8")
# convert to yaml
config = yaml.safe_load(file_contents)
return config
except:
verbose_proxy_logger.error(f"Error retrieving file contents: {str(e)}")
return None
# # Example usage
# bucket_name = 'litellm-proxy'
# object_key = 'litellm_proxy_config.yaml'