gh(hUdZddlmZddlZddlmZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZdd lmZdd lmZmZdd lmZdd lmZe d Zd,dZd-dZd.dZd,dZd.dZ e!e"dfZ#d/dZ$d0d"Z%ej&eed Z'd#e(d$<ej)ed%Z*d#e(d&<d1d(Z+d2d)Z,Gd*d+eeZ-dS)3zBasePrompt schema definition.) annotationsN)ABC) Formatter)AnyCallableLiteral) BaseModel create_model) PromptValueStringPromptValue)BasePromptTemplate)get_colored_textmustache) formatter)is_interactive_env)f-stringrjinja2templatestrkwargsrreturnc  ddlm}n$#t$r}d}t||d}~wwxYw||jdi|S)aFormat a template using jinja2. *Security warning*: As of LangChain 0.0.329, this method uses Jinja2's SandboxedEnvironment by default. However, this sand-boxing should be treated as a best-effort approach rather than a guarantee of security. Do not accept jinja2 templates from untrusted sources as they may lead to arbitrary Python code execution. https://jinja.palletsprojects.com/en/3.1.x/sandbox/ Args: template: The template string. **kwargs: The variables to format the template with. Returns: The formatted string. Raises: ImportError: If jinja2 is not installed. r)SandboxedEnvironmentajinja2 not installed, which is needed to use the jinja2_formatter. Please install it with `pip install jinja2`.Please be cautious when using jinja2 templates. Do not expand jinja2 templates using unverified or user-controlled inputs as that can result in arbitrary Python code execution.N)jinja2.sandboxr ImportError from_stringrender)rrremsgs Z/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/langchain_core/prompts/string.pyjinja2_formatterr"s, &7777777 &&& L #A%&" ?   ! ! - -h 7 7 > H H H HHs *%*input_variables list[str]Nonect|}t|}||z }||z }d}|r |d|dz }|r|d|z }|r*tj|ddSdS)zValidate that the input variables are valid for the template. Issues a warning if missing or extra variables are found. Args: template: The template string. input_variables: The input variables. zMissing variables:  zExtra variables: ) stacklevelN)set#_get_jinja2_variables_from_templatewarningswarnstrip)rr#input_variables_setvalid_variablesmissing_variablesextra_variableswarning_messages r!validate_jinja2r5Aso..9(CCO'*==)O;OOFE1BEEEEA@@@@= o++--!<<<<<<==set[str]c ddlm}m}n$#t$r}d}t||d}~wwxYw|}||}||S)Nr) Environmentmetazojinja2 not installed, which is needed to use the jinja2_formatter. Please install it with `pip install jinja2`.)rr9r:rparsefind_undeclared_variables)rr9r:rr envasts r!r,r,Zs&,,,,,,,,, &&& ; #A% & +--C ))H  C  ) )# . ..s ,',c ,tj||S)zFormat a template using mustache. Args: template: The template string. **kwargs: The variables to format the template with. Returns: The formatted string. )rr)rrs r!mustache_formatterr@hs ?8V , ,,r6ct}d}tj|D]X\}}|dkr|dz}n>|dvr:|dkr4|dkr.||dd|dvr|dz }Y|S)zGet the variables from a mustache template. Args: template: The template string. Returns: The variables from the template. rend)variablesectioninverted section no escape.rErF)r+rtokenizeaddsplit)rvars section_depthtypekeys r!mustache_template_varsrQusUUDM&x00   c 5== Q MM L L Ls "" HHSYYs^^A& ' ' ' 2 2 2 Q M Kr6Defstype[BaseModel]c Ni}d}g}tj|D]\}}|dkr |dkr|r|})|dvr@|||t |dz}d||<m|dvr(d||t |dz<i}|r^|\}}|} |dd D]} | | i} | |d |rd ni|^td |S) zGet the variables from a mustache template. Args: template: The template string. Returns: The variables from the template as a Pydantic model. rrHrBrIF)rDrGTNr' PromptInput) rrJpopappendtuplerLpopitem setdefault_create_model_recursive) rfieldsprefix section_stackrOrPdefsfieldis_leafcurrentparts r!mustache_schemaresgF F+-M&x00 : : c #::  5== -&**,, 4 4 4   ( ( (eCIIcNN333F"F6NN . . .59F6E#))C..111 2D =))w#2#J 3 3D((r22GG59G&;bb<<< = #=$ 7 77r6namer`rOcTt|fid|DS)Nc`i|]+\}}||rt||dfnt|df,SN)r\rO.0kvs r! z+_create_model_recursive..sR   1 N'1--t44Q   r6)r items)rfr`s r!r\r\sE            r6zdict[str, Callable]DEFAULT_FORMATTER_MAPPING)rrDEFAULT_VALIDATOR_MAPPINGtemplate_formatc t|}n=#t$r0}d|dttd}t ||d}~wwxYw |||dS#tt f$r}d|d}t ||d}~wwxYw)a_Check that template string is valid. Args: template: The template string. template_format: The template format. Should be one of "f-string" or "jinja2". input_variables: The input variables. Raises: ValueError: If the template format is not supported. ValueError: If the prompt schema is invalid. zInvalid template format z, should be one of rHNzMInvalid prompt schema; check for mismatched or missing input parameters from )rqKeyErrorlistrp ValueError IndexError)rrrr#validator_funcexcr s r!check_valid_templaterzs'2?C ''' 3 3 3.// 3 3 3 oo3& ' 'x11111 j !''' ($ ( ( ( oo3& 's,  A +AA  AB-BBc|dkrt|}n\|dkr,dt|D}n*|dkrt|}nd|}t |t |S)a3Get the variables from the template. Args: template: The template string. template_format: The template format. Should be one of "f-string" or "jinja2". Returns: The variables from the template. Raises: ValueError: If the template format is not supported. rrc h|] \}}}}| | Srir)rk_rms r! z)get_template_variables..s(   !Q1q}A}}}r6rzUnsupported template format: )r,rr;rQrvsorted)rrrr#r s r!get_template_variablesrs(""=hGG J & &  ) 1 1( ; ;    J & &0::?o??oo / " ""r6cNeZdZdZeddZddZdd Z dddZddZ dS)StringPromptTemplatezAString prompt that exposes the format method, returning a prompt.rr$c gdS)z*Get the namespace of the langchain object.) langchainpromptsbaser)clss r!get_lc_namespacez%StringPromptTemplate.get_lc_namespace s0///r6rrr c 8t|jdi|S)zFormat the prompt with the inputs. Args: kwargs: Any arguments to be passed to the prompt template. Returns: A formatted string. textr)r formatselfrs r! format_promptz"StringPromptTemplate.format_prompts'!kdk&;&;F&;&;<<<.5s3   6?Is ^+c1   r6c6i|]\}}|t|dS)yellow)rrjs r!rnz4StringPromptTemplate.pretty_repr..9s659Q#Ax00r6r)r#ror)rr dummy_varss r! pretty_reprz StringPromptTemplate.pretty_repr(sq  CGCW     =G=M=M=O=OJt{((Z(((r6r%cdt|tdS)z,Print a pretty representation of the prompt.)rN)printrr)rs r! pretty_printz!StringPromptTemplate.pretty_print>s. d$6$8$899:::::r6N)rr$)rrrr )F)rrrr)rr%) __name__ __module__ __qualname____doc__ classmethodrrrrrrr6r!rr sKK000[0 = = = = D D D D))))),;;;;;;r6r)rrrrrr)rrr#r$rr%)rrrr7)rrrrS)rfrr`rRrrO)rrrrrr#r$rr%)rrrrrrr$).r __future__rr-abcrstringrtypingrrrpydanticr r langchain_core.prompt_valuesr r langchain_core.prompts.baser langchain_core.utilsrrlangchain_core.utils.formattingr$langchain_core.utils.interactive_envrPromptTemplateFormatr"r5r,r@rQdictrrRrer\rrp__annotations__validate_input_variablesrqrzrrrr6r!rsR###"""""")))))))))),,,,,,,,GGGGGGGG::::::;;;;;;;;555555CCCCCC?@)I)I)I)IX====2 / / / / - - - -6 CK!8!8!8!8H "22222 ''''@####:6;6;6;6;6;-s6;6;6;6;6;r6