o Zh'@s<dZddlZddlZddlZddlZddlZddlmZmZddlZddl Z ddl Z ddl m Z ddl m Z ddl mZdd Zd ed eefd d ZdedefddZdedeeefddfddZejdddZGddde jZGdddeejZdefddZd eeefdeeeffddZd efd!d"ZdS)#zCommon utilities for the SDK.N)AnyUnion)alias_generators) _api_client)errorsc Cs|durdSt|ddD]\}}|dru|dd}||vr>t|tr4ddtt|D||<n td|d|t|tr_t||D]\}}t|||d d||qIdS||D]}t|||d d|qcdS|d r|dd }||vrig||<t||d ||d d|dS||i}q| |d}|dur|sdS||krdSt|t rt|t r| |dStd |dd|d|d|||d<dS)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_r r K/var/www/html/lang_env/lib/python3.10/site-packages/google/genai/_common.py 3sz%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) datakeysvalueikeykey_namejdZ existing_datar r rr"sR           rrr 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] _selfNr r cs"g|] }t|ddqS)rN)get_value_by_path)r r&r"r r rrps"z%get_value_by_path..rrrr)rrr(r BaseModelhasattrgetattr)rr r#r$r r)rr(`s*     "  r(objreturncCsPt|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|qSr convert_to_dict)r r#r!r r r sz#convert_to_dict..cSg|]}t|qSr r0)r itemr r rrz#convert_to_dict..)rpydanticr* model_dumpritemsr)r-r r rr1s   r1modelresponsecCst|}|D]b\}}dd|jD}||jvr%||vr%||q|||}|j|j}t|tur?t |d}t |t rQt|t urQt ||qt |trj|D]}t |t rit t |d|qXqdS)zfRemoves extra fields from the response that are not in the model. Mutates the response in place. cSsi|]\}}|j|qSr )alias)r r#Z field_infor r rr2s z(_remove_extra_fields..rN) rr8Z model_fieldspopr annotationtyping get_originrget_argsrr_remove_extra_fields)r9r: key_valuesr#r!Z alias_mapr=r4r r rrAs(        rATr*)boundc @sxeZdZejejdddddddejfd Z e dej e de eefde eefd e fd d Zd e eeffd d ZdS)r*Tr Zforbidbase64) Zalias_generatorZpopulate_by_nameZfrom_attributesZprotected_namespacesextraZarbitrary_types_allowedZser_json_bytesZval_json_bytesZ ignored_typesclsr:kwargsr.cCst||||}|SN)rAZmodel_validate)rGr:rHZvalidated_responser r r_from_responses  zBaseModel._from_responsecCs|jdddS)NTjson)r/mode)r7)selfr r r to_json_dictszBaseModel.to_json_dictN)__name__ __module__ __qualname__r6Z ConfigDictrZto_camelr>TypeVarZ model_config classmethodTyperCrstrobjectrJrNr r r rr*s.     cs$eZdZdZefddZZS)CaseInSensitiveEnumzCase insensitive enum.c sz||WStyIz ||WYStyHt|d|jzt||}t||_ ||_ |WYYSYYYdSww)Nz is not a valid ) upperKeyErrorlowerwarningswarnrOsuper__new__rU_name__value_)rGr!Zunknown_enum_val __class__r r _missing_s     zCaseInSensitiveEnum._missing_)rOrPrQ__doc__rSrc __classcell__r r rarrWsrWcCs0tjd}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_idr r rtimestamped_unique_namesrocCsi}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|tVqdSrI)rbytesr vr r r sz.encode_unserializable_types..cSsg|] }t|dqS)rp)rEurlsafe_b64encodedecoderrr r rrsz/encode_unserializable_types..css|] }t|tjVqdSrI)rrgrrr r rrt!scSsg|]}|qSr ) isoformatrrr r rr"r5cSr3r )encode_unserializable_typesrrr r rr$r5) rrr8rqrErurvrgrwrxrall)rZprocessed_datar#r!r r rrxs(       rxmessagecsfdd}|S)z&Experimental warning, only warns once.cs"dtfdd}|S)NFcs(s dtjtjdd|i|S)NT)rzcategory stacklevel)r[r\rZExperimentalWarning)argsrH)funcrz warning_doner rwrapper.sz8experimental_warning..decorator..wrapper) functoolswraps)rrrz)rrr decorator,s z'experimental_warning..decoratorr )rzrr rrexperimental_warning*s r)rdrErgenumrr>rrrjr[r6rrrrrrUr(rVr1rrArRrCr*EnumrWrorxrr r r rs:   >#  '" &