o i2^@sUdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z m Z mZmZmZddlZddlZddlZddlmZddlmZedZeee fZeed<Gdd d eZd e ee e fd eed e d dfddZ ddd e d eede d e fddZ!d e deeefd dfddZ"d e deedeede#de$ed df ddZ%dTdede&d efd d!Z'dUd#e(d$e&d e fd%d&Z)d'e*d e&fd(d)Z+d*e d+eee(fd dfd,d-Z,e j-d.d/d0Z.dd1d2d3d4dd5d#e d6e#d7e#d8e#d9e#d:e#d;e e e#d efdd?Z0Gd@d/d/ej1Z1GdAdBdBeej2Z3d efdCdDZ4d eee(fd eee(ffdEdFZ5dGed e e dHe fge dHe fffdIdJZ6dKed efdLdMZ7dNedOed efdPdQZ8dNedOed dfdRdSZ9dS)VzCommon utilities for the SDK.N)AnyCallable FrozenSetOptionalUnionget_args get_origin)alias_generators) TypeAliaszgoogle_genai._common StringDictc@seZdZdZdS)ExperimentalWarningz"Warning for experimental features.N)__name__ __module__ __qualname____doc__rrZ/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/google/genai/_common.pyr &sr datakeysvaluereturnc Cs4|durdSt|ddD]\}}|dr|dd}|durB||vrBt|tr8ddtt|D||<n td|d|t|trg|durgt||D]\}}t|||d d||qQdS|dur~||D]}t|||d d|qodS|d r|dd }|dur||vrig||<|durt||d ||d d|dS|dur||i}q|dur| |d}|dur|sdS||krdSt|t rt|t r| |dStd |dd|d|d|ddkrt|t rt|t r| |dS|||d<dSdS)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[]cSsg|]}iqSrr).0_rrr =sz%set_value_by_path..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_namejd existing_datarrrr)*sj            r) default_valuer5csdgkr|StD]r\}|sS|dr7|dd}||vr3fdd||DSS|dra|dd}||vr]||r]t||d d dd SS||vrj||}q t|trzt||rzt||}q S|S) 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!rNrcs&g|]}t|dddqS)rNr4)get_value_by_path)rr2r5r.rrrrsz%get_value_by_path..rrrrr4)r"r#r6r$ BaseModelhasattrgetattr)rrr5r/r0rr7rr6ts2         r6pathsc Cs|D]W\}}|d}|d}t}d}t|D] \}} | dkr'|}nq|dkrSt||krSt|t|D]}||} | dkrR| dsR| dsR|| q9t|||d|qdS)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.r/rrrmove_value_by_paths$   rHrDrEkey_idxrFcsB|t|krdS||}|dr9|dd}|vr3t|tr5|D]}t||||dq&dSdSdS|dkrttrfddtD}fdd |D} | D]$\} } g} ||dD]} | dkrt| | qh| | qht | | q\|D]} | =qdSdS|vrt||||ddSdS) z>Recursively moves values from source path to destination path.Nrrrr<cs"g|] }|ds|vr|qS)r) startswithrk)rFrrrsz)_move_value_recursive..csi|]}||qSrrrK)rrr z)_move_value_recursive..) r'r#r$r%rAr,rr=appendr))rrDrErIrFr/r0item keys_to_movevalues_to_moverLv new_dest_keysdkr)rrFrrAsD         rAT snake_strconvertcCs|s|Stddd|S)z>Converts a snake_case string to CamelCase, if convert is True.z _([a-zA-Z])cSs|dS)Nr)groupupper)matchrrrsz&maybe_snake_to_camel..)resub)rVrWrrrmaybe_snake_to_camelsr^Fobj convert_keyscs^t|tjrt|jddSt|trfdd|DSt|tr-fdd|DS|S)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_nonecs i|] \}}t|t|qSr)r^convert_to_dict)rr/rr`rrrMsz#convert_to_dict..csg|]}t|qSr)rb)rrPrcrrrrNz#convert_to_dict..)r$pydanticr8rb model_dumpr,r=r%)r_r`rrcrrbs    rb annotationcCst|}t|}|turdS|rt|dkrdS|d}t|}t|}|tur*dS|r2t|dkr4dS|\}}|tuo@|tjuS)zChecks if the given annotation is list[dict[str, typing.Any]] or typing.List[typing.Dict[str, typing.Any]]. This maps to Struct type in the API. Frr)rrr%r'r,strtypingr)rf outer_origin outer_argsinner_annotation inner_origin inner_argskey_type value_typerrr_is_struct_typesrqmodelresponsecCst|}|D]g\}}dd|jD}||jvr%||vr%||q|||}|j|j}t|tur?t |d}t |t rQt|t urQt ||qt |trot |r[q|D]}t |t rnt t |d|q]qdS)zfRemoves extra fields from the response that are not in the model. Mutates the response in place. cSsi|]\}}|j|qSr)alias)rr/ field_inforrrrMHs z(_remove_extra_fields..rN)r%r= model_fieldspopr+rfrirrrr$r,_remove_extra_fieldsrq)rrrs key_valuesr/r alias_maprfrPrrrrx=s,        rxTr8)boundrgd indent_level indent_deltamax_len max_itemsdepthvisitedrrrrrrc Cs|durt}t|}||vrdS|dkrdStt||g}d|}d||} t|tjr|jj} g} tt |j } | D]>} t |j | }|j sLq?zt || }Wn t y\Yq?w|durbq?t|||||||d|d}| | | d|q?| s| d S| d d | d |d St|trd |vr|dd}d|dSt |St|trt||krt |d|ddddSt |St|tjjrb|sdS|dkrt|dt|dkrdnd}d|dSt||krdt|dSg} z t|td}Wnty t|}Ynw|D]1}||}t|||||||d|d}t|||||||d|d}| | |d|q#dd | d |dSt|tttfrvt|||||||dSt |}|d d | S)z-Returns a representation of the given object.Nz<... Circular reference ...>rz<... Max depth ...> rr=z()z( ,  )z"""z\"\"\"rz...'z{} itemsz{<... z at Max depth ...>}z )r/z: z{ }) frozensetidr%r$rdr8 __class__r sortedtypervreprr:AttributeError _pretty_reprrOjoinrhreplacebytesr' collectionsabcMappingr TypeErrortupler?_format_collection)r_rrrrrrobj_idindentnext_indent_strcls_namer=fields field_namerur value_represcapeditem_count_str sorted_keysrLrSk_reprv_reprraw_reprrrrris            ""    rc Csnt|tr d}|}n&t|trd}t|}nt|tr't|}|r$d}n d}n tdt||s:|d|dS|dkr[t|dd t|dk} |dd | d |dSd |} d ||} g} tt||} t| D]}||}| | t |||||||d|d qrt||kr| | d t||d|ddd | d| |dS)z(Formats a collection (list, tuple, set).)[])(r){r)zset(rzUnsupported collection type: rrrrz<... z at Max depth ...>rrz more items ...>rr) r$r%rr?r(rr'minr&rOrr)r_rrrrrrbrackets internal_objrrrelements num_to_showr.elemrrrrsR       *rc seZdZejejdddddddejfd Z de ffdd Z e d ej ed ee efd ee efdefd d Zdee effddZZS)r8Trforbidbase64) alias_generatorpopulate_by_namefrom_attributesprotected_namespacesextraarbitrary_types_allowedser_json_bytesval_json_bytes ignored_typesrcs(zt|WStytYSwN)r Exceptionsuper__repr__selfrrrr4s   zBaseModel.__repr__clsrskwargscCs^|duo d|vo |dduo t|dto d|dvo |dd}|s(t||||}|S)Nconfiginclude_all_fields)r$r,rxmodel_validate)rrsrshould_skip_removing_fieldsvalidated_responserrr_from_response:s      zBaseModel._from_responsecCs|jdddS)NTjson)ramode)rerrrr to_json_dictVszBaseModel.to_json_dict)r rrrd ConfigDictr to_camelriTypeVar model_configrhr classmethodTyper{r,objectrr __classcell__rrrrr8%s0    cs.eZdZdZededeffdd ZZS)CaseInSensitiveEnumzCase insensitive enum.rrc sz||WStyIz ||WYStyHt|d|jzt||}t||_ ||_ |WYYSYYYdSww)Nz is not a valid ) rYKeyErrorlowerwarningswarnr r__new__rh_name__value_)rrunknown_enum_valrrr _missing_]s     zCaseInSensitiveEnum._missing_)r rrrrrrrrrrrrZs rcCs0tjd}tjdd}|d|S)z]Composes a timestamped unique name. Returns: A string representing a unique name. z %Y%m%d%H%M%Srr~r)datetimenowstrftimeuuiduuid4hex) timestamp unique_idrrrtimestamped_unique_nameqsrcCsi}t|ts |S|D]e\}}t|tr!t|d||<q t|tjr.|||<q t|tr:t |||<q t|t rnt dd|DrQdd|D||<t dd|Drddd|D||<q dd|D||<q |||<q |S) 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). asciicss|]}t|tVqdSr)r$rrrSrrr sz.encode_unserializable_types..cSsg|] }t|dqS)r)rurlsafe_b64encodedecoderrrrrsz/encode_unserializable_types..css|] }t|tjVqdSr)r$rrrrrrscSsg|]}|qSr) isoformatrrrrrcSsg|]}t|qSr)encode_unserializable_typesrrrrrr) r$r,r=rrrrrrrr%all)rprocessed_datar/rrrrr|s(       rmessage.cs*dtdtfdtdtfffdd }|S)z&Experimental warning, only warns once.func.rcs0dtdtdtdtffdd }|S)NFargsrrcs&s dtjtdd|i|S)NTrg)rcategory stacklevel)rrr )rr)rr warning_donerrwrappersz8experimental_warning..decorator..wrapper) functoolswrapsr)rrr)rrr decorators  z'experimental_warning..decorator)rr)rrrrrexperimental_warnings&rkey_strcCs|ddS)z?Normalizes a key for case-insensitive and snake/camel matching.rr)rr)rrrr_normalize_key_for_matchingsr  target_dict update_dictcCsi}dd|D}|D]@\}}t|}||vr ||}n|}t|tr9t||tr9t|||||<qt|trKt||trK|||<q|||<q|S)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. cSsi|]}t||qSr)r )rr/rrrrMs z"align_key_case..)rr=r r$r,r+align_key_caser%)r r aligned_update_dicttarget_keys_mapr/rnormalized_update_key aligned_keyrrrr s*        r c Cst||}|D]G\}}||vr%t||tr%t|tr%t|||q ||vrLt||t|sLtd|dt||dt|d|||<q |||<q dS)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)r r r r/rrrrrs&     r)T)F):rrcollections.abcrrenumrloggingr\rirrrrrrrrrrdr typing_extensionsr getLoggerrr,rhr __annotations__Warningr r%r)r6rHintr?rAboolr^rrbrrqrxrr{rrr8Enumrrrrr r rrrrrs$    K *$ 8!)  ~ >5" &  )