o پiW'@sfddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZeGdddZde d efd d Zd ed e e eefd efddZd efddZde fddZde fddZde fddZde fddZde fddZejdefddZe e!de fdd Z"e e#ede efd!d"Z$d#d$Z%dS)%N) dataclass)AnyCallableDictListOptionalTypecast) EnvironmentStrictUndefined) BaseModelc@sBeZdZUdZeed<ejed<ddZdefddZ d d Z d S) PromptzRepresents a prompt function. We return a `Prompt` class instead of a simple function so the template defined in prompt functions can be accessed. template signaturecCs"t|jj|_t|j|_dSN)listr parameterskeyscreate_jinja_templaterjinja_environmentselfrD/home/ubuntu/.local/lib/python3.10/site-packages/outlines/prompts.py __post_init__szPrompt.__post_init__returncOs.|jj|i|}||jjdi|jS)z}Render and return the template. Returns ------- The rendered template as a Python ``str``. Nr)rbindapply_defaultsrrender arguments)rargskwargsbound_argumentsrrr__call__szPrompt.__call__cCs|jSr)rrrrr__str__)szPrompt.__str__N) __name__ __module__ __qualname____doc__str__annotations__inspect Signaturerr#r$rrrrr s   r fnrcCs4t|}|j}|durtdtt|}t||S)aDecorate a function that contains a prompt template. This allows to define prompts in the docstring of a function and simplify their manipulation by providing some degree of encapsulation. It uses the `render` function internally to render templates. >>> import outlines >>> >>> @outlines.prompt >>> def build_prompt(question): ... "I have a ${question}" ... >>> prompt = build_prompt("How are you?") This API can also be helpful in an "agent" context where parts of the prompt are set when the agent is initialized and never modified later. In this situation we can partially apply the prompt function at initialization. >>> import outlines >>> import functools as ft ... >>> @outlines.prompt ... def solve_task(name: str, objective: str, task: str): ... '''Your name is {{name}}. .. Your overall objective is to {{objective}}. ... Please solve the following task: {{task}} ... ''' ... >>> hal = ft.partial(solve_task, "HAL", "Travel to Jupiter") Returns ------- A `Prompt` callable class which will render the template when called. Nz6Could not find a template in the function's docstring.)r+rr( TypeErrorr r)r )r-r docstringrrrrprompt-s %  r0rvaluescKst|}|jdi|S)a) Parse a Jinaj2 template and translate it into an Outlines graph. This function removes extra whitespaces and linebreaks from templates to allow users to enter prompts more naturally than if they used Python's constructs directly. See the examples for a detailed explanation. Examples -------- Outlines follow Jinja2's syntax >>> import outlines >>> outline = outlines.render("I like {{food}} and {{sport}}", food="tomatoes", sport="tennis") I like tomatoes and tennis If the first line of the template is empty, `render` removes it >>> from outlines import render >>> >>> tpl = ''' ... A new string''' >>> tpl ... '\nA new string' >>> render(tpl) ... 'a new string' Similarly, `render` ignores linebreaks introduced by placing the closing quotes underneath the text: >>> tpl = ''' ... A new string ... ''' >>> tpl ... '\nA new string\n' >>> render(tpl) ... 'A new string' If you want to insert a linebreak at the end of the rendered template, you will need to leave an empty line at the end of the template: >>> tpl = ''' ... A new string ... ... ''' >>> tpl ... '\nA new string\n\n' >>> render(tpl) ... 'A new string\n' `render` removes the identation in docstrings. This is particularly important when using prompt functions >>> tpl = ''' ... a string ... and another string''' >>> tpl ... '\n a string\n and another string' >>> render(tpl) ... 'a string\nand another string' The indentation of the first line is assumed to be the same as the second line's >>> tpl = '''a string ... and another''' >>> tpl ... 'a string\n and another' >>> render(tpl) ... 'a string\nand another' To get a different indentation for the first and the second line, we can start the prompt on the string's second line: >>> tpl = ''' ... First line ... Second line''' >>> render(tpl) ... 'First Line\n Second Line' Parameters ---------- template A string that contains a template written with the Jinja2 syntax. **values Map from the variables in the template to their value. Returns ------- A string that contains the rendered template. Nr)rr)rr1jinja_templaterrrr_s[rcCst|}|ddd}|r|d7}tdd|}tdddtd}t|j d<t |j d <t |j d <t |j d <t |j d <t|j d <||}|S)N z  z(?![\r\n])(\b\s+)T) trim_blocks lstrip_blockskeep_trailing_newline undefinedname descriptionsourcerschemar )r+cleandocreplaceendswithresubr r get_fn_namefiltersget_fn_description get_fn_sourceget_fn_signature get_schema get_fn_args from_string)rcleaned_templateends_with_linebreakenvr2rrrrs&        rcCs2t|stdt|dst|j}|S|j}|S)zReturns the name of a callable.z,The `name` filter only applies to callables.r%)callabler.hasattrtyper%)r-r:rrrrCs  rCcCs@t|stdg}t|}dd|jD}d|}|S)zTReturns the arguments of a function with annotations and default values if provided.z,The `args` filter only applies to callables.cSsg|]}t|qSr)r)).0paramrrr szget_fn_args..z, )rNr.r+rrr1join)r- arg_str_listrarg_strrrrrIs  rIcCs@t|stdt|}|durd}|S|dd}|S)z1Returns the first line of a callable's docstring.z3The `description` filter only applies to callables.Nr4r5r)rNr.r+getdocsplitstrip)r-r/r;rrrrEs rEcCsTt|stdtt|}ttdtj |}|dur&| d}|Std)z%Return the source code of a callable..The `source` filter only applies to callables.z (\bdef\b.*)Nrz)Could not read the function's source code) rNr.textwrapdedentr+ getsourcerAsearchcompileDOTALLgroup)r-r< re_searchrrrrFs rFcCsPt|stdtt|}ttd|}|dur!d}|S| d}|S)z#Return the signature of a callable.rZz \(([^)]+)\)Nr4) rNr.r[r\r+r]rAr^r_ra)r-r<rbrrrrrGs rGmodelcCstdt|d)Nz.No schema rendering function defined for type .)NotImplementedErrorrPrdrrrrH#srHcCstj|ddS)z"Return a pretty-printed dictionaryindent)jsondumpsrgrrrget_schema_dict*srmcCs`t|ttks tdt|drd}|}nd}|}||d}t||}tj |ddS)z&Return the schema of a Pydantic model.z4The `schema` filter only applies to Pydantic models.model_json_schemaz$defs definitionsNrhri) rPr r.rOrnr=getparse_pydantic_schemarkrl)rddef_key raw_schemaror=rrrget_schema_pydantic0s    rtcCsri}|dD].\}}d|vr|d||<qd|vr.|dd}t||d|||<qd|d||<q|S)zParse the output of `Basemodel.[schema|model_json_schema]()`. This recursively follows the references to other schemas in case of nested models. Other schemas are stored under the "definitions" key in the schema of the top-level model. propertiesr;z$ref/rh<>)itemsrXrq)rsro simple_schemar:valuerefsrrrrqCs  rq)& functoolsr+rkrAr[ dataclassesrtypingrrrrrrr jinja2r r pydanticr r r0r)rrrCrIrErFrGsingledispatchrHregisterdictrmrPrtrqrrrrs4 $ "2_