
    h                     d    d dl mZmZmZ d dlmZ d dlmZmZ d dl	m
Z
 d dlmZ  G d dee      Zy)	    )ListOptionalTuple)CustomLogger)PromptManagementBasePromptManagementClient)AllMessageValues)StandardCallbackDynamicParamsc                       e Zd Z	 	 ddedee   dedee   dee   dedee   d	ee	   d
e
eee   ef   fdZed
efd       Zdeded
efdZ	 	 ddedee   dedee   d	ee	   d
efdZy)CustomPromptManagementNmodelmessagesnon_default_params	prompt_idprompt_variablesdynamic_callback_params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)
         )	selfr   r   r   r   r   r   r   r   s	            i/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/integrations/custom_prompt_management.pyget_chat_completion_promptz1CustomPromptManagement.get_chat_completion_prompt   s    " h 222    c                      y)Nzcustom-prompt-managementr   )r   s    r   integration_namez'CustomPromptManagement.integration_name    s    )r   c                      y)NTr   )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   s         r   _compile_prompt_helperz-CustomPromptManagement._compile_prompt_helper+   s     "M
 	
r   )NN)__name__
__module____qualname__strr   r	   dictr   r
   intr   r   propertyr   boolr   r   r"   r   r   r   r   r      s    '+(,33 '(3 !	3
 C=3 #4.3 "?3 sm3 !3 
sD)*D0	13& *# * * "? 
	 '+(,



 #4.

 "?	


 sm

 !

 
 

r   r   N)typingr   r   r   "litellm.integrations.custom_loggerr   +litellm.integrations.prompt_management_baser   r   litellm.types.llms.openair	   litellm.types.utilsr
   r   r   r   r   <module>r0      s*    ( ( ; 7 =)
\+? )
r   