o پi{@sddlZddlmZddlmZmZmZmZm Z ddl Z ddl m Z m Z er2ddlmZddlmZeGdddZd efd d Zd ed edeefddZdS)N) dataclass) TYPE_CHECKINGCallableOptionalTupleUnion)generatemodels)SequenceGenerator)Promptc@sleZdZUdZded<eeeefed<eed<dZ e ded<e dd ed efd d Z ddZ ddZdS)Functiona)Represents an Outlines function. Functions are a convenient way to encapsulate a prompt template, a language model and a Pydantic model that define the output structure. Once defined, the function can be called with arguments that will be used to render the prompt template. r prompt_templateschema model_nameNr generatorfn program_path function_namecCst|}t||}|S)z Load a function stored on GitHub)download_from_githubextract_function_from_file)clsrrprogram_contentfunctionrE/home/ubuntu/.local/lib/python3.10/site-packages/outlines/function.py from_githubs zFunction.from_githubcCs t|j}t||j|_dS)z,Load the model and initialize the generator.N)r transformersrrjsonrr)selfmodelrrrinit_generator&s zFunction.init_generatorcOs,|jdur ||j|i|}||S)a>Call the function. .. warning:: This currently does not support batching. Parameters ---------- args Values to pass to the prompt template as positional arguments. kwargs Values to pass to the prompt template as keyword arguments. N)rr r )rargskwargspromptrrr__call__+s  zFunction.__call__)r)__name__ __module__ __qualname____doc____annotations__rstrrobjectrr classmethodrr r$rrrrr s   r short_pathc Csd}d}|d}t|dkrtd|dddkrtd |d }|d }|d d}d||||g|d}t|}|jd krG|jS|jdkrTtd|d|dS)zs   2