"j2 SrSSKJr SSKrSSKrSSKrSSKJrJrJ r SSKJ r J r J r J r Jr SSKJrJrJrJrJrJrJrJr SSKJr SSKJrJr SS KJrJr SS K J!r! \ (aSS KJ"r" \ "S 5r#SS jr$"SS\5r%"SS\\&\'\&\ 4-5r("SS\%5r)SS.SSjjr*SSS.SSjjr+SSjr,SS.SSjjr-g) aShared spec utilities for constructing objects from YAML/JSON/dict specifications. This module provides the `NamedSpec` class (generalized from `EvaluatorSpec` in pydantic_evals) and registry/loading utilities that can be reused by both the evaluator system and the capability system. ) annotationsN)CallableMappingSequence) TYPE_CHECKINGAnyLiteralTypeVarcast) BaseModel ConfigDict RootModelValidationErrorfield_validatormodel_serializermodel_validator with_config)to_jsonable_python)SerializationInfoSerializerFunctionWrapHandler) NotRequired TypedDict)get_function_type_hints)ModelWrapValidatorHandlerTch[USS9n[U[5=(a [SU55$)aCheck if a value would serialize to a dict with all string keys. When serialize() uses the compact tuple form (arguments = (value,)), the serialized output becomes {Name: value}. On deserialization, _SerializedNamedSpec._args treats any dict with all-string keys as kwargs. This means a single positional argument that is itself a dict (like ModelSettings) would be incorrectly unpacked as kwargs on the round-trip. We avoid the compact form in this case. T)serialize_unknownc3B# UHn[U[5v M g7fN isinstancestr.0ks N/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_ai/_spec.py 2serializes_as_string_keyed_dict../s-S(QjC.@.@()rr!dictall)valuejsonables r&serializes_as_string_keyed_dictr.%s-"%4@H h % S#-S(-S*SSc\rSrSr%SrS\S'S\S'\SSj5r\SSj5r\ "S S 9\ SS j55r \ "S S 9SS j5r S rg) NamedSpec2uA specification for constructing a named object from serialized arguments. Supports three short forms: * `'MyClass'` — no arguments * `{'MyClass': single_arg}` — a single positional argument * `{'MyClass': {k1: v1, k2: v2}}` — keyword arguments r"name"None | tuple[Any] | dict[str, Any] argumentscZ[UR[5(a UR$g)zGet the positional arguments.)r!r5tupleselfs r&argsNamedSpec.argsDs" dnne , ,>> !r/c\[UR[5(a UR$0$)zGet the keyword arguments.)r!r5r*r9s r&kwargsNamedSpec.kwargsKs$ dnnd + +>> ! r/wrap)modecU"U5$![aBn[RU5nO![a Uef=fURU5sSnA$SnAff=f)z-Deserialize a NamedSpec from various formats.N)r_SerializedNamedSpecmodel_validate to_named_spec)clsr,handlerexc deserializeds r& deserializeNamedSpec.deserializeRs[ 35> ! 3 3BB5I "   --c2 2  3s' A+A 8A AAc[UR[5(aURRS5(aURc UR $[UR[ 5(a@[URS5(aU"U5$UR URS0$UR UR0$U"U5$)z7Serialize using the appropriate short-form if possible.use_short_formr)r!contextr*getr5r3r8r.)r:rGinfos r& serializeNamedSpec.serialize_s dllD ) )dll.>.>?O.P.P~~%yy DNNE2234>>!3DEE"4=( 4>>!#455 4>>224= r/r7N)returnztuple[Any, ...])rSdict[str, Any])r,rrGz$ModelWrapValidatorHandler[NamedSpec]rSr1)rGrrPrrSr)__name__ __module__ __qualname____firstlineno____doc____annotations__propertyr;r>r classmethodrJrrQ__static_attributes__r7r/r&r1r12s~ I-11   &! 3" 36"!#!r/r1cz\rSrSrSr\"S5\S Sj55r\S Sj5r \S Sj5r \ 4S Sjjr Sr g )rCqz?Internal class for handling the serialized form of a NamedSpec.rootc[U[5(aU$[U5S:wa%[S[ UR 5535eU$)zBEnforce that the root value has exactly one key when it is a dict.z.s4t!:a%%tr)) r!r`r"rjrkvaluesr*rerfr+r r)r:r,rfs r&_args_SerializedNamedSpec._argss dii % %T$))**,-. eT " ""5::<0D4t444DcNE22S% ""r/c8U"URURS9$)N)r3r5)rlrrr:rFs r&rE"_SerializedNamedSpec.to_named_specs djj99r/r7N)r,zstr | dict[str, Any]rSr)rSr")rSr4)rFztype[NamedSpec]rSr1)rUrVrWrXrYrr\rgr[rlrrr1rEr]r7r/r&rCrCqs[IV,,  # #4=::r/rCc\rSrSrSrSrg)CapabilitySpeczA capability specification, distinguishable from other NamedSpec types for schema generation. In JSON schemas, fields typed as CapabilitySpec are replaced with the full capability Union (the same set of types used in `AgentSpec.capabilities`). r7N)rUrVrWrXrYr]r7r/r&rxrxsr/rx)validatec0nUHPnUbU"U5 U"U5nUc[SUSURS35eXu;a[SUSU<35eXeU'MR UH!nU"U5nUcMURXv5 M# U$)aCreate a registry of types from default and custom types. Args: custom_types: Additional classes to include in the registry. defaults: Default classes to include (can be overridden by custom types). get_name: Callable to get the serialization name from a class. Return None to opt out. label: Human-readable label for error messages. validate: Optional callback to validate each custom type. Returns: A mapping from names to classes. zCustom z class z. has opted out of serialization (name is None)z Duplicate z class name: )rdrU setdefault) custom_typesdefaultsget_namelabelrzregistryrFr3s r&build_registryrs($&H   SM} <wugWS\\NBpqr r  z% dXFG G}      *  Or/)rN instantiatecURnURU5nUcC[UR5SU<SUS[ UR 55SUSUS3 5eUbU"XqR UR5$U"UR 0URD6$![a5nU(aSU3OS n [S USUR<U S U35UeSnAff=f) aLoad an object from the registry based on a specification. Args: registry: Mapping from names to classes. spec: Specification of the object to load. label: Human-readable label for error messages. custom_types_param: Name of the parameter for custom types, used in error messages. context: Optional context for error messages. instantiate: Optional callback to instantiate the class. Default: `cls(*args, **kwargs)`. Returns: An initialized instance. N z is not in the provided `z`. Valid choices: z$. If you are trying to use a custom z$, you must include its type in the `z ` argument.z for zFailed to instantiate z: ) r3rOrd capitalizererfr;r> Exception) rspecrcustom_types_paramrNrr3rFedetails r&load_from_registryrs, 99D ,,t C {!"!D8+DEWDXXjkopxp}p}plAkBB2278\]o\pp{ }  ]  "sIIt{{; ; 1T[[1 1 ]&-5 "21%$))vhbQRPSTU[\\]s%B!B!! C +0CC c*[U[5(ag[R"U5nU[LagU[R Ld[U[ R5(ap[R"U5nUVs/sHn[U5=ncMUPM nnU(dg[U5S:XaUS$[R [U5$[R"U5nU(a[SU55(agU$s snf)zFilter a type to only include members that can be represented in JSON schema. For Union types, removes non-serializable members (TypeVars, Callables). Returns None if the type is entirely non-serializable. Nrbrc3<# UHn[U5SLv M g7fr)filter_serializable_type)r$as r&r'+filter_serializable_type.. sFA,Q/47s) r!r typing get_originrUniontypes UnionTypeget_argsrrcr8any)tporiginr;rfafiltereds r&rrs"g   r "FB!@!@r" $X1/G/J)JBX x=A A; ||E(O,, ??2 D FFFF IYs DD)get_schema_targetc^/nUR5GHumnUbU"U5OUn[U5nURSS5 UR5VVs0sHupg[U5=ncMXh_M nnn0n [R "U5R R5GH%n U RS;aDU RU RU R4;aURU RS5 MXU RU RU R4;aURU RS5 MU RU;aMURU R[5 U R U R"La$[$XZRXZR'GM XZRXR'GM( S U4Sjjn ['U5S:XdU (dUR)[*T5 ['U5S:Xa-UR5un UR)U "STU 055 O;['U 5S:Xa,U R5un UR)U "STU 055 ['U5S:dGM~U "SU5n UR)U "S TU 055 GM U$s snnf) adBuild a list of schema types from a registry for JSON schema generation. Args: registry: Mapping from names to classes. get_schema_target: Optional callback to get the schema target (e.g. `from_spec` method) from a class. Default: use the class itself. Returns: A list of types suitable for use in a Union for JSON schema generation. NrSrucX>[UST3U5n[[SSS95"U5$)N_forbidT)extraarbitrary_types_allowed)rrr )cls_name_prefixfieldstdr3s r&_make_typed_dict,build_schema_types.._make_typed_dict<s4o.av6?BzRVWXY[\ \r/rrb short_spec spec_paramsr)rr"rrTrSr)itemsrpoprinspect signature parametersrqr3kindPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORDVAR_POSITIONAL VAR_KEYWORDr|rdefaultemptyrrcappendr )rr schema_typesrFtarget type_hintsr%vfvrequired_type_hintsprtype_hint_type params_tdr3s @r&build_schema_typesrsW!L^^% c+<+H"3'c,V4 x&)3(8(8(:n(:E]^_E`?`reae(: n.0""6*55<<>Avv(QVV8I8I1KbKb7c-cqvvt,vv!**AMM::qvvt,vvZ'  ! !!&&# .yy'%0FF1C%D 66".8.@#FF+?" ] z?a ':    . z?a )002 ^    0n?U V W $ % *299; ^    0n?U V W z?Q ( CI    0$ 9J K L]&` Uos J<0J<)r,rrSbool) r}Sequence[type[T]]r~rrzCallable[[type[T]], str | None]rr"rzz Callable[[type[T]], None] | NonerSMapping[str, type[T]])rrrr1rr"rr"rNz str | Nonerz>Callable[[type[T], tuple[Any, ...], dict[str, Any]], T] | NonerSr)rrrSz Any | None)rzMapping[str, type[Any]]rz!Callable[[type[Any]], Any] | NonerSz list[Any]).rY __future__rrrrcollections.abcrrrrrr r r pydanticr r rrrrrr pydantic_corerpydantic_core.core_schemarrtyping_extensionsrrpydantic_ai._utilsrrrr.r1r"r*rCrxrrrrr7r/r&rs] # 77==   -V462 CL T<! <!~#:9S4S>%9:#:LY26 &#& &. &  & / &&^RV$]#$] $]  $]  $]  $]P$]$]NJ<@@%@9@ @r/