관리-도구
편집 파일: json_schema.cpython-38.pyc
U /?�f9� � @ s� U d Z ddlmZ ddlZddlZddlZddlZddlZddl m Z ddl m Z ddlm Z ddlmZ ddlmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z ddl Z dd l m Z m!Z!m"Z"m#Z# dd l$m%Z% dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, dd l-m.Z. d dl/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 d dl8m9Z9 d dl:m;Z;m<Z<m=Z= d dl>m?Z?m@Z@mAZA e�r�d dlBmCZC d dlDmEZEmFZF d dlGmHZH d dlImJZJ d dlKmLZL e(e"jMe"jNf ZOeePef ZQe(d ZRddd�ZSdeTd< e+d dd �d d d d �d!d"��ZUe(d# ZVG d$d%� d%eW�ZXd&ZYed'eP�ZZed(eP�Z[ed)eP�Z\eeZeRf Z]e d*ed+�Z^ej_f e4j`�G d,d-� d-��ZaG d.d/� d/�Zbd0eYebd1fd2d3d4d5d6d d7�d8d9�Zcd0ddeYebd:�d;d3d<d<d4d5d=d>�d?d@�Zde eeefePegdedA eeePdBf dCf f ZhdDeTdB< dEdFdG�dHdI�ZidJdBdK�dLdM�Zjdfd d<d dN�dOdP�Zkej_f e4j`�G dQdR� dR��Zlej_f e4j`�G dSdT� dT��ZmdUdVdW�dXdY�Zne dZ�Zoe�r�e'eodCf Zpn ej_f e4j`�G d[d\� d\��Zpd]d^d_�d`da�Zqd^dbdc�ddde�ZrdS )ga� Usage docs: https://docs.pydantic.dev/2.5/concepts/json_schema/ The `json_schema` module contains classes and functions to allow the way [JSON Schema](https://json-schema.org/) is generated to be customized. In general you shouldn't need to use this module directly; instead, you can use [`BaseModel.model_json_schema`][pydantic.BaseModel.model_json_schema] and [`TypeAdapter.json_schema`][pydantic.TypeAdapter.json_schema]. � )� annotationsN)� defaultdict)�deepcopy)� is_dataclass)�Enum)� TYPE_CHECKING�Any�Callable�Counter�Dict�Hashable�Iterable�NewType�Pattern�Sequence�Tuple�TypeVar�Union�cast)� CoreSchema� PydanticOmit� core_schema�to_jsonable_python)� ComputedField)� Annotated�Literal� TypeAlias� assert_never� deprecated�final)�PydanticDeprecatedSince26� )�_config�_core_metadata� _core_utils� _decorators�_internal_dataclass� _mock_val_ser�_schema_generation_shared� _typing_extra)�GetJsonSchemaHandler)�JsonDict�JsonSchemaExtraCallable� JsonValue)� PydanticInvalidForJsonSchema� PydanticSchemaGenerationError�PydanticUserError)� ConfigDict)�CoreSchemaField�CoreSchemaOrField)�PydanticDataclass)�GetJsonSchemaFunction�� BaseModel)� validation� serializationZInput�Outputzdict[JsonSchemaMode, str]�_MODE_TITLE_MAPPINGz\`update_json_schema` is deprecated, use a simple `my_dict.update(update_dict)` call instead.��category�JsonSchemaValue�dict[str, Any])�schema�updates�returnc C s | � |� | S )aW Update a JSON schema in-place by providing a dictionary of updates. This function sets the provided key-value pairs in the schema and returns the updated schema. Args: schema: The JSON schema to update. updates: A dictionary of key-value pairs to set in the schema. Returns: The updated JSON schema. )�update)r@ rA � rD �B/opt/hc_python/lib/python3.8/site-packages/pydantic/json_schema.py�update_json_schema^ s rF )�skipped-choice�non-serializable-defaultc @ s e Zd ZdZdS )�PydanticJsonSchemaWarninga This class is used to emit warnings produced during JSON schema generation. See the [`GenerateJsonSchema.emit_warning`][pydantic.json_schema.GenerateJsonSchema.emit_warning] and [`GenerateJsonSchema.render_warning_message`][pydantic.json_schema.GenerateJsonSchema.render_warning_message] methods for more details; these can be overridden to control warning behavior. N)�__name__� __module__� __qualname__�__doc__rD rD rD rE rI { s rI z#/$defs/{model}�CoreRef�DefsRef�JsonRef�JsonSchemaKeyT)�boundc @ sf e Zd ZU ded< ded< edddd d�d d ��Zd d d �d d�Zddd �dd�Zddd�dd�ZdS )�_DefinitionsRemappingzdict[DefsRef, DefsRef]�defs_remappingzdict[JsonRef, JsonRef]�json_remappingz dict[DefsRef, list[DefsRef]]zdict[DefsRef, JsonRef]z dict[DefsRef, JsonSchemaValue])�prioritized_choices� defs_to_json� definitionsrB c s� t |�}d|i}td�D ]�}tt�� |D ](}| | }|D ]}� | �|| � q8q(� �� D ]\}} t� | �� |< qZi } i } |D ]:} | | }t� fdd�|D ��} | | | < || | || < q�t| | �}|� d|i�}||kr�| S |}qt d��dS )z� This function should produce a remapping that replaces complex DefsRef with the simpler ones from the prioritized_choices such that applying the name remapping would result in an equivalent JSON schema. �$defs�d c 3 s" | ]}t � | �d kr|V qdS )r! N)�len��.0�x�Zschemas_for_alternativesrD rE � <genexpr>� s zA_DefinitionsRemapping.from_prioritized_choices.<locals>.<genexpr>z.Failed to simplify the JSON schema definitionsN) r �ranger �list�append�items�_deduplicate_schemas�nextrS �remap_json_schemar. )rV rW rX Zcopied_definitions�definitions_schemaZ_iter�defs_refZ alternatives� alternative�schemasrT rU Zoriginal_defs_refZremapped_defs_refZ remappingZnew_definitions_schemarD r_ rE �from_prioritized_choices� s. z._DefinitionsRemapping.from_prioritized_choicesrO )�refrB c C s | j �||�S �N)rT �get��selfrm rD rD rE �remap_defs_ref� s z$_DefinitionsRemapping.remap_defs_refrP c C s | j �||�S rn )rU ro rp rD rD rE �remap_json_ref� s z$_DefinitionsRemapping.remap_json_refr �r@ rB c s� t |t�r� �t|��S t |t�r4� fdd�|D �S t |t�r�|�� D ]d\}}|dkrtt |t�rt� �t|��|d< qF|dkr�� fdd�|d �� D �|d< qF� �|�||< qF|S )zH Recursively update the JSON schema replacing all $refs c s g | ]}� � |��qS rD )rg �r] �item�rq rD rE � <listcomp>� s z;_DefinitionsRemapping.remap_json_schema.<locals>.<listcomp>�$refrY c s& i | ] \}}� � t|��� �|��qS rD )rr rO rg )r] �key�valuerw rD rE � <dictcomp>� s � z;_DefinitionsRemapping.remap_json_schema.<locals>.<dictcomp>)� isinstance�strrs rP rb �dictrd rg )rq r@ rz r{ rD rw rE rg � s � z'_DefinitionsRemapping.remap_json_schemaN) rJ rK rL �__annotations__� staticmethodrl rr rs rg rD rD rD rE rS � s .rS c @ s� e Zd ZU dZdZdhZded< defddd �d d �Ze d d �dd��Z e dd �dd��Z dd �dd�Z ddd�dd�Z �d8d dd d �d d �Zd!d d"�d#d$�Zd%d d"�d&d'�Zd(d d"�d)d*�Zd+d d"�d,d-�Zd.d d"�d/d0�Zd1d d"�d2d3�Zd4d d"�d5d6�Zd7d d"�d8d9�Zd:d d"�d;d<�Zd=d d"�d>d?�Zd@d d"�dAdB�ZdCd d"�dDdE�ZdFd d"�dGdH�ZdId d"�dJdK�Z dLd d"�dMdN�Z dOd d"�dPdQ�Z dRd d"�dSdT�Z dUd d"�dVdW�Z dXd d"�dYdZ�Z!e"d[d\d]�e#d^d d"�d_d`���Z$e"dad\d]�e#d^d d"�dbdc���Z%d^d d"�ddde�Z&dfd d"�dgdh�Z'did d"�djdk�Z(dld d"�dmdn�Z)dod d"�dpdq�Z*drd d"�dsdt�Z+dud d"�dvdw�Z,dxd d"�dydz�Z-d{d d"�d|d}�Z.d~d d"�dd��Z/d�d d"�d�d��Z0d�d d"�d�d��Z1d�d d"�d�d��Z2d�d d"�d�d��Z3d�d d"�d�d��Z4d�d�d�d��d�d��Z5d�d d"�d�d��Z6d�d d"�d�d��Z7d�d d"�d�d��Z8d�d d"�d�d��Z9e:d�d�d��d�d���Z;d�d d��d�d��Z<d�ddd��d�d��Z=d�d d"�d�d��Z>d�d d"�d�d��Z?d�d d"�d�d��Z@d�d d"�d�d��ZAd�d d"�d�d��ZBd d�d�d�d�d d��d�dZCd d dÜd�dńZDd�d d"�d�dȄZEd�ddɜd�d˄ZFd�ddd͜d�dτZGd�d d"�d�d҄ZHd�d d"�d�dՄZId�d d"�d�dZJd�d�d dۜd�d݄ZKd�d�d dޜd�d�ZLd�dd�d�d�ZMd�d d"�d�d�ZNd�d d"�d�d�ZOd�d d"�d�d�ZPd�d d"�d�d�ZQd�d d"�d�d�ZRd�d d"�d�d��ZSd�d d"�d�d��ZTd�d d"�d�d��ZUd�d�d"�d��d �ZVdd�d��d�d�ZWd!dd"��d�d�ZXdd�d��d�d�ZY�d�d �d ��d �d �ZZ�d �d�d��d�d�Z[d d dÜ�d�d�Z\�dd��d��d�d�Z]�d�d�d��d�d�Z^d d �d �d �d ��d �d �Z_G �d!�d"� �d"�Z`�d#d �d$��d%�d&�Zad �d'dÜ�d(�d)�Zbd!dd �d*��d+�d,�Zc�d-d�d �d.��d/�d0�Zd�d-dd��d.��d1�d2�Ze�d3d ��d4�d5�Zfd �d d"��d6�d7�Zgd\S (9 �GenerateJsonSchemaa& Usage docs: https://docs.pydantic.dev/2.8/concepts/json_schema/#customizing-the-json-schema-generation-process A class for generating JSON schemas. This class generates JSON schemas based on configured parameters. The default schema dialect is [https://json-schema.org/draft/2020-12/schema](https://json-schema.org/draft/2020-12/schema). The class uses `by_alias` to configure how fields with multiple names are handled and `ref_template` to format reference names. Attributes: schema_dialect: The JSON schema dialect used to generate the schema. See [Declaring a Dialect](https://json-schema.org/understanding-json-schema/reference/schema.html#id4) in the JSON Schema documentation for more information about dialects. ignored_warning_kinds: Warnings to ignore when generating the schema. `self.render_warning_message` will do nothing if its argument `kind` is in `ignored_warning_kinds`; this value can be modified on subclasses to easily control which warnings are emitted. by_alias: Whether to use field aliases when generating the schema. ref_template: The format string used when generating reference names. core_to_json_refs: A mapping of core refs to JSON refs. core_to_defs_refs: A mapping of core refs to definition refs. defs_to_core_refs: A mapping of definition refs to core refs. json_to_defs_refs: A mapping of JSON refs to definition refs. definitions: Definitions in the schema. Args: by_alias: Whether to use field aliases in the generated schemas. ref_template: The format string to use when generating reference names. Raises: JsonSchemaError: If the instance of the class is inadvertently re-used after generating a schema. z,https://json-schema.org/draft/2020-12/schemarG zset[JsonSchemaWarningKind]�ignored_warning_kindsT�boolr~ ��by_alias� ref_templatec C sr || _ || _i | _i | _i | _i | _i | _t�t� i ��| _ d| _ i | _ t t�| _i | _| �� | _i | _d| _d S )Nr8 F)r� r� �core_to_json_refs�core_to_defs_refs�defs_to_core_refs�json_to_defs_refsrX r" ZConfigWrapperStackZ ConfigWrapper�_config_wrapper_stack�_mode� _prioritized_defsref_choicesr �int�_collision_counter�_collision_index�build_schema_type_to_method�_schema_type_to_method�"_core_defs_invalid_for_json_schema�_used)rq r� r� rD rD rE �__init__ s zGenerateJsonSchema.__init__z_config.ConfigWrapper�rB c C s | j jS rn )r� �tailrw rD rD rE r" 3 s zGenerateJsonSchema._config�JsonSchemaModec C s | j jd k r| j jS | jS d S rn )r" Zjson_schema_mode_overrider� rw rD rD rE �mode7 s zGenerateJsonSchema.modezKdict[CoreSchemaOrFieldType, Callable[[CoreSchemaOrField], JsonSchemaValue]]c C s� i }t �t�}|D ]p}|�dd�� d�}zt| |�||< W q tk r� } z(td|�dt| �j� d|� d��|�W 5 d}~X Y qX q|S ) aO Builds a dictionary mapping fields to methods for generating JSON schemas. Returns: A dictionary containing the mapping of `CoreSchemaOrFieldType` to a handler method. Raises: TypeError: If no method has been defined for generating a JSON schema for a given pydantic core schema type. �-�_Z_schemaz9No method for generating JsonSchema for core_schema.type=z (expected: �.�)N) r) Zall_literal_values�CoreSchemaOrFieldType�replace�getattr�AttributeError� TypeError�typerJ )rq �mappingZcore_schema_typesrz � method_name�erD rD rE r� > s � ��z.GenerateJsonSchema.build_schema_type_to_methodzGSequence[tuple[JsonSchemaKeyT, JsonSchemaMode, core_schema.CoreSchema]]zctuple[dict[tuple[JsonSchemaKeyT, JsonSchemaMode], JsonSchemaValue], dict[DefsRef, JsonSchemaValue]])�inputsrB c C s� | j r tdt| �j� d�dd��|D ]\}}}|| _| �|� q$| �� }i }|D ],\}}}|| _| �|�}|�|�|||f< qPd| ji}|�|�}d| _ |t |d �fS )a� Generates JSON schema definitions from a list of core schemas, pairing the generated definitions with a mapping that links the input keys to the definition references. Args: inputs: A sequence of tuples, where: - The first element is a JSON schema key type. - The second element is the JSON mode: either 'validation' or 'serialization'. - The third element is a core schema. Returns: A tuple where: - The first element is a dictionary whose keys are tuples of JSON schema key type and JSON mode, and whose values are the JSON schema corresponding to that pair of inputs. (These schemas may have JsonRef references to definitions that are defined in the second returned element.) - The second element is a dictionary whose keys are definition references for the JSON schemas from the first returned element, and whose values are the actual JSON schema definitions. Raises: PydanticUserError: Raised if the JSON schema generator has already been used to generate a JSON schema. �nThis JSON schema generator has already been used to generate a JSON schema. You must create a new instance of � to generate a new JSON schema.�json-schema-already-used��coderY T) r� r0 r� rJ r� �generate_inner� _build_definitions_remappingrg rX �_sort_json_schema)rq r� rz r� r@ �definitions_remapping�json_schemas_map� json_schemarD rD rE �generate_definitionsX s$ � z'GenerateJsonSchema.generate_definitionsr8 r r> )r@ r� rB c C s� || _ | jr&tdt| �j� d�dd��| �|�}| �|�}tt|� d��}|dk r�| � |�}|| dksp|dkr�dd|igi}n|� � }|| d8 < tt|� d��}qJ| � |� | � � }| jr�| j|d < |�|�}d | _t|�S ) a� Generates a JSON schema for a specified schema in a specified mode. Args: schema: A Pydantic model. mode: The mode in which to generate the schema. Defaults to 'validation'. Returns: A JSON schema representing the specified schema. Raises: PydanticUserError: If the JSON schema generator has already been used to generate a JSON schema. r� r� r� r� ry Nr! �allOfrY T)r� r� r0 r� rJ r� �get_json_ref_countsr rP ro �get_schema_from_definitions�copy� _garbage_collect_definitionsr� rX rg r� )rq r@ r� r� Zjson_ref_countsrm �ref_json_schemar� rD rD rE �generate� s. � zGenerateJsonSchema.generater3 rt c sB d|krFt |d �}|�jf}|�jkrF�j| �jkrFd�j| iS t�|�}dddd��fdd� �ddd�d d �� d dd �� ��fd d� }t��|�}|j � dd�D ].}||fd dddd��fdd� }t��|�}q�|j � dd�D ]0}||fd dddd�� �fdd� }t��|�}q�||�} t � |��r>�|| �} � | �} | S )z�Generates a JSON schema for a given core schema. Args: schema: The given core schema. Returns: The generated JSON schema. rm ry r r> )r r� rB c sh d| krdt | d �}� �|�\}}t|d �}|� j|< |�dd �|kr`|� j|< � j�|d � |}|S )Nrm ry )rN �get_cache_defs_ref_schemarP r� ro rX r� �pop)r r� �core_refri r� �json_refrw rD rE � populate_defs� s z8GenerateJsonSchema.generate_inner.<locals>.populate_defs�r� rB c S s@ d| kr<t | �� �dkr<| �� } | �d�}dd|igi| �} | S )Nry r! r� )r[ �keysr� r� )r� rm rD rD rE �convert_to_all_of� s z<GenerateJsonSchema.generate_inner.<locals>.convert_to_all_ofr3 )�schema_or_fieldrB c s� d}�j dkr(d| kr(| d }��|�}|dkrjt�| �sDt�| �r\�j| d }|| �}ntd| � ���t�| �r��| |�}� |�}|S )a@ Generate a JSON schema based on the input schema. Args: schema_or_field: The core schema to generate a JSON schema from. Returns: The generated JSON schema. Raises: TypeError: If an unexpected schema type is encountered. Nr9 r� � Unexpected schema type: schema=)r� � ser_schemar$ �is_core_schema�is_core_schema_fieldr� r� )r� r� r� Zgenerate_for_schema_type�r� r� rq rD rE � handler_func� s z7GenerateJsonSchema.generate_inner.<locals>.handler_funcZpydantic_js_functionsrD r* r5 )r� �current_handler�js_modify_functionrB c sJ || |�}t �| �r � | |�}|�|�}|�dd �}|rF|rF|�|� |S )Nry )r$ r� Zresolve_ref_schemar� rC )r� r� r� r� Zoriginal_schemarm )r� rD rE �new_handler_func s z;GenerateJsonSchema.generate_inner.<locals>.new_handler_funcZ pydantic_js_annotation_functionsc s* || |�}t �| �r&�| |�}� |�}|S rn )r$ r� )r� r� r� r� )r� r� rD rE r� s ) rN r� r� rX r� r# �CoreMetadataHandlerr( ZGenerateJsonSchemaHandler�metadataro r$ r� ) rq r@ r� � core_mode_refZmetadata_handlerr� r� r� r� r� rD r� rE r� � s2 �� z!GenerateJsonSchema.generate_innerzcore_schema.AnySchemac C s i S )z�Generates a JSON schema that matches any value. Args: schema: The core schema. Returns: The generated JSON schema. rD �rq r@ rD rD rE � any_schema0 s z GenerateJsonSchema.any_schemazcore_schema.NoneSchemac C s ddiS )z�Generates a JSON schema that matches `None`. Args: schema: The core schema. Returns: The generated JSON schema. r� �nullrD r� rD rD rE � none_schema; s z GenerateJsonSchema.none_schemazcore_schema.BoolSchemac C s ddiS )z�Generates a JSON schema that matches a bool value. Args: schema: The core schema. Returns: The generated JSON schema. r� �booleanrD r� rD rD rE � bool_schemaF s z GenerateJsonSchema.bool_schemazcore_schema.IntSchemac C s0 ddi}| � ||| jj� dd� |�� D �}|S )z�Generates a JSON schema that matches an int value. Args: schema: The core schema. Returns: The generated JSON schema. r� �integerc S s( i | ] \}}|t jt j hkr||�qS rD ��math�inf�r] �k�vrD rD rE r| \ s z1GenerateJsonSchema.int_schema.<locals>.<dictcomp>��update_with_validations�ValidationsMapping�numericrd �rq r@ r� rD rD rE � int_schemaQ s z GenerateJsonSchema.int_schemazcore_schema.FloatSchemac C s0 ddi}| � ||| jj� dd� |�� D �}|S )z�Generates a JSON schema that matches a float value. Args: schema: The core schema. Returns: The generated JSON schema. r� �numberc S s( i | ] \}}|t jt j hkr||�qS rD r� r� rD rD rE r| j s z3GenerateJsonSchema.float_schema.<locals>.<dictcomp>r� r� rD rD rE � float_schema_ s z GenerateJsonSchema.float_schemazcore_schema.DecimalSchemac C s� | � t� � �}| jdkr�|�d�}|�d�}|�d�}|�d�}|�d�}d| �tj|�d�|d krhd nt|�|d krzd nt|�|d kr�d nt|�|d kr�d nt|�|d kr�d nt|�d ��|gi}|S ) z�Generates a JSON schema that matches a decimal value. Args: schema: The core schema. Returns: The generated JSON schema. r8 � multiple_of�le�ge�lt�gt�anyOf� allow_inf_nanN)r� r� r� r� r� r� )� str_schemar r� ro r� �float)rq r@ r� r� r� r� r� r� rD rD rE �decimal_schemam s, �� ��z!GenerateJsonSchema.decimal_schemazcore_schema.StringSchemac C s> ddi}| � ||| jj� t|�d�t�r:|�d�j|d< |S )z�Generates a JSON schema that matches a string value. Args: schema: The core schema. Returns: The generated JSON schema. r� �string�pattern)r� r� r� r} ro r r� r� rD rD rE r� � s z GenerateJsonSchema.str_schemazcore_schema.BytesSchemac C s0 d| j jdkrdndd�}| �||| jj� |S )z�Generates a JSON schema that matches a bytes value. Args: schema: The core schema. Returns: The generated JSON schema. r� �base64Z base64url�binary�r� �format)r" �ser_json_bytesr� r� �bytesr� rD rD rE � bytes_schema� s z GenerateJsonSchema.bytes_schemazcore_schema.DateSchemac C s ddd�}| � ||| jj� |S )z�Generates a JSON schema that matches a date value. Args: schema: The core schema. Returns: The generated JSON schema. r� �dater� )r� r� r� r� rD rD rE � date_schema� s z GenerateJsonSchema.date_schemazcore_schema.TimeSchemac C s ddd�S )z�Generates a JSON schema that matches a time value. Args: schema: The core schema. Returns: The generated JSON schema. r� �timer� rD r� rD rD rE � time_schema� s z GenerateJsonSchema.time_schemazcore_schema.DatetimeSchemac C s ddd�S )z�Generates a JSON schema that matches a datetime value. Args: schema: The core schema. Returns: The generated JSON schema. r� z date-timer� rD r� rD rD rE �datetime_schema� s z"GenerateJsonSchema.datetime_schemazcore_schema.TimedeltaSchemac C s | j jdkrddiS ddd�S )z�Generates a JSON schema that matches a timedelta value. Args: schema: The core schema. Returns: The generated JSON schema. r� r� r� r� �durationr� )r" �ser_json_timedeltar� rD rD rE �timedelta_schema� s z#GenerateJsonSchema.timedelta_schemazcore_schema.LiteralSchemac C s� dd� |d D �}dd� |D �}d|i}t |�dkr@|d |d< d d � |D �}|thkrbd |d < nf|thkrvd |d < nR|thkr�d|d < n>|thkr�d|d < n*|thkr�d|d < n|td�hkr�d|d < |S )z�Generates a JSON schema that matches a literal value. Args: schema: The core schema. Returns: The generated JSON schema. c S s g | ]}t |t�r|jn|�qS rD )r} r r{ �r] r� rD rD rE rx � s z5GenerateJsonSchema.literal_schema.<locals>.<listcomp>�expectedc S s g | ] }t |��qS rD )r r rD rD rE rx � s �enumr! r �constc S s h | ] }t |��qS rD �r� �r] r� rD rD rE � <setcomp>� s z4GenerateJsonSchema.literal_schema.<locals>.<setcomp>r� r� r� r� r� �arrayNr� )r[ r~ r� r� r� rb r� )rq r@ r �result�typesrD rD rE �literal_schema� s& z!GenerateJsonSchema.literal_schemazcore_schema.EnumSchemac C s |d }|j sdn t�|j �}|dkr*d}|j|d�}dd� |�� D �}dd� |d D �}||d < t|�d krz|d |d < dd� |D �}t|t�s�|thkr�d|d< ndt|t�s�|thkr�d|d< nFt|t �s�|t hkr�d|d< n(|t hkr�d|d< n|t hk�r d|d< |S )z�Generates a JSON schema that matches an Enum value. Args: schema: The core schema. Returns: The generated JSON schema. �clsNzAn enumeration.)�title� descriptionc S s i | ]\}}|d k r||�qS rn rD r� rD rD rE r| s z2GenerateJsonSchema.enum_schema.<locals>.<dictcomp>c S s g | ]}t |j��qS rD )r r{ r rD rD rE rx s z2GenerateJsonSchema.enum_schema.<locals>.<listcomp>�membersr r! r r c S s h | ] }t |��qS rD r r rD rD rE r s z1GenerateJsonSchema.enum_schema.<locals>.<setcomp>r� r� r� r� r� r ) rM �inspect�cleandocrJ rd r[ r} r~ r� r� r� rb )rq r@ Z enum_typer r r r rD rD rE � enum_schema� s. � z GenerateJsonSchema.enum_schemaz core_schema.IsInstanceSchemac C s | � |d|d � d��S )a Handles JSON schema generation for a core schema that checks if a value is an instance of a class. Unless overridden in a subclass, this raises an error. Args: schema: The core schema. Returns: The generated JSON schema. z core_schema.IsInstanceSchema (r r� �� handle_invalid_for_json_schemar� rD rD rE �is_instance_schema! s z%GenerateJsonSchema.is_instance_schemaz core_schema.IsSubclassSchemac C s i S )aI Handles JSON schema generation for a core schema that checks if a value is a subclass of a class. For backwards compatibility with v1, this does not raise an error, but can be overridden to change this. Args: schema: The core schema. Returns: The generated JSON schema. rD r� rD rD rE �is_subclass_schema. s z%GenerateJsonSchema.is_subclass_schema�core_schema.CallableSchemac C s | � |d�S )z�Generates a JSON schema that matches a callable value. Unless overridden in a subclass, this raises an error. Args: schema: The core schema. Returns: The generated JSON schema. r r r� rD rD rE �callable_schema< s z"GenerateJsonSchema.callable_schemazcore_schema.ListSchemac C s: d|kr i n | � |d �}d|d�}| �||| jj� |S )z�Returns a schema that matches a list schema. Args: schema: The core schema. Returns: The generated JSON schema. � items_schemar �r� rd �r� r� r� r �rq r@ r r� rD rD rE � list_schemaI s z GenerateJsonSchema.list_schema�D`tuple_positional_schema` is deprecated. Use `tuple_schema` instead.Nr< zcore_schema.TupleSchemac C s t jdtdd� | �|�S )�Replaced by `tuple_schema`.r � �� stacklevel��warnings�warnr � tuple_schemar� rD rD rE �tuple_positional_schemaW s �z*GenerateJsonSchema.tuple_positional_schema�B`tuple_variable_schema` is deprecated. Use `tuple_schema` instead.c C s t jdtdd� | �|�S )r r( r r! r# r� rD rD rE �tuple_variable_schemab s �z(GenerateJsonSchema.tuple_variable_schemac s� ddi}d|kr�|d }|dkrJ||d< � fdd�|d d |� D �|d < |d t |d �krv� �|d | �|d < q�d |d < n:� fdd�|d D �}|r�||d < t |�|d< t |�|d< � �||� jj� |S )z�Generates a JSON schema that matches a tuple schema e.g. `Tuple[int, str, bool]` or `Tuple[int, ...]`. Args: schema: The core schema. Returns: The generated JSON schema. r� r �variadic_item_indexr �minItemsc s g | ]}� � |��qS rD �r� ru rw rD rE rx | s z3GenerateJsonSchema.tuple_schema.<locals>.<listcomp>r N� prefixItemsr! rd Tc s g | ]}� � |��qS rD r, ru rw rD rE rx � s �maxItems)r[ r� r� r� r )rq r@ r� r* r- rD rw rE r& m s$ � z GenerateJsonSchema.tuple_schemazcore_schema.SetSchemac C s | � |�S )z�Generates a JSON schema that matches a set schema. Args: schema: The core schema. Returns: The generated JSON schema. ��_common_set_schemar� rD rD rE � set_schema� s z GenerateJsonSchema.set_schemazcore_schema.FrozenSetSchemac C s | � |�S )z�Generates a JSON schema that matches a frozenset schema. Args: schema: The core schema. Returns: The generated JSON schema. r/ r� rD rD rE �frozenset_schema� s z#GenerateJsonSchema.frozenset_schemaz3core_schema.SetSchema | core_schema.FrozenSetSchemac C s< d|kr i n | � |d �}dd|d�}| �||| jj� |S )Nr r T)r� Z uniqueItemsrd r r rD rD rE r0 � s z%GenerateJsonSchema._common_set_schemazcore_schema.GeneratorSchemac C s: d|kr i n | � |d �}d|d�}| �||| jj� |S )z�Returns a JSON schema that represents the provided GeneratorSchema. Args: schema: The schema. Returns: The generated JSON schema. r r r r r rD rD rE �generator_schema� s z#GenerateJsonSchema.generator_schemazcore_schema.DictSchemac C s� ddi}d|kr"| � |d ��� ni }|�dd�}d|krL| � |d ��� ni }|�dd� |sh|dk r�|dkrz||d< n ||i|d < | �||| jj� |S ) z�Generates a JSON schema that matches a dict schema. Args: schema: The core schema. Returns: The generated JSON schema. r� �object� keys_schemar� N� values_schemar �additionalPropertiesZpatternProperties)r� r� r� r� r� r4 )rq r@ r� r5 Z keys_patternr6 rD rD rE � dict_schema� s z GenerateJsonSchema.dict_schemaz _core_utils.AnyFunctionSchemac C s0 t �|�r| �|d �S | �|d|d � d��S )Nr@ z*core_schema.PlainValidatorFunctionSchema (�functionr� )r$ � is_function_with_inner_schemar� r r� rD rD rE �_function_schema� s �z#GenerateJsonSchema._function_schemaz)core_schema.BeforeValidatorFunctionSchemac C s | � |�S )z�Generates a JSON schema that matches a function-before schema. Args: schema: The core schema. Returns: The generated JSON schema. �r; r� rD rD rE �function_before_schema� s z)GenerateJsonSchema.function_before_schemaz(core_schema.AfterValidatorFunctionSchemac C s | � |�S )z�Generates a JSON schema that matches a function-after schema. Args: schema: The core schema. Returns: The generated JSON schema. r< r� rD rD rE �function_after_schema� s z(GenerateJsonSchema.function_after_schemaz(core_schema.PlainValidatorFunctionSchemac C s | � |�S )z�Generates a JSON schema that matches a function-plain schema. Args: schema: The core schema. Returns: The generated JSON schema. r< r� rD rD rE �function_plain_schema� s z(GenerateJsonSchema.function_plain_schemaz'core_schema.WrapValidatorFunctionSchemac C s | � |�S )z�Generates a JSON schema that matches a function-wrap schema. Args: schema: The core schema. Returns: The generated JSON schema. r< r� rD rD rE �function_wrap_schema s z'GenerateJsonSchema.function_wrap_schemaz core_schema.WithDefaultSchemac C s� | � |d �}d|kr|S |d }| jdkrr|d �di � }rr|�d� }rr|�d�dkrr|�d�dkrr||�}z| �|�}W n. tjk r� | �d d |� d �� | Y S X d |kr�|g|d �S ||d< |S dS )z�Generates a JSON schema that matches a schema with a default value. Args: schema: The core schema. Returns: The generated JSON schema. r@ �defaultr9 r9 r� �function-plainZinfo_argFrH zDefault value z= is not JSON serializable; excluding default from JSON schemary )r� rA N)r� r� ro �encode_default� pydantic_core�PydanticSerializationError� emit_warning)rq r@ r� rA r� Zser_funcZencoded_defaultrD rD rE �default_schema s4 � � � � � z!GenerateJsonSchema.default_schemazcore_schema.NullableSchemac C s4 ddi}| � |d �}||kr"|S | �||g�S dS )z�Generates a JSON schema that matches a schema that allows null values. Args: schema: The core schema. Returns: The generated JSON schema. r� r� r@ N)r� �get_flattened_anyof)rq r@ Z null_schemaZinner_json_schemarD rD rE �nullable_schemaB s z"GenerateJsonSchema.nullable_schemazcore_schema.UnionSchemac C s� g }|d }|D ]v}t |t�r&|d n|}z|�| �|�� W q tk rV Y qY q tk r� } z| �d|j� W 5 d}~X Y qX qt|�dkr�|d S | � |�S )z�Generates a JSON schema that matches a schema that allows values matching any of the given schemas. Args: schema: The core schema. Returns: The generated JSON schema. �choicesr rG Nr! ) r} �tuplerc r� r r. rF �messager[ rH )rq r@ � generatedrJ �choiceZ choice_schema�excrD rD rE � union_schemaU s " z GenerateJsonSchema.union_schemaz core_schema.TaggedUnionSchemac C s� i }|d � � D ]z\}}t|t�r(|j}z| �|��� |t|�< W q tk rZ Y qY q tk r� } z| � d|j � W 5 d}~X Y qX qt |� � �}d|i}| � ||�}|dk r�|dd� |� � D �d�|d< |S ) ac Generates a JSON schema that matches a schema that allows values matching any of the given schemas, where the schemas are tagged with a discriminator field that indicates which schema should be used to validate the value. Args: schema: The core schema. Returns: The generated JSON schema. rJ rG NZoneOfc S s i | ]\}}||� d |��qS )ry �ro r� rD rD rE r| � s z:GenerateJsonSchema.tagged_union_schema.<locals>.<dictcomp>)Z propertyNamer� � discriminator)rd r} r r{ r� r� r~ r r. rF rL re �values�_extract_discriminator) rq r@ rM r� r� rO �one_of_choicesr� �openapi_discriminatorrD rD rE �tagged_union_scheman s$ " � z&GenerateJsonSchema.tagged_union_schema�list[JsonDict]� str | None)r@ rU rB c C s d}t |d t�r|d S t |d t��r t|d �dkrXt |d d t�rX|d d S |d D ]�}t |t�st �q t|�dkr�q`|d }t |t�s�q`d}|D ]\}d|kr�t |d t�s�t�| �t|d ��p�i }q�|�di �}t |t�r�||kr�d} q�q�|r`|} �q q`|S ) zvExtract a compatible OpenAPI discriminator from the schema and one_of choices that end up in the final schema.NrR r! r Try � propertiesF) r} r~ rb r[ �AssertionErrorr� rP ro r ) rq r@ rU rV Z alias_path�aliasZ alias_is_present_on_all_choicesrN rZ rD rD rE rT � s6 " z)GenerateJsonSchema._extract_discriminatorzcore_schema.ChainSchemac C s$ | j dkrdnd}| �|d | �S )a� Generates a JSON schema that matches a core_schema.ChainSchema. When generating a schema for validation, we return the validation JSON schema for the first step in the chain. For serialization, we return the serialization JSON schema for the last step in the chain. Args: schema: The core schema. Returns: The generated JSON schema. r8 r ���Zsteps)r� r� )rq r@ Z step_indexrD rD rE � chain_schema� s z GenerateJsonSchema.chain_schemaz core_schema.LaxOrStrictSchemac C s0 |� dd�}|r | �|d �S | �|d �S dS )z�Generates a JSON schema that matches a schema that allows values matching either the lax schema or the strict schema. Args: schema: The core schema. Returns: The generated JSON schema. �strictFZ strict_schemaZ lax_schemaN)ro r� )rq r@ Z use_strictrD rD rE �lax_or_strict_schema� s z'GenerateJsonSchema.lax_or_strict_schemaz core_schema.JsonOrPythonSchemac C s | � |d �S )a� Generates a JSON schema that matches a schema that allows values matching either the JSON schema or the Python schema. The JSON schema is used instead of the Python schema. If you want to use the Python schema, you should override this method. Args: schema: The core schema. Returns: The generated JSON schema. r� r, r� rD rD rE �json_or_python_schema� s z(GenerateJsonSchema.json_or_python_schema�core_schema.TypedDictSchemac s� |� dd��� �fdd�|d �� D �}� jdkrJ|�� �|� dg ��� t|�}t|�}� j�|�� � � |�}W 5 Q R X |� d�}|� d �}|d kr�|� d d �}|d k r�|� d �p�|j }� � |||||�}n"|dkr�d|d< n|dkr�d|d< |S )z�Generates a JSON schema that matches a schema that defines a typed dict. Args: schema: The core schema. Returns: The generated JSON schema. �totalTc s, g | ]$\}}� � |�r|� �|��|f�qS rD ��field_is_present�field_is_required�r] �name�field�rq rc rD rE rx � s �z8GenerateJsonSchema.typed_dict_schema.<locals>.<listcomp>�fieldsr9 �computed_fields�json_schema_extraZextra_behaviorN�extra�ignorer �forbidFr7 �allow) ro rd r� �extend� _name_required_computed_fields�_get_typed_dict_cls�_get_typed_dict_configr� �push� _named_required_fields_schemarJ �_update_class_schema) rq r@ �named_required_fieldsr �configr� rm rn r rD rj rE �typed_dict_schema� s, � z$GenerateJsonSchema.typed_dict_schemazlist[ComputedField]z1list[tuple[str, bool, core_schema.ComputedField]])rl rB c C s dd� | D �S )Nc S s g | ]}|d d|f�qS )Z property_nameTrD �r] ri rD rD rE rx s zEGenerateJsonSchema._name_required_computed_fields.<locals>.<listcomp>rD )rl rD rD rE rs s z1GenerateJsonSchema._name_required_computed_fieldsz+Sequence[tuple[str, bool, CoreSchemaField]])ry rB c C s� i }g }|D ]�\}}}| j r(| �||�}z| �|��� }W n tk rR Y q Y nX d|krx| �|�rx| �|�}||d< | �|�}|||< |r |�|� q d|d�} |r�|| d< | S )Nr r4 �r� rZ �required) r� �_get_alias_namer� r� r �field_title_should_be_set�get_title_from_name�handle_ref_overridesrc ) rq ry rZ Zrequired_fieldsrh r~ ri Zfield_json_schemar r� rD rD rE rw s( z0GenerateJsonSchema._named_required_fields_schemar2 )ri rh rB c C s� |d dkr|� d|�}n$| jdkr2|� d|�}n |� d|�}t|t�rN|}nXt|t�r�td|�}|D ]4}t|t�rft|�dkrft|d t�rf|d } q�qfnt|� |S ) Nr� �computed-fieldr\ r8 Zvalidation_aliasZserialization_aliaszlist[str] | strr! r )ro r� r} r~ rb r r[ r )rq ri rh r\ �pathrD rD rE r 1 s $z"GenerateJsonSchema._get_alias_namezcore_schema.TypedDictFieldc C s | � |d �S )z�Generates a JSON schema that matches a schema that defines a typed dict field. Args: schema: The core schema. Returns: The generated JSON schema. r@ r, r� rD rD rE �typed_dict_field_schemaF s z*GenerateJsonSchema.typed_dict_field_schemazcore_schema.DataclassFieldc C s | � |d �S )z�Generates a JSON schema that matches a schema that defines a dataclass field. Args: schema: The core schema. Returns: The generated JSON schema. r@ r, r� rD rD rE �dataclass_field_schemaQ s z)GenerateJsonSchema.dataclass_field_schemazcore_schema.ModelFieldc C s | � |d �S )z�Generates a JSON schema that matches a schema that defines a model field. Args: schema: The core schema. Returns: The generated JSON schema. r@ r, r� rD rD rE �model_field_schema\ s z%GenerateJsonSchema.model_field_schemazcore_schema.ComputedFieldc C s | � |d �S )z�Generates a JSON schema that matches a schema that defines a computed field. Args: schema: The core schema. Returns: The generated JSON schema. � return_schemar, r� rD rD rE �computed_field_schemag s z(GenerateJsonSchema.computed_field_schemazcore_schema.ModelSchemac C s� t d|d �}|j}|�d�}| j�|�� | �|d �}W 5 Q R X |�d�}|jrx|jd j}|rp|rpt d��|rx|}| � |||�dd �||�}|S ) z�Generates a JSON schema that matches a schema that defines a model. Args: schema: The core schema. Returns: The generated JSON schema. ztype[BaseModel]r r r@ rm �rootzz"model_config['json_schema_extra']" and "Field.json_schema_extra" on "RootModel.root" field must not be set simultaneouslyrn N) r Z model_configro r� rv r� Z__pydantic_root_model__Z model_fieldsrm � ValueErrorrx )rq r@ r rz r r� rm Zroot_json_schema_extrarD rD rE � model_schemar s �z GenerateJsonSchema.model_schemaz-Literal[('allow', 'ignore', 'forbid')] | Nonez type[Any]z)JsonDict | JsonSchemaExtraCallable | None)r� r rn r rm rB c C s� d|kr | � t|d ��p |}n|}|d k r8|�d|� d|krb|dkrRd|d< n|dkrbd|d< t|ttf�rz|�|�}t|t�r�|�|� nJt |�r�t t � |�j �dkr�|||� q�||� n|d k r�td |� d ���t|d �r�d|d < |S ) Nry r r7 rq Trp Fr! z"model_config['json_schema_extra']=z$ should be a dict, callable, or None�__deprecated__r )r� rP � setdefaultr} r� � classmethod�__get__r rC �callabler[ r � signature� parametersr� �hasattr)rq r� r rn r rm �schema_to_updaterD rD rE rx � s2 � z'GenerateJsonSchema._update_class_schemar� c C sF d|kr>| � t|d ��}|dkr4td|d � ���| �|�S |}|S )z�Resolve a JsonSchemaValue to the non-ref schema if it is a $ref schema. Args: json_schema: The schema to resolve. Returns: The resolved schema. ry Nz(Cannot update undefined schema for $ref=)r� rP � RuntimeError�resolve_schema_to_update)rq r� r� rD rD rE r� � s z+GenerateJsonSchema.resolve_schema_to_updatez core_schema.ModelFieldsSchemac sv � fdd�|d � � D �}� jdkr<|�� �|�dg ��� � �|�}|�dd�}|dk rr� �|�}� �|�|d< |S ) z�Generates a JSON schema that matches a schema that defines a model's fields. Args: schema: The core schema. Returns: The generated JSON schema. c s. g | ]&\}}� � |�r|� j|d d�|f�qS )T�rc rd rg rw rD rE rx � s �z:GenerateJsonSchema.model_fields_schema.<locals>.<listcomp>rk r9 rl � extras_schemaNr7 )rd r� rr rs ro rw r� r� )rq r@ ry r� r� r� rD rw rE �model_fields_schema� s � z&GenerateJsonSchema.model_fields_schema)ri rB c C s2 | j dkr|�d� S | j dkr$dS t| j � dS )a Whether the field should be included in the generated JSON schema. Args: field: The schema for the field itself. Returns: `True` if the field should be included in the generated JSON schema, `False` otherwise. r9 �serialization_excluder8 TN)r� ro r )rq ri rD rD rE re � s z#GenerateJsonSchema.field_is_presentzPcore_schema.ModelField | core_schema.DataclassField | core_schema.TypedDictField)ri rc rB c C sJ | j dkr | jjr |�d� S |d dkr6|�d|�S |d d dkS dS ) aY Whether the field should be marked as required in the generated JSON schema. (Note that this is irrelevant if the field is not present in the JSON schema.). Args: field: The schema for the field itself. total: Only applies to `TypedDictField`s. Indicates if the `TypedDict` this field belongs to is total, in which case any fields that don't explicitly specify `required=False` are required. Returns: `True` if the field should be marked as required in the generated JSON schema, `False` otherwise. r9 r� r� ztyped-dict-fieldr~ r@ rA N)r� r" Z+json_schema_serialization_defaults_requiredro )rq ri rc rD rD rE rf � s z$GenerateJsonSchema.field_is_requiredz core_schema.DataclassArgsSchemac sB � fdd�|d D �}� j dkr8|�� �|�dg ��� � �|�S )z�Generates a JSON schema that matches a schema that defines a dataclass's constructor arguments. Args: schema: The core schema. Returns: The generated JSON schema. c s. g | ]&}� � |�r|d � j|dd�|f�qS )rh Tr� rd r| rw rD rE rx s �z<GenerateJsonSchema.dataclass_args_schema.<locals>.<listcomp>rk r9 rl )r� rr rs ro rw )rq r@ ry rD rw rE �dataclass_args_schema s � z(GenerateJsonSchema.dataclass_args_schemazcore_schema.DataclassSchemac C s� |d }t |dtdi ��}|�d�p(|j}| j�|�� | �|d ��� }W 5 Q R X |�d�}| �|||�dd�||�}t |�r�t |d �s�d}n|j dkr�dn t � |j �}|r�||d < |S ) z�Generates a JSON schema that matches a schema that defines a dataclass. Args: schema: The core schema. Returns: The generated JSON schema. r �__pydantic_config__r1 r r@ rm rn NZ__pydantic_validator__r )r� r ro rJ r� rv r� r� rx r r� rM r r )rq r@ r rz r r� rm r rD rD rE �dataclass_schema% s z#GenerateJsonSchema.dataclass_schemazcore_schema.ArgumentsSchemac C s� t �|�j}|�d�}|d }dd� |D �}dd� |D �}dd� |D �}|�d�}|�d�} |r�| oj| } | r�| �|| |�S | o�| } | r�| �|| | �S |s�| o�| } | r�| �|| |�S td ��d S ) z�Generates a JSON schema that matches a schema that defines a function's arguments. Args: schema: The core schema. Returns: The generated JSON schema. Z'pydantic_js_prefer_positional_arguments�arguments_schemac S s g | ]}|� d �dkr|�qS )r� � keyword_onlyrQ �r] �arD rD rE rx P s z7GenerateJsonSchema.arguments_schema.<locals>.<listcomp>c S s g | ]}|� d �dkr|�qS )r� > N�positional_or_keywordrQ r� rD rD rE rx Q s c S s g | ]}|� d �dkr|�qS )r� Zpositional_onlyrQ r� rD rD rE rx R s �var_args_schema�var_kwargs_schemazfUnable to generate JSON schema for arguments validator with positional-only and keyword-only argumentsN)r# r� r� ro �p_arguments_schema�kw_arguments_schemar. ) rq r@ r� Zprefer_positional� argumentsZkw_only_argumentsZkw_or_p_argumentsZp_only_argumentsr� r� Zpositional_possibleZkeyword_possiblerD rD rE r� C s, �z#GenerateJsonSchema.arguments_schemaz$list[core_schema.ArgumentsParameter]zCoreSchema | None)r� r� rB c C s� i }g }|D ]P}| � |�}| �|d ��� }| �|�|d< |||< |d d dkr |�|� q d|d�}|rt||d< |r�| �|�} | r�| |d< nd |d< |S ) z�Generates a JSON schema that matches a schema that defines a function's keyword arguments. Args: arguments: The core schema. Returns: The generated JSON schema. r@ r r� rA r4 r} r~ r7 F)�get_argument_namer� r� r� rc ) rq r� r� rZ r~ �argumentrh �argument_schemar� Z additional_properties_schemarD rD rE r� h s$ z&GenerateJsonSchema.kw_arguments_schema)r� r� rB c C s� g }d}|D ]P}| � |�}| �|d ��� }| �|�|d< |�|� |d d dkr |d7 }q d|d�}|rt||d < |r�| �|�} | r�| |d < n t|�|d < |S ) z�Generates a JSON schema that matches a schema that defines a function's positional arguments. Args: arguments: The core schema. Returns: The generated JSON schema. r r@ r r� rA r! r )r� r- r+ rd r. )r� r� r� r� rc r[ ) rq r� r� Z prefix_itemsZ min_itemsr� rh r� r� r rD rD rE r� � s$ z%GenerateJsonSchema.p_arguments_schemaz core_schema.ArgumentsParameter)r� rB c C s, |d }| j r(|�d�}t|t�r(|}n |S )z�Retrieves the name of an argument. Args: argument: The core schema. Returns: The name of the argument. rh r\ )r� ro r} r~ )rq r� rh r\ rD rD rE r� � s z$GenerateJsonSchema.get_argument_namezcore_schema.CallSchemac C s | � |d �S )z�Generates a JSON schema that matches a schema that defines a function call. Args: schema: The core schema. Returns: The generated JSON schema. r� r, r� rD rD rE � call_schema� s z GenerateJsonSchema.call_schemaz core_schema.CustomErrorSchemac C s | � |d �S )z�Generates a JSON schema that matches a schema that defines a custom error. Args: schema: The core schema. Returns: The generated JSON schema. r@ r, r� rD rD rE �custom_error_schema� s z&GenerateJsonSchema.custom_error_schemazcore_schema.JsonSchemac C s: |� d�pt�� }| �|�}| jdkr2dd|d�S |S dS )z�Generates a JSON schema that matches a schema that defines a JSON object. Args: schema: The core schema. Returns: The generated JSON schema. r@ r8 r� zapplication/json)r� ZcontentMediaTypeZ contentSchemaN)ro r r� r� r� )rq r@ Zcontent_core_schemaZcontent_json_schemarD rD rE r� � s z GenerateJsonSchema.json_schemazcore_schema.UrlSchemac C s" dddd�}| � ||| jj� |S )z�Generates a JSON schema that matches a schema that defines a URL. Args: schema: The core schema. Returns: The generated JSON schema. r� �urir! �r� r� � minLength�r� r� r� r� rD rD rE � url_schema� s z GenerateJsonSchema.url_schemaz core_schema.MultiHostUrlSchemac C s" dddd�}| � ||| jj� |S )z�Generates a JSON schema that matches a schema that defines a URL that can be used with multiple hosts. Args: schema: The core schema. Returns: The generated JSON schema. r� zmulti-host-urir! r� r� r� rD rD rE �multi_host_url_schema� s z(GenerateJsonSchema.multi_host_url_schemazcore_schema.UuidSchemac C s ddd�S )z�Generates a JSON schema that matches a UUID. Args: schema: The core schema. Returns: The generated JSON schema. r� �uuidr� rD r� rD rD rE � uuid_schema s z GenerateJsonSchema.uuid_schemaz core_schema.DefinitionsSchemac C sx |d D ]`}z| � |� W q tk rf } z.t|d �}|| j| �|| jf�< W Y �qW 5 d}~X Y qX q| � |d �S )z�Generates a JSON schema that matches a schema that defines a JSON object with definitions. Args: schema: The core schema. Returns: The generated JSON schema. rX rm Nr@ )r� r. rN r� � get_defs_refr� )rq r@ Z definitionr� r� rD rD rE rh s z%GenerateJsonSchema.definitions_schemaz%core_schema.DefinitionReferenceSchemac C s t |d �}| �|�\}}|S )z�Generates a JSON schema that matches a schema that references a definition. Args: schema: The core schema. Returns: The generated JSON schema. Z schema_ref)rN r� )rq r@ r� r� r� rD rD rE �definition_ref_schema& s z(GenerateJsonSchema.definition_ref_schemazVcore_schema.SerSchema | core_schema.IncExSeqSerSchema | core_schema.IncExDictSerSchema�JsonSchemaValue | Nonec C sr |d }|dks|dkr6|� d�}|dk rn| �|�S n8|dksF|dkrT| �t�� �S |d dkrn| �|d �S dS ) z�Generates a JSON schema that matches a schema that defines a serialized object. Args: schema: The core schema. Returns: The generated JSON schema. r� rB z function-wrapr� Nr� z to-string�modelr@ )ro r� r� r )rq r@ Z schema_typer� rD rD rE r� 3 s z GenerateJsonSchema.ser_schema)rh rB c C s |� � �dd�S )z�Retrieves a title from a name. Args: name: The name to retrieve a title from. Returns: The title. r� � )r r� �rq rh rD rD rE r� N s z&GenerateJsonSchema.get_title_from_namec C s� t �|�r2|d dkr |d }n|d }| �|�S t �|�r�|�d�rJdS |d dkrd| �|d �S t �|�r|| �|d �S |d dkr�dS d S td |� ���d S ) a� Returns true if a field with the given schema should have a title set based on the field name. Intuitively, we want this to return true for schemas that wouldn't otherwise provide their own title (e.g., int, float, str), and false for those that would (e.g., BaseModel subclasses). Args: schema: The schema to check. Returns: `True` if the field should have a title set, `False` otherwise. r� r� r� r@ rm F> �nullablerA rX zdefinition-refTr� N)r$ r� r� r� ro r: r. )rq r@ Z field_schemarD rD rE r� Y s z,GenerateJsonSchema.field_title_should_be_setc C s t �dd|��dd�S )z�Normalizes a name to be used as a key in a dictionary. Args: name: The name to normalize. Returns: The normalized name. z[^a-zA-Z0-9.\-_]r� r� �__)�re�subr� r� rD rD rE �normalize_name| s z!GenerateJsonSchema.normalize_name� CoreModeRefrO )r� rB c C s |\}}t �d|�}dd� |D �}d�|�}dd� |D �}d�|�}t| }t| �|��}t| �|�d|� � �} t| �|��} t| � d|� ��} t| �|��} | j�| �} | dkr�| j| d7 < | j| } | j| < t| � d | � ��}t| � d | � ��}|| | | ||g| j |< |S ) z�Override this method to change the way that definitions keys are generated from a core reference. Args: core_mode_ref: The core reference. Returns: The definitions key. z([\][,])c S s g | ]}|� d d�d �qS )�:r! r )�rsplitr\ rD rD rE rx � s z3GenerateJsonSchema.get_defs_ref.<locals>.<listcomp>� c S s g | ]}t �d d|��qS )z (?:[^.[\]]+\.)+((?:[^.[\]]+))z\1)r� r� r\ rD rD rE rx � s r� Nr! r� ) r� �split�joinr; rO r� r� ro r� r� )rq r� r� r� � componentsZcore_ref_no_idZ short_refZ mode_titlerh Z name_modeZmodule_qualnameZmodule_qualname_modeZmodule_qualname_idZoccurrence_indexZmodule_qualname_occurrenceZ module_qualname_occurrence_moderD rD rE r� � s4 � z GenerateJsonSchema.get_defs_refrN z tuple[DefsRef, JsonSchemaValue])r� rB c C s� || j f}| j�|�}|dk r4| j| }|d|ifS | �|�}|| j|< || j|< t| jj|d��}|| j|< || j |< d|i}||fS )a� This method wraps the get_defs_ref method with some cache-lookup/population logic, and returns both the produced defs_ref and the JSON schema that will refer to the right definition. Args: core_ref: The core reference to get the definitions reference for. Returns: A tuple of the definitions reference and the JSON schema that will refer to it. Nry �r� ) r� r� ro r� r� r� rP r� r� r� )rq r� r� Zmaybe_defs_refr� ri r� rD rD rE r� � s z,GenerateJsonSchema.get_cache_defs_ref_schemac C s� d|kr�|� � }| �t|d ��}|dkrjt|�dkrf|� � }|�dg � |d �d|d i� |d= |S t|�� �D ],\}}|dkr�qv||krv|| |krv||= qvt|�dkr�t|d �}|d= d|ks�t�d|ig|d< |S )aQ It is not valid for a schema with a top-level $ref to have sibling keys. During our own schema generation, we treat sibling keys as overrides to the referenced schema, but this is not how the official JSON schema spec works. Because of this, we first remove any sibling keys that are redundant with the referenced schema, then if any remain, we transform the schema from a top-level '$ref' to use allOf to move the $ref out of the top level. (See bottom of https://swagger.io/docs/specification/using-ref/ for a reference about this behavior) ry Nr! r� ) r� r� rP r[ r� rc rb rd r[ )rq r� Zreferenced_json_schemar� r� r� rD rD rE r� � s* z'GenerateJsonSchema.handle_ref_overridesrP )r� rB c C s, | j | }|| jkr | j| �| j�|d �S rn )r� r� rX ro )rq r� Zdef_refrD rD rE r� � s z.GenerateJsonSchema.get_schema_from_definitionsr )�dftrB c C s� ddl m}m} | j}z0|t|��r(|n|t|�|jd�j|dd�}W n$ tk rj t� d|� ���Y nX tj ||j |j d�S )a Encode a default value to a JSON-serializable value. This is used to encode default values for fields in the generated JSON schema. Args: dft: The default value to encode. Returns: The encoded default value. r! )� TypeAdapter�_type_has_config)rz �json�r� z Unable to encode default value )Ztimedelta_modeZ bytes_mode) Z type_adapterr� r� r" r� Z config_dictZ dump_pythonr/ rD rE r r� r� )rq r� r� r� rz rA rD rD rE rC s ���z!GenerateJsonSchema.encode_defaultzdict[str, str]�None)r� r r� rB c C s* |� � D ] \}}||kr|| ||< qdS )a� Update the json_schema with the corresponding validations specified in the core_schema, using the provided mapping to translate keys in core_schema to the appropriate keys for a JSON schema. Args: json_schema: The JSON schema to update. core_schema: The core schema to get the validations from. mapping: A mapping from core_schema attribute names to the corresponding JSON schema attribute names. N)rd )rq r� r r� Zcore_keyZjson_schema_keyrD rD rE r� s z*GenerateJsonSchema.update_with_validationsc @ sX e Zd ZdZdddddd�Zdd d �Zdd d d �Zd dd �Zddd �Zddddd�Z dS )z%GenerateJsonSchema.ValidationsMappingag This class just contains mappings from core_schema attribute names to the corresponding JSON schema attribute names. While I suspect it is unlikely to be necessary, you can in principle override this class in a subclass of GenerateJsonSchema (by inheriting from GenerateJsonSchema.ValidationsMapping) to change these mappings. Z multipleOf�maximum�minimumZexclusiveMaximumZexclusiveMinimum)r� r� r� r� r� r� Z maxLength)� min_length� max_lengthr� )r� r� r� r+ r. Z minPropertiesZ maxProperties)r� r� r� r� N) rJ rK rL rM r� r� r� r r4 r� rD rD rD rE r� . s2 ������r� zlist[JsonSchemaValue]�rk rB c C s` g }|D ]2}t |�dkr0d|kr0|�|d � q|�|� qt|�}t |�dkrX|d S d|iS )Nr! r� r )r[ rr rc re )rq rk r r@ rD rD rE rH T s z&GenerateJsonSchema.get_flattened_anyofzdict[JsonRef, int]c s* t � �ddd�� ��fdd� � � |� �S )zKGet all values corresponding to the key '$ref' anywhere in the json_schema.r r� rt c s� t | t�r�d| krxt| d �}t |t�s,d S |�k}�| d7 < |rLd S �j| }|�jkrj�j| �� �j| � | �� D ] }� |� q�n t | t�r�| D ] }� |� q�d S )Nry r! ) r} r rP r~ r� r� rX rS rb )r@ r� Zalready_visitedri r� ��_add_json_refsZ json_refsrq rD rE r� d s$ z>GenerateJsonSchema.get_json_ref_counts.<locals>._add_json_refs)r )rq r� rD r� rE r� ` s z&GenerateJsonSchema.get_json_ref_counts)r@ � error_inforB c C s t d|� ���d S )Nz!Cannot generate a JsonSchema for )r. )rq r@ r� rD rD rE r | s z1GenerateJsonSchema.handle_invalid_for_json_schema�JsonSchemaWarningKind)�kind�detailrB c C s$ | � ||�}|dk r t�|t� dS )zfThis method simply emits PydanticJsonSchemaWarnings based on handling in the `warning_message` method.N)�render_warning_messager$ r% rI )rq r� r� rL rD rD rE rF s z GenerateJsonSchema.emit_warningc C s || j krdS |� d|� d�S )am This method is responsible for ignoring warnings as desired, and for formatting the warning messages. You can override the value of `ignored_warning_kinds` in a subclass of GenerateJsonSchema to modify what warnings are generated. If you want more control, you can override this method; just return None in situations where you don't want warnings to be emitted. Args: kind: The kind of warning to render. It can be one of the following: - 'skipped-choice': A choice field was skipped because it had no valid choices. - 'non-serializable-default': A default value was skipped because it was not JSON-serializable. detail: A string with additional details about the warning. Returns: The formatted warning message, or `None` if no warning should be emitted. Nz [�])r� )rq r� r� rD rD rE r� � s z)GenerateJsonSchema.render_warning_messagerS c C sJ i }| j �� D ](}|D ] }t| jj|d��}|||< qqt�| j || j�S )Nr� )r� rS rP r� r� rS rl rX )rq rW Z defs_refsri r� rD rD rE r� � s �z/GenerateJsonSchema._build_definitions_remappingc sl t � � t|�}|rN|�� }| j| }|� kr.q� �|� |�t| j| �� q� fdd�| j�� D �| _d S )Nc s i | ]\}}|� kr||�qS rD rD r� �Zvisited_defs_refsrD rE r| � s zCGenerateJsonSchema._garbage_collect_definitions.<locals>.<dictcomp>)�set�_get_all_json_refsr� r� �addrC rX rd )rq r@ Zunvisited_json_refsZ next_json_refZ next_defs_refrD r� rE r� � s z/GenerateJsonSchema._garbage_collect_definitions)r8 )hrJ rK rL rM Zschema_dialectr� r� �DEFAULT_REF_TEMPLATEr� �propertyr" r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r' r) r& r1 r2 r0 r3 r8 r; r= r> r? r@ rG rI rP rW rT r^ r` ra r{ r� rs rw r r� r� r� r� r� rx r� r� re rf r� r� r� r� r� r� r� r� r� r� r� r� rh r� r� r� r� r� r� r� r� r� rC r� r� rH r� r rF r� r� r� rD rD rD rE r� � s� $14s ! & # 5%(& !, %%( # 0 ) & r� Tr8 z)type[BaseModel] | type[PydanticDataclass]r� r~ ztype[GenerateJsonSchema]r� )r r� r� �schema_generatorr� rB c C sf ddl m} |||d�}t| jtj�r0| j�� | |kr@td��t| jtj�rVtd��|j | j|d�S )a� Utility function to generate a JSON Schema for a model. Args: cls: The model class to generate a JSON Schema for. by_alias: If `True` (the default), fields will be serialized according to their alias. If `False`, fields will be serialized according to their attribute name. ref_template: The template to use for generating JSON Schema references. schema_generator: The class to use for generating the JSON Schema. mode: The mode to use for generating the JSON Schema. It can be one of the following: - 'validation': Generate a JSON Schema for validating data. - 'serialization': Generate a JSON Schema for serializing data. Returns: The generated JSON Schema. r! r6 r� zTmodel_json_schema() must be called on a subclass of BaseModel, not BaseModel itself.z this is a bug! please report itr� ) �mainr7 r} �__pydantic_core_schema__r' �MockCoreSchema�rebuildr� r[ r� )r r� r� r� r� r7 Zschema_generator_instancerD rD rE �model_json_schema� s r� )r� r r r� r� zJSequence[tuple[type[BaseModel] | type[PydanticDataclass], JsonSchemaMode]]rY zotuple[dict[tuple[type[BaseModel] | type[PydanticDataclass], JsonSchemaMode], JsonSchemaValue], JsonSchemaValue])�modelsr� r r r� r� rB c C s~ | D ] \}}t |jtj�r|j�� q|||d�}dd� | D �} |�| �\} } i } | r^| | d< |rj|| d< |rv|| d< | | fS )a? Utility function to generate a JSON Schema for multiple models. Args: models: A sequence of tuples of the form (model, mode). by_alias: Whether field aliases should be used as keys in the generated JSON Schema. title: The title of the generated JSON Schema. description: The description of the generated JSON Schema. ref_template: The reference template to use for generating JSON Schema references. schema_generator: The schema generator to use for generating the JSON Schema. Returns: A tuple where: - The first element is a dictionary whose keys are tuples of JSON schema key type and JSON mode, and whose values are the JSON schema corresponding to that pair of inputs. (These schemas may have JsonRef references to definitions that are defined in the second returned element.) - The second element is a JSON schema containing all definitions referenced in the first returned element, along with the optional title and description keys. r� c S s g | ]\}}|||j f�qS rD )r� )r] �mr� rD rD rE rx � s z&models_json_schema.<locals>.<listcomp>rY r r )r} r� r' r� r� r� ) r� r� r r r� r� r r� �instancer� r� rX r� rD rD rE �models_json_schema� s �r� )�_HashableJsonValue.r� .r zIterable[JsonDict]rX r� c C s t dd� | D ��� �S )Nc S s i | ]}t |�|�qS rD ��_make_json_hashable)r] r@ rD rD rE r| s z(_deduplicate_schemas.<locals>.<dictcomp>)rb rS )rk rD rD rE re s re r- )r{ rB c C sH t | t�r$ttdd� | �� D ���S t | t�r@tdd� | D ��S | S d S )Nc s s | ]\}}|t |�fV qd S rn r� r� rD rD rE r` s z&_make_json_hashable.<locals>.<genexpr>c s s | ]}t |�V qd S rn r� r rD rD rE r` s )r} r rK �sortedrd rb )r{ rD rD rE r� s r� )r{ � parent_keyrB c C s� t | t�rPi }| �� }|dkr.|dkr.t|�}|D ]}t| | |d�||< q2|S t | t�r|g }| D ]}|�t||�� qb|S | S d S )NrZ rA )r� )r} r r� r� r� rb rc )r{ r� Z sorted_dictr� rz � sorted_listrv rD rD rE r� s r� c @ sF e Zd ZU dZded< dZded< ddd d �d d �Zd d�dd�ZdS )�WithJsonSchemaa, Usage docs: https://docs.pydantic.dev/2.8/concepts/json_schema/#withjsonschema-annotation Add this as an annotation on a field to override the (base) JSON schema that would be generated for that field. This provides a way to set a JSON schema for types that would otherwise raise errors when producing a JSON schema, such as Callable, or types that have an is-instance core schema, without needing to go so far as creating a custom subclass of pydantic.json_schema.GenerateJsonSchema. Note that any _modifications_ to the schema that would normally be made (such as setting the title for model fields) will still be performed. If `mode` is set this will only apply to that schema generation mode, allowing you to set different json schemas for validation and serialization. r� r� N�/Literal[('validation', 'serialization')] | Noner� �core_schema.CoreSchemar* r> �r �handlerrB c C s8 | j p |j }||j kr ||�S | jd kr.t�n| jS d S rn )r� r� r )rq r r� r� rD rD rE � __get_pydantic_json_schema__D s z+WithJsonSchema.__get_pydantic_json_schema__r� r� c C s t t| j��S rn ��hashr� r� rw rD rD rE �__hash__P s zWithJsonSchema.__hash__�rJ rK rL rM r� r� r� r� rD rD rD rE r� 2 s r� c @ sF e Zd ZU dZded< dZded< ddd d �d d �Zd d�dd�ZdS )�Examplesa Add examples to a JSON schema. Examples should be a map of example names (strings) to example values (any valid JSON). If `mode` is set this will only apply to that schema generation mode, allowing you to add different examples for validation and serialization. r? �examplesNr� r� r� r* r> r� c C sJ | j p |j }||�}||j kr"|S |�di �}|�t| j�� ||d< |S )Nr� )r� ro rC r r� )rq r r� r� r� r� rD rD rE r� b s z%Examples.__get_pydantic_json_schema__r� r� c C s t t| j��S rn r� rw rD rD rE r� n s zExamples.__hash__r� rD rD rD rE r� T s r� r z set[JsonRef]�rv rB c C s� t � }| g}|r�|�� }t|t�r�|�� D ]T\}}|dkrTt|t�rT|�t|�� q*t|t�rj|�|� q*t|t �r*|� |� q*q t|t �r |� |� q |S )z6Get all the definitions references from a JSON schema.ry ) r� r� r} r rd r~ r� rP rc rb rr )rv �refs�stack�currentrz r{ rD rD rE r� r s r� �AnyTypec @ s@ e Zd ZdZddd�dd�Zdddd �d d �Zd d �dd�ZdS )�SkipJsonSchemaa Usage docs: https://docs.pydantic.dev/2.8/concepts/json_schema/#skipjsonschema-annotation Add this as an annotation on a field to skip generating a JSON schema for that field. Example: ```py from typing import Union from pydantic import BaseModel from pydantic.json_schema import SkipJsonSchema from pprint import pprint class Model(BaseModel): a: Union[int, None] = None # (1)! b: Union[int, SkipJsonSchema[None]] = None # (2)! c: SkipJsonSchema[Union[int, None]] = None # (3)! pprint(Model.model_json_schema()) ''' { 'properties': { 'a': { 'anyOf': [ {'type': 'integer'}, {'type': 'null'} ], 'default': None, 'title': 'A' }, 'b': { 'default': None, 'title': 'B', 'type': 'integer' } }, 'title': 'Model', 'type': 'object' } ''' ``` 1. The integer and null types are both included in the schema for `a`. 2. The integer type is the only type included in the schema for `b`. 3. The entirety of the `c` field is omitted from the schema. r r� c C s t || � f S rn )r )r rv rD rD rE �__class_getitem__� s z SkipJsonSchema.__class_getitem__r r* r> r� c C s t �d S rn )r )rq r r� rD rD rE r� � s z+SkipJsonSchema.__get_pydantic_json_schema__r� r� c C s t t| ��S rn )r� r� rw rD rD rE r� � s zSkipJsonSchema.__hash__N)rJ rK rL rM r r� r� rD rD rD rE r � s 1r rb ztype[Any] | Nonert c C s t �| �j}|�d�}|S )NZpydantic_typed_dict_cls)r# r� r� ro )r@ r� r rD rD rE rt � s rt r1 )r rB c C s0 | d k r,zt �| d�W S tk r* Y nX i S )Nr� )r% Zget_attribute_from_basesr� )r rD rD rE ru � s ru )N)srM � __future__r Z _annotations� dataclassesr r� r� r$ � collectionsr r� r r r r �typingr r r r r r r r r r r r r r rD r r r r Zpydantic_core.core_schemar �typing_extensionsr r r r r r Zpydantic.warningsr � _internalr" r# r$ r% r& r' r( r) Zannotated_handlersr* rz r+ r, r- �errorsr. r/ r0 r� r1 Z_internal._core_utilsr2 r3 Z_internal._dataclassesr4 Z#_internal._schema_generation_sharedr5 r� r7 ZCoreSchemaTypeZCoreSchemaFieldTyper� r~ r> r� r; r� rF r� � UserWarningrI r� rN rO rP r� rQ � dataclassZ slots_truerS r� r� r� r� r� r� r� re r� r� r� r� r� r r rt ru rD rD rD rE �<module> s� @ ( � P _�(� 3$� ! >