o پi@sddlmZddlmZmZddlmZmZddlm Z m Z ee fde de fddZ e ee fde de fd d Ze ee fd ede de fd d ZdS))singledispatch)SequenceGeneratorAdapterVisionSequenceGeneratorAdapter)OpenAITransformersVision)Sampler multinomial regex_strsamplercC&ddlm}|||jd}t|||S)aGenerate structured text in the language of a regular expression. Parameters ---------- model: An instance of `Transformer` that represents a model from the `transformers` library. regex_str: The regular expression that the output must follow. sampler: The sampling algorithm to use to generate token ids from the logits distribution. Returns ------- A `SequenceGeneratorAdapter` instance that generates text constrained by the regular expression. rRegexLogitsProcessor tokenizer)outlines.processorsr rrmodelr r r logits_processorrK/home/ubuntu/.local/lib/python3.10/site-packages/outlines/generate/regex.pyregex s  rcCr )Nrr r)rr rrrrrr regex_vision&s  rrcCstd)NzdCannot use regex-structured generation with an OpenAI modeldue to the limitations of the OpenAI API.)NotImplementedError)rr r rrr regex_openai2srN) functoolsroutlines.generate.apirroutlines.modelsrroutlines.samplersrrstrrregisterrrrrrrs,