Ë
    tPêjÆ  ã                   ó\   — d dl mZ d dlmZmZ d dlmZ d dlmZ d dl	m
Z
  G d„ dee«      Zy)	é    )ÚCustomLogger)ÚPromptManagementBaseÚPromptManagementClient)ÚAllMessageValues)Ú
PromptSpec)ÚStandardCallbackDynamicParamsc                   ó€  — e Zd Z	 	 d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edefd„«       Zdedz  de
dz  d
e	defd„Z	 	 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	 	 	 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y)ÚCustomPromptManagementÚignore_prompt_manager_modelNÚ%ignore_prompt_manager_optional_paramsc                 ó    — || _         || _        y )N)r   r   )Úselfr   r   Úkwargss       úr/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/custom_prompt_management.pyÚ__init__zCustomPromptManagement.__init__   s   € ð ,GˆÔ(Ø5ZˆÕ2ó    ÚmodelÚmessagesÚnon_default_paramsÚ	prompt_idÚprompt_variablesÚdynamic_callback_paramsÚprompt_specÚprompt_labelÚprompt_versionÚreturnc                 ó   — |||fS )ax  
        Returns:
        - model: str - the model to use (can be pulled from prompt management tool)
        - messages: List[AllMessageValues] - the messages to use (can be pulled from prompt management tool)
        - non_default_params: dict - update with any optional params (e.g. temperature, max_tokens, etc.) to use (can be pulled from prompt management tool)
        © )r   r   r   r   r   r   r   r   r   r   r   r   s               r   Úget_chat_completion_promptz1CustomPromptManagement.get_chat_completion_prompt   s   € ð( �hÐ 2Ð2Ð2r   c                  ó   — y)Nzcustom-prompt-managementr   )r   s    r   Úintegration_namez'CustomPromptManagement.integration_name+   s   € à)r   c                  ó   — y)NTr   )r   r   r   r   s       r   Úshould_run_prompt_managementz3CustomPromptManagement.should_run_prompt_management/   s   € ð r   c                 ó   — t        d«      ‚)Nz?Custom prompt management does not support compile prompt helper©ÚNotImplementedError)r   r   r   r   r   r   r   s          r   Ú_compile_prompt_helperz-CustomPromptManagement._compile_prompt_helper7   s   € ô "Ð"cÓdÐdr   c              ƒ   ó    K  — t        d«      ‚­w)NzECustom prompt management does not support async compile prompt helperr%   )r   r   r   r   r   r   r   s          r   Úasync_compile_prompt_helperz2CustomPromptManagement.async_compile_prompt_helperB   s   è ø€ ô "Ð"iÓjÐjùs   ‚)FF)NNNFF)NN)NNN)Ú__name__Ú
__module__Ú__qualname__Úboolr   ÚstrÚlistr   Údictr   r   ÚintÚtupler   Úpropertyr!   r#   r   r'   r)   r   r   r   r
   r
      s  „ ð 49Ø=Bñ[à%)¨D¡[ð[ð 04°d©{ó[ð" *.Ø#'Ø%)Ø38Ø=Bñ3àð3ð Ð'Ñ(ð3ð !ð	3ð
 ˜‘:ð3ð  ™+ð3ð "?ð3ð   $Ñ&ð3ð ˜D‘jð3ð ˜d™
ð3ð &*¨D¡[ð3ð 04°d©{ð3ð 
ˆs�DÐ)Ñ*¨DÐ0Ñ	1ó3ð, ð* #ò *ó ð*ðà˜‘:ðð   $Ñ&ðð "?ð	ð
 
óð $(Ø%)ñ	eà˜‘:ð	eð   $Ñ&ð	eð  ™+ð		eð
 "?ð	eð ˜D‘jð	eð ˜d™
ð	eð 
 ó	eð  *.Ø#'Ø%)ñ	kà˜‘:ð	kð  ™+ð	kð "?ð		kð
   $Ñ&ð	kð ˜D‘jð	kð ˜d™
ð	kð 
 ô	kr   r
   N)Ú"litellm.integrations.custom_loggerr   Ú+litellm.integrations.prompt_management_baser   r   Úlitellm.types.llms.openair   Ú"litellm.types.prompts.init_promptsr   Úlitellm.types.utilsr   r
   r   r   r   ú<module>r9      s+   ðÝ ;÷õ 7Ý 9Ý =ô@k˜\Ð+?õ @kr   