o ưi2@stdZddlZddlmZddlmZmZmZmZm Z m Z ddl Z ddl m Z mZmZGdddZGdd d ZdS) zq Based on Google's GenAI Kit dotprompt implementation: https://google.github.io/dotprompt/reference/frontmatter/ N)Path)AnyDictListOptionalTupleUnion) DictLoader Environmentselect_autoescapec@sDeZdZdZ  d dedeeeefdeefddZdd Z dS) PromptTemplatez>Represents a single prompt template with metadata and content.Ncontentmetadata template_idcCs||_|pi|_||_gd}|jd|_|jdidi|_|jdid|_|jdidi|_i|_|j D]}||vrO|j||j|<qAdS)N)modelinputoutputrrschemarformat) r rrgetr input_schemaZ output_formatZ output_schemaZoptional_paramskeys)selfr rrZrestricted_keyskeyrd/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/dotprompt/prompt_manager.py__init__s zPromptTemplate.__init__cCsd|jd|jdS)NzPromptTemplate(id='z ', model='z'))rrrrrr__repr__%szPromptTemplate.__repr__NN) __name__ __module__ __qualname____doc__strrrrrrrrrrr s r c @s>eZdZdZ    d2deedeedeeeeeeffdeefddZd3d d Z d4deeeeeffdeed dfd d Z de ee fded e fddZded eeeefeffddZ  d5dedeeeefdeed efddZdeeefdeeefd dfddZded e eeffddZ d4dedeed ee fdd Zd eefd!d"Zded eeeeffd#d$Zd3d%d&Z d4deded'eeeefd dfd(d)Zde ee fd eeeffd*d+Zdeeefd efd,d-Zd eeeeefffd.d/Zdeeeeeffd dfd0d1Z dS)6 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 Cs|rt|nd|_i|_||_ttitddgddddddd |_|jr)||jr>|s2t d | |j|}||j|<|rH| ||dSdS) Nhtmlxmlz{{z}}z{%z%}z{#z#})loaderZ autoescapeZvariable_start_stringZvariable_end_stringZblock_start_stringZblock_end_stringZcomment_start_stringZcomment_end_stringz2prompt_id is required when prompt_file is provided) rr'promptsr)r r r jinja_env _load_prompts ValueError_load_prompt_file_load_prompts_from_json)rr&r'r(r)templaterrrr4s.   zPromptManager.__init__returnc Csp|jr|jstd|jt|jd}|D]}z|j}|||}||j|<Wqty5YqwdS)z1Load all .prompt files from the prompt directory.z!Prompt directory does not exist: z*.promptN) r'existsr0listglobstemr1r- Exception)rZ prompt_filesr)r&r3rrrr/Ys   zPromptManager._load_promptsc Csh|r||i}|D]'\}}z|dd}|di}t|||d}||j|<Wq ty1Yq wdS)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 r rr rrN)itemsrr r-r9)rr(r&Z prompt_infor rr3rrrr2ls    z%PromptManager._load_prompts_from_json file_pathcCs>t|tr t|}|jdd}||\}}t|||dS)z%Load and parse a single .prompt file.utf-8encodingr;) isinstancer$r read_text_parse_frontmatterr strip)rr=r&r frontmattertemplate_contentrrrr1s  zPromptManager._load_prompt_filer c Cs~d}t||tj}|r7|d}|d}z t|pi}W||fStjy6}ztd|d}~wwi}|}||fS)z+Parse YAML frontmatter from prompt content.z^---\s*\n(.*?)\n---\s*\n(.*)$zInvalid YAML frontmatter: N)rematchDOTALLgroupyamlZ safe_loadZ YAMLErrorr0)rr Zfrontmatter_patternrJfrontmatter_yamlrFrEerrrrCs  z PromptManager._parse_frontmatterprompt_variablesversionc Cs|j||d}|dur)t|j}|rd|dnd}td|d|d||p,i}|jr7|||jz|j|j }|j d i|} | WSt y_} z t d |d | d} ~ ww) 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&rQNz (version )r:zPrompt ''z not found. Available prompts: zError rendering template 'z': r) get_promptr6r-rKeyErrorr_validate_inputr.Z from_stringr renderr9r0) rr&rPrQr3Zavailable_prompts version_str variablesZjinja_templaterenderedrOrrrrWs$zPromptManager.renderrYrc Csf|D],\}}||vr0||}||}t||s0td|dt|dt|dt|jqdS)z3Basic validation of input variables against schema.zInvalid type for field 'z ': expected r z, got N)r<_get_python_typerAr0getattrr$typer )rrYr field_nameZ field_typevalue expected_typerrrrVs  zPromptManager._validate_input schema_typec Cs2ttttftttttttttd }||tS)z*Convert schema type string to Python type.) stringr$numberintegerintfloatbooleanboolarrayr6objectdict)r$rerfrhr6rkrlower)rraZ type_mappingrrrr[szPromptManager._get_python_typecCs6|dur|d|}||jvr|j|S|j|S)a3 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 Nz.v)r-r)rr&rQZ versioned_idrrrrTs     zPromptManager.get_promptcCst|jS)z'Get a list of all available prompt IDs.)r6r-rrrrr list_promptsszPromptManager.list_promptscCs|j|}|r |jSdS)z#Get metadata for a specific prompt.N)r-rr)rr&r3rrrget_prompt_metadata"s z!PromptManager.get_prompt_metadatacCs |j|jr|dSdS)zBReload all prompts from the directory (if directory was provided).N)r-clearr'r/rrrrreload_prompts's  zPromptManager.reload_promptsrcCs t||pi|d}||j|<dS)z'Add a prompt template programmatically.r;N)r r-)rr&r rr3rrr add_prompt-s zPromptManager.add_promptcCs0t|}|jdd}||\}}||dS)zConvert a .prompt file to JSON format. Args: file_path: Path to the .prompt file Returns: Dictionary with 'content' and 'metadata' keys r>r?r r)rrBrCrD)rr=r rErFrrrprompt_file_to_json6s z!PromptManager.prompt_file_to_jsoncCsF|dd}|di}|s|Sddl}|j|dd}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 r r:rrNF)Zdefault_flow_stylez--- )rrMdump)rr(r rrMrNrrrjson_to_prompt_fileGs z!PromptManager.json_to_prompt_filecCs.i}|jD] \}}|j|jd||<q|S)zyGet all loaded prompts in JSON format. Returns: Dictionary mapping prompt_id to prompt data rr)r-r<r r)rresultr&r3rrrget_all_prompts_as_json^s  z%PromptManager.get_all_prompts_as_jsoncCs||dS)zFLoad additional prompts from JSON data (merges with existing prompts).N)r2)rr(rrrload_prompts_from_json_datalsz)PromptManager.load_prompts_from_json_data)NNNN)r4N)Nr)!r r!r"r#rr$rrrr/r2rrr r1rrCrerWrVr]tupler[rTrrmrnrprqrsrurwrxrrrrr%)s  % &  " -      " r%)r#rIpathlibrtypingrrrrrrrMZjinja2r r r r r%rrrrs