i|2dZddlZddlmZddlmZmZmZmZm 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/.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 \eZdZdZ d!deedeedeeeeeeffdeefdZd"d Z d#deeeeeffdeeddfd Z d e ee fdede fd Zd edeeeefeffdZ d$dedeeeefdeedefdZdeeefdeeefddfdZdede eeffdZ d#dedeedee fdZdeefdZdedeeeeffdZd"dZ d#ded edeeeefddfdZd e ee fdeeeffdZdeeefdefdZdeeeeefffdZdeeeeeffddfd Z y)% 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) rr1promptsr3r r r jinja_env _load_prompts ValueError_load_prompt_file_load_prompts_from_json)rr0r1r2r3templates 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) r1existsrBlistglobstemrCr? Exception)r prompt_filesr3r0rEs r!rAzPromptManager._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?rL)rr2r0 prompt_inforrrEs r!rDz%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-8encodingrP) isinstancer,r read_text_parse_frontmatterrstrip)rrSr0r frontmattertemplate_contents r!rCzPromptManager._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 YAMLErrorrB)rrfrontmatter_patternrbfrontmatter_yamlr]r\es r!rZz PromptManager._parse_frontmatters?,gryyA ${{1~ ${{1~  C"nn-=>D" ,,,K& ,,,>> C #=aS!ABB Cs A22BBBprompt_variablesversionc|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 )r0rlNz (version )rOzPrompt ''z not found. Available prompts: zError rendering template 'z': r-) get_promptrIr?rKeyErrorr_validate_inputr@ from_stringrrenderrLrB) rr0rkrlrEavailable_prompts version_str variablesjinja_templaterenderedrjs r!rtzPromptManager.renders,??Y?H   $T\\%6%6%8 9 52+M# +M +MZ c3h 15c3h   :C:E$+4F:(8<++'/}+ . !+,)d3i)7S7Xd38n5M7 !RV++'*+6>tCH~6N+ +NU39-=N$sCx.N"8tCH~8#8. c4S>.A)B 2T#s(^ 342 2r#r/)r+rapathlibrtypingrrrrrr rejinja2r r r rr/r-r#r!rs9 :: ==PP8G2G2r#