tPj5dZddlZddlmZddlmZmZddlZddlm Z m Z ddl m Z de de dzfd ZGd d ZGd d Zy)zq Based on Google's GenAI Kit dotprompt implementation: https://google.github.io/dotprompt/reference/frontmatter/ N)Path)AnyFinal) DictLoaderselect_autoescape)ImmutableSandboxedEnvironment prompt_idreturncZ|jd\}}}|r|r|jr|Sy)N.v) rpartitionisdigit)r base separatorversions r/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/dotprompt/prompt_manager.pystrip_version_suffixrs/(33D9D)WTgoo/ cFeZdZdZ ddedeeefdzdedzfdZdZy) PromptTemplatez>Represents a single prompt template with metadata and content.Ncontentmetadata template_idc||_|xsi|_||_gd}|jjd|_|jjdijdi|_|jjdijd|_|jjdijdi|_i|_|jD]#}||vs|j||j|<%y)N)modelinputoutputrrschemarformat) rrrgetr input_schema output_format output_schemaoptional_params)selfrrrrestricted_keyskeys r__init__zPromptTemplate.__init__s   B &">]]&&w/  MM--gr:>>xL!]]..x<@@J!]]..x<@@2N!== ?C/),0MM#,>$$S) ?rc<d|jd|jdS)NzPromptTemplate(id='z ', model='z'))rrr%s r__repr__zPromptTemplate.__repr__-s"$T%5%5$6j BOOrNN) __name__ __module__ __qualname____doc__strdictrr(r+rrrrsEH +/"& ??sCx.4'?4Z ?*Prrc JeZdZdZ d!dedzdedzdeeeeeffdzdedzfdZd"d Zd#deeeeeffdedzddfd Z d ee zdede fd Z d ede eeefeffdZ d$dedeeefdzdedzdefdZdeeefdeeefddfdZdedee zfdZd#dededzde dzfdZdeefdZdedeeefdzfdZd"dZd#ded edeeefdzddfdZd ee zdeeeffdZdeeefdefdZdeeeeefffdZdeeeeeffddfd Zy)% PromptManagera Manager for loading and rendering .prompt files following the Dotprompt specification. Supports: - YAML frontmatter for metadata - Handlebars-style templating (using Jinja2) - Input/output schema validation - Model configuration Nr prompt_directory prompt_data prompt_filec |r t|nd|_i|_||_t t it ddgdddddd |_|jr|j|jr8|s td |j|j|}||j|<|r|j||yy) Nhtmlxmlz{{z}}z{%z%}z{#z#})loader autoescapevariable_start_stringvariable_end_stringblock_start_stringblock_end_stringcomment_start_stringcomment_end_stringz2prompt_id is required when prompt_file is provided) rr6promptsr8rrr jinja_env _load_prompts ValueError_load_prompt_file_load_prompts_from_json)r%r r6r7r8templates rr(zPromptManager.__init__<s;K%5 6PT24 &7b>(&%9"& $#!!%#          !UVV"44T5E5EyQH&.DLL #   ( (i @ rr cP|jr|jjstd|jt|jj d}|D]0} |j }|j ||}||j|<2y#t$rY?wxYw)z1Load all .prompt files from the prompt directory.z!Prompt directory does not exist: z*.promptN) r6existsrGlistglobstemrHrD Exception)r% prompt_filesr8r rJs rrFzPromptManager._load_promptsds$$D,A,A,H,H,J@AVAV@WXY Y"4#8#8#=#=j#IJ ' K ',, 11+yI*2 Y'     s)-B B%$B%c|rd|vr||in|}|jD]G\}} |jdd}|jdi}t|||}||j|<Iy#t$rYVwxYw)a:Load prompts from JSON data structure. Expected format: { "prompt_id": { "content": "template content", "metadata": {"model": "gpt-4", "temperature": 0.7, ...} } } or { "content": "template content", "metadata": {"model": "gpt-4", "temperature": 0.7, ...} } + prompt_id A dict carrying a "content" key is a single flat template registered under prompt_id; anything else is treated as already keyed by template ID. rrrrrN)itemsr rrDrP) r%r7r keyed_promptsr prompt_inforrrJs rrIz%PromptManager._load_prompts_from_jsonus*)-(?(?(H% %$**, !  rrc6d}tj||tj}|r@|jd}|jd} t j |xsi}||fSi}|}||fS#tj $r}td|d}~wwxYw)z+Parse YAML frontmatter from prompt content.z^---\s*\n(.*?)\n---\s*\n(.*)$zInvalid YAML frontmatter: N)rematchDOTALLgroupyaml safe_load YAMLErrorrG)r%rfrontmatter_patternrgfrontmatter_yamlrbraes rr_z PromptManager._parse_frontmatters&Fxx 3WbiiH &+kk!n ${{1~  C"nn-=>D" ,,,K& ,,,>> C #=aS!ABB Cs A22BBBprompt_variablesrc|j||}|At|jj}|rd|dnd}t d|d|d||xsi}|j r|j ||j  |jj|j}|jd i|} | S#t$r} td |d | d} ~ wwxYw) a Render a prompt template with the given variables. Args: prompt_id: The ID of the prompt template to render prompt_variables: Variables to substitute in the template version: Optional version number. If provided, looks for {prompt_id}.v{version} Returns: The rendered prompt string Raises: KeyError: If prompt_id is not found ValueError: If template rendering fails )r rNz (version )rSzPrompt ''z not found. Available prompts: zError rendering template 'z': r3) get_promptrMrDkeysKeyErrorr!_validate_inputrE from_stringrrenderrPrG) r%r rprrJavailable_prompts version_str variablesjinja_templaterenderedros rryzPromptManager.renders,//Iw/O  '+DLL,=,=,?'@ ]^o]pqr r+1r    H,A,A B M$(NN$>$>x?O?O$PN3n33@i@HO M9)CsKL L Ms8B?? CCCr|rc |jD]g\}}||vs ||}|j|}t||r.td|dt |dt |dt |jy)z3Basic validation of input variables against schema.zInvalid type for field 'z ': expected r-z, got N)rU_get_python_typer]rGgetattrr1typer-)r%r|r field_name field_typevalue expected_types rrwzPromptManager._validate_inputs&,lln  "J Y&!*- $ 5 5j A !%7$2:,?$$+M:s=GY$Z#[[abfglbmbvbvawy r schema_typec ttttftttttttt t d }|j |jtS)z*Convert schema type string to Python type.) stringr1numberintegerintfloatbooleanboolarrayrMobjectr2)r1rrrrMr2r lower)r%r type_mappings rrzPromptManager._get_python_typesQEl 8  1 1 3S99rc|$|d|}||jvr|j|S|jj|}||St|}|r|jj|SdS)a# Get a prompt template by ID and optional version. Args: prompt_id: The base prompt ID version: Optional version number. If provided, looks for {prompt_id}.v{version} Returns: The prompt template if found, None otherwise Nr )rDr r)r%r r versioned_id direct_matchbase_prompt_ids rrtzPromptManager.get_prompt s{  %.Kr'";Lt||+||L11"ll..y9  #  4Y ?3At||/KtKrcHt|jjS)z'Get a list of all available prompt IDs.)rMrDrur*s r list_promptszPromptManager.list_prompts"sDLL%%'((rcX|jj|}|r |jSdS)z#Get metadata for a specific prompt.N)rDr r)r%r rJs rget_prompt_metadataz!PromptManager.get_prompt_metadata&s),,**95$,x  6$6rcr|jj|jr|jyy)zBReload all prompts from the directory (if directory was provided).N)rDclearr6rFr*s rreload_promptszPromptManager.reload_prompts+s,       !rrcFt||xsi|}||j|<y)z'Add a prompt template programmatically.rTN)rrD)r%r rrrJs r add_promptzPromptManager.add_prompt1s#(8>r_hi"* Yrct|}|jd}|j|\}}|j|dS)zConvert a .prompt file to JSON format. Args: file_path: Path to the .prompt file Returns: Dictionary with 'content' and 'metadata' keys rZr[rr)rr^r_r`)r%rXrrarbs rprompt_file_to_jsonz!PromptManager.prompt_file_to_json6sLO ",,g,>)-(?(?(H% %+113MMrc|jdd}|jdi}|s|Sddl}|j|d}d|d|S) zConvert JSON prompt data to .prompt file format. Args: prompt_data: Dictionary with 'content' and 'metadata' keys Returns: String content in .prompt file format rrSrrNF)default_flow_stylez--- )r rjdump)r%r7rrrjrns rjson_to_prompt_filez!PromptManager.json_to_prompt_fileGsY%B7%//*b9N "+$))H"O'(gY77rci}|jjD]!\}}|j|jd||<#|S)zyGet all loaded prompts in JSON format. Returns: Dictionary mapping prompt_id to prompt data r)rDrUrr)r%resultr rJs rget_all_prompts_as_jsonz%PromptManager.get_all_prompts_as_json^sO #'<<#5#5#7  Ix#++$--!F9    rc&|j|y)zFLoad additional prompts from JSON data (merges with existing prompts).N)rI)r%r7s rload_prompts_from_json_dataz)PromptManager.load_prompts_from_json_datals $$[1r)NNNN)r N)Nr,)r-r.r/r0r1r2rr(rFrIrrrHtupler_rryrwrrrtrMrrrrrrrrr3rrr5r51s!%'+8<"& &A:&A*&A#tCH~-.5 &A 4Z &AP"#4T#s(^8K3L#Y\_cYc#os#J 3: # . -#-%S#X8K2L-037" )M)MsCx.4/)Mt )M )MV c3h c3h TX :C:D5L:&LCL#*LY]H]L0)d3i)7S7T#s(^d5J7 ! +C+#+c3hRVAV+bf+ NS4ZNDcNN"8tCH~8#8. c4S>.A)B 2tCc3hrsM  08CC$JPP8}2}2r