g.ddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZmZddlmZddlZddlmZmZmZmZdd lmZmZe d Ze d ee d f Z!e de e"e f Z#e de e  Z$e de d ef Z%d:dZ& d;dZ' ddZ*d?dZ+d@dZ,dAdZ-dBdZ.dCdZ/dDdZ0dEdZ1dFdZ2dGdZ3dd d!dHd"Z4dId#Z5dJd$Z6e d%Z7e d&Z8edKd'Z9edLd(Z9edMd)Z9dNd*Z9dOd+Z:dPd,Z;dQd-ZdTd0Z?dUd1Z@dVd2ZAdWd3ZBdXd4ZCdYd5ZDd6d7dZd8ZEd[d9ZFy)\) annotationsN) AnyTupleMappingTypeVarCallableIterableSequencecastoverload)Path)datedatetime) TypeGuard)NotGiven FileTypes NotGivenOr HeadersLike) parse_dateparse_datetime_T_TupleT.)bound _MappingT _SequenceT CallableTc@|Dcgc] }|D]}| c}}Scc}}wN)tsublistitems G/var/www/openai/venv/lib/python3.12/site-packages/groq/_utils/_utils.pyflattenr%"s" ! 6WgdDgD 66 6sc Tg}|D] }|jt||dd"|S)zRecursively extract files from the given dictionary based on specified paths. A path may look like this ['foo', 'files', '', 'data']. Note: this mutates the given dictionary. rNindex flattened_key)extend_extract_items)querypathsfilespaths r$ extract_filesr0&s/*,E ^E4qMN Lc  ||}|dz }t|rC t||k(r|j|}n||}||}n |d|dz }t||||St|r4|dk7rgSt|Dcgc]}t|||||d znd c}SgS#t$r>t|trgcYSddlm}||||J|t t |fgcYSwxYw#t$rgcYSwxYwcc}w) Nr)assert_is_file_content)key[]r'zz[]) IndexError isinstancer_filesr3r ris_dictlenpopKeyErrorr+is_listr%)objr/r(r)r4r3r#s r$r+r+9sX 75k QJEs| D e#wws|3x  M qQZ 'M  '    ) I   D :G:S-$"6Y]       Ik  7 c8 $I 4 s 6(((Y 4566 7* I  ( s.B%C%5C6C"8'C"!C"% C32C3c$t|t Sr)r9rr@s r$is_givenrCzs#x( ((r1c"t|tSrr9tuplerBs r$is_tuplerG c5 !!r1c"t|tSrrErBs r$ is_tuple_trJrHr1c"t|tSrr9r rBs r$ is_sequencerM c8 $$r1c"t|tSrrLrBs r$ is_sequence_trPrNr1c"t|tSrr9rrBs r$ is_mappingrS c7 ##r1c"t|tSrrRrBs r$ is_mapping_trVrTr1c"t|tSr)r9dictrBs r$r;r; c4  r1c"t|tSr)r9listrBs r$r?r?rYr1c"t|tSr)r9r rBs r$ is_iterabler]rNr1c  t|r;tt|jDcic]\}}|t |c}}St |r'tt|Dcgc] }t |c}S|Scc}}wcc}w)zMinimal reimplementation of copy.deepcopy() that will only copy certain object types: - mappings, e.g. `dict` - list This is done for performance reasons. )rSr ritemsdeepcopy_minimalr?)r#kventrys r$r`r`st$BDJJLILDAq,Q//LIJJt}BdCdU)%0dCDD KJCs A; B z, or)delimfinalct|}|dk(ry|dk(r|dS|dk(r|dd|d|dS|j|ddd|d|dzS)Nrr5r )r<join)seqrerfsizes r$ human_joinrnsv s8D qy qy1v  qya&5'3q6(++ ::c#2h AeWAc"gY"7 77r1cd|dS)zOAdd single quotation marks around the given string. Does *not* do any escaping.'r )strings r$quoterrs vha=r1cdfd }|S)a4Decorator to enforce a given set of arguments or variants of arguments are passed to the decorated function. Useful for enforcing runtime validation of overloaded functions. Example usage: ```py @overload def foo(*, a: str) -> str: ... @overload def foo(*, b: bool) -> str: ... # This enforces the same constraints that a static type checker would # i.e. that either a or b must be passed to the function @required_args(["a"], ["b"]) def foo(*, a: str | None = None, b: bool | None = None) -> str: ... ``` c$tjj}|jDcgc]+\}}|j|j |j hvr|-c}}tjdfd }|Scc}}w)Ncl t t|D]\}} j ||jD]} j|D]}t fd|D}|snt dkDrTtDcgc],}dt|Dcgc] }t|c}dzd z.c}}}d |d } t | t d kDsJttd  z } t | dkDr0d t| Dcgc] }t|c}} t | dt| d } t |  |i|S#t$r2t j dt dt |ddwxYwcc}wcc}}wcc}w)Nz () takes z argument(s) but z were givenc3&K|]}|v ywrr ).0param given_paramss r$ z@required_args..inner..wrapper..sJ'u 4'sr5(and)rf)z,Missing required arguments; Expected either z arguments to be givenrzMissing required arguments: zMissing required argument: ) set enumerateaddr8 TypeError__name__r<keysallrnrrr[)argskwargsi_r4variantmatchesarg variationsmsgmissingryfunc positionalvariantss @r$wrapperz-required_args..inner..wrappers%(UL!$1  $$Z]3({{}  %%$J'JK$ x=1$!+muvmubiz*I#5:*IQVWWZ]]muv"JI TjkC n$x=1,,,#3x{#3l#BCG7|a' .inners""4(33 &||~ - ezz%%++ -   ! ) ! )F] s0B )rrrrr )rrs` r$ required_argsrs,0d Lr1_K_Vcyrr rBs r$strip_not_givenrs(+r1cyrr rBs r$rr"sFIr1cyrr rBs r$rr&s,/r1c|yt|s|S|jDcic]\}}t|tr||c}}Scc}}w)zHRemove all top-level keys where their values are instances of `NotGiven`N)rSr_r9r)r@r4values r$rr*sE { c? ), X:3Juhz%get_required_header..vs##))A,STI[I[I]:]r1zCould not find z header) lowerrVr_r9strresub capitalizerget ValueError)headersheader lower_headerrarbintercaps_headernormalized_headerrs r$get_required_headerrms<<>LGMMODAqwwyL(Z3-?$ vvn.]_e_p_p_rs$lFLLNDTU -. LV vhg6 77r1cJ tjS#t$rYywxYw)Nfalse)sniffiocurrent_async_library Exceptionr r1r$get_async_libraryrs(,,.. s  ""maxsizecNtj|}tt|S)zmA version of functools.lru_cache that retains the type signature for the wrapped function arguments. r)r lru_cacher r)rrs r$rrs&!!G W r1cht|r6|jDcic]\}}t|t|c}}St|r4t |t t tfs|Dcgc] }t|c}St |ttfr|jS|Scc}}wcc}w)zvTranslates a mapping / sequence recursively in the same fashion as `pydantic` v2's `model_dump(mode="json")`. ) rSr_ json_safer]r9rbytes bytearrayrr isoformat)datar4rr#s r$rrs$CG::<Prs"    #'AAP T] )5#5 6 KwsF{'; < \&)9 : KxS'9 : 7  #  ! &> > >  >  > ! >B)""%%$$!!%  48d 8 HV T] T] + + I I / /Y6   ! 8&(+ r1