o Ci<@sddlmZddlmZmZmZmZddlmZm Z ddl m Z ddl m Z mZer6ddlmZddlmZed e eZGd d d eeZddddZ ddddZdS)) annotations) TYPE_CHECKINGCallableGenericTypeVar)SchemaSerializerSchemaValidator)Literal)PydanticErrorCodesPydanticUserError)PydanticDataclass) BaseModelValSerc@s8eZdZdZdZdddddZdddZdddZdS) MockValSerzMocker for `pydantic_core.SchemaValidator` or `pydantic_core.SchemaSerializer` which optionally attempts to rebuild the thing it's mocking when one of its methods is accessed and raises an error if that fails. )_error_message_code _val_or_ser_attempt_rebuildN)attempt_rebuild error_messagestrcoder val_or_ser"Literal['validator', 'serializer']r"Callable[[], ValSer | None] | NonereturnNonecCs(||_|dkr tnt|_||_||_dS)N validator)rrrrrr)selfrrrrr x/var/www/html/fyndo/python/python_agents/rag_suite/venv/lib/python3.10/site-packages/pydantic/_internal/_mock_val_ser.py__init__s zMockValSer.__init__itemcCs@d}|jr|}|durt||St|j|t|j|jd)NTr)rgetattrrr rr)rr#__tracebackhide__val_serr r r! __getattr__&s  zMockValSer.__getattr__ ValSer | NonecCs.|jr|}|dur |St|j|jddS)Nr$)rr rr)rr'r r r!rebuild1s zMockValSer.rebuild) rrrr rrrrrr)r#rrr)rr))__name__ __module__ __qualname____doc__ __slots__r"r(r*r r r r!rs  rall referenced typesclstype[BaseModel]cls_namerundefined_namerrcs\d|d|d|d}dfdd }t|d d |d _dfd d }t|d d|d _dS)a&Set `__pydantic_validator__` and `__pydantic_serializer__` to `MockValSer`s on a model. Args: cls: The model class to set the mocks on cls_name: Name of the model class, used in error messages undefined_name: Name of the undefined thing, used in error messages `*` is not fully defined; you should define z , then call `z.model_rebuild()`.rSchemaValidator | Nonecjddddur jSdSNF) raise_errors_parent_namespace_depth) model_rebuild__pydantic_validator__r r1r r!attempt_rebuild_validatorHz2set_model_mocks..attempt_rebuild_validatorclass-not-fully-definedrrrrSchemaSerializer | Nonecr8r9)r=__pydantic_serializer__r r?r r!attempt_rebuild_serializerUrAz3set_model_mocks..attempt_rebuild_serializer serializerNrr7rrD)rr>rEr1r3r4undefined_type_error_messager@rFr r?r!set_model_mocks;s$  rLtype[PydanticDataclass]cslddlmd|d|d|d}dfd d }t|d d |d _dfdd }t|d d |d _dS)a*Set `__pydantic_validator__` and `__pydantic_serializer__` to `MockValSer`s on a dataclass. Args: cls: The model class to set the mocks on cls_name: Name of the model class, used in error messages undefined_name: Name of the undefined thing, used in error messages r )rebuild_dataclassr5r6z4, then call `pydantic.dataclasses.rebuild_dataclass(z)`.rr7cddddur jSdSr9)r>r r1rNr r!r@trAz6set_dataclass_mocks..attempt_rebuild_validatorrBrrCrDcrOr9)rEr rPr r!rFrAz7set_dataclass_mocks..attempt_rebuild_serializerNrHrI) dataclassesrNrr>rErJr rPr!set_dataclass_mockscs&  rRN)r0)r1r2r3rr4rrr)r1rMr3rr4rrr) __future__rtypingrrrr pydantic_corerrtyping_extensionsr errorsr r rQr mainrrrrLrRr r r r!s      ))