gKrUdZddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZddlZddlmZddlZddlmZd d lmZmZd d lmZe rdd lmZdd lmZmZddlm Z m!Z!ejDjGdZ$e$jKdZ&GddedZ'GddedZ(GddZ)d.dZ*eGddZ+e+Z,dZ-de.d<e/de-DZ0de.d <da1d!e.d"<d/d#Z2d0d$Z3d1d%Z4e d&Z5e d'Z6ed(Z7ed)Z8d2d*Z9d3d+Z:d4d,Z;d5d-Z $$v3DMM33T:DM r#c<ddlmjdk(r"tjd fd }|Sjdk(r"tjd fd }|Sjdk(sJtjd fd }|S) zEDecorator which wraps a schema validator method with instrumentation.rrallctjs  |g|i|Sj|}  |g|i|}jdj |||S#$r*}jdj ||d}~wt $r*}jdj||d}~wwxYwNT)successF)r _initialized _on_enter_count_validation _on_success_on_error_span Exception_on_exception_span) input_dataargskwargsspanresulterror exceptionrr9 validators r$wrapped_validatorz4_ValidateWrapper.__call__..wrapped_validatorqs$11 %ZA$A&AA ~~j1 "&zCDCFCF**4*8$$T62!M'**5*9''e4 **5*9++D)<s# A""C'%B  C%B==Cfailurec@tjs |g|i|S |g|i|}jd|S#$r)}jdj|d}~wt$r)}jdj |d}~wwxYwrB)rrDrF _on_error_logrI_on_exception_log) rKrLrMrOrPrQrr9rRs r$rSz4_ValidateWrapper.__call__..wrapped_validators$11$ZA$A&AA "&zCDCFCF**4*8!M'**5*9&&u- **5*9**95s! ?B$A(( B4$BBmetricsctjs |g|i|S |g|i|}jd|S#t$rjdwxYwrB)rrDrFrI)rKrLrMrOr9rRs r$rSz4_ValidateWrapper.__call__..wrapped_validatorsw$11$ZA$A&AA"&zCDCFCF **4*8!M !**5*9s ?A)rKrrLrrMrreturnr)pydanticrr0 functoolswraps)r9rRrSrs`` @r$__call__z_ValidateWrapper.__call__js, <<5 __Y ' "( "H! Q\\Y &__Y ' "( "H! #<<9, ,, __Y ' "( "! r#c |jjd|j|j|dd|jj S)Nz*Pydantic {schema_name} {validation_method}infoz pydantic.)r/r.rK_level _span_name)r1rNr/r. __enter__)r9rKs r$rEz_ValidateWrapper._on_entersS}}!! 8(("44!"4#9#9":; "  )+ r#cT|j|dd||jdddy)NT succeeded)rCstatusrO)_set_span_attributes__exit__)r9rNrOs r$rGz_ValidateWrapper._on_successs3 !!  " dD$'r#c |jjdd|j|j|j ddy)Nwarnz"Validation on {schema_name} failedF include_url)r/ error_counterrors)level msg_template attributes)r1logr/rmrn)r9rPs r$rVz_ValidateWrapper._on_error_logsG =#//$002,,5,9  r#c |j|dd|j|jd|jd|j dddy)NFfailedrk)rCrfrmrnrj)rgrmrn set_levelrh)r9rNrPs r$rHz_ValidateWrapper._on_error_spansV !! ))+<r^rErGrVrHrWrJrgrFr"r#r$r-r-Es<I;Z;;# ; : ; ) ;5; ;6L!\(   (  K)  r#r-c|ddvr|djS|ddvrt|dS|ddk(rE|d}|ddk(r-|d}|dDcgc] }|d |k(r|c}\}t|St|S|dScc}w) aFind the best name to use for a schema. The follow rules are used: * If the schema represents a model or dataclass, use the name of the class. * If the root schema is a wrap/before/after validator, look at its `schema` property. * Otherwise use the schema's `type` property. Args: schema: The schema to get the name for. Returns: The name of the schema. ry>modelrcls> function-wrapfunction-afterfunction-beforer: definitionszdefinition-ref schema_refref)rr3)r: inner_schemar definitionschema_definitions r$r3r3sf~//e}%%% Q Qvh/00 = (h'  #3 3*<8J#)"7#"7Je$ 2"7#   ##45 5"<0 0f~#s A>ceZdZdZeej edksejjddk(r ddZ y d dZ y) LogfirePydanticPluginaImplements a new API for pydantic plugins. Patches Pydantic to accept this new API shape. Set the `LOGFIRE_PYDANTIC_RECORD` environment variable to `"off"` to disable the plugin, or `PYDANTIC_DISABLE_PLUGINS` to `true` to disable all Pydantic plugins. z2.5.0LOGFIRE_PYDANTIC_RECORDoffcy)zBackwards compatibility for Pydantic < 2.5.0. This method is called every time a new `SchemaValidator` is created, and is a NO-OP for Pydantic < 2.5.0. NNNr")r9___s r$new_schema_validatorz*LogfirePydanticPlugin.new_schema_validator-s$r#c t|jd}|r d|vr|d}ntj}|dk(ryt |r9t t d|||||t d|||||t d|||||fSy)a.This method is called every time a new `SchemaValidator` is created. Args: schema: The schema to validate against. schema_type: The original type which the schema was created from, e.g. the model class. schema_type_path: Path defining where `schema_type` was defined, or where `TypeAdapter` was called. schema_kind: The kind of schema to validate against. config: The config to use for validation. plugin_settings: The plugin settings. Returns: A tuple of decorator factories for each of the three validation methods - `validate_python`, `validate_json`, `validate_strings` or a tuple of three `None` if recording is `off`. rr+rrvalidate_python validate_jsonvalidate_strings)_patch_PluggableSchemaValidatorr5get_pydantic_plugin_configr+_include_model_patch_build_wrapperr-) r9r: schema_typer= schema_kindconfigplugin_settingslogfire_settingsr+s r$rz*LogfirePydanticPlugin.new_schema_validator7s2 , -.229= H0@$@)(335<<'./$&$%6Yikqr$_ffoWgiop$%7Zjlrs $r#N)rrrrrZ/tuple[_ValidateWrapper, ...] | tuple[None, ...])r:rrrr=rrrrrrzdict[str, Any]rZr) rrrr rr[ __version__osenvironr5rr"r#r$rr!s8''(;w+??2::>>RkClpuCu $ $!$ $ < $, $, $, $- , $ $ , $ & , $,, $=, $r#r)fastapilogfire_backendfastuiztuple[str, ...]IGNORED_MODULESc#&K|] }|d yw).Nr").0modules r$ rjs0\O&F81OsIGNORED_MODULE_PREFIXESPydanticPlugin | None_pydantic_plugin_config_valuecNttStjjS)zGet the Pydantic plugin config.)rr param_managerpydantic_pluginr"r#r$rros $0,,**:::r#c|ay)zSet the pydantic plugin config.N)r) plugin_configs r$set_pydantic_plugin_configrws %2!r#ct}|j}|j}jj t st vry|rtfd|Dry|rtfd|DSy)z-Check whether a model should be instrumented.Fc3nK|],}tj|ddj.yw$z::Nresearchnamerpatternrr=s r$rz!_include_model..s9melZaryyG9A6("=M=R=R.s9iahV]299y]vhb9I9N9N8O,PQahrT)rincludeexcluder startswithrrany)r=rrrrs` @r$rr}ss ' )FnnGnnG $ $F 01V5N3melmmiahiii r#c&ddlm}t|_y)a%The old pydantic plugin API required managing state between event handler methods. This was messy, especially in a way that handles concurrency and nested validation: see https://pydantic.slack.com/archives/C05AF4A4WRM/p1710503400257589. The 'new API' simply requires decorating the validation methods, returning a new wrapper function. At the time of writing, this API doesn't actually exist yet in pydantic. We patch it here so that this will also work for older versions of pydantic without needing to support both APIs. r_schema_validatorN)pydantic.pluginr_build_wrapper build_wrapperrs r$rrs2&4#r#cddlm}tj|jj dj k(r dd}||_yy)zPatch a 'bug' in PluggableSchemaValidator. Getting an attribute before proper initializing (e.g. when using cloudpickle) leads to infinite recursion trying to get _schema_validator. r)PluggableSchemaValidatorzg def __getattr__(self, name: str) -> Any: return getattr(self._schema_validator, name) cN|dk(r t|t|j|S)Nr)AttributeErrorgetattrr)r9rs r$ __getattr__z4_patch_PluggableSchemaValidator..__getattr__s+**$T**41148 8r#N)r9rrr7rZr)!pydantic.plugin._schema_validatorrinspect getsourcerstrip)rrs r$rrsPK 2>>?EEG     90; ,# r#PRc^|D]'}t||}|r|}t|s ||})|SN)_wrap_with_old_handlercallable)funcevent_handlershandler old_wrappeds r$rrs:!-T7; D g 4=D" Kr#cyrr")rrs r$_nooprs r#cddlmt|dt|dt|dt|dcxurcxur cxurturyt j dfd }|S) Nrron_enter on_successon_error on_exceptionc|i| |i|} ||S#$r}|d}~wt$r} |d}~wwxYwr)rI) rLrMrOrPrQrrrrrrs r$wrapperz'_wrap_with_old_handler..wrappersc$!&! 4*6*F v M  UO    #  sA + A AA)rLzP.argsrMzP.kwargsrZr)r[r_get_handler_methodrr\r])rrrrrrrrs` @@@@@r$rrs("7J7H$Wl;J"7J7H&w?L:BB\BUBC__T    Nr#cZt||d}|tS|jdk(rtS|S)Nzpydantic.plugin)rrr)r method_names r$rrs3g{D1G   0 0 r#)r:rrZr7)rZr)rrrZr)r=rrZr)rCallable[P, R]rz list[Any]rZr)rrrr)rrrrrZzCallable[P, R] | None)rrrr7rZzCallable[..., None])=r __future__rr\rrr dataclassesrrtypingrrrr r r r[typing_extensionsr rr _internal.configrr_internal.utilsrrrrr pydantic_corerr_meter_provider get_meterMETERcreate_counterrrrr-r3rpluginrr!tuplerrrrrrrrrrrrrr"r#r$rsN4" !LL'<)(:4%%//0BC))*@A0Ye0(iu$w w tD A$A$ A$H  $JI+00\O0\+\\7;4;;2 *  5  5 ; ;:cN CL <r#