o Thr:@sdZddlZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z ddl Z ddlZddlZddlmZddlmZddlmZedZd e eeefd eed ed dfd dZd ed eed efddZded efddZded efddZdedeeefd dfddZ ej!dddZ"Gdddej#Z#Gdd d eej$Z%d efd!d"Z&d eeefd eeeffd#d$Z'd%ed eed&efged&efffd'd(Z(d)ed efd*d+Z)d,eeefd-eeefd eeeffd.d/Z*d,eeefd-eeefd dfd0d1Z+dS)2zCommon utilities for the SDK.N)AnyCallableOptionalUnion get_originget_args)alias_generators) _api_client)errorszgoogle_genai._commondatakeysvaluereturnc Cs|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|||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|]}iqSr).0_rrW/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/google/genai/_common.py 5sz%set_value_by_path..zvalue z$ must be a list given an array path r [0]rz+Cannot set value for an existing key. Key: z; Existing value: z ; New value: .) enumerateendswith isinstancelistrangelen ValueErrorset_value_by_path setdefaultgetdictupdate) r r rikeykey_namejd existing_datarrrr"$s^           r"csdgkr|StD]k\}|sdS|dr4|dd}||vr1fdd||DSdS|dr[|dd}||vrX||rXt||d d dSdS||vrd||}q t|trtt||rtt||}q dS|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] _selfNrrcs"g|] }t|ddqS)r N)get_value_by_path)rr+r'r rrrus"z%get_value_by_path..rrrr )rrr.r BaseModelhasattrgetattr)r r r(r)rr/rr.es*     "  r.objcCsPt|tjr |jddSt|trdd|DSt|tr&dd|DS|S)adRecursively 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. 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_nonecSsi|] \}}|t|qSrconvert_to_dict)rr(rrrr sz#convert_to_dict..cSg|]}t|qSrr5)ritemrrrrz#convert_to_dict..)rpydanticr0 model_dumpr%itemsr)r3rrrr6s   r6 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. Fr r)rrrr r%strtypingr)r> outer_origin outer_argsinner_annotation inner_origin inner_argskey_type value_typerrr_is_struct_typesrImodelresponsecCst|}|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_inforrrr7s z(_remove_extra_fields..rN)rr= model_fieldspopr$r>rArrrrr%_remove_extra_fieldsrI)rJrK key_valuesr(r alias_mapr>r9rrrrPs,        rPTr0)boundc @sxeZdZejejdddddddejfd Z e dej e de eefde eefd e fd d Zd e eeffd d ZdS)r0Trforbidbase64) alias_generatorpopulate_by_namefrom_attributesprotected_namespacesextraarbitrary_types_allowedser_json_bytesval_json_bytes ignored_typesclsrKkwargsrcCs^|duo d|vo |dduo t|dto d|dvo |dd}|s(t||||}|S)Nconfiginclude_all_fields)rr%rPmodel_validate)r`rKrashould_skip_removing_fieldsvalidated_responserrr_from_responses       zBaseModel._from_responsecCs|jdddS)NTjson)r4mode)r<)selfrrr to_json_dictszBaseModel.to_json_dictN)__name__ __module__ __qualname__r; ConfigDictrto_camelrATypeVar model_config classmethodTyperSr%r@objectrgrkrrrrr0s.    cs.eZdZdZededeffdd ZZS)CaseInSensitiveEnumzCase insensitive enum.rrc sz||WStyIz ||WYStyHt|d|jzt||}t||_ ||_ |WYYSYYYdSww)Nz is not a valid ) upperKeyErrorlowerwarningswarnrlsuper__new__r@_name__value_)r`runknown_enum_val __class__rr _missing_s     zCaseInSensitiveEnum._missing_)rlrmrn__doc__rsrr __classcell__rrrrrvs rvcCs0tjd}tjdd}|d|S)z]Composes a timestamped unique name. Returns: A string representing a unique name. z %Y%m%d%H%M%Srr)datetimenowstrftimeuuiduuid4hex) timestamp unique_idrrrtimestamped_unique_name/srcCsi}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|tVqdSN)rbytesrvrrr Ssz.encode_unserializable_types..cSsg|] }t|dqS)r)rVurlsafe_b64encodedecoderrrrrTsz/encode_unserializable_types..css|] }t|tjVqdSr)rrrrrrrWscSsg|]}|qSr) isoformatrrrrrXr:cSr8r)encode_unserializable_typesrrrrrZr:) rr%r=rrVrrrrrrall)r processed_datar(rrrrr:s(       rmessage.cs*dtdtfdtdtfffdd }|S)z&Experimental warning, only warns once.func.rcs0dtdtdtdtffdd }|S)NFargsrarcs(s dtjtjdd|i|S)NTr?)rcategory stacklevel)rzr{r ExperimentalWarning)rra)rr warning_donerrwrapperdsz8experimental_warning..decorator..wrapper) functoolswrapsr)rrr)rrr decoratorbs  z'experimental_warning..decorator)rr)rrrrrexperimental_warning`s&rkey_strcCs|ddS)z?Normalizes a key for case-insensitive and snake/camel matching.r)replacery)rrrr_normalize_key_for_matchingssr 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(rrrr7sz"align_key_case..)r r=rrr%r$align_key_caser)rraligned_update_dicttarget_keys_mapr(rnormalized_update_key aligned_keyrrrrxs    rc 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)rr=rr%recursive_dict_updatetypeloggerwarning)rrrr(rrrrrs&     r),rrVrenumrloggingrArrrrrrrrzr;rrr r getLoggerrr%rr@r"r.rur6rboolrIrPrqrSr0EnumrvrrrrrrrrrrsR     *A#   *," ,&.