j/dZddlZddlmZddlmZmZmZmZm Z m Z ddl Z ddl m Z mZmZGddZGdd Zy) zq Based on Google's GenAI Kit dotprompt implementation: https://google.github.io/dotprompt/reference/frontmatter/ N)Path)AnyDictListOptionalTupleUnion) DictLoader Environmentselect_autoescapecFeZdZdZ ddedeeeefdeefdZdZ y) PromptTemplatez>Represents a single prompt template with metadata and content.Ncontentmetadata template_idc0||_|xsi|_||_gd}|jjd|_|jjdijdi|_|jjdijd|_|jjdijdi|_i|_|jjD]#}||vs|j||j|<%y)N)modelinputoutputrrschemarformat) rrrgetr input_schema output_format output_schemaoptional_paramskeys)selfrrrrestricted_keyskeys /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/litellm/integrations/dotprompt/prompt_manager.py__init__zPromptTemplate.__init__s   B &7]]&&w/  MM--gr:>>xL!]]..x<@@J!]]..x<@@2N!==%%'C/),0MM#,>$$S)(c<d|jd|jdS)NzPromptTemplate(id='z ', model='z'))rrrs r!__repr__zPromptTemplate.__repr__%s"$T%5%5$6j BOOr#)NN) __name__ __module__ __qualname____doc__strrrrr"r&r#r!rr sEH .2%) ??4S>*?c] ?*Pr#rc BeZdZdZ d deedeedeeeeeeffdeefdZd!d Z d"deeeeeffdeeddfd Z d e ee fdede fd Zd edeeeefeffdZ d"dedeeeefdefdZdeeefdeeefddfdZdede eeffdZdedee fdZdeefdZdedeeeeffdZd!dZ d"ded edeeeefddfdZd e ee fdeeeffdZdeeefdefdZdeeeeefffdZdeeeeeffddfdZy)# 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 N prompt_idprompt_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) rr0promptsr2r r r jinja_env _load_prompts ValueError_load_prompt_file_load_prompts_from_json)rr/r0r1r2templates r!r"zPromptManager.__init__4s;K%5 6PT24 &$b>(&%9"& $#!!%#          !UVV--d.>.> JH&.DLL #   ( (i @ r#returncP|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) r0existsrAlistglobstemrBr> Exception)r prompt_filesr2r/rDs r!r@zPromptManager._load_promptsYs$$D,A,A,H,H,J3D4I4I3JK D1166zBC 'K ',, 11+yI*2 Y' (   s)-B B%$B%c|r||i}|jD]G\}} |jdd}|jdi}t|||}||j|<Iy#t$rYVwxYw)aLoad 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 rrrrrN)itemsrrr>rK)rr1r/ prompt_inforrrDs r!rCz%PromptManager._load_prompts_from_jsonls( $k2K&1&7&7&9 "I{ %//)R8&??:r:)#% ) +3 Y'':  sAA"" A.-A. file_pathct|tr t|}|jd}|j |\}}t |j ||S)z%Load and parse a single .prompt file.utf-8encodingrO) isinstancer+r read_text_parse_frontmatterrstrip)rrRr/r frontmattertemplate_contents r!rBzPromptManager._load_prompt_filesa i %YI%%w%7)-(?(?(H% %$**, !  r#rc6d}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 YAMLErrorrA)rrfrontmatter_patternrafrontmatter_yamlr\r[es r!rYz PromptManager._parse_frontmatters?,gryyA ${{1~ ${{1~  C"nn-=>D" ,,,K& ,,,>> C #=aS!ABB Cs A22BBBprompt_variablesc||jvr4t|jj}td|d||j|}|xsi}|jr|j ||j |j j|j}|jdi|}|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 Returns: The rendered prompt string Raises: KeyError: If prompt_id is not found ValueError: If template rendering fails zPrompt 'z ' not found. Available prompts: zError rendering template 'z': Nr,) r>rHrKeyErrorr_validate_inputr? from_stringrrenderrKrA) rr/rjavailable_promptsrD variablesjinja_templaterenderedris r!rozPromptManager.renders" DLL ( $T\\%6%6%8 9 9+%EFWEXY << *$*    H,A,A B M!^^778H8HIN,~,,9y9HO M9)CsKL L Ms8B:: CCCrqrc |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)rP_get_python_typerWrAgetattrr+typer')rrqr field_name field_typevalue expected_types r!rmzPromptManager._validate_inputs'-lln "J Y&!*- $ 5 5j A !%7$2:,?$$+M:s=GY$Z#[[abfglbmbvbvawy '5r# schema_typec ttttftttttttt t d }|j |jtS)z*Convert schema type string to Python type.) stringr+numberintegerintfloatbooleanboolarrayrHobjectdict)r+rrrrHrrlower)rr| type_mappings r!ruzPromptManager._get_python_typesQEl 7  1 1 3S99r#c8|jj|S)zGet a prompt template by ID.)r>r)rr/s r! get_promptzPromptManager.get_prompts|| **r#cHt|jjS)z'Get a list of all available prompt IDs.)rHr>rr%s r! list_promptszPromptManager.list_promptssDLL%%'((r#cX|jj|}|r |jSdS)z#Get metadata for a specific prompt.N)r>rr)rr/rDs r!get_prompt_metadataz!PromptManager.get_prompt_metadata s)<<##I.$,x  6$6r#cr|jj|jr|jyy)zBReload all prompts from the directory (if directory was provided).N)r>clearr0r@r%s r!reload_promptszPromptManager.reload_promptss,       !r#rcFt||xsi|}||j|<y)z'Add a prompt template programmatically.rON)rr>)rr/rrrDs r! add_promptzPromptManager.add_prompts)"hn") #+ Yr#ct|}|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 rTrUrr)rrXrYrZ)rrRrr[r\s r!prompt_file_to_jsonz!PromptManager.prompt_file_to_jsonsLO %%w%7)-(?(?(H% %+113MMr#c|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 rrNrrNF)default_flow_stylez--- )rrddump)rr1rrrdrhs r!json_to_prompt_filez!PromptManager.json_to_prompt_file.sY//)R0??:r2N $499X%H'(gY77r#ci}|jjD]!\}}|j|jd||<#|S)zyGet all loaded prompts in JSON format. Returns: Dictionary mapping prompt_id to prompt data r)r>rPrr)rresultr/rDs r!get_all_prompts_as_jsonz%PromptManager.get_all_prompts_as_jsonEsJ #'<<#5#5#7 Ix#++$--!F9 $8  r#c&|j|y)zFLoad additional prompts from JSON data (merges with existing prompts).N)rC)rr1s r!load_prompts_from_json_dataz)PromptManager.load_prompts_from_json_dataSs $$[1r#)NNNN)rEN)N) r'r(r)r*rr+rrr"r@rCr rrrBrrYrormrwtuplerurrrrrrrrrrr,r#r!r.r.)s$(*.;?%) #AC=#A#3-#Ad3S#X#678 #A c] #AJ(RV$T#s(^ 34$AI#$ $L sDy) 69  $-#-%S#X8K2L-,LP$M$M08c3h0H$M $ML c3h 15c3h   :C:E$+4F:&+C+H^,D+)d3i)7S7Xd38n5M7 !RV++'*+6>tCH~6N+ +NU39-=N$sCx.N"8tCH~8#8. c4S>.A)B 2T#s(^ 342 2r#r.)r*r`pathlibrtypingrrrrrr rdjinja2r r r rr.r,r#r!rs9 :: ==PP8n2n2r#