Ë
    tPêjÑ	  ã                   óL   — d dl Z d dlZd dlmZmZ d dlmZ d dlZ G d„ d«      Zy)é    N)ÚdatetimeÚtimezone)ÚFinalc                   ó   — e Zd Zd„ Zd„ Zy)ÚGreenscaleLoggerc                 óŒ   — dd l }|j                  d«      | _        | j                  ddœ| _        |j                  d«      | _        y )Nr   ÚGREENSCALE_API_KEYzapplication/json)zapi-keyzContent-TypeÚGREENSCALE_ENDPOINT)ÚosÚgetenvÚgreenscale_api_keyÚheadersÚgreenscale_logging_url)Úselfr   s     úd/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/greenscale.pyÚ__init__zGreenscaleLogger.__init__
   s@   € Ûà"$§)¡)Ð,@Ó"AˆÔà×.Ñ.Ø.ñ
ˆŒð ')§i¡iÐ0EÓ&FˆÕ#ó    c                 ó¸  — 	 |r|j                  «       ni }|j                  d«      |j                  di «      j                  d«      |j                  di «      j                  d«      dœ}t        j                  t        j
                  «      j                  d«      |d<   t        |«      t        u r3t        |«      t        u r"t        ||z
  j                  «       dz  «      |d	<   g }|j                  d
i «      j                  di «      }	|	j                  «       D ]Z  \  }
}|
j                  d«      sŒ|
dk(  r||d<   Œ#|
dk(  r||d<   Œ.|j                  |
j                  dd«      t        |«      dœ«       Œ\ ||d<   | j                  €t!        d«      ‚t"        j$                  j'                  | j                  | j(                  t+        j,                  |t        ¬«      ¬«      }|j.                  dk7  r# |d|j0                  › d|j.                  › �«       y  |d|j0                  › �«       y # t         $ r*} |d|› dt3        j4                  «       › �«       Y d }~y d }~ww xY w)NÚmodelÚusageÚprompt_tokensÚcompletion_tokens)ÚmodelIdÚinputTokenCountÚoutputTokenCountz%Y-%m-%dT%H:%M:%SZÚ	timestampiè  ÚinvocationLatencyÚlitellm_paramsÚmetadataÚ
greenscaleÚgreenscale_projectÚprojectÚgreenscale_applicationÚapplicationÚgreenscale_Ú )ÚkeyÚvalueÚtagsz.Greenscale Logger Error - No logging URL found)Údefault)r   ÚdataéÈ   zGreenscale Logger Error - z, zGreenscale Logger Succeeded - z, Stack trace: )Ú
model_dumpÚgetr   Únowr   ÚutcÚstrftimeÚtypeÚintÚtotal_secondsÚitemsÚ
startswithÚappendÚreplaceÚstrr   Ú	ExceptionÚlitellmÚmodule_level_clientÚpostr   ÚjsonÚdumpsÚstatus_codeÚtextÚ	tracebackÚ
format_exc)r   ÚkwargsÚresponse_objÚ
start_timeÚend_timeÚprint_verboseÚresponse_jsonr+   r)   r   r'   r(   ÚresponseÚes                 r   Ú	log_eventzGreenscaleLogger.log_event   s1  € ð'	cÙ@L <×#:Ñ#:Ô#<ÐRTˆMà!Ÿ:™: gÓ.Ø#0×#4Ñ#4°W¸bÓ#A×#EÑ#EÀoÓ#VØ$1×$5Ñ$5°g¸rÓ$B×$FÑ$FÐGZÓ$[ñˆDô
 !)§¡¬X¯\©\Ó :× CÑ CÐDXÓ YˆD�Ñä�H‹~¤Ñ)¬d°:Ó.>Ä(Ñ.JÜ,/°¸JÑ1F×0UÑ0UÓ0WÐZ^Ñ0^Ó,_�Ð(Ñ)ð ˆDØ$Ÿj™jÐ)9¸2Ó>×BÑBÀ:ÈrÓRˆHØ&Ÿn™nÓ.ò b‘
��UØ—>‘> ,Õ/ØÐ2Ò2Ø*/˜˜YšØÐ 8Ò8Ø.3˜˜]Ò+àŸ™¨C¯K©K¸ÀrÓ,JÔUXÐY^ÓU_Ñ$`Õaðbð  ˆD�‰Là×*Ñ*Ð2ÜÐ PÓQÐQä%×9Ñ9×>Ñ>Ø×+Ñ+ØŸ™Ü—Z‘Z ¬cÔ2ð ?ó ˆHð
 ×#Ñ# sÒ*ÙÐ :¸8¿=¹=¸/ÈÈH×L`ÑL`ÐKaÐbÕcáÐ >¸x¿}¹}¸oÐNÕOøÜò 	cÙÐ6°q°c¸Ì×I]ÑI]ÓI_ÐH`Ða×bÑbûð	cús%   ‚D+H& Ä.C!H& ÈH& È&	IÈ/ IÉIN)Ú__name__Ú
__module__Ú__qualname__r   rL   © r   r   r   r   	   s   „ òGó(cr   r   )r>   rB   r   r   Útypingr   r;   r   rP   r   r   ú<module>rR      s    ðÛ Û ß 'Ý ã ÷3cò 3cr   