qiUbUddlZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z mZddlmZmZmZmZmZmZmZmZmZmZmZmZ ddlmZmZd d l!m"Z"m#Z#m$Z$eegdfZ%ed Z&ed e'e(ef Z)e*Z+edee&dee&fdZ,eddde-deee&gee&ffdZ, d=dddeee&de-deee&eee&gee&fffdZ,de e)de'ddfdZ.e de e)defdZ/de e)de'e(effdZ0eedddddfdee1e%e%fdeeefdeegefefde-d e-d!ee-d"e-d#ee'd$edefd%Z2de%fd&Z3d'e(d(ed)eddfd*Z4d'e(d(ed+e5ed,fddfd-Z6d'e(d(ed+e5ed,fddfd.Z7d'e(d(ed+e5ed,fddfd/Z8d'e(d(ed+e5ed,fddfd0Z9d'e(d(ed+e5ed,fddfd1Z:d'e(d(ed+e5ed,fddfd2Z;d'e(d(ed+e5ed,fddfd3Zdede-fd7Z?d8ede-fd9Z@ee6ee7e1e8e'e9e5e:eAe;ejjeN) _MISSING_TYPEMISSINGFieldfieldfieldsmake_dataclass) lru_cachewraps) AnnotatedAnyCallable ForwardRefLiteralOptionalTypeTypeVarUnionget_args get_originoverload) NotRequiredRequiredrr)#StrictDataclassClassValidationErrorStrictDataclassDefinitionError#StrictDataclassFieldValidationErrorT TypedDictType)boundclsreturncyNr)r!s M/opt/pipecat/venv/lib/python3.12/site-packages/huggingface_hub/dataclasses.pystrictr&2s%(F accept_kwargsr)cyr$rr(s r%r&r&6sLOr'c\dttdttffd }|||S|S)a Decorator to add strict validation to a dataclass. This decorator must be used on top of `@dataclass` to ensure IDEs and static typing tools recognize the class as a dataclass. Can be used with or without arguments: - `@strict` - `@strict(accept_kwargs=True)` Args: cls: The class to convert to a strict dataclass. accept_kwargs (`bool`, *optional*): If True, allows arbitrary keyword arguments in `__init__`. Defaults to False. Returns: The enhanced dataclass with strict validation on field assignment. Example: ```py >>> from dataclasses import dataclass >>> from huggingface_hub.dataclasses import as_validated_field, strict, validated_field >>> @as_validated_field >>> def positive_int(value: int): ... if not value >= 0: ... raise ValueError(f"Value must be positive, got {value}") >>> @strict(accept_kwargs=True) ... @dataclass ... class User: ... name: str ... age: int = positive_int(default=10) # Initialize >>> User(name="John") User(name='John', age=10) # Extra kwargs are accepted >>> User(name="John", age=30, lastname="Doe") User(name='John', age=30, *lastname='Doe') # Invalid type => raises >>> User(name="John", age="30") huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': TypeError: Field 'age' expected int, got str (value: '30') # Invalid value => raises >>> User(name="John", age=-1) huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': ValueError: Value must be positive, got -1 ``` r!r"c tdstdjdi}tD]}g}|j t ||j jd}|Rt|ts|g}|D])}t|rtd|jd|d|j||||j<|_ jdtd t d td dffd }|_rj"}t%|d td dffd }|_tdsd td dfd} | _j(t%d t ffd } j*j,dur| _g} t/D]} | j1dst3| } t5| s-t7t9j:| j<dk7rtdjd| d| j | | _dt@d dffd }d|_!tdr0t3jDddstdjd|_"j"t%dfd }tGd|S)N__dataclass_fields__zClass 'z.' must be a dataclass before applying @strict. validatorzInvalid validator for field '': z.. Must be a callable taking a single argument.selfnamevaluer"c|jj|gD] } || |||y#ttf$r}t |||d}~wwxYw)z1Custom __setattr__ method for strict dataclasses.)rcauseN)__validators__get ValueError TypeErrorr)r0r1r2r.eoriginal_setattrs r%__strict_setattr__z0strict..wrap..__strict_setattr__so"0044T2> Z Ze$ Z T4 / #I.Z=DPQRXYYZs7AAAkwargsct Dchc]}|j}}|jDcic] \}}||vs ||}}}t|dkDr%t d j dt|dt D]}|j|vr%t ||j||j6|jtur"t ||j|jj|jtur&t ||j|jtd|jdi}|jD]\} } | |vs | || <|jdi|ycc}wcc}}w)NrzWhen `accept_kwargs=True`, z, accepts only keyword arguments, but found `z` positional args.zMissing required field - ''r) rr1itemslenr7__name__setattrdefaultrdefault_factoryr8 __post_init__) r0argsr<fdataclass_fieldskvstandard_kwargsadditional_kwargsr1r2r!s r%__init__z&strict..wrap..__init__sw5;3K#@qAFF#@ #@4:LLN"\DAqaK[F[1a4"\"\ t9q=$5cll^D&&)$i[0BD  PAvv0affoaff.EF'1affaii8**'9affa.?.?.AB'*DQVVHA(NOOP%'!#)<<>8KD%#3327)$/8#""7%67=$A"\sE= FFrEcN|jD]\}}t|||y)z2Default __post_init__ to accept additional kwargs.N)r?rB)r0r<r1r2s r%rEz+strict..wrap..__post_init__s''-||~3 edE23r'c|}|jjDcgc]\}}|jvrd|d|}}}dj|}|r |ddd|dS|Scc}}w)N*=z, ))__dict__r?r-join)r0 standard_reprrIrJrLadditional_reprr! original_reprs r%__repr__z&strict..wrap..__repr__s!.d 3 !% 3 3 5%1 8 88s!A5M%!% #')),=">GX-,-R/@Bj]jj%s A*T validate_rz' has a class validator 'z' that takes more than one argument. Class validators must take only 'self' as an argument. Methods starting with 'validate_' are considered to be class validators.cjD] } || y#ttf$r}t|j||d}~wwxYw)z%Run class validators on the instance.)r.r4N)__class_validators__r7r8rrA)r0r.r9r!s r%validatez&strict..wrap..validatesW 55 l ldO l#I.l= HZHZbcdjkklsA AA r]"__is_defined_by_strict_decorator__Fz' already implements a method called 'validate'. This method name is reserved when using the @strict decorator on a dataclass. If you want to keep your own method, please rename it.c@|g|i|j|y)z*Run class validators after initialization.N)r])r0rFr<r! initial_inits r%init_with_validatez0strict..wrap..init_with_validates#  / / / LL r'rM)r"N)$hasattrrrArappend_create_type_validatormetadatar6 isinstancelist _is_validatorr1extendr5 __setattr__r strrMr rErY__dataclass_params__reprdir startswithgetattrcallabler@inspect signature parametersr\rr^r]rB)r!field_validatorsrG validatorscustom_validatorr.r; original_initrMrErYclass_validatorsr1methodr]rar`rXr:r)s` @@@r%wrapzstrict..wraptss230#,,'UV  :< 2AJ   4Q7 8 zz~~k: +!"2D9(8'9$!1I(3<;AFF83ykQA !!"23'1 QVV $ 2.?? 0S 0 0C 0D 0- LLM = ! 8 8 8" 8D$CL303#3$3 %2! LLM = ! k# k" k '',,4' H ,D??;/S$'FF#7$$V,778A=4cll^+DTFK>>  # #F + ,$4  l1 l l7;3 3 #3<<)MuU4cll^,NN   || |     Z!34 r')rr)r!r)r{s ` r%r&r&:s8th$q'hd1ghV491T1r'schemadatac*t|}|di|y)a Validate that a dictionary conforms to the types defined in a TypedDict class. Under the hood, the typed dict is converted to a strict dataclass and validated using the `@strict` decorator. Args: schema (`type[TypedDictType]`): The TypedDict class defining the expected structure and types. data (`dict`): The dictionary to validate. Raises: `StrictDataclassFieldValidationError`: If any field in the dictionary does not conform to the expected type. Example: ```py >>> from typing import Annotated, TypedDict >>> from huggingface_hub.dataclasses import validate_typed_dict >>> def positive_int(value: int): ... if not value >= 0: ... raise ValueError(f"Value must be positive, got {value}") >>> class User(TypedDict): ... name: str ... age: Annotated[int, positive_int] >>> # Valid data >>> validate_typed_dict(User, {"name": "John", "age": 30}) >>> # Invalid type for age >>> validate_typed_dict(User, {"name": "John", "age": "30"}) huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': TypeError: Field 'age' expected int, got str (value: '30') >>> # Invalid value for age >>> validate_typed_dict(User, {"name": "John", "age": -1}) huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': ValueError: Value must be positive, got -1 ``` Nr)!_build_strict_cls_from_typed_dict)r|r} strict_clss r%validate_typed_dictr"sX36:Jr'c t|}t|dds|jD]t\}}t|}|turDt |^}}t |s t|}tt|gt|z||<]t |rit|||<vg}|jD]p\}}t|tur7t |^}}|j||ttd|difN|j||ttfrtt|j|S)N __total__Tr.r)rCre)rC)_get_typed_dict_annotationsrpr?rr r_is_required_or_notrequiredrtuplergrcr_TYPED_DICT_DEFAULT_VALUEr&rrA)r| type_hintskeyr2originbasemetars r%rrTs9-V4J 6; -$**, 5JC&F"&uo t248&t,D"+E4&4:2E,F"G 307"-e"4 3 5F &&(R U e  )"5/KD4 FMM3e4MYdfjklfmXn&op q FMM3u5N'OP Q R .&9 ::r'c  ddl}|j|S#t$rU|jj dij Dcic]\}}|||n t dncc}}wc}}cYSwxYw)z0Extract type annotations from a TypedDict class.rN__annotations__) annotationlibget_annotations ImportErrorrTr6r?type)r|rr1r2s r%rrss}  ,,V44  &223DbIOOQ  e 5,%$t* <    s6A5 A(' A54A5Tr.rCrDinitrmhashcomparerer<c `t|ts|g}|i}||d<td|||||||d|S)a< Create a dataclass field with a custom validator. Useful to apply several checks to a field. If only applying one rule, check out the [`as_validated_field`] decorator. Args: validator (`Callable` or `list[Callable]`): A method that takes a value as input and raises ValueError/TypeError if the value is invalid. Can be a list of validators to apply multiple checks. **kwargs: Additional arguments to pass to `dataclasses.field()`. Returns: A field with the validator attached in metadata r.rCrDrrmrrrer)rfrgr) r.rCrDrrmrrrer<s r%validated_fieldrs[4 i &K %H[  '       r'cttdddddfdtttfdttgtftfdt dt dt t dt d t td tffd }|S) a Decorates a validator function as a [`validated_field`] (i.e. a dataclass field with a custom validator). Args: validator (`Callable`): A method that takes a value as input and raises ValueError/TypeError if the value is invalid. TNrCrDrrmrrrer<c ,tf|||||||d|S)Nr)r) rCrDrrmrrrer<r.s r%_innerz"as_validated_field.._inners9   +    r')rrr rr boolrdict)r.rs` r%as_validated_fieldrs.5CJ##' sM)* xC0-?@    tn    4.  , Mr'r1r2 expected_typect|}t|}|turytj |x}r ||||yt |t rt|||yt |tst |try|tur(|turtd|dt|||dy|tur|turyt|||dytd|d|)z.Validate that 'value' matches 'expected_type'.NField 'z' is required but missing.rzUnsupported type for field 'r/)rrr _BASIC_TYPE_VALIDATORSr6rfr_validate_simple_typerrkrrr8type_validatorr)r1r2rrrFr.s r%rrs  &F M "D,008 8 8$t$ M4 (dE=9 M: .*]C2P 8  - -gdV+EFG GtUDG, ;  - - tUDG,6tfC OPPr'rF.c g}|D]} t|||ytd|dt |d|ddj |#t$r$}|jt|Yd}~kd}~wwxYw)z8Validate that value matches one of the types in a Union.Nrz ' with value z doesn't match any type in z . Errors: z; )rr8rcrkrmrU)r1r2rFerrorstr9s r%_validate_unionrs F " " 4 * "  $}T%[M1LTFR\]a]f]fgm]n\op  " MM#a& ! ! "s A A5A00A5c4||vrtd|d|d|y)zValidate Literal type.rz' expected one of , got N)r8)r1r2rFs r%_validate_literalrs. D'$'9$veWMNNr'c  t|ts$td|dt|j|d}t |D]\}} t |d|d||y #t$r}td|d|d|d }~wwxYw) zValidate list[T] type.rz' expected a list, got r[]Invalid item at index z in list 'r>N)rfrgr8rrA enumeraterr1r2rF item_typeiitemr9s r%_validate_listrs eT "'$'>tE{?S?S>TUVVQIU#R4 R dV1QCq>4 ;R R4QCz$qIJPQ Q Rs A"" B+A>>Bc0t|ts$td|dt|j|\}}|j D])\}} t |d||t |d|d||+y#t$r}td|d|d}~wwxYw) zValidate dict[K, V] type.rz' expected a dict, got z.keyrrzInvalid key or value in dict 'r>N)rfrr8rrAr?r)r1r2rFkey_type value_typerIrJr9s r%_validate_dictr s eT "'$'>tE{?S?S>TUVV Hj M1 M dV4=!X 6 dV1QE+Q ;M MNz' expected a tuple of length r) rfrr8rrAr@Ellipsisrrzip)r1r2rFrrr9expecteds r%_validate_tuplersi eU #'$'?U @T@T?UVWW 4yA~$q'X- ' WGAt W$q1~tT!W= W Tc%j '$'DSYKvVYZ_V`Uabcc#,S-=#> W Ah W$q1~tX> W  W"8;tfA NOUVV W W"8;tfA NOUVV Ws0 C+D+ D 4DD  D0D++D0ct|ts$td|dt|j|d}t |D]\}} t |d||y#t$r}td|d|d}~wwxYw)zValidate set[T] type.rz' expected a set, got rz itemzInvalid item in set 'r>N)rfsetr8rrArrrs r% _validate_setr2s eS !'$'=d5k>R>R=STUUQIU#D4 D dV5>4 ;D D3D6;) rf collectionsabcSequencer8rrArrrs r%_validate_sequencer@s e[__55 6'$'B4;CWCWBXYZZ QIU#V4 V dV1QCq>4 ;V V4QC~dV1MNTU U Vs#A99 BBBc t||s>td|d|jdt|jdt |d y)z&Validate simple type (int, str, etc.).rz ' expected rz (value: rSN)rfr8rArrm)r1r2rs r%rrRsX e] +dV;}'='=&>fT%[EYEYDZZcdhindocppq r   ,r'rc$dtddffd }|S)z-Create a type validator function for a field.r2r"NcHtj|jyr$)rr1r)r2rs r%r.z)_create_type_validator..validator^suzz5%**5r')r )rr.s` r%rdrdZs666 r'ct|sytj|}t|jj }t |dk(ry|djtjjtjjtjjfvry|ddD]+}|jtjjk(s+yy)zCheck if a function is a validator. A validator is a Callable that can be called with a single positional argument. The validator can have more arguments with default values. Basically, returns True if `validator(value)` is possible. FrrNT)rqrrrsrgrtvaluesr@kind ParameterPOSITIONAL_OR_KEYWORDPOSITIONAL_ONLYVAR_POSITIONALrCempty)r.rsrt parameters r%rhrhds I !!),Ii**1134J :!!}//))((" ^    1 1 7 7 7 r' type_hintcP|ttfvxst|ttfvS)z2Helper to check if a type is Required/NotRequired.)rrr)rs r%rrs' ;/ / eJy4IhXcMd4der'r) )r&rr Validator_Trrrr$)Icollections.abcrrrsystypes dataclassesrrrrrr functoolsr r typingr r r rrrrrrrrrrrrtyping_extensionsrrrrrrrrrkrobjectrr&rrrrrgrrrrrrrrrrrrrdrhrrrrrr version_info UnionType__all__rr'r%rs TT&     2,ud{#  CLtCH~> "H (Q(DG( ( $)OTOhQy$q'7I.JO O"e2