gx&UdZddlmZddlZddlmZmZmZmZm Z ddl m Z m Z m Z mZddlmZmZer ddlmZmZmZmZed Ze d Zed Zeej6Zd ed < d ddZGddZddZ ddZ!y)z(Pluggable schema validator for pydantic.) annotationsN) TYPE_CHECKINGAnyCallableIterableTypeVar) CoreConfig CoreSchemaSchemaValidatorValidationError)Literal ParamSpec)BaseValidateHandlerProtocolPydanticPluginProtocol SchemaKindSchemaTypePathPR)on_validate_pythonon_validate_jsonon_validate_stringsz list[Event]eventsc |ddlm}ddlm}|} | rt |||||||| |xsiSt ||S)zCreate a `SchemaValidator` or `PluggableSchemaValidator` if plugins are installed. Returns: If plugins are installed then return `PluggableSchemaValidator`, otherwise return `SchemaValidator`. r)r) get_plugins)r_loaderrPluggableSchemaValidatorr ) schema schema_typeschema_type_moduleschema_type_name schema_kindconfigplugin_settingsrrpluginss V/var/www/openai/venv/lib/python3.12/site-packages/pydantic/plugin/_schema_validator.pycreate_schema_validatorr(sS!$mG'   -/? @     !r  vv..cDeZdZdZdZ ddZddZy)rzPluggable schema validator.)_schema_validator validate_jsonvalidate_pythonvalidate_stringsc Pt|||_g}g} g} |D]W} | j||||||\} } }| |j| | | j| |G| j|Yt|jj||_ t|jj| |_ t|jj| |_ y#t$r8}td| jd| j j d||d}~wwxYw)NzError using plugin `:z`: ) r r+new_schema_validator TypeError __module__ __class____name__append build_wrapperr-r,r.)selfrr schema_type_pathr#r$r&r%python_event_handlersjson_event_handlersstrings_event_handlerspluginpjses r'__init__z!PluggableSchemaValidator.__init__:s8"1!@CEACDFF u 55K)9;P_1a }%,,Q/}#**1-}&--a0 -T-C-C-S-SUjk*4+A+A+O+OQde -d.D.D.U.UWm n u"6v7H7H6I6K[K[KdKdJeehijhk lmstt usC$$ D%-3D  D%c.t|j|SN)getattrr+)r8names r' __getattr__z$PluggableSchemaValidator.__getattr__[st--t44r)N)rr r rr9rr#rr$CoreConfig | Noner&z Iterable[PydanticPluginProtocol]r%zdict[str, Any]returnNone)rFstrrIr)r5r3 __qualname____doc__ __slots__rBrGr)r'rr5sl%[Iooo) o  o " o2o(o oB5r)rc|sStd|Dtd|Dtd|Dtd|Dtjdfd }|S)Nc3NK|]}t|ds|jyw)on_enterN)filter_handlersrR.0hs r' z build_wrapper..c _nPQS]@^!**n%%c3NK|]}t|ds|jyw) on_successN)rSr[rTs r'rWz build_wrapper..ds f>a_UVXdEeQ\\>rYc3NK|]}t|ds|jyw)on_errorN)rSr]rTs r'rWz build_wrapper..erXrYc3NK|]}t|ds|jyw) on_exceptionN)rSr_rTs r'rWz build_wrapper..fs knXY[iHjannnrYc D] }||i|   |i|} D] }|| |S#t$r} D] }|| d}~wt$r} D] }|| d}~wwxYwrD)r Exception)argskwargson_enter_handlerresulton_success_handlererroron_error_handler exceptionon_exception_handlerfunc on_enters on_errors on_exceptions on_successess r'wrapperzbuild_wrapper..wrapperhs$-  $1&1%. t.v.+7&&v.+7 # (1$$U+)2 ,9((3-: s + A%A A%A  A%)rbzP.argsrczP.kwargsrIr)tuple functoolswraps)rkevent_handlersrprlrmrnros` @@@@r'r7r7_sr  _n__ f>ff _n__ knkk      &r)cDt||d}|y|jdk(ryy)zFilter out handler methods which are not implemented by the plugin directly - e.g. are missing or are inherited from the protocol. NFzpydantic.pluginT)rEr3) handler_cls method_namehandlers r'rSrSs1k;5G   0 0r))NN)rr r rr!rKr"rKr#rr$rHr%zdict[str, Any] | NonerIz*SchemaValidator | PluggableSchemaValidator)rkCallable[P, R]rtz!list[BaseValidateHandlerProtocol]rIry)rvrrwrKrIbool)"rM __future__rrrtypingrrrrr pydantic_corer r r r typing_extensionsr rrrrrrrrEventlist__args__r__annotations__r(rr7rSrOr)r'rs."BBRR0aacN CLOP5>>* *!%-1/ /// /  /  /+/0/@'5'5T@ r)