Ë
    tPêj­&  ã            
       óÈ   — d dl mZmZ d dlmZmZm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«      Zd
edz  dedz  dedz  deeef   fd„Z G d„ de«      Zy)é    )ÚABCÚabstractmethod)ÚTYPE_CHECKINGÚAnyÚFinal)Ú	TypedDict)ÚAllMessageValues)Ú
PromptSpec)ÚStandardCallbackDynamicParams)ÚLoggingc                   ón   — e Zd ZU edz  ed<   ee   ed<   edz  ed<   eeef   dz  ed<   ee   dz  ed<   y)ÚPromptManagementClientNÚ	prompt_idÚprompt_templateÚprompt_template_modelÚprompt_template_optional_paramsÚcompleted_messages)	Ú__name__Ú
__module__Ú__qualname__ÚstrÚ__annotations__Úlistr	   Údictr   © ó    úp/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/prompt_management_base.pyr   r      sF   … Ø�T‰zÓØÐ*Ñ+Ó+Ø ™:Ó%Ø%)¨#¨s¨(¡^°dÑ%:Ó:ØÐ-Ñ.°Ñ5Ô5r   r   Úprompt_specNÚignore_prompt_manager_modelÚ%ignore_prompt_manager_optional_paramsÚreturnc                 óÂ   — | �| j                   nd }t        |«      xs t        |d uxr |j                  «      t        |«      xs t        |d uxr |j                  «      fS ©N)Úlitellm_paramsÚboolr   r    )r   r   r    Úspec_paramss       r   Ú#resolve_prompt_manager_ignore_flagsr'      sl   € ð
 8CÐ7N˜×3Ò3ÐTX€KäÐ(Ó)Òv¬T°+ÀTÐ2IÒ2uÈk×NuÑNuÓ-vÜÐ2Ó3ò 	_Ü� 4Ð'Ò]¨K×,]Ñ,]Ó^ðð r   c                    ó6  — e Zd Zeedefd„«       «       Zededz  dedz  dede	fd„«       Z
e	 	 ddedz  dedz  dedz  ded	edz  d
edz  defd„«       Ze	 	 	 d dedz  dedz  dededz  d	edz  d
edz  defd„«       Zdee   dee   dee   fd„Z	 	 	 d dededz  dee   ded	edz  d
edz  dedz  defd„Z	 	 	 d dedz  dedz  dee   dededz  d	edz  d
edz  defd„Zdededefd„Z	 	 d!dedee   dedede	dz  de	dz  fd„Z	 	 	 	 	 d"dedee   dededz  dedz  dededz  d	edz  d
edz  de	dz  de	dz  deeee   ef   fd„Z	 	 	 	 	 	 d#dedee   dededz  dedz  dedddedz  dee   dz  d	edz  d
edz  de	dz  de	dz  deeee   ef   fd„Zy)$ÚPromptManagementBaser!   c                  ó   — y r#   r   )Úselfs    r   Úintegration_namez%PromptManagementBase.integration_name$   s   € ð 	r   r   Nr   Údynamic_callback_paramsc                  ó   — y r#   r   )r+   r   r   r-   s       r   Úshould_run_prompt_managementz1PromptManagementBase.should_run_prompt_management)   s   € ð 	r   Úprompt_variablesÚprompt_labelÚprompt_versionc                  ó   — y r#   r   )r+   r   r   r0   r-   r1   r2   s          r   Ú_compile_prompt_helperz+PromptManagementBase._compile_prompt_helper2   s   € ð 	r   c              ƒ   ó   K  — y ­wr#   r   )r+   r   r0   r-   r   r1   r2   s          r   Úasync_compile_prompt_helperz0PromptManagementBase.async_compile_prompt_helper>   s   è ø€ ð 	ùs   ‚r   Úclient_messagesc                 ó   — ||z   S r#   r   )r+   r   r7   s      r   Úmerge_messagesz#PromptManagementBase.merge_messagesJ   s   € ð
  Ñ0Ð0r   c                 ó”   — | j                  ||||||¬«      }	 |d   |z   }	|	|d<   |S # t        $ r}
t        d|
› d|› �«      ‚d }
~
ww xY w©N)r   r   r0   r-   r1   r2   r   zError compiling prompt: z. Prompt id=r   )r4   Ú	ExceptionÚ
ValueError)r+   r   r0   r7   r-   r1   r2   r   Úcompiled_prompt_clientÚmessagesÚes              r   Úcompile_promptz#PromptManagementBase.compile_promptQ   s…   € ð )-×(CÑ(CØØ#Ø-Ø$;Ø%Ø)ð )Dó )
Ðð	TØ4Ð5FÑGÈ/ÑYˆHð 8@ÐÐ3Ñ4Ø%Ð%øô	 ò 	TÜÐ7¸°s¸,ÀyÀkÐRÓSÐSûð	Tús   ™( ¨	A±AÁAc              ƒ   ó°   K  — | j                  ||||||¬«      ƒ d {  –—† }	 |d   |z   }	|	|d<   |S 7 Œ# t        $ r}
t        d|
› d|› �«      ‚d }
~
ww xY w­wr;   )r6   r<   r=   )r+   r   r0   r7   r-   r   r1   r2   r>   r?   r@   s              r   Úasync_compile_promptz)PromptManagementBase.async_compile_promptl   s’   è ø€ ð /3×.NÑ.NØØ#Ø-Ø$;Ø%Ø)ð /Oó /
÷ )
Ðð	TØ4Ð5FÑGÈ/ÑYˆHð 8@ÐÐ3Ñ4Ø%Ð%ð)
ùô ò 	TÜÐ7¸°s¸,ÀyÀkÐRÓSÐSûð	Tüs-   ‚A�2žA£4 «A´	A½AÁAÁAÚprompt_management_clientÚmodelc                 óT   — |d   �|d   S |j                  | j                  › d�d«      S )Nr   ú/Ú )Úreplacer,   )r+   rD   rE   s      r   Ú_get_model_from_promptz+PromptManagementBase._get_model_from_prompt‡   s9   € Ø#Ð$;Ñ<ÐHØ+Ð,CÑDÐDà—=‘= D×$9Ñ$9Ð#:¸!Ð!<¸bÓAÐAr   r?   Únon_default_paramsr   r    c                 ót   — |d   xs |}|d   xs i }i |¥|s|ni ¥}	|s| j                  ||¬«      }n|}|||	fS )Nr   r   )rD   rE   )rJ   )
r+   r   r?   rK   rE   r   r    r   r   Úupdated_non_default_paramss
             r   Úpost_compile_prompt_processingz3PromptManagementBase.post_compile_prompt_processing�   st   € ð %4Ð4HÑ$IÒ$UÈXÐà1@ÐAbÑ1cÒ1iÐgiÐ'ð-
Ø ð-
á6[Ñ.Ðacð-
Ð"ñ
 +Ø×/Ñ/ÈÐ`eÐ/Óf‰EàˆEàÐ(Ð*DÐDÐDr   c                 óÀ   — |€|||fS | j                  |||¬«      s|||fS | j                  ||||||	¬«      }t        ||
|¬«      \  }}| j                  ||||||¬«      S )N©r   r   r-   )r   r0   r7   r-   r1   r2   ©r   r   r    ©r   r?   rK   rE   r   r    )r/   rA   r'   rN   )r+   rE   r?   rK   r   r0   r-   r   r1   r2   r   r    r   Úresolved_ignore_modelÚresolved_ignore_optional_paramss                  r   Úget_chat_completion_promptz/PromptManagementBase.get_chat_completion_prompt¦   s¶   € ð ÐØ˜(Ð$6Ð6Ð6Ø×0Ñ0ØØ#Ø$;ð 1ô 
ð
 ˜(Ð$6Ð6Ð6à!%×!4Ñ!4ØØ-Ø$Ø$;Ø%Ø)ð "5ó "
ˆô BeØ#Ø(CØ2WôB
Ñ>ÐÐ>ð
 ×2Ñ2Ø+ØØ1ØØ(=Ø2Qð 3ó 
ð 	
r   Úlitellm_logging_objÚLiteLLMLoggingObjÚtoolsc           	   ƒ   óÐ   K  — | j                  |||¬«      s|||fS | j                  ||||||
|¬«      ƒ d {  –—† }t        |||¬«      \  }}| j                  ||||||¬«      S 7 Œ,­w)NrP   )r   r0   r7   r-   r   r1   r2   rQ   rR   )r/   rC   r'   rN   )r+   rE   r?   rK   r   r0   r-   rV   r   rX   r1   r2   r   r    r   rS   rT   s                    r   Ú async_get_chat_completion_promptz5PromptManagementBase.async_get_chat_completion_promptÔ   s´   è ø€ ð  ×0Ñ0ØØ#Ø$;ð 1ô 
ð
 ˜(Ð$6Ð6Ð6à'+×'@Ñ'@ØØ-Ø$Ø$;Ø#Ø%Ø)ð (Aó (
÷ "
ˆô BeØ#Ø(CØ2WôB
Ñ>ÐÐ>ð
 ×2Ñ2Ø+ØØ1ØØ(=Ø2Qð 3ó 
ð 	
ð"
ús   ‚5A&·A$¸-A&)NN)NNN)FF)NNNFF)NNNNFF)r   r   r   Úpropertyr   r   r,   r
   r   r%   r/   r   Úintr   r4   r6   r   r	   r9   rA   rC   rJ   rN   ÚtuplerU   rZ   r   r   r   r)   r)   #   s–  „ ØØð #ò ó ó ðð ðà˜‘:ðð   $Ñ&ðð "?ð	ð
 
òó ðð ð $(Ø%)ñ	à˜‘:ð	ð   $Ñ&ð	ð  ™+ð		ð
 "?ð	ð ˜D‘jð	ð ˜d™
ð	ð 
 ò	ó ð	ð ð *.Ø#'Ø%)ñ	à˜‘:ð	ð  ™+ð	ð "?ð		ð
   $Ñ&ð	ð ˜D‘jð	ð ˜d™
ð	ð 
 ò	ó ð	ð1àÐ.Ñ/ð1ð Ð.Ñ/ð1ð 
ÐÑ	ó	1ð $(Ø%)Ø)-ñ&àð&ð  ™+ð&ð Ð.Ñ/ð	&ð
 "?ð&ð ˜D‘jð&ð ˜d™
ð&ð   $Ñ&ð&ð 
 ó&ðB *.Ø#'Ø%)ñ&à˜‘:ð&ð  ™+ð&ð Ð.Ñ/ð	&ð
 "?ð&ð   $Ñ&ð&ð ˜D‘jð&ð ˜d™
ð&ð 
 ó&ð6BÐ?Uð BÐ^að BÐfió Bð 49Ø=BñEà/ðEð Ð'Ñ(ðEð !ð	Eð
 ðEð &*¨D¡[ðEð 04°d©{óEðB *.Ø#'Ø%)Ø38Ø=Bñ,
àð,
ð Ð'Ñ(ð,
ð !ð	,
ð
 ˜‘:ð,
ð  ™+ð,
ð "?ð,
ð   $Ñ&ð,
ð ˜D‘jð,
ð ˜d™
ð,
ð &*¨D¡[ð,
ð 04°d©{ð,
ð 
ˆs�DÐ)Ñ*¨DÐ0Ñ	1ó,
ðn *.Ø#'Ø#'Ø%)Ø38Ø=Bñ-
àð-
ð Ð'Ñ(ð-
ð !ð	-
ð
 ˜‘:ð-
ð  ™+ð-
ð "?ð-
ð 1ð-
ð   $Ñ&ð-
ð �D‰z˜DÑ ð-
ð ˜D‘jð-
ð ˜d™
ð-
ð &*¨D¡[ð-
ð 04°d©{ð-
ð 
ˆs�DÐ)Ñ*¨DÐ0Ñ	1ô-
r   r)   )Úabcr   r   Útypingr   r   r   Útyping_extensionsr   Úlitellm.types.llms.openair	   Ú"litellm.types.prompts.init_promptsr
   Úlitellm.types.utilsr   Ú*litellm.litellm_core_utils.litellm_loggingr   rW   r   r%   r]   r'   r)   r   r   r   ú<module>re      sy   ðß #ß ,Ñ ,å 'å 6Ý 9Ý =áÝWô6˜Yô 6ð
Ø˜dÑ"ð
à!%¨¡ð
ð ,0°$©;ð
ð ˆ4�ˆ:Ñó	
ô^
˜3õ ^
r   