o پi @sddlZddlZddlZddlmZddlmZmZmZddl m Z m Z dee e ee fde fddZd ede fd d Zd eede fd dZdS)N)Enum)CallableTypeUnion) BaseModel create_model json_schemareturncCs^t|tr t|}|St|tr|}|St|tr#t|}|Std|ddd)a.Convert a JSON schema to a string. Parameters ---------- json_schema The JSON schema. Returns ------- str The JSON schema converted to a string. Raises ------ ValueError If the schema is not a dictionary, a string or a Pydantic class. zCannot parse schema z. The schema must be either zBa Pydantic class, a dictionary or a string that contains the JSON zschema specification) isinstancedictjsondumpsstr issubclassrmodel_json_schema ValueError)r schema_strrL/home/ubuntu/.local/lib/python3.10/site-packages/outlines/fsm/json_schema.pyconvert_json_schema_to_str s     rfnc Cst|}i}|jD]\}}|jtjkrtd|jdf||<q z|j}WntyE}zd}t j d|t dWYd}~nd}~wwt |fi|}| S)zTurn a function signature into a JSON schema. Every JSON object valid to the output JSON Schema can be passed to `fn` using the ** unpacking syntax. z)Each argument must have a type annotation. ArgumentszwThe function name could not be determined. Using default name 'Arguments' instead. For debugging, here is exact error: )categoryN)inspect signature parametersitems annotation_emptyr__name__ Exceptionwarningswarn UserWarningrr)rr argumentsnameargfn_nameemodelrrrget_schema_from_signature+s$   r*myenumcCs<t|dkrtd|jddd|D}|j|d}|S)NrzYour enum class z has 0 members. If you are working with an enum of functions, do not forget to register them as callable (using `partial` for instance)cSs,g|]}t|jrt|jjnd|jiqS)const)callablevaluer*func).0eltrrr Ls z(get_schema_from_enum..)titleoneOf)lenrr)r+choicesschemarrrget_schema_from_enumGs   r8)rr r!enumrtypingrrrpydanticrrr rrr*typer8rrrrs  !