Ë
    uPêj·  ã                   ó    — d dl Z d dlmZmZmZ d dlZd dlmZ d dlm	Z	 d dl
mZ d dlmZmZ ddlmZ erd dlZd d	lmZ eZneZ G d
„ de	«      Zy)é    N)ÚTYPE_CHECKINGÚAnyÚFinal)ÚBaseLLMException)ÚOpenAIGPTConfig)ÚAllMessageValues)ÚModelResponseÚUsageé   ©ÚOobaboogaError)ÚLoggingc                   óì   — e Zd Z	 ddededeej                  z  dz  defd„Z		 	 ddedej                  d	ed
ededee   dededddedz  dedz  defd„Z	 	 ddededee   dedededz  dedz  defd„Zy)ÚOobaboogaConfigNÚerror_messageÚstatus_codeÚheadersÚreturnc                 ó   — t        |||¬«      S )N)r   Úmessager   r   )Úselfr   r   r   s       úo/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/oobabooga/chat/transformation.pyÚget_error_classzOobaboogaConfig.get_error_class   s   € ô ¨+¸}ÐV]Ô^Ð^ó    ÚmodelÚraw_responseÚmodel_responseÚlogging_objÚrequest_dataÚmessagesÚoptional_paramsÚlitellm_paramsÚencodingztiktoken.Encoding | NoneÚapi_keyÚ	json_modec                 óf  — |j                  ||
|j                  d|i¬«       	 |j                  «       }d|v rt	        |d   |j
                  ¬«      ‚	 |d   d   d   d   |j                  d   j                  _        t        t        j                  «       «      |_        ||_        t        |d	   d
   |d	   d   |d	   d   ¬«      }t        |d	|«       |S # t        $ r" t	        |j                  |j
                  ¬«      ‚w xY w# t        $ r%}t	        t        |«      |j
                  ¬«      ‚d }~ww xY w)NÚcomplete_input_dict)Úinputr$   Úoriginal_responseÚadditional_args)r   r   ÚerrorÚchoicesr   r   ÚcontentÚusageÚprompt_tokensÚcompletion_tokensÚtotal_tokens)r/   r0   r1   )Ú	post_callÚtextÚjsonÚ	Exceptionr   r   r,   r   r-   ÚstrÚintÚtimeÚcreatedr   r
   Úsetattr)r   r   r   r   r   r   r    r!   r"   r#   r$   r%   Úcompletion_responseÚer.   s                  r   Útransform_responsez"OobaboogaConfig.transform_response    s_  € ð 	×ÑØØØ*×/Ñ/Ø2°LÐAð	 	ô 	
ð	bØ)5×):Ñ):Ó)<Ðð Ð)Ñ)Ü Ø+¨GÑ4Ø(×4Ñ4ôð ð
Ø<OÐPYÑ<ZÐ[\Ñ<]Ð^gÑ<hÐirÑ<s�×&Ñ& qÑ)×1Ñ1Ô9ô "%¤T§Y¡Y£[Ó!1ˆÔØ$ˆÔÜØ-¨gÑ6°ÑGØ1°'Ñ:Ð;NÑOØ,¨WÑ5°nÑEô
ˆô
 	� ¨Ô/ØÐøô1 ò 	bÜ ¨×):Ñ):È×H`ÑH`ÔaÐað	bûô ò Ü$Ü ›FØ ,× 8Ñ 8ôð ûðús#   £C Á*D Ã+C?Ä	D0Ä D+Ä+D0Úapi_basec                 ó$   — dddœ}|�d|› �|d<   |S )Nzapplication/json)Úacceptzcontent-typezToken ÚAuthorization© )r   r   r   r    r!   r"   r$   r>   s           r   Úvalidate_environmentz$OobaboogaConfig.validate_environmentS   s/   € ð )Ø.ñ
ˆð ÐØ)/°¨yÐ'9ˆG�OÑ$Øˆr   )N)NN)Ú__name__Ú
__module__Ú__qualname__r6   r7   ÚdictÚhttpxÚHeadersr   r   ÚResponser	   ÚLoggingClassÚlistr   Úboolr=   rC   rB   r   r   r   r      sD  „ ð
 04ñ	_àð_ð ð_ð ˜Ÿ™Ñ%¨Ñ,ð	_ð
 
ó_ð& #Ø!%ñ1àð1ð —n‘nð1ð &ð	1ð
 "ð1ð ð1ð Ð'Ñ(ð1ð ð1ð ð1ð -ð1ð �t‘ð1ð ˜$‘;ð1ð 
ó1ðt #Ø#ñàðð ðð Ð'Ñ(ð	ð
 ðð ðð �t‘ðð ˜‘*ðð 
ôr   r   )r8   Útypingr   r   r   rH   Ú)litellm.llms.base_llm.chat.transformationr   Ú+litellm.llms.openai.chat.gpt_transformationr   Úlitellm.types.llms.openair   Úlitellm.types.utilsr	   r
   Úcommon_utilsr   ÚtiktokenÚ*litellm.litellm_core_utils.litellm_loggingr   ÚLiteLLMLoggingObjrK   r   rB   r   r   ú<module>rW      sA   ðÛ ß ,Ñ ,ã å FÝ GÝ 6ß 4å )áÛåWà$�Là€LôL�oõ Lr   