o
    AhI                     @  sh  d Z ddlmZ ddlZddlZddlZddlZddlm	Z	m
Z
mZmZmZmZmZmZ ddlmZ ddlmZmZ ddlmZmZmZmZ dd	lmZ dd
lmZ ddlm Z  ddl!m"Z" ddl#m$Z$m%Z%m&Z& e	rvddl'm(Z( ddl)m*Z* dZ+edZ,ej-dkreej.e$e&fdeddddddddddd
dMd(d)Z/eej.e$e&fdeddddddddddd
dNd-d)Z/n4eej.e$e&fdeddddddddd.dOd/d)Z/eej.e$e&fdeddddddddd.dPd0d)Z/eej.e$e&fd	dQddddddddddd
dRd3d)Z/dSd5d6Z0ee1Z2ej-d7k r"dTd<d=Z3e3ej4_5ddd>dd?dUdFdGZ6dVdKdLZ7dS )Wz7Provide an enhanced dataclass that performs validation.    )annotationsN)TYPE_CHECKINGAnyCallableGenericLiteralNoReturnTypeVaroverload)warn)	TypeGuarddataclass_transform   )_config_decorators_namespace_utils_typing_extra)_dataclasses)getattr_migration)
ConfigDict)PydanticUserError)Field	FieldInfoPrivateAttr)PydanticDataclass)MappingNamespace)	dataclassrebuild_dataclass_T   
   )field_specifiersFT.
initrepreqorderunsafe_hashfrozenconfigvalidate_on_initkw_onlyslotsr$   Literal[False]r%   boolr&   r'   r(   r)   r*    ConfigDict | type[object] | Noner+   bool | Noner,   r-   return-Callable[[type[_T]], type[PydanticDataclass]]c        
   
      C     d S N r#   r6   r6   U/var/www/html/openai_agents/venv/lib/python3.10/site-packages/pydantic/dataclasses.pyr         r   _clstype[_T]type[PydanticDataclass]c       
         C  r4   r5   r6   )r9   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r6   r6   r7   r   0   r8   r$   r%   r&   r'   r(   r)   r*   r+   c                 C  r4   r5   r6   r<   r6   r6   r7   r   C      c          	      C  r4   r5   r6   )	r9   r$   r%   r&   r'   r(   r)   r*   r+   r6   r6   r7   r   R   r=   type[_T] | NoneGCallable[[type[_T]], type[PydanticDataclass]] | type[PydanticDataclass]c       
      	     sj   |du sJ d|dusJ dt jdkr|	dni d fd
d}| du r1|S || S )a  !!! abstract "Usage Documentation"
        [`dataclasses`](../concepts/dataclasses.md)

    A decorator used to create a Pydantic-enhanced dataclass, similar to the standard Python `dataclass`,
    but with added validation.

    This function should be used similarly to `dataclasses.dataclass`.

    Args:
        _cls: The target `dataclass`.
        init: Included for signature compatibility with `dataclasses.dataclass`, and is passed through to
            `dataclasses.dataclass` when appropriate. If specified, must be set to `False`, as pydantic inserts its
            own  `__init__` function.
        repr: A boolean indicating whether to include the field in the `__repr__` output.
        eq: Determines if a `__eq__` method should be generated for the class.
        order: Determines if comparison magic methods should be generated, such as `__lt__`, but not `__eq__`.
        unsafe_hash: Determines if a `__hash__` method should be included in the class, as in `dataclasses.dataclass`.
        frozen: Determines if the generated class should be a 'frozen' `dataclass`, which does not allow its
            attributes to be modified after it has been initialized. If not set, the value from the provided `config` argument will be used (and will default to `False` otherwise).
        config: The Pydantic config to use for the `dataclass`.
        validate_on_init: A deprecated parameter included for backwards compatibility; in V2, all Pydantic dataclasses
            are validated on init.
        kw_only: Determines if `__init__` method parameters must be specified by keyword only. Defaults to `False`.
        slots: Determines if the generated class should be a 'slots' `dataclass`, which does not allow the addition of
            new attributes after instantiation.

    Returns:
        A decorator that accepts a class as its argument and returns a Pydantic `dataclass`.

    Raises:
        AssertionError: Raised if `init` is not `False` or `validate_on_init` is `False`.
    Fz7pydantic.dataclasses.dataclass only supports init=Falsez-validate_on_init=False is no longer supportedr   )r,   r-   cls	type[Any]r2   r;   c              
     s  ddl m} || rtd| j ddd| }tdd | jD }|s8d	ur8t| d
r8td| j dtdd d	ur>nt	| d
d	}t
|}tj| }|| | j}t| rwd	}| f}t| trpt| j }	||	f }t| j|} d	ur |jrtd| jdtdd n|jpd t| }
|
D ]}t	| |d	}t|trt| |t| qt|  t j!| fd	 d} W d	   n1 sw   Y  |j"r
| j#t$%| j#d) fdd}|&d	| | _#r
t| ds
d*d!d"}d+d$d%}|| _'|| _(d| _)|| _*|| _t	|d&d	}|j+| _+t,j-d'kr.|d	ur.||_.|| _.|j/| _/t0t1| _2d| _3tj4| |dd( | S ),zCreate a Pydantic dataclass from a regular dataclass.

        Args:
            cls: The class to create the Pydantic dataclass from.

        Returns:
            A Pydantic dataclass.
        r   )is_model_classz(Cannot create a Pydantic dataclass from z" as it is already a Pydantic modelzdataclass-on-model)codec                 s  s    | ]}t |V  qd S r5   )dataclassesis_dataclass).0baser6   r6   r7   	<genexpr>   s    z6dataclass.<locals>.create_dataclass.<locals>.<genexpr>N__pydantic_config__z[`config` is set via both the `dataclass` decorator and `__pydantic_config__` for dataclass zK. The `config` specification from `dataclass` decorator will take priority.   )category
stacklevelzN`frozen` is set via both the `dataclass` decorator and `config` for dataclass zW.This is not recommended. The `frozen` specification on `dataclass` will take priority.FT)r$   r%   r&   r'   r(   r)   instancer   namestrvaluer   r2   Nonec                  sp    r| ||S t | }t||d }t|tr|| | d S t|tjr.| j|| d S |j	
| || d S r5   )typegetattr
isinstanceproperty__set__	functoolscached_property__dict____setitem____pydantic_validator__validate_assignment)rM   rN   rP   inst_clsattrfrozen_original_setattrr6   r7   validated_setattr   s   
z>dataclass.<locals>.create_dataclass.<locals>.validated_setattr__setstate__self	list[Any]c                   s    fddt  D S )Nc                   s   g | ]}t  |jqS r6   )rS   rN   )rF   frd   r6   r7   
<listcomp>  s    zTdataclass.<locals>.create_dataclass.<locals>._dataclass_getstate.<locals>.<listcomp>)rD   fieldsrg   r6   rg   r7   _dataclass_getstate  s   z@dataclass.<locals>.create_dataclass.<locals>._dataclass_getstatestatec                 S  s.   t t| |D ]\}}t| |j| qd S r5   )ziprD   ri   object__setattr__rN   )rd   rk   fieldrP   r6   r6   r7   _dataclass_setstate  s   z@dataclass.<locals>.create_dataclass.<locals>._dataclass_setstate__firstlineno__)r       )raise_errors)rM   r   rN   rO   rP   r   r2   rQ   )rd   r   r2   re   )rd   r   rk   re   r2   rQ   )5_internal._utilsrB   r   __name__any	__bases__hasattrr   UserWarningrS   r   ConfigWrapperr   DecoratorInfosbuildupdate_from_config__doc___pydantic_dataclassesis_stdlib_dataclass
issubclassr   __parameters__types	new_classr)   r   safe_get_annotationsrT   r   setattras_dataclass_fieldpatch_base_fieldsrD   r   r\   rn   rW   wraps__get____getstate__rc   __is_pydantic_dataclass____pydantic_decorators__
__module__sysversion_inforq   __qualname__classmethod_pydantic_fields_complete__pydantic_fields_complete____pydantic_complete__complete_dataclass)r@   rB   original_clshas_dataclass_baseconfig_dictconfig_wrapper
decoratorsoriginal_docbasesgeneric_basecls_anns
field_namefield_valuerb   rj   rp   firstlinenor*   r&   r)   kwargsr'   r%   r-   r(   r_   r7   create_dataclass   s   	








	



z#dataclass.<locals>.create_dataclassN)r@   rA   r2   r;   )r   r   )r9   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r   r6   r   r7   r   b   s   /
 !r@   c                 C  s   t dd | j D S )zReturn whether the fields where successfully collected (i.e. type hints were successfully resolves).

    This is a private property, not meant to be used outside Pydantic.
    c                 s  s    | ]}|j V  qd S r5   )	_complete)rF   
field_infor6   r6   r7   rH   A  s    z,_pydantic_fields_complete.<locals>.<genexpr>)all__pydantic_fields__values)r@   r6   r6   r7   r   <  s   r   )r       argsr   r   r   c                  O  s   t d)a9  This function does nothing but raise an error that is as similar as possible to what you'd get
        if you were to try calling `InitVar[int]()` without this monkeypatch. The whole purpose is just
        to ensure typing._type_check does not error if the type hint evaluates to `InitVar[<parameter>]`.
        z 'InitVar' object is not callable)	TypeError)r   r   r6   r6   r7   _call_initvarJ  s   r   rJ   )forcers   _parent_namespace_depth_types_namespacer   rs   r   intr   MappingNamespace | Nonec                C  s   |s| j rdS dD ]}|| jv rt| | q	d| _ |dur |}n|dkr.tj|ddp,i }ni }tj|d}tj| t	j
| jdd||dd	S )
ax  Try to rebuild the pydantic-core schema for the dataclass.

    This may be necessary when one of the annotations is a ForwardRef which could not be resolved during
    the initial attempt to build the schema, and automatic rebuilding fails.

    This is analogous to `BaseModel.model_rebuild`.

    Args:
        cls: The class to rebuild the pydantic-core schema for.
        force: Whether to force the rebuilding of the schema, defaults to `False`.
        raise_errors: Whether to raise errors, defaults to `True`.
        _parent_namespace_depth: The depth level of the parent namespace, defaults to 2.
        _types_namespace: The types namespace, defaults to `None`.

    Returns:
        Returns `None` if the schema is already "complete" and rebuilding was not required.
        If rebuilding _was_ required, returns `True` if rebuilding was successful, otherwise `False`.
    N)__pydantic_core_schema__r[   __pydantic_serializer__Fr   T)parent_depthr   )parent_namespace)check)rs   ns_resolver_force_build)r   rY   delattrr   parent_frame_namespacer   
NsResolverr   r   r   rz   rI   )r@   r   rs   r   r   r^   
rebuild_nsr   r6   r6   r7   r   T  s,   


r   class_rA   "TypeGuard[type[PydanticDataclass]]c                C  s,   zd| j v o
t| W S  ty   Y dS w )zWhether a class is a pydantic dataclass.

    Args:
        class_: The class.

    Returns:
        `True` if the class is a pydantic dataclass, `False` otherwise.
    r   F)rY   rD   rE   AttributeError)r   r6   r6   r7   is_pydantic_dataclass  s
   	r   )r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r/   r*   r0   r+   r1   r,   r/   r-   r/   r2   r3   )r9   r:   r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r1   r*   r0   r+   r1   r,   r/   r-   r/   r2   r;   )r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r1   r*   r0   r+   r1   r2   r3   )r9   r:   r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r1   r*   r0   r+   r1   r2   r;   r5   )r9   r>   r$   r.   r%   r/   r&   r/   r'   r/   r(   r/   r)   r1   r*   r0   r+   r1   r,   r/   r-   r/   r2   r?   )r@   r;   r2   r/   )r   r   r   r   r2   r   )r@   r;   r   r/   rs   r/   r   r   r   r   r2   r1   )r   rA   r2   r   )8r~   
__future__r   _annotationsrD   rW   r   r   typingr   r   r   r   r   r   r	   r
   warningsr   typing_extensionsr   r   	_internalr   r   r   r   r   r   
_migrationr   r*   r   errorsr   ri   r   r   r   _internal._dataclassesr   _internal._namespace_utilsr   __all__r   r   ro   r   r   ru   __getattr__r   InitVar__call__r   r   r6   r6   r6   r7   <module>   s    (
 
Z
=