Ë
    uPêjì  ã                   óŽ   — U d dl mZmZ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«       Zeed	<    G d
„ d«      Zy)é    )ÚAsyncIteratorÚ	CoroutineÚMapping)ÚFinalÚcastN)ÚGenerateContentCompletionKwargs)ÚGenericLiteLLMParams)ÚModelResponseé   )ÚGoogleGenAIAdapterÚGOOGLE_GENAI_ADAPTERc                   ó°  — e Zd ZdZe	 	 	 	 ddedeeeef      eeef   z  deeef   dz  de	de
dz  deeef   dz  d	efd
„«       Ze	 	 ddedeeeef      eeef   z  de
deeef   dz  de	ded	eeef   ee   z  fd„«       Ze	 	 	 ddedeeeef      eeef   z  de
deeef   dz  de	de	ded	eeef   ee   z  eddeeef   ee   z  f   z  fd„«       Zy)Ú"GenerateContentToCompletionHandlerzaHandler for transforming generate_content calls to completion format when provider config is NoneNÚmodelÚcontentsÚconfigÚstreamÚlitellm_paramsÚextra_kwargsÚreturnc                 ó¬   — t        j                  d| |||dœ|xs i ¤Ž}t        |«      }|�d|v r|d   |d<   d|v r|d   |d<   |r||d<   t        di |¤ŽS )z1Prepare kwargs for litellm.completion/acompletion)r   r   r   r   ÚmetadataÚextra_headersr   © )r   Ú(translate_generate_content_to_completionÚdictr   )r   r   r   r   r   r   Úcompletion_requestÚcompletion_kwargss           új/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/google_genai/adapters/handler.pyÚ_prepare_completion_kwargsz=GenerateContentToCompletionHandler._prepare_completion_kwargs   s˜   € ô %9×$aÑ$að %
ØØØØ)ñ	%
ð
 Ò!˜rñ%
Ðô $(Ð(:Ó#;Ðð Ð#à˜\Ñ)Ø0<¸ZÑ0HÐ! *Ñ-à ,Ñ.Ø5AÀ/Ñ5RÐ! /Ñ2áØ*0Ð˜hÑ'ä.ÑCÐ1BÑCÐCó    Úkwargsc              ‹   ó¶  K  — t         j                  | |||||¬«      }	 t        j                  di |¤Žƒ d{  –—† }|rUt	        |d«      s%t
        j                  t        t        |«      «      }|S t
        j                  |«      }	|	�|	S t        d«      ‚t
        j                  t        t        |«      «      }|S 7 Œ€# t        $ r}
t        d|
› �«      ‚d}
~
ww xY w­w)zDHandle generate_content call asynchronously using completion adapter©r   r   r   r   r   r   NÚ	__aiter__ú&Failed to transform streaming responsez8Error calling litellm.acompletion for generate_content: r   )r   r    ÚlitellmÚacompletionÚhasattrr   Ú(translate_completion_to_generate_contentr   r
   Ú,translate_completion_output_params_streamingÚ
ValueErrorÚ	Exception)r   r   r   r   r   r"   r   Úcompletion_responseÚgenerate_content_responseÚtransformed_streamÚes              r   Úasync_generate_content_handlerzAGenerateContentToCompletionHandler.async_generate_content_handler6   s
  è ø€ ô $F×#`Ñ#`ØØØØØ)Øð $aó $
Ðð	]Ü/6×/BÑ/BÑ/WÐEVÑ/W×)WÐáô Ð2°KÔ@ä0D×0mÑ0mÜœ]Ð,?Ó@ó1Ð-ð 5Ð4ô 1E×0qÑ0qØ+ó1Ð&ð *Ð5Ø1Ð1Ü$Ð%MÓNÐNô -A×,iÑ,iÜœÐ(;Ó<ó-Ð)ð 1Ð0ð1 *Xùô4 ò 	]ÜÐWÐXYÐWZÐ[Ó\Ð\ûð	]üsP   ‚CŸB: ·B8¸6B: Á.CÁ/B: ÂCÂ/B: Â7CÂ8B: Â:	CÃCÃCÃCÚ	_is_asyncc           	      óÔ  — |rt        j                  d| ||||dœ|¤ŽS t         j                  | |||||¬«      }	 t        j                  di |¤Ž}|rUt        |d«      s%t        j                  t        t        |«      «      }	|	S t        j                  |«      }
|
�|
S t        d«      ‚t        j                  t        t        |«      «      }	|	S # t        $ r}t        d|› �«      ‚d}~ww xY w)z5Handle generate_content call using completion adapter)r   r   r   r   r   r$   Ú__iter__Nr&   z7Error calling litellm.completion for generate_content: r   )r   r2   r    r'   Ú
completionr)   r   r*   r   r
   r+   r,   r-   )r   r   r   r   r   r3   r"   r   r.   r/   r0   r1   s               r   Úgenerate_content_handlerz;GenerateContentToCompletionHandler.generate_content_handlerh   s2  € ñ Ü5×TÑTð ØØ!ØØØ-ñð ñð ô $F×#`Ñ#`ØØØØØ)Øð $aó $
Ðð	\Ü)0×);Ñ);Ñ)PÐ>OÑ)PÐáô Ð2°JÔ?ä0D×0mÑ0mÜœ]Ð,?Ó@ó1Ð-ð 5Ð4ô 1E×0qÑ0qØ+ó1Ð&ð *Ð5Ø1Ð1Ü$Ð%MÓNÐNô -A×,iÑ,iÜœÐ(;Ó<ó-Ð)ð 1Ð0øäò 	\ÜÐVÐWXÐVYÐZÓ[Ð[ûð	\ús$   ºAC ÂC Â/C Ã	C'ÃC"Ã"C')NFNN)NF)NFF)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚstaticmethodÚstrÚlistr   ÚobjectÚboolr	   r   r   r    r   Úbytesr2   r   r7   r   r!   r   r   r      s,  „ Ùkàð ,0ØØ6:Ø48ñ!DØð!Dà�t˜C ˜KÑ(Ñ)¨D°°f°Ñ,=Ñ=ð!Dð �S˜&�[Ñ! DÑ(ð!Dð ð	!Dð
 -¨tÑ3ð!Dð ˜c 6˜kÑ*¨TÑ1ð!Dð 
)ò!Dó ð!DðF ð
 ,0Øñ/]Øð/]à�t˜C ˜KÑ(Ñ)¨D°°f°Ñ,=Ñ=ð/]ð -ð/]ð �S˜&�[Ñ! DÑ(ð	/]ð
 ð/]ð ð/]ð 
ˆc�6ˆkÑ	˜]¨5Ñ1Ñ	1ò/]ó ð/]ðb ð
 ,0ØØñ:\Øð:\à�t˜C ˜KÑ(Ñ)¨D°°f°Ñ,=Ñ=ð:\ð -ð:\ð �S˜&�[Ñ! DÑ(ð	:\ð
 ð:\ð ð:\ð ð:\ð 
ˆc�6ˆkÑ	˜]¨5Ñ1Ñ	1°I¸dÀDÈ$ÈsÐTZÈ{ÑJ[Ð^kÐlqÑ^rÑJrÐ>rÑ4sÑ	sò:\ó ñ:\r!   r   )Úcollections.abcr   r   r   Útypingr   r   r'   Ú#litellm.types.google_genai.adaptersr   Úlitellm.types.routerr	   Úlitellm.types.utilsr
   Útransformationr   r   Ú__annotations__r   r   r!   r   ú<module>rI      s;   ðß =Ò =ß ã Ý OÝ 5Ý -å .ñ 1Ó2Ð �eÓ 2÷T\ò T\r!   