tPjhddlZddlmZddlZddlmZdZdZdededed e fd Z dededed e fd Z y) N)Final)verbose_proxy_loggercl ddl}ddlm}ddlm}|j }|j d|j|j|j}tjd|||j||}tjd||d jjd }tjd tj |} | S#t"$r } tj$d | Yd} ~ yd} ~ wt&$r } tj$d | Yd} ~ yd} ~ wwxYw)Nr Credentials) bedrock_converse_chat_completions3aws_access_key_idaws_secret_access_keyaws_session_tokenz Retrieving %s from S3 bucket: %sBucketKeyz Response: %sBodyutf-8zFile contents retrieved from S3ImportError: %s"Error retrieving file contents: %s)boto3botocore.credentialsr litellm.mainrget_credentialsclient access_key secret_keytokenrdebug get_objectreaddecodeyaml safe_load ImportErrorerror Exception) bucket_name object_keyrrr credentials s3_clientresponse file_contentsconfiges q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/common_utils/load_config_utils.pyget_file_contents_from_s3r/ s4A*J*Z*Z*\  << )44"-"8"8)// ( ""#EzS^_#..kz.R"">8< (/446==gF ""#DE}5 9""#4a88 ""#GKs$C C## D3,D D3D..D3c(K ddlm}||}|j|d{}|td||j d}t j |}|S7<#t$r }tjd|Yd}~yd}~wwxYww)NrGCSBucketLoggerr&File contents are None for rr) *litellm.integrations.gcs_bucket.gcs_bucketr2download_gcs_objectr%r r!r"rr$)r&r'r2 gcs_bucketr+r,r-s r.!get_config_file_contents_from_gcsr8,sN+# )<B#A&A$;A&#B$A&& B/B B BBr&r'local_file_pathreturnc ddl}ddlm}ddlm}|}|j }|j d|j|j|j}tjd|||j||} | d jjd } tjd | tj tj"j%|d t'|d5} | j)| dddtjd|y #1swY xYw#t*$r } tj,d| Yd} ~ yd} ~ wt.$r } tj0d| Yd} ~ yd} ~ wwxYw)aA Download a Python file from S3 and save it to local filesystem. Args: bucket_name (str): S3 bucket name object_key (str): S3 object key (file path in bucket) local_file_path (str): Local path where file should be saved Returns: bool: True if successful, False otherwise rNr) BaseAWSLLMr r z-Downloading Python file %s from S3 bucket: %srrrzFile contents: %sTexist_okw)Python file downloaded successfully to %srFz!Error downloading Python file: %s)rrr!litellm.llms.bedrock.base_aws_llmr<rrrrrrrrrr osmakedirspathdirnameopenwriter#r$r% exception) r&r'r9rrr< base_aws_llmr(r)r*r+fr-s r.download_python_file_from_s3rKAsX &4@(l *6*F*F*H  << )44"-"8"8)// (  ""#RT^`kl#..kz.R (/446==gF ""#6 F BGGOOO4tD/3 ' #1 GGM " # ""#NP_`  # # ""#4a8 &&'JANs<CD/D;/D84D;; F E F +FF cK ddlm}||}|j|d{}|td||j d}t j t jj|dt|d 5}|j|dddtjd |y7#1swY"xYw#t$r }tjd |Yd}~y d}~wwxYww) aD Download a Python file from GCS and save it to local filesystem. Args: bucket_name (str): GCS bucket name object_key (str): GCS object key (file path in bucket) local_file_path (str): Local path where file should be saved Returns: bool: True if successful, False otherwise rr1r3Nr4rTr=r?r@z*Error downloading Python file from GCS: %sF)r5r2r6r%r rBrCrDrErFrGrrrH)r&r'r9r2r7r+rJr-s r.download_python_file_from_gcsrMzs N+# )<A%C CC =C8>C C C C5C0+C80C55C8) rBtypingrr!litellm._loggingrr/r8strboolrKrMr.rTso  1 F*6666 6r)))) )rS