o پi @sddlZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZmZddlmZdd lmZdd lmZmZd d lmZeedfd e eeefdedeedefddZeeefd e eefdefddZdS)N)Enum)singledispatch)CallableOptionalUnion)build_regex_from_schema) BaseModel)get_schema_from_enumget_schema_from_signature)SequenceGeneratorAdapter)OpenAI)Sampler multinomial)regex schema_objectsamplerwhitespace_patternreturncstttr"t}t||}t|||}fdd|_|Sttt rBtt }t||}t|||}dd|_|St r_tt }t||}t|||}dd|_|Stt rx}t||}t|||}dd|_|Stdddd ) aj Generate structured JSON data with a `Transformer` model based on a specified JSON Schema. Parameters ---------- model: An instance of `Transformer` that represents a model from the `transformers` library. schema_object: The JSON Schema to generate data for. Can be a JSON string, a Pydantic model, or a callable that returns a JSON schema. sampler: The sampling algorithm to use to generate token ids from the logits distribution. whitespace_pattern Pattern to use for JSON syntactic whitespace (doesn't impact string literals) Example: allow only a single space or newline with `whitespace_pattern=r"[ ]?"` Returns ------- A `SequenceGenerator` instance that generates text constrained by the schema_object and transforms the result if BaseModel is used. c |SN parse_rawxrJ/home/ubuntu/.local/lib/python3.10/site-packages/outlines/generate/json.py4 zjson..cS t|Srpyjsonloadsrrrrr9rcSr rr!rrrrr>rcSr rr!rrrrrCrCannot parse schema . The schema must be either Aa Pydantic object, a function or a string that contains the JSON Schema specification) isinstancetyperr"dumpsmodel_json_schemarrformat_sequencerr callabler str ValueError)modelrrrschema regex_str generatorrrrjsons>             r4cst|ts tdtttrt}fdd}nttr*}dd}n t dddd|j d d d t |d d d}||_ |S)NzZThe OpenAI API does not support any other sampling algorithm than the multinomial sampler.crrrrrrrrZrzjson_openai..cSr rr!rrrrr]rr$r%r&r' json_schemadefaultT)namestrictr1)r)r5)response_format) r(rNotImplementedErrorr)rr"r*r+r.r/new_with_replacementsr#r,)r0rrr1r,r3rrr json_openaiNs6     r<)r4r"enumr functoolsrtypingrrroutlines_core.fsm.json_schemarpydanticroutlines.fsm.json_schemar r outlines.generate.apir outlines.modelsr outlines.samplersr rrr.objectregisterr<rrrrs<         <