ci 4dZddlZddlmZdZGddZy)z4Prompt template for creating prompts with variables.N)Setz\{([_a-zA-Z][_a-zA-Z0-9]*)\}cNeZdZdZdefdZdeefdZd dZdefdZ defdZ y ) PromptTemplateaA prompt template for creating prompts with variables. The `PromptTemplate` class allows users to define a template string with variables represented in curly braces `{variable}`. The variable names cannot contain spaces and must start with a letter or underscore, followed by letters, digits, or underscore. These variables can be replaced with specific values using the `assemble` method, providing flexibility in generating dynamic prompts. Usage: ``` template_str = "Hello, {name}! Today is {day}. How are you?" prompt_template = PromptTemplate(template_str) completed_prompt = prompt_template.assemble(name="John", day="Monday") print(completed_prompt) ``` templatecNt||_|j|_y)zInitializes the PromptTemplate with a given template. Args: template: The template string with variables. Variables should be represented in curly braces `{variable}`. N)strr_get_variables variables)selfrs J/tmp/pip-target-z3e9_cxr/lib/python/vertexai/evaluation/prompt_template.py__init__zPromptTemplate.__init__-sH  ,,.returnc\ttjt|jS)z>Extracts and return a set of variable names from the template.)setrefindall_VARIABLE_NAME_REGEXrr s r r zPromptTemplate._get_variables7s2::2DMMBCCrc |j}|jD]Q\}}||jvrtd|d|jd|zdz}|j |t |}St |S)a8Replaces only the provided variables in the template with specific values. Args: **kwargs: Keyword arguments where keys are placeholder names and values are the replacements. Returns: A new PromptTemplate instance with the updated template string. zInvalid variable name 'z'. Valid variables are: {})ritemsr ValueErrorreplacerr)r kwargsassembled_string variable_namevalue placeholders r assemblezPromptTemplate.assemble;s ==$*LLN Q M5DNN2 -m_=,,0NN+;= -3K/77 SZP  Q.//rc|jS)zReturns the template string.rrs r __str__zPromptTemplate.__str__Ps }}rc"d|jdS)z6Returns a string representation of the PromptTemplate.zPromptTemplate('z')r#rs r __repr__zPromptTemplate.__repr__Ts!$--33rN)rr) __name__ __module__ __qualname____doc__rr rr r!r$r&rr rrsB&//DCD0*4#4rr)r*rtypingrrrr+rr r-s "; 6=4=4r