Li2^tUdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z m Z mZmZmZddlZddlZddlZddlmZddlmZej.dZeee fZeed<Gdd eZd e ee e fd eed e d dfdZ ddd e d eede d e fdZ!d e deeefd dfdZ"d e deedeede#de$ed df dZ%dBdede&d efdZ'dCde(de&d e fdZ)de*d e&fd Z+d!e d"eee(fd dfd#Z,e jZd$d%&Z.dd'd(d)d*dd+de d,e#d-e#d.e#d/e#d0e#d1e e e#d efd2Z/de d,e#d-e#d.e#d/e#d0e#d1e e#d efd3Z0Gd4d%ejbZ1Gd5d6eejdZ3d efd7Z4d eee(fd eee(ffd8Z5d9ed e e d:e fge d:e fffd;Z6ded?ed efd@Z8d>ed?ed dfdAZ9y)DzCommon utilities for the SDK.N)AnyCallable FrozenSetOptionalUnionget_args get_origin)alias_generators) TypeAliaszgoogle_genai._common StringDictceZdZdZy)ExperimentalWarningz"Warning for experimental features.N)__name__ __module__ __qualname____doc__R/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/google/genai/_common.pyrr&s*rrdatakeysvaluereturnc |yt|ddD]\}}|jdr|dd}|K||vrGt|tr&t t |Dcgc]}ic}||<nt d|d|t|tr0|.t||D]\}}t|||dzd||y|||D]}t|||dzd|y|jdr.|dd }| ||vrig||<|t||d ||dzd|y||j|i}"||j|d} | S|sy|| k(ryt| tr"t|tr| j|yt d |dd | d |d|ddk(r2t|tr"t|tr|j|y|||d<yycc}w)a9Examples: set_value_by_path({}, ['a', 'b'], v) -> {'a': {'b': v}} set_value_by_path({}, ['a', 'b[]', c], [v1, v2]) -> {'a': {'b': [{'c': v1}, {'c': v2}]}} set_value_by_path({'a': {'b': [{'c': v1}, {'c': v2}]}}, ['a', 'b[]', 'd'], v3) -> {'a': {'b': [{'c': v1, 'd': v3}, {'c': v2, 'd': v3}]}} N[]zvalue z$ must be a list given an array path [0]rz+Cannot set value for an existing key. Key: z; Existing value: z ; New value: ._self) enumerateendswith isinstancelistrangelen ValueErrorset_value_by_path setdefaultgetdictupdate) rrrikeykey_name_jd existing_datas rr*r**sL ] $s)$&fa ||DSbh hd2 eT "(-c%j(9:1B:$x.ugA#G   E4 T%5d8n- 8DAq AtAEG}eAh 7 8    > 7a aa!eg 6 7 e Sbh hd2X $x.+T!a%']EB   __S" %d5&8 HHT"X&M  M !  mT *z%/FU#9$r(D -mE7! E  r(g t$% ET"X;/;s G$ default_valuer7c |dgk(r|St|D]\}}|s|cS|jdr4|dd}||vr'||Dcgc]}t|||dzd|c}cS|cS|jdr.|dd}||vr!||rt||d ||dzd|cS|cS||vr||}t|trt ||r t ||}|cS|Scc}w) zExamples: get_value_by_path({'a': {'b': v}}, ['a', 'b']) -> v get_value_by_path({'a': {'b': [{'c': v1}, {'c': v2}]}}, ['a', 'b[]', 'c']) -> [v1, v2] r"rNrrr6rr r)r#r$get_value_by_pathr% BaseModelhasattrgetattr)rrr7r/r0r1r4s rr9r9ts# gY K$fa   ||DSbh T (^  aa!egm L   e Sbh T d8n N1 tAEG}M   Cy dI &74+=tS!56 ++ sCpathsc|jD]\}}|jd}|jd}t}d}t|D]\}} | dk(s |}n|dk7rht ||kDrZt |t |D]B}||} | dk7s| j dr | j dr2|j| Dt|||d|y)a#Moves values from source paths to destination paths. Examples: move_value_by_path( {'requests': [{'content': v1}, {'content': v2}]}, {'requests[].*': 'requests[].request.*'} ) -> {'requests': [{'request': {'content': v1}}, {'request': {'content': v2}}]} r!r*rrrN) itemssplitsetr#r(r'r$add_move_value_recursive) rr= source_path dest_path source_keys dest_keys exclude_keys wildcard_idxr/r0s rmove_value_by_pathrKs!& Ik9##C(K$I5LLK(3    rc)n|;\3y>2 !l #:cll40e9L   3  $ Y<H-IrrGrHkey_idxrIc|t|k\ry||}|jdr;|dd}||vr1t||tr||D]}t ||||dz|yyy|dk(rt|t rt|j Dcgc]}|jds||vr|} }| Dcic]}||| } }| jD]F\}} g} ||dD]*} | dk(r| j|| j| ,t|| | H| D]}||=yy||vrt |||||dz|yycc}wcc}w)z>Recursively moves values from source path to destination path.Nrrrr?r2) r(r$r%r&rDr-r startswithr@appendr*)rrGrHrLrIr0r1itemk keys_to_movevalues_to_movev new_dest_keysdks rrDrDs K   G#\\$3BxH4JtH~t<x. $ +y'A+|  =  cz$ $c"q '< l-99q47 9n9!&&(2$!Q GH% %B 3Y   #   $  % $ q12! G/6 d{ s)[)Wq[,3:s E5 E  snake_strconvertc:|s|Stjdd|S)z>Converts a snake_case string to CamelCase, if convert is True.z _([a-zA-Z])c@|jdjS)Nr)groupupper)matchs rz&maybe_snake_to_camel..sekk!n.B.B.Dr)resub)rWrXs rmaybe_snake_to_camelras    Di PProbj convert_keyscjt|tjrt|j d|St|t r7|j Dcic]\}}t||t|c}}St|tr|Dcgc]}t||c}S|Scc}}wcc}w)aRecursively converts a given object to a dictionary. If the object is a Pydantic model, it uses the model's `model_dump()` method. Args: obj: The object to convert. convert_keys: Whether to convert the keys from snake case to camel case. Returns: A dictionary representation of the object, a list of objects if a list is passed, or the object itself if it is not a dictionary, list, or Pydantic model. T) exclude_none) r%pydanticr:convert_to_dict model_dumpr-r@rar&)rbrcr0rrPs rrgrgsX''( 3>>t>rz<... Max depth ...> rr=z()z( ,  )z"""z\"\"\"rz...'z{} itemsz{<... z at Max depth ...>}z )r0z: z{ }) frozensetidr&r%rfr: __class__rsortedtyperyreprr<AttributeError _pretty_reprrOjoinrlreplacebytesr( collectionsabcMappingr TypeErrortuplerB_format_collection)rbrrrrrrobj_idindentnext_indent_strcls_namer@fields field_namer~r value_represcapeditem_count_str sorted_keysrQrTk_reprv_reprraw_reprs rrris _kG c7& w ) QY d7mvh. /'  &<,67/X''(}}%%H E DI** +FB 9))*5j __Z(  #l2# j llo&zl!J<@A-B0 z_Zs ejj/ /Bvha. @@#s s{ E;/g7)3  9#u 3x'S'!)_%cr*+4 00 9#{../   zc(5CA 2(FGn~&&: ;; 3x)#c(1 %% E%388:3/k; a&a #l2# f #l2# f llo&vhb9:);*UZZ& &2fXR 88#eS)*  !! CyH   DB&7"8 99iZ %$k%s$ K-K=- K:9K:="L"!L"c t|trd}|}nXt|trd}t|}n:t|trt|}|rd}nd}nt dt ||s |d|dzS|dkr/t |dd t |dk7z} |dd | d |dSd |z} d ||zz} g} tt ||} t| D]1}||}| j| t|||z||||dz | z3t ||kDr#| j| d t ||z d|dddj| zd| |dzS)z(Formats a collection (list, tuple, set).)[])(r){r)zset(rzUnsupported collection type: rrrrz<... z at Max depth ...>rrz more items ...>rr) r%r&rrBr)rr(minr'rOrr)rbrrrrrrbrackets internal_objrrrelements num_to_showr/elems rrrsTHL#uH9L#s9L hh 4T#YK@ AA  A;! $$ aZL)*%S5F5I0J/KLNqk]%//A(1+ OO  &<,67/ (C %y1+   a ?D OO  % 4%!)     " OO  5\!2Y!> ??OP QK= ejj2 2s6(8A;-5P PPrc eZdZejej dddddddejf Z de ffd Z e dejed ee efd ee efdefd Zdee effd ZxZS) r:Trforbidbase64) alias_generatorpopulate_by_namefrom_attributesprotected_namespacesextraarbitrary_types_allowedser_json_bytesval_json_bytes ignored_typesrcV t|S#t$rt| cYSwxYwN)r Exceptionsuper__repr__)selfrs rrzBaseModel.__repr__4s0 $   W   s ((clsrwkwargsc|duxr5d|vxr/|dduxr&t|dtxrd|dvxr|dd}|s t|||j|}|S)Nconfiginclude_all_fields)r%r-r|model_validate)rrwrshould_skip_removing_fieldsvalidated_responses r_from_responsezBaseModel._from_response:s d 3   3 8 D ( 3 vh' . 3 !F8$4 4  3 8 1 2   '3)++H5 rc(|jddS)NTjson)remode)rh)rs r to_json_dictzBaseModel.to_json_dictVs ??6? ::r)rrrrf ConfigDictr to_camelrmTypeVar model_configrlr classmethodTyperr-objectrr __classcell__rs@rr:r:%s$$$&// "^^% ,   ;;q>S&[!3;   6;Df-;rc6eZdZdZededeffd ZxZS)CaseInSensitiveEnumzCase insensitive enum.rrcB ||jS#t$r ||jcYS#t$r\tj|d|j  t |||}t||_ ||_ |cYcYS#YYYyxYwwxYwwxYw)Nz is not a valid ) r\KeyErrorlowerwarningswarnrr__new__rl_name__value_)rrunknown_enum_valrs r _missing_zCaseInSensitiveEnum._missing_]s     5;;=!!   / ~>? #W_S%8 $'J  !%*  "! !   sG B5B-B#(B B BBBBBB)rrrrrrrrrs@rrrZs&CCrrctjjjd}tjj dd}|d|S)z]Composes a timestamped unique name. Returns: A string representing a unique name. z %Y%m%d%H%M%Srrr2)datetimenowstrftimeuuiduuid4hex) timestamp unique_ids rtimestamped_unique_namerqsL ##%..~>)jjlq#) ;a { ##rci}t|ts|S|jD]:\}}t|tr(t j |j d||<?t|tjr|j||<mt|trt|||<t|trtd|Dr5|Dcgc]&}t j |j d(c}||<td|Dr#|Dcgc]}|jc}||<|Dcgc] }t|c}||<6|||<=|Scc}wcc}wcc}w)aConverts unserializable types in dict to json.dumps() compatible types. This function is called in models.py after calling convert_to_dict(). The convert_to_dict() can convert pydantic object to dict. However, the input to convert_to_dict() is dict mixed of pydantic object and nested dict(the output of converters). So they may be bytes in the dict and they are out of `ser_json_bytes` control in model_dump(mode='json') called in `convert_to_dict`, as well as datetime deserialization in Pydantic json mode. Returns: A dictionary with json.dumps() incompatible type (e.g. bytes datetime) to compatible type (e.g. base64 encoded string, isoformat date string). asciic3<K|]}t|tywr)r%r.0rTs r z.encode_unserializable_types..s 1aZ5 ! 1sc3PK|]}t|tj ywr)r%rrs rrz.encode_unserializable_types..s =!Z8,, - =s$&) r%r-r@rrurlsafe_b64encodedecoder isoformatencode_unserializable_typesr&all)rprocessed_datar0rrTs rr r |sB'). D$  KJJL"jc5%"44U;BB7KnS E8,, -!OO-nS E4 7>nS E4 15 1 1AF <=F $ $Q ' . .w 7 s =u = =6;<q{{}<sGLM!:1=Ms!nS#"$  =Ms+E%E*E/message.cPdtdtfdtdtfffd }|S)z&Experimental warning, only warns once.func.rctdtjdtdtdtffd }|S)NFargsrrcTsdtjtd|i|S)NTrk)r category stacklevel)rrr)rrrr  warning_dones rwrapperz8experimental_warning..decorator..wrappers4  ( 4 "6 ""r) functoolswrapsr)rrrr s` @r decoratorz'experimental_warning..decorators?L__T #s #c #c # # Nr)rr)r rs` rexperimental_warningrs0 hsCx(Xc3h-?" rkey_strcB|jddjS)z?Normalizes a key for case-insensitive and snake/camel matching.r2r)rr)rs r_normalize_key_for_matchingrs b ! ' ' ))r target_dict update_dictci}|jDcic]}t||}}|jD]\}}t|}||vr||}n|}t|tr2t|j |trt |||||<_t|tr%t|j |tr|||<|||<|Scc}w)aAligns the keys of update_dict to the case of target_dict keys. Args: target_dict: The dictionary with the target key casing. update_dict: The dictionary whose keys need to be aligned. Returns: A new dictionary with keys aligned to target_dict's key casing. )rrr@r%r-r,align_key_caser&)rraligned_update_dictr0target_keys_maprnormalized_update_key aligned_keys rr r s%'7B7G7G7I03!#&+/ %%'/jc57</#$9:kk%: $d$*8 k "E*+& E4 Z $d&*/+&).+&)/* 3sCc pt||}|jD]\}}||vr3t||tr t|trt |||=||vrRt||t |s:t jd|dt ||dt |d|||<|||<y)aRecursively updates a target dictionary with values from an update dictionary. We don't enforce the updated dict values to have the same type with the target_dict values except log warnings. Users providing the update_dict should be responsible for constructing correct data. Args: target_dict (dict): The dictionary to be updated. update_dict (dict): The dictionary containing updates. zType mismatch for key 'z'. Existing type: z , new type: z. Overwriting.N)r r@r%r-recursive_dict_updaterloggerwarning)rrr!r0rs rr&r&s$'{K@'--/jc5 { {3' . ud #K,e4 J{3/?e$M nn #C5);s#$ %\$u+n Nk#k#r)T)F):rrcollections.abcrrenumrloggingr_rmrrrrrrr rrrfr typing_extensionsr getLoggerr'r-rlr __annotations__Warningrr&r*r9rKintrBrDboolrarrgrrur|rrrrr:Enumrrr rrr r&rrrr3s $   RRR %'   1 2S#X I&+'+G 4S> "G*.s)G