o 0Xxi]%@s<dZddlZddlZddlZddlZddlZddlmZddlZddlZddl Z ddl m Z ddl m Z ddl m Z dd Zd ed eefd d ZdeeefdeeeffddZde jdeeefddfddZGddde jZGdddeejZdefddZd eeefdeeeffddZdefdd ZdS)!zCommon utilities for the SDK.N)Union)alias_generators) _api_client)errorsc Cst|durdSt|ddD]m\}}|dru|dd}||vr>t|tr4ddtt|D||<n td|d|t|tr_t||D]\}}t|||d d||qIdS||D]}t|||d d|qcdS||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 S/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/google/genai/_common.py 3sz%set_value_by_path..zvalue z$ must be a list given an array path rz+Cannot set value for an existing key. Key: z; Existing value: z ; New value: .) enumerateendswith isinstancelistrangelen ValueErrorset_value_by_path setdefaultgetdictupdate) datakeysvalueikeykey_namejd existing_datar r r r"sF       rrrcsdgkr|StD]D\}|sdS|dr4|dd}||vr1fdd||DSdS||vr=||}q t|trMt||rMt||}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] _selfNrr cs"g|] }t|ddqS)rN)get_value_by_path)r r#rrr r rjs"z%get_value_by_path..)rrr BaseModelhasattrgetattr)rrr r!r r'r r&Zs     r&objreturncCsPt|tjr |jddSt|trdd|DSt|tr&dd|DS|S)zRecursively 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. T) exclude_nonecSsi|] \}}|t|qSr convert_to_dict)r r rr r r sz#convert_to_dict..cSg|]}t|qSr r.)r itemr r r rz#convert_to_dict..)rpydanticr( model_dumpritemsr)r+r r r r/ws   r/modelresponsecCst|}|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 field_infor r r r0s z(_remove_extra_fields..rN) rr6 model_fieldspopr annotationtyping get_originrget_argsrr_remove_extra_fields)r7r8 key_valuesr r alias_mapr=r2r r r rAs(        rAc @sheZdZejejddddddddZede e e fde e e fddfd d Z de e e ffd d Z d S)r(Tr forbidbase64)alias_generatorpopulate_by_namefrom_attributesprotected_namespacesextraarbitrary_types_allowedser_json_bytesval_json_bytesr8kwargsr,cCst||||}|SN)rAmodel_validate)clsr8rNvalidated_responser r r _from_responses  zBaseModel._from_responsecCs|jdddS)NTjson)r-mode)r5)selfr r r to_json_dictszBaseModel.to_json_dictN)__name__ __module__ __qualname__r4 ConfigDictrto_camel model_config classmethodrstrobjectrSrWr r r r r(s(     r(cs$eZdZdZefddZZS)CaseInSensitiveEnumzCase insensitive enum.c sz||WStyIz ||WYStyHt|d|jzt||}t||_ ||_ |WYYSYYYdSww)Nz is not a valid ) upperKeyErrorlowerwarningswarnrXsuper__new__r__name__value_)rQrunknown_enum_val __class__r r _missing_s     zCaseInSensitiveEnum._missing_)rXrYrZ__doc__r^rn __classcell__r r rlr rasracCs0tjd}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 r timestamped_unique_namesrzcCsi}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|tVqdSrO)rbytesr vr r r sz.encode_unserializable_types..cSsg|] }t|dqS)r{)rEurlsafe_b64encodedecoder}r r r r sz/encode_unserializable_types..css|] }t|tjVqdSrO)rrrr}r r r rscSsg|]}|qSr ) isoformatr}r r r rr3cSr1r )encode_unserializable_typesr}r r r rr3) rrr6r|rErrrrrrrall)rprocessed_datar rr r r rs(       rmessagecsfdd}|S)z&Experimental warning, only warns once.cs"dtfdd}|S)NFcs(s dtjtjdd|i|S)NT)rcategory stacklevel)rerfrExperimentalWarning)argsrN)funcr warning_doner r wrappersz8experimental_warning..decorator..wrapper) functoolswraps)rrr)rrr decorators z'experimental_warning..decoratorr )rrr rr experimental_warnings r)rorErrenumrr>rrurer4rrrrr`rr_r&rr/r(rAEnumrarzrrr r r r s8    8"  (" &