Ë
    uPêj*
  ã                   ó„   — d Z ddlmZ ddl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 erddlmZ  G d„ dee«      Zy)z†
Handler file for calls to Azure OpenAI's o1/o3 family of models

Written separately to handle faking streaming for o1 and o3 models.
é    )ÚCallable)ÚTYPE_CHECKINGÚOptionalN)ÚLogging)ÚModelResponseé   )ÚOpenAIChatCompletioné   )ÚBaseAzureLLM)ÚClientSessionc            *       óü   ‡ — e Zd Zdddddddddddi dddddfdedeej                  z  dededede	dz  d	e
dz  d
edz  de	dz  de	dz  de	dz  dedz  de	dz  dededz  dede	dz  de	dz  dedz  ded   f(ˆ fd„Zˆ xZS )ÚAzureOpenAIO1ChatCompletionNFÚmodel_responseÚtimeoutÚoptional_paramsÚlitellm_paramsÚlogging_objÚmodelÚmessagesÚprint_verboseÚapi_keyÚapi_baseÚapi_versionÚdynamic_paramsÚazure_ad_tokenÚacompletionÚheadersÚcustom_prompt_dictÚorganizationÚcustom_llm_providerÚdrop_paramsÚshared_sessionr   c                 ó|   •— | j                  ||	|
|||¬«      }t        ‰| �	  |||||||||	|
||||||||||||¬«      S )N)r   r   r   r   ÚclientÚ	_is_async)r   r   r   r   r   r   r   r   r   r   r   r   r   r   Ú	logger_fnr   r   r$   r   r    r!   r"   )Úget_azure_openai_clientÚsuperÚ
completion)Úselfr   r   r   r   r   r   r   r   r   r   r   r   r   r   r&   r   r   r$   r   r    r!   r"   Ú	__class__s                          €úm/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/azure/chat/o_series_handler.pyr)   z&AzureOpenAIO1ChatCompletion.completion   s‡   ø€ ð2 ×-Ñ-Ø)ØØØ#ØØ!ð .ó 
ˆô ‰wÑ!Ø)ØØ+Ø)Ø#ØØØ'ØØØ#Ø)Ø)Ø#ØØØ1ØØ%Ø 3Ø#Ø)ð- "ó 
ð 	
ó    )Ú__name__Ú
__module__Ú__qualname__r   ÚfloatÚhttpxÚTimeoutÚdictÚLiteLLMLoggingObjÚstrÚlistr   Úboolr   r)   Ú__classcell__)r+   s   @r,   r   r      sF  ø„ ð !Ø $Ø)-Ø"Ø#Ø"&Ø&*Ø%)Ø!ØØ#Ø#%ØØ#'Ø*.Ø#'Ø48ñ/8
à%ð8
ð ˜Ÿ™Ñ&ð8
ð ð	8
ð
 ð8
ð 'ð8
ð �T‰zð8
ð ˜‘+ð8
ð   $‘ð8
ð �t‘ð8
ð ˜‘*ð8
ð ˜4‘Zð8
ð ˜t™ð8
ð ˜d™
ð8
ð ð8
ð" ˜‘ð#8
ð$ !ð%8
ð( ˜D‘jð)8
ð* ! 4™Zð+8
ð, ˜D‘[ð-8
ð. ! Ñ1÷/8
ñ 8
r-   r   )Ú__doc__Úcollections.abcr   Útypingr   r   r2   Ú*litellm.litellm_core_utils.litellm_loggingr   r5   Úlitellm.types.utilsr   Úopenai.openair	   Úcommon_utilsr   Úaiohttpr   r   © r-   r,   ú<module>rC      s6   ðñõ %ß *ã å SÝ -å 1Ý 'áÝ%ô9
 ,Ð0Dõ 9
r-   