tPj7dZddlZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZerdd lmZneZd d lmZmZGd d eZy)z Dotprompt manager that integrates with LiteLLM's prompt management system. Builds on top of PromptManagementBase to provide .prompt file support. N) TYPE_CHECKINGAnyFinal)CustomPromptManagement)PromptManagementClient)AllMessageValues) PromptSpec)StandardCallbackDynamicParams)Logging) PromptManagerPromptTemplatec "eZdZdZ d,dedzdedzdeezdzdedzfdZedefd Zede fd Z dedzd e dzd e de fd Z d-dedzd e dzdedzd e dedzdedzdefdZ d.dedzdedzd e d e dzdedzdedzdefdZ d/dedeedededzdedzd e d e dzdedzdedzde dzde dzdeeeeeffdZ d0dedeedededzdedzd e ded e dzdeedzdedzdedzde dzde dzdeeeeeffdZdedeefdZdededefd Zd!edefd"Zdeddfd#Zd1d$Zded%eeefddfd&Z d'eeeeeffddfd(Z!deeeeefffd)Z"d*edeeeffd+Z#y)2DotpromptManageran Dotprompt manager that integrates with LiteLLM's prompt management system. This class enables using .prompt files with the litellm completion() function by implementing the PromptManagementBase interface. Usage: # Set global prompt directory litellm.prompt_directory = "path/to/prompts" # Use with completion response = litellm.completion( model="dotprompt/gpt-4", prompt_id="my_prompt", prompt_variables={"variable": "value"}, messages=[{"role": "user", "content": "This will be combined with the prompt"}] ) Nprompt_directory prompt_file prompt_data prompt_idcddl}|xs |j|_t|trt j ||_n |xsi|_d|_||_ ||_ y)Nr) litellmglobal_prompt_directoryr isinstancestrjsonloadsr_prompt_managerrr)selfrrrrrs u/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/dotprompt/dotprompt_manager.py__init__zDotpromptManager.__init__+sY  0 SG4S4S k3 '#zz+6D *0bD 59&"returncy)z>       ##%)(C2W?  r litellm_logging_objtoolschKddlm}|j|||||||||| | | | | d{S7w)zY Async version - delegates to PromptManagementBase async implementation. rrO) rr0rSr+r*rTr1r2rLrMN)rQrP async_get_chat_completion_prompt)rr?rDrKrr0r+rSr*rTr1r2rLrMrPs rrVz1DotpromptManager.async_get_chat_completion_promptsY& U)JJ    - 3$;#%)(C2WK    s )202rCc |j}g}d}g}|jd}|D]l}|j}|jdrY|r@|r>|j|j |dj |jd}|ddjg}~|jdrY|r@|r>|j|j |dj |jd}|ddjg}|jd rZ|r@|r>|j|j |dj |jd }|d djg}S|r|j|hd}|g}o|rD|rBdj |j}|r!|j|j |||s#|r!|j|j d||S) a9 Convert rendered prompt content to chat messages. This method supports multiple formats: 1. Simple text -> converted to user message 2. Text with role prefixes (System:, User:, Assistant:) -> parsed into separate messages 3. Already formatted as a single message N zSystem:systemzUser:userz Assistant: assistant )stripsplit startswithappend_create_messagejoin) rrCcontentrD current_rolecurrent_contentlinesline content_texts rr>z%DotpromptManager._convert_to_messagess*//1 }}T* -D::>#3"4)OOOD$8$8tyyQ`GaGgGgGi$jk% #'8>>#3"4.OOOD$8$8tyyQ`GaGgGgGi$jk* #'9??#4"5 #**40$*L'+fO5 -: O"&))O"<"B"B"DL 4 4\< PQG OOD00A Br rolerec ||dS)z5Create a message with the specified role and content.)rkrer$)rrkres rrcz DotpromptManager._create_message9s  r rAcXi}|j|j|j|S)z Extract optional parameters from the prompt template metadata. Includes parameters like temperature, max_tokens, etc. )rFupdate)rrArFs rr@z)DotpromptManager._extract_optional_params@s1 "$  # # /  " "8#;#; <r c ||_d|_y)z,Set the prompt directory and reload prompts.N)rr)rrs rset_prompt_directoryz%DotpromptManager.set_prompt_directoryNs 0#r cR|jr|jjyy)z&Reload all prompts from the directory.N)rreload_promptsr%s rrrzDotpromptManager.reload_promptsSs"     / / 1 r json_datac|jdd}|jdi}|jj|||y)zAdd a prompt from JSON data.rer7metadataN)getr) add_prompt)rrrsrerus radd_prompt_from_jsonz%DotpromptManager.add_prompt_from_jsonXs:"y"5#-- B7 &&y'8Dr prompts_datac:|jj|y)z%Load multiple prompts from JSON data.N)r)load_prompts_from_json_data)rrys rload_prompts_from_jsonz'DotpromptManager.load_prompts_from_json^s 77 Er c6|jjS)zGet all prompts in JSON format.)r)get_all_prompts_as_jsonr%s rget_prompts_as_jsonz$DotpromptManager.get_prompts_as_jsonbs""::<rcrr@rprrrrxr|rrr$r rrrs(*(,"&)- $ #*#4Z#CZ$& # : #(#$ $$ : $&"?   2$(%)6K:6K $&6K+ 6K "? 6K Dj 6Kd 6K 6Kz*.#'%) : + "?   $&  Dj  d    B*.#'%)38=B  '( !  :  +  "?  $& Dj d  &*D[ 04d{  sD)*D0 1 P*.#'#'%)38=B$ $ '($ ! $ : $ + $ "?$ /$  $&$ DzD $ Dj$ d $ &*D[$ 04d{$  sD)*D0 1$ L9S9TBR=S9v C # :J   D $S$T$ 2 EcEd38nEQUE F4T#s(^8K3LFQUF=T#tCH~*=%>=BSBT#s(^Br r)rrtypingrrr-litellm.integrations.custom_prompt_managementrrQrlitellm.types.llms.openair"litellm.types.prompts.init_promptsr litellm.types.utilsr *litellm.litellm_core_utils.litellm_loggingr rr)r rrr$r rrsD  ,,PN69=W9QB-QBr