hHCddlZddlmZmZmZmZmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZmZe e gdfZedZedeed eefd Zed d d ed e eegeeffdZ d-d d deeed ed eeee eegeefffdZeedddddfdee eefdee efdee ge fefdededeededee de d e fdZdefdZ de!de de d dfdZ"de!de d ee d!fd dfd"Z#de!de d ee d!fd dfd#Z$de!de d ee d!fd dfd$Z%de!de d ee d!fd dfd%Z&de!de d ee d!fd dfd&Z'de!de d ee d!fd dfd'Z(de!de de)d dfd(Z*d)ed efd*Z+de d efd+Z,ee#ee$e-e%e.e&e/e'e0e(iZ1gd,Z2dS).N) _MISSING_TYPEMISSINGFieldfieldfields)wraps) AnyCallableDictListLiteralOptionalTupleTypeTypeVarUnionget_args get_originoverload)#StrictDataclassClassValidationErrorStrictDataclassDefinitionError#StrictDataclassFieldValidationErrorTclsreturncdSN)rs c/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/huggingface_hub/dataclasses.pystrictr! s%(SF accept_kwargsr$cdSrrr#s r r!r!$sLOCr"cvdttdttffd }| ||n|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 ``` rrc ztdstdjdi}tD]}g}|t ||jd}|\t|ts|g}|D],}t|std|j d|d-| ||||j <|_ jdtd t d td dffd }|_rdjt%d td dffd }|_jt%d t ffd }|_g} t)D]} | dst-| } t/| s8t1t3j| jdkrtdjd| d| | | _dt:d dffd } d| _tdr.t-jddstdjd| _jt%dfd } tAd| 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.selfnamevaluerc|j|gD]9} ||#ttf$r}t |||d}~wwxYw|||dS)z1Custom __setattr__ method for strict dataclasses.)rcauseN)__validators__get ValueError TypeErrorr)r+r,r-r)eoriginal_setattrs r __strict_setattr__z0strict..wrap..__strict_setattr__}s"044T2>> Z Z ZIe$$$$"I.ZZZ=DPQRRRXYYZ  T4 / / / / /s ,AAAkwargscdtDfd|D}|fi||D]\}}|vr||| dS)Nch|] }|j Sr)r,).0fs r z9strict..wrap..__init__..s#@#@#@qAF#@#@#@r"c$i|] \}}|v || Srr)r:kvdataclass_fieldss r z:strict..wrap..__init__..s*"\"\"\DAqaK[F[F[1aF[F[F[r")ritems __setattr__)r+r7standard_kwargsr,r-r@r original_inits @r __init__z&strict..wrap..__init__s$A#@F3KK#@#@#@ "\"\"\"\FLLNN"\"\"\ d66o666$*<<>>66KD%#333((u55566r"c|}fd|jD}d|}|r|ddd|dn|S)Nc8g|]\}}|jvd|d|S)*=)r()r:r>r?rs r z:strict..wrap..__repr__..sC%%%1 888"MMAMM888r"z, ))__dict__rBjoin)r+ standard_repradditional_kwargsadditional_reprr original_reprs r __repr__z&strict..wrap..__repr__s!. d 3 3 %%%%!% 3 3 5 5%%%! #')),=">">GXj-,BBBBBB]jjr" 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]>} ||#ttf$r}t|j||d}~wwxYwdS)z%Run class validators on the instance.)r)r/N)__class_validators__r2r3r__name__)r+r)r4rs r validatez&strict..wrap..validatesz 5 l l lIdOOOO"I.lll= HZbcdddjkkl l ls AAATrY"__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.cJ|g|Ri||dS)z*Run class validators after initialization.N)rY)r+argsr7r initial_inits r init_with_validatez0strict..wrap..init_with_validates= L / / / / / / / LL     r"rF)rN)!hasattrrrXrappend_create_type_validatormetadatar1 isinstancelist _is_validatorr,extendr0rCr strrFrrTdir startswithgetattrcallableleninspect signature parametersrWrrZrYsetattr)rfield_validatorsr; validatorscustom_validatorr)r6rFrTclass_validatorsr,methodrYr^r]rErSr5r$s` @@@@r wrapzstrict..wrapbss233 0V#,VVV  :< 2 2AJ   4Q77 8 8 8 z~~k:: +!"2D99:(8'9$!1I(33<AAFAAyAAA!!"2333'1 QV $ $-? 0S 0 0C 0D 0 0 0 0 0 0- ( $LM = ! ! 6 6 6 6 6 6 6 6" ! 6$CL LM = ! ! k# k k k k k k" ! k $CLHH , ,D??;// S$''FF## 7$V,,788A==4>cl>>T>>>  # #F + + + +#3  l1 l l l l l l l7;3 3 # # 3<)MuUU 4NclNNN   | |            Z!3444 r")rr)rr$rvs ` r r!r!(sYtH$q'Hd1gHHHHHHV44999T1r"Tr)defaultdefault_factoryinitreprhashcomparerbr7c lt|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 Nr)rwrxryrzr{r|rbr)rcrdr) r)rwrxryrzr{r|rbr7s r validated_fieldrsm4 i & & K %H[  '         r"c6ttdddddfdtttfdttgtftfdt 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. TNrwrxryrzr{r|rbr7c .tf|||||||d|S)Nr~)r) rwrxryrzr{r|rbr7r)s r _innerz"as_validated_field.._inner$sC   +      r")rrr rr boolrr )r)rs` r as_validated_fieldrs.5CJ##'  sM)* xC0-?@    tn    4.       , Mr"r,r- expected_typec$t|}t|}|turdSt|x}r||||dSt |t rt|||dStd|d|)z.Validate that 'value' matches 'expected_type'.NzUnsupported type for field 'r*) rrr _BASIC_TYPE_VALIDATORSr1rctype_validate_simple_typer3)r,r-roriginr\r)s r type_validatorr=s  & &F M " "D,0088 8Q $t$$$$$ M4 ( (QdE=99999OtOO OOPPPr"r\.c  g}|D]O} t|||dS#t$r,}|t|Yd}~Hd}~wwxYwtd|dt |d|dd|)z8Validate that value matches one of the types in a Union.NField 'z ' with value z doesn't match any type in z . Errors: z; )rr3r`rgrzrO)r,r-r\errorstr4s r _validate_unionrLs F "" " 4 * * * FF " " " MM#a&& ! ! ! ! ! ! ! ! " p$ppT%[[ppTpp]a]f]fgm]n]npp  s A"A  Ac>||vrtd|d|d|dS)zValidate Literal type.rz' expected one of , got N)r3)r,r-r\s r _validate_literalr[s< DM$MM$MMeMMNNNr"c :t|ts'td|dt|j|d}t |D]F\}} t |d|d||#t$r}td|d|d|d }~wwxYwd S) zValidate List[T] type.rz' expected a list, got r[]Invalid item at index z in list ''N)rcrdr3rrX enumeraterr,r-r\ item_typeiitemr4s r _validate_listras eT " "WU$UUtE{{?SUUVVVQIU##RR4 R d>>Q>>>4 ; ; ; ; R R RIQII$IIIJJPQ Q RRRsA22 B<BBc`t|ts'td|dt|j|\}}|D]W\}} t |d||t |d|d||2#t$r}td|d|d}~wwxYwdS) zValidate Dict[K, V] type.rz' expected a dict, got z.keyrrzInvalid key or value in dict 'rN)rcdictr3rrXrBr)r,r-r\key_type value_typer>r?r4s r _validate_dictros eT " "WU$UUtE{{?SUUVVV Hj MM1 M d===!X 6 6 6 d++Q+++Q ; ; ; ; M M MDTDDDEE1 L M MMs+B B+B&&B+c t|ts'td|dt|jt |dkrm|dt ur^t|D]L\}} t|d|d||d$#t$r}td|d |d |d }~wwxYwd St |t |kr2td|d t |d t |tt||D]I\}\}} t|d|d||!#t$r}td|d |d |d }~wwxYwd S)zValidate Tuple type.rz' expected a tuple, got rrrrrz in tuple 'rNz' expected a tuple of length r) rctupler3rrXrlEllipsisrrzip)r,r-r\rrr4expecteds r _validate_tupler~s eU # #XV$VVU @TVVWWW 4yyA~~$q'X-- '' W WGAt W$~~~~~tT!W==== W W W N N Nt N N NOOUVV W W W Tc%jj b$bbSYYbbVYZ_V`V`bbccc#,S-=-=#>#> W W Ah W$~~~~~tX>>>> W W W N N Nt N N NOOUVV W W Ws04B B8B33B85E  E3E..E3c.t|ts'td|dt|j|d}t |D]@\}} t |d||#t$r}td|d|d}~wwxYwdS)zValidate Set[T] type.rz' expected a set, got rz itemzInvalid item in set 'rN)rcsetr3rrXrrrs r _validate_setrs eS ! !VT$TTd5kk>RTTUUUQIU##DD4 D d>>>4 ; ; ; ; D D D;D;;;<tj|jdSr)rr,r)r-rs r r)z)_create_type_validator..validatorsuz5%*55555r")r )rr)s` r raras766666666 r"ct|sdStj|}t|j}t |dkrdS|djtjj tjj tjj fvrdS|ddD]}|j tjj krdS dS)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)rkrmrnrdrovaluesrlkind ParameterPOSITIONAL_OR_KEYWORDPOSITIONAL_ONLYVAR_POSITIONALrwempty)r)rnro parameters r reres I  u!),,Ii*113344J :!u!}/)(" u^   1 7 7 755 8 4r")r!r Validator_Trrrr)3rm dataclassesrrrrr functoolsrtypingr r r r r rrrrrrrrrrrrrrr!rrrrgrrrrrrrrrrarerdrrrr__all__rr"r rsVDDDDDDDDDDDDDD ud{#  GCLL (Q(DG((( ( $)OOOTOhQy$q'7I.JOOO O"E2