
    Cj2              
       v    S SK J r   S SKJrJr  S SKrS SKJrJr  S SKJ	r	   " S S5      r
S\S	\S
\ S\S\4
S jrg)    )datetime)OptionalcastN)print_verboseverbose_logger)StandardLoggingPayloadc                   @    \ rS rSr           SS jrS rS rSrg)S3Logger   Nc                 n   SS K n [        R                  " S[        R                   35        Sn[        R                  Gb  [        R                  R                  5        H\  u  nn[        U[        5      (       d  M  UR                  S5      (       d  M5  [        R                  " U5      [        R                  U'   M^     [        R                  R                  S5      n[        R                  R                  S5      n[        R                  R                  S5      n[        R                  R                  SS	5      n[        R                  R                  S
5      n[        R                  R                  S5      n[        R                  R                  S5      n[        R                  R                  S5      n	[        R                  R                  S5      n
[        R                  R                  S5      n[        R                  R                  S5      n[        [        R                  R                  SS5      5      nXl        Xl        X l        [        R                  " SU 35        UR                  "  SUUUUUUU	U
US.	UD6U l        g ! ["         a  n[%        S[        U5       35        UeS nAff = f)Nr   z'in init s3 logger - s3_callback_params Fzos.environ/s3_bucket_names3_region_names3_api_version
s3_use_sslT	s3_verifys3_endpoint_urls3_aws_access_key_ids3_aws_secret_access_keys3_aws_session_token	s3_configs3_paths3_use_team_prefixzs3 logger using endpoint url )	region_nameendpoint_urlapi_versionuse_sslverifyaws_access_key_idaws_secret_access_keyaws_session_tokenconfigz Got exception on init s3 client )s3)boto3r   debuglitellms3_callback_paramsitems
isinstancestr
startswith
get_secretgetboolr   bucket_namer   client	s3_client	Exceptionr   )selfr   r   r   r   r   r   r   r   r   r   r   kwargsr#   r   keyvaluees                     i/home/rurouni/.local/share/pipx/venvs/strix-agent/lib/python3.13/site-packages/litellm/integrations/s3.py__init__S3Logger.__init__   sU    	6	  9':T:T9UV "'))5")"<"<"B"B"DJC!%--%2B2B=2Q2Q:A:L:LU:S2237 #E ")!;!;!?!?@P!Q!(!;!;!?!?@P!Q!(!;!;!?!?@P!Q$77;;L$O
#66::;G	")"<"<"@"@AR"S'.'A'A'E'E*($ ,3+E+E+I+I.,( (/'A'A'E'E*($ $66::;G	!4488C%)..223GO&" '9#-"L  #@@Q!RS"\\*,*" "6&>"6  DN  	<SVHEFG	s%   A2J <J G7J 
J4J//J4c                 2   #    U R                  XX4U5        g 7f)N)	log_event)r2   r3   response_obj
start_timeend_timer   s         r7   _async_log_eventS3Logger._async_log_eventW   s      	vZ=Qs   c           
      "    [         R                  " SU 35        UR                  S0 5      nUR                  S0 5      =(       d    0 n0 n[        U[        5      (       a%  UR                  5        H  u  pU	S;   a  M  XU	'   M     [        [        [           UR                  SS 5      5      nUc  g US   R                  S5      nSn[        R                  (       a  U R                  (       a  Ub  U S3n[        R                  R                  X;5      =(       d    Sn[        [        [        [           U R                   5      =(       d    SUUU5      nS	UR#                  S
5      -   S-   US   -   S-   nSSKJn  U" U5      nU" SU 35        U R(                  R+                  U R,                  UUSSSU S3SS9nU" S[        U5       35        U" SU 35        U$ ! [.         a,  n[         R0                  " S[        U5       35         S nAg S nAff = f)Nz/s3 Logging - Enters logging function for model litellm_paramsmetadata)headersendpointcaching_groupsprevious_modelsstandard_logging_objectuser_api_key_team_alias /ztime-z%Y-%m-%dT%H-%M-%S-%f_id.jsonr   )
safe_dumpsz
s3 Logger - Logging payload = zapplication/jsonenzinline; filename=""z0private, immutable, max-age=31536000, s-maxage=0)BucketKeyBodyContentTypeContentLanguageContentDispositionCacheControlzResponse from s3:z*s3 Layer Logging - final response object: zs3 Layer Error - )r   r$   r,   r(   dictr'   r   r   r   r%   enable_preview_featuresr   utilsget_logging_idget_s3_object_keyr)   r   strftime*litellm.litellm_core_utils.safe_json_dumpsrO   r0   
put_objectr.   r1   	exception)r2   r3   r<   r=   r>   r   rB   rC   clean_metadatar4   r5   payload
team_aliasteam_alias_prefixs3_file_names3_object_keys3_object_download_filenamerO   payload_strresponser6   s                        r7   r;   S3Logger.log_event\   s7   V	  A&J $ZZ(8"=N"":r28b   N(D))"*.."2JC   !.3s+ #3 9=/0

4d;9G
  ,001JKJ "//++*'1l!$4!"==77
LRPRL-Xc]DLL17R!	M %%&<=> $-  	 ( N$W-K<[MJK~~00''! . $%78S7TTU#VO 1 H -c(m_=>F|nUVO 	$$'8Q%AB	s   B,G /D(G 
H""H		H)r.   r0   r   r   )NNNNTNNNNNN)__name__
__module____qualname____firstlineno__r8   r?   r;   __static_attributes__     r7   r
   r
      s8     !!%!GRR
Wrr   r
   r   prefixr=   rf   returnc                     U (       a  U R                  S5      S-   OSU-   UR                  S5      -   S-   U-   nUS-  nU$ )NrK   rJ   z%Y-%m-%drN   )rstripr^   )r   rs   r=   rf   rg   s        r7   r]   r]      s_     '.	s	"2
	


j
)	* 	 		  WMrr   )r   typingr   r   r%   litellm._loggingr   r   litellm.types.utilsr   r
   r)   r]   rq   rr   r7   <module>rz      sV     !  : 6g gT  	
 	rr   