+j UdZddlZddlmZmZddlmZmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZmZmZddlmZddlmZdd lmZejrddlZddl Z dd l!m"Z"dd l!mZdd l!mZe j#d Z$e j%dkrej&Z'ejej&fZ(ddlm)Z)nej*Z'ejfZ(ddl+m)Z)ejr ddl,m-Z.ddl/m0Z1n5eZ.ej2e3d<eZ1ej2e3d< ddl,m-Z.ddl/m0Z1n#e4$rdZ.dZ1YnwxYwddddZ5de56DZ7ddde"j8de9de9de:de9f d Z;de"j8d!e9de9fd"Zd&ej?de@ej?fd'ZAd#e"j8d!e9dee9fd(ZBd)edejCfd*ZDd+ejEdejFfd,ZGd-eejEeejEfde@ejFfd.ZHd/eejIe@ejIfde@ejCfd0ZJd1ejIdejCfd2ZKd1ejIdejCfd3ZLd1ejMdejCfd4ZNd5eejOdejPfd6ZQd7eee@ejOejOe@ejPfde@ejPfd8ZRd9eeee9efdee@ejSfd:ZTeejUejVejOfZWd;eeWejVdfdejUfd<ZXde"j8d=ee@ejUe@ejVeWfdee@e9e@ejUffd>ZYd=eeejZej[ejUfde@ejUfd?Z\d@ej?ddfdAZ]dBeddfdCZ^d@edee"j8ddfdDZ_ dqdEddFd@ej?dee"j8dGeej?dHe=dIee`e:ddf dJZadKedee"j8dejbfdLZcdBeejdefde)ej?fdMZedee"j8dBeejdefdeejbfdNZfdBeejgefdeejhfdOZidBejjdeejhfdPZkde"j8dBedeeej0effdQZlde"j8dBe@ede@ej0fdRZmde"j8dSe9de9fdTZnde"j8dUejodejpfdVZqde"j8dUejrdejsfdWZtdXee9ejufdejvfdYZwdXexe9efdexe9effdZZyde"j8dSe9de9fd[Zzd\e9de9fd]Z{d^Z|d_Z}d`Z~daZd#e"j8dbej?defdcZdSeee9ejejejfde9fddZdee9deeje9ffdfZd;ejdejUfdgZd=eeejejfde@ejUfdhZ drdieeeejejfdje=dejfdkZdleejEeejEfdejfdmZdne@ejde@exe9effdoZd!e9de=fdpZdS)sz"Transformers for Google GenAI SDK.N)IterableMapping)EnumEnumMeta)Any GenericAliasListOptionalSequenceUnion)mcp_to_gemini_tool)get_value_by_path)is_duck_type_of) _api_client)_common)typeszgoogle_genai._transformers) ) TypeGuard) ClientSession)ToolMcpClientSessionMcpToolexactMatchSpecbleuSpec rougeSpec) exact_matchbleu rouge_specci|]\}}|| Sr").0kvs ^/home/wildlama/visual-decline/.venv/lib/python3.11/site-packages/google/genai/_transformers.py r'IsLLLDAq1aLLL)collection_hierarchy_depthclient resource_namecollection_identifierr*returnc||d o |d|ddz|k}|jr|dr|S|dr d|jd|S||drd|jd|jd|S|rd|jd|jd|d|S|S|r|d|S|S)a|Prepends resource name with project, location, collection_identifier if needed. The collection_identifier will only be prepended if it's not present and the prepending won't violate the collection hierarchy depth. When the prepending condition doesn't meet, returns the input resource_name. Args: client: The API client. resource_name: The user input resource name to be completed. collection_identifier: The collection identifier to be prepended. See collection identifiers in https://google.aip.dev/122. collection_hierarchy_depth: The collection hierarchy depth. Only set this field when the resource has nested collections. For example, `users/vhugo1802/events/birthday-dinner-226`, the collection_identifier is `users` and collection_hierarchy_depth is 4. See nested collections in https://google.aip.dev/122. Example: resource_name = 'cachedContents/123' client.vertexai = True client.project = 'bar' client.location = 'us-west1' _resource_name(client, 'cachedContents/123', collection_identifier='cachedContents') returns: 'projects/bar/locations/us-west1/cachedContents/123' Example: resource_name = 'projects/foo/locations/us-central1/cachedContents/123' # resource_name = 'locations/us-central1/cachedContents/123' client.vertexai = True client.project = 'bar' client.location = 'us-west1' _resource_name(client, resource_name, collection_identifier='cachedContents') returns: 'projects/foo/locations/us-central1/cachedContents/123' Example: resource_name = '123' # resource_name = 'cachedContents/123' client.vertexai = False _resource_name(client, resource_name, collection_identifier='cachedContents') returns 'cachedContents/123' Example: resource_name = 'some/wrong/cachedContents/resource/name/123' resource_prefix = 'cachedContents' client.vertexai = False # client.vertexai = True _resource_name(client, resource_name, collection_identifier='cachedContents') returns: 'some/wrong/cachedContents/resource/name/123' Returns: The completed resource name. /r projects/z locations/ /locations/) startswithcountvertexaiprojectlocation)r+r,r-r*$should_prepend_collection_identifiers r&_resource_namer9LsOJ  " "&;#>#>#> ? ??$# 4 4] 4 4 : :3 ? ?! C # $ ' _ ,,    ! !, / / 9 9 9- 9 99  ! !%:"="="= > > U U UFO U Um U UU - m m mFO m mF[ m m^k m mm +% 7 7 7 77 r(modelc|stdd|vsd|vsd|vrtd|jrk|ds*|ds|dr|Sd |vr!|d d \}}d|d |Sd |S|dr|S|d r|Sd|S)Nzmodel is required.z..?&zinvalid model parameter.r1models/ publishers/r0r z/models/zpublishers/google/models/z tunedModels/) ValueErrorr5r3split)r+r: publishermodel_ids r&t_modelrDs + ) * ** U]]cUllcUll / 0 00 _ %% 1   I & & 1   M * * 1 l !KKQ//i 89 8 8h 8 88 0 0 00  "" l   . ) ) l u  r( api_client base_modelsc(|jr|rdSdS|rdSdS)Nzpublishers/google/modelsmodels tunedModels)r5)rErFs r& t_models_urlrJs4  ' ' X X ]r(responsecf|sgS|d}||S|d}||S|d}||S|d|dgStdtd|gS)NrHrIpublisherModels httpHeaders jsonPayloadz!Cannot determine the models type.z1Cannot determine the models type for response: %s)getloggerwarningdebug)rKrH tuned_modelspublisher_modelss r&t_extract_modelsrVs  I/7||H/E/E&  M5=\\-5P5P, 9A::! ll=!!- ,,} % % - I NN6777 LLDhOOO Ir(ct||}|sdS|dr|jrd|jd|jd|S|dr|jrd|jd|jd|S|S)Nr?r1r2r0r>z/publishers/google/)rDr3r5r6r7)rEr:s r&t_caches_modelrXs *e $ $%  4 m$$)< QJ&PP:3FPPPP  ""z': ez) e ej6I e e^c e ee Lr(imagecbd}t}|jdkr2t|ddr!|jdvrd}|dt j}|j||fi||}d| }tj || S) NPNGJPEGfilename)1LRGBRGBXCMYKkeep)qualityimage/)data mime_type) dictformatgetattrmodeupdateioBytesIOsavegetvaluelowerrBlob)rY image_format save_paramsimage_io image_bytesrhs r& pil_to_blobrxs, $+ lf %R ( ( *9 9 9Lv&&& Z\\( %*X|33{333!!##+-|))++--)  : : ::r(function_responsec|stdt|trtj|St |tjr|Stdt|)Nzfunction_response is required.zOCould not parse input as FunctionResponse. Unsupported function_response type: ) r@ isinstancerirFunctionResponsemodel_validater TypeErrortype)rys r&t_function_responsers 7 5 6 66!4((  ! 0 01B C CC(%*@AA   >$():$;$; > >  r(function_responsesc|stdt|tr d|DSt|gS)Nz function_responses are required.c,g|]}t|Sr")r)r#rKs r& z(t_function_responses...s! M M Mh  ) ) M M Mr()r@r{r rrs r&t_function_responsesr%sU 9 7 8 88"H--5 M M:L M M MM  2 3 3 44r(blobscdt|tr d|DSt|gS)Nc,g|]}t|Sr")t_blob)r#blobs r&rzt_blobs..7 + + +TF4LL + + +r()r{listr)rs r&t_blobsr3s8t + +U + + ++ 5MM?r(rc|stdt|tjr|St |t rtj|Sd|jj vrE ddl }|j j }n#t$rd}YnwxYw|t ||rt|Stdt|)Nzblob is required.rYrz6Could not parse input as Blob. Unsupported blob type: )r@rrrsr{rir} __class____name__rr PIL.ImageImage ImportErrorrxr~r)rPIL PIL_Images r&rr<s * ( ) ))T5:&& Kd+ : $ $T * ** '--////)/ii iiiD)!? = > >> :  DN  K KKdGG Z & &t , ,,  #GGG Z%."?"?"E"E F F FFFFGT5:&& K '--////)/ii iiiD)!.rr()r{rr@rrs r&t_partsrs^  ]z%..]u] 2 3 33t + +U + + ++ 5MM?r( predictionsc |sdSg}|D]k}|drT|tjtj|dd|ddl|S)NrYgcsUri imageBytes)gcs_urirw)rY)rPappendrGeneratedImager)rimages predictions r&t_image_predictionsrs  4 &  j~~g mm  K$W-h7(1,?    -r(contentc|tdt|tjr|St |t rw tj|S#tj$rEt|}|j rtj |gntj |gcYSwxYwt|tj r#tj t|gSt|tjr3|j rtj |gntj |gStj |S)Nzcontent is required.r)r@rrContentr{rir}rrr function_call ModelContent UserContentrr)r possible_parts r& t_contentrsb _ + , ,,Wem,, N  ] ) )' 2 22  #Woom ( 8% M? 3 3 3 3 777Wej))6  F7OO#4 5 5 55Wej))   0'++++  gY / / /   ) ) ))sA##AB76B7contentsct|tr d|D}nt|g}|jrg}|D]}|t|trt j|}|jF|jD]>}|j r| |j $t d?|S|S)Nc,g|]}t|Sr")rr#rs r&rz(t_contents_for_embed..s GGG7Ig..GGGr(z/Non-text part found, only returning text parts.) r{rrr5rirrr}rrrrQrR)r+rtransformed_contents text_partsrrs r&t_contents_for_embedrs$1GGhGGG%h//0 _ J' Q Q   gt $ $ :M0099' = $m Q Qdy Q ****nnOPPPP  r(c|t|tr|stdt|tst|gSg}g}dtt jtfdtt jfd}dt j dtfddtt j dtffd dtt j d tt j ddffd dtt j d tt j d t jddffd }|D]}t|t j st|tr`||t|tr)| t j| || |||r||||t|tr3| t j |tdt#||||S)Nzcontents are required.rr.c\t|ts4t|tjst|tjrdSt|t rs|sdS tj|dS#tj $r; tj |YdS#tj $rYYdSwxYwwxYwd|j j vr8 ddl}|jj}n#t $rd}YnwxYw|t||rdSdS)NTFrYr)r{rrrrrrir}rrrrrrrrrrrs r&_is_partzt_contents.._is_partsl 4 4 , , 4 , , T$  u !!$'''t  %  . ' ' - - -'     $.)//1111IO    :dI#>#> t 5s<&BCB99C C C  C5D DDc|j SN)r)rs r& _is_user_partz!t_contents.._is_user_parts! !!r(rc:tfd|DS)Nc3.K|]}|VdSrr")r#rrs r& z6t_contents.._are_user_parts..s-55t}}T""555555r()all)rrs r&_are_user_partsz#t_contents.._are_user_partss& 5555u555 5 55r(resultaccumulated_partsc|sdS||rtj|ntj|g|dd<dS)Nr)rrrr)rrrs r&$_append_accumulated_parts_as_contentz8t_contents.._append_accumulated_parts_as_contentsx   f MM ?, - - 9 12222  &7 8 8 8 aaar( current_partct|}||kr||dS|||g|dd<dSr)rr)rrrrrrs r&_handle_current_partz(t_contents.._handle_current_part!sw ,''L}\""oo6G&H&HHH|,,,,,**63DEEE*^r(rzUnsupported content type: )r{rr@rr r PartUnionDictrrrboolrrrrrir}r) rrrrrrrrrs @@@r& t_contentsrs *Xt44X - . .. Hd # #! h    "&(*# %%s* +#$%####J"%*"""""6T%*-6$666666 5=! ej)         ,5=! ,ej) ,' , , , , , , , , ,& E Egw .. E*Wd2K2K E**63DEEE GT " " e'g6667777 g '  E6#4g>>>> GT " "E mmEM0099:::: CDMMCC D DD&&v/@AAA -r(schemacb|dddkr d|d<|d=dSd|vr|dD]}}d|vru|ddkrid|d<|dddit|ddkr.|ddD] \}}|||< |d=|dSdS) aProcess null fields in the schema so it is compatible with OpenAPI. The OpenAPI spec does not support 'type: 'null' in the schema. This function handles this case by adding 'nullable: True' to the null field and removing the {'type': 'null'} entry. https://swagger.io/docs/specification/v3_0/data-models/data-types/#null Example of schema properties before and after handling null fields: Before: { "name": { "title": "Name", "type": "string" }, "total_area_sq_mi": { "anyOf": [ { "type": "integer" }, { "type": "null" } ], "default": None, "title": "Total Area Sq Mi" } } After: { "name": { "title": "Name", "type": "string" }, "total_area_sq_mi": { "type": "integer", "nullable": true, "default": None, "title": "Total Area Sq Mi" } } rNnullTnullableanyOfr r)rPremovelenitems)ritemkeyvals r&handle_null_fieldsrHsX ZZ''F:v&w 4DLF22!zw/000 vg  1 $ $ /!,2244  hc3F3KKWor(originc&td|)z2Raises an error if the schema type is unsupported.Unsupported schema type: )r@rs r&"_raise_for_unsupported_schema_typers7v77888r(c|r@|js;|ds|drtddSdSdS)NadditionalPropertiesadditional_propertiesztadditionalProperties is only supported in Gemini Enterprise Agent Platform mode, , not in Gemini Developer API mode.)r5rPr@)rr+s r&'_raise_for_unsupported_mldev_propertiesrs MoM **+ , , M ZZ/ 0 0 M L M MMMMMMMMr(Torder_propertiesvisited_dicts_pathdefsrrctt|vrdSt||ddkr|d=t |dD]"\}}||dx}|||<#D|diD]\}} t| t||ddx} 4| | dd d tj d tj ffd |d x} 5fd| D|d < t|dS|d} t| tr| j} t| t"r| } |d} | | dkr | g|d<|d=nt'd| dkr|dx}t)|D]\}} | ||<t+|dkr*r(d|vr$t)||d<|dx}#t|t.r||d<nS| dkrM|dx}||d<|dx}fd|D|d< t|dS)a{Updates the schema and each sub-schema inplace to be API-compatible. - Inlines the $defs. Example of a schema before and after (with mldev): Before: `schema` { 'items': { '$ref': '#/$defs/CountryInfo' }, 'title': 'Placeholder', 'type': 'array' } `defs` { 'CountryInfo': { 'properties': { 'continent': { 'title': 'Continent', 'type': 'string' }, 'gdp': { 'title': 'Gdp', 'type': 'integer'} }, } 'required':['continent', 'gdp'], 'title': 'CountryInfo', 'type': 'object' } } After: `schema` { 'items': { 'properties': { 'continent': { 'title': 'Continent', 'type': 'string' }, 'gdp': { 'title': 'Gdp', 'type': 'integer' }, } 'required':['continent', 'gdp'], 'title': 'CountryInfo', 'type': 'object' }, 'type': 'array' } NtitlePlaceholderLiteralEnum))rr)any_ofr) prefix_items prefixItems)property_orderingpropertyOrderingz$defsr$refdefs/ sub_schemar.c|ddx}4|dd}t|vriSt||S)z@Returns the processed `sub_schema`, resolving its '$ref' if any.rNrrr)poprAidprocess_schema)rrefr+rrrs r&_recursez process_schema.._recurses|~~fd+++8 '**2./j J- - -  )-  r(rc&g|] }|Sr"r")r#rr s r&rz"process_schema..s#EEE xx ++EEEr(rconstSTRINGenumzLiteral values must be strings.OBJECT propertiesr rrrARRAYrrc&g|] }|Sr"r")r#prefixr s r&rz"process_schema..@s#GGGFxx//GGGr()setr addrPrrrr rrmrAr StringDictrr{rvaluerupperr@rrkeysri)rr+rrr from_nameto_namer_rr r schema_typerrname additionalrprefixesr s ```` @r&r r s HZZ%%% FF$$$ ZZ444w)&&999 i It,,,9fWo \ ::gr " "D   :   +/ V ZZ % %%c2 MM$syy))"-.///7-'2D  7###f0EEEEfEEEF7Objj))) F 6""+ T""$#K S!!&##%%K **W  % hwfVn // 8 9 99Hjj... ;":#3#3#5#56600 $ #8J// 4 j  ! !A % % & ..&*:??+<+<&=&="#jj!7888 E J % %>)1*)=)=%&gG$$$1 fWoJJ}---:GGGGhGGGf]BvJJ'''''r(rc  d}|D]e}t|jtrd}t|jts,t d|jdt |jf||r*d|D}t|j|t}|Gfddtj }| }t|||d d }tj|S) NFTz Enum member z( value must be a string or integer, got c6g|]}t|jSr")rr)r#members r&rz!_process_enum..Ts"8883v|$$888r()rceZdZUed<dS)"_process_enum..Placeholder placeholderNr __module__ __qualname____annotations__)enum_to_processsr& Placeholderr'Xs      r(r.rr()r{rintrr~r rrrr BaseModelmodel_json_schemar rSchemar}) rr+is_integer_enumr% str_membersstr_enumr. enum_schemar-s @r& _process_enumr7DsE/f&,$$oo  c * *  # # #6<   # #   /884888KDM;S999HO!!!!!!!H&!!!--//+f%%%L)-8+  $ $[ 1 11r(c^t|totd|DS)zEVerifies the schema is of type dict[str, Any] for mypy type checking.c3@K|]}t|tVdSr)r{r)r#rs r&rz(_is_type_dict_str_any..es=**"jc******r()r{rirrs r&_is_type_dict_str_anyr:asC FD ! ! c**&,***''r(csdSttr>tr/t|tjSttrt|Sttjrttt jkrt d}t||tj|Sttsrttr]ttjrC}t||tj|SttsEtts0tt$st'jt*vrkGfddtj}|}t|||dd}tj|St-d)NT) exclude_unsetceZdZUed<dS)t_schema..Placeholderr(Nr)rsr&r.r>sr(r.rr(r)r{rir:r rr2r}rr7rr model_dumprr issubclassrr0r1VersionedUnionTypetyping get_origin _UNION_TYPESr@)r+rrr.s ` r&t_schemarEjsZ  4/"7"?"?/66""" < & &v . ..!!)  ( ((VU\**/ F||tELNN++++)000   T  2 2F66""" < & &v . .. V\ * */ VT " " / VX/ 0 0 / % % ' 'F66""" < & &v . ..&& / FD ! ! / F. / / /  6 " "l 2 2h( * * , ,F66""" L !- 0F < & &v . ..7v77888r(c|sdSt|tjr|St|tr;tjtjtj|St|trtj|Stdt|)N) voice_name)prebuilt_voice_config) voice_configzUnsupported speechConfig type: ) rr SpeechConfigr{r VoiceConfigPrebuiltVoiceConfigrir}r@rrs r&t_speech_configrMs  4VU/00 M  &"'";v"N"N"N       5   , ,V 4 44CT&\\CCDDDr(ct|tjr|}t|trtj|}|jtd|S)Nz&55f==M-9 F   r(c|sdStj|stj|r5tjtj||gSt$t|trt|St|trtj |S|S)N)r+callablefunction_declarations) inspect isfunctionismethodrrFunctionDeclaration from_callablerrrr{rir})r+rs r&t_toolr[s  4  7#3F#;#;  :  % 3 3 4      vw?? f % %%&$ : $ $V , ,, Mr(c|sgStjg}g}|D]}t||}||jrc|j\|xj|jz c_|}d|_|dr||~|||jr|||S)NrTT) exclude_none)rrr[rU model_copyr?r)r+r function_tooltoolstooltransformed_tool tool_copys r&t_toolsrds  I*2666- %''dfd++#  0 '1=++  2 ++%//11 *. '   T  2 2 " ,,y ! ! ! %&&&(  LL ,r(r c&t||dS)NcachedContents)r-)r9)r+r s r&t_cached_content_namergs  D( T!M  H   J <c<< = ==r(destct|trtjdi|}|St|trh|drtjd|S|drtjd|St d|t|tjr|St d|) NrlrmrnrorprqzUnsupported destination: r")r{rirBatchJobDestinationrr3r@r)rs r&t_batch_job_destinationrEsd9  $ , ,t , ,D K$9 w ;  &  ! !;  & 9499 : ::tU6779 K 777 8 88r(c|didg}|s|S|D]9}|di}|sd|vr|d|d<n:|S)NinlinedResponsesrK embeddinginlinedEmbedContentResponses)rPr)rinline_responsesrKinner_responses r&t_recv_batch_job_destinationr^sXX0"5599"  K"  h\\*b11N n$$-1XX6H-I-Id )* e% +r(cP|jsEd}tj||r|ddSt d|dd}tj||r|ddS|r|St d|d)Nzbatches/[^/]+$r0rzInvalid batch job name: .z:^projects/[^/]+/locations/[^/]+/batchPredictionJobs/[^/]+$)r5rxryrAr@isdigit)r+r  mldev_patternvertex_patterns r&t_batch_job_nameros ;%M x t$$; ZZ__R  9$999 : ::P.Xnd##9 ::c??2  ||~~9 K 7777 8 88r(statecv|dkrdS|dkrdS|dkrdS|dkrdS|d krd S|d krd S|d krdS|S)NBATCH_STATE_UNSPECIFIEDJOB_STATE_UNSPECIFIEDBATCH_STATE_PENDINGJOB_STATE_PENDINGBATCH_STATE_RUNNINGJOB_STATE_RUNNINGBATCH_STATE_SUCCEEDEDJOB_STATE_SUCCEEDEDBATCH_STATE_FAILEDJOB_STATE_FAILEDBATCH_STATE_CANCELLEDJOB_STATE_CANCELLEDBATCH_STATE_EXPIREDJOB_STATE_EXPIREDr")rs r& t_job_staters ''' " " %%%   %%%   '''  $$$   '''  %%%   Lr(g?g4@g @g?structc|dx}rd|vr|}d}t}|ddkr|tkrtd|d||d|i }t j|||z }t|tzt}|ddk|d x}rtd|d |d ||d S|S)Nr z /operations/gdoneTz Operation z timed out. GET) http_methodpath request_dicterrorz failed with error: z. rK) rP!LRO_POLLING_INITIAL_DELAY_SECONDSLRO_POLLING_TIMEOUT_SECONDS RuntimeErrorrequesttimesleepminLRO_POLLING_MULTIPLIER!LRO_POLLING_MAXIMUM_DELAY_SECONDS)rErr  operation total_seconds delay_secondsrs r&t_resolve_operationrsJjj   dn&<&<$*IM5M --  4 ' ' 4 4 4FFF9FFGGG$$$R%i j}$m 0 0 +m --  4 ' ' g&&&u  Ft F F F F9 F F   == $ $$ Mr(ct|tjr|j}nRt|tjr|j}n0t|tjr|j |jj}nd}|tdt|ts tdt|d| drZ| dd}tjd|}|td||d }n0| dr| dd}|S) NzFile name is required.z"Could not convert object of type `z` to a file name.zhttps://rtr z [a-z0-9]+z&Could not extract file name from URI: r)rrrr VideorGeneratedVideovideor@r{rrr3rArxrygroup)r suffixrys r& t_file_namersPT5:&& 9DDtU[)) 8DDtU122 z Z^dd d \ - . .. D#   JT$ZZJJJ   __Z  # ZZ ! !! $F H[& ) )E } FFF G GG ;;q>>DD x  # ::h   "D +r(statusc|dkrtjjS|dkrtjjS|dkrtjjS|dkrtjjStjD]}t |j|kr|cS|S)NSTATE_UNSPECIFIEDCREATINGACTIVEFAILED)rJobStaterrrrrr)rrs r&t_tuning_job_statusrs """ > //  > ++  > --  > ** U[  V # #  $ Mr(ct|trtj|St |tjr|St dt|d)NCould not convert input (type "z") to `types.Content`)r{rirrr}rr@r)rs r&t_content_strictrsi = ' ' 0 00w .. N N$w--NNN  r(cdt|tr d|DSt|gS)Nc,g|]}t|Sr")rrs r&rz%t_contents_strict..s! > > >' W % % > > >r()r{r rrs r&t_contents_strictrs;(##( > >X > > >> X & & ''r(turns turn_completec|tj|S tjt||S#t$r&}t dt |d|d}~wwxYw)N)rr)rrrz") to `types.LiveClientContent`)rLiveClientContentr Exceptionr@r)rres r&t_client_contentrs  ]  " ? ? ??   "///#        $$u++ $ $ $     s#= A-!A((A-inputc|std| tjt|S#t$r&}tdt |d|d}~wwxYw)Nz#A tool response is required, got: rrz$") to `types.LiveClientToolResponse`)r@rLiveClientToolResponserrr)rrs r&t_tool_responsers E CECC D DD  '/5III        )$u++ ) ) )     s"9 A)!A$$A)metricsc0g}|D]}t|tr6 tj|}n#t j$rYnwxYwt|tjr1|}ddg|d<||i}ddg|d<t|dg }|dkri|d<n|dkri|d<n| d r| d d }d |i|d <njt|drH|jrAd|ji}t|dg}|r||d<t|dg} | rd| i|d<||d<ntd||||S)a Prepares the metric payload for the evaluation request. Args: request_dict: The dictionary containing the request details. resolved_metrics: A list of resolved metric objects. Returns: The updated request dictionary with the prepared metric payload. AVERAGESTANDARD_DEVIATIONaggregation_metricsr rexact_match_specr bleu_specrougerr^ rouge_typer prompt_templatemetric_prompt_templatejudge_model_system_instructionsystem_instructionreturn_raw_outputcustom_output_format_configpointwise_metric_specz0Unsupported metric type or invalid metric name: )r{rir UnifiedMetricr}rrr?rgetvrrr3replacehasattrrr@) rmetrics_payloadmetricunified_metric_payloadmetric_payload_item metric_namerpointwise_specrrs r& t_metricsr'sIO4242 FD ! ! &55f==&&'    $  FE/ 0 0171B1B1D1D  9 45 5666,.  4/0 &**0022k  % %24.// & +-K((  ! !' * *  ((b11 -9:,FL)) 6, - - &2H 2F4JK! 56    D1C.- . ()    !#4; .6 78F344 O+ O O   01111 s>AAcd|vr|dkpd|vS)Ngeminizgemini-embedding-001maasr")r:s r&t_is_vertex_embed_content_modelrms(%;E%;; 5 r(r)NT)__doc__base64collections.abcrrrrrrVrnloggingrxsysrr builtin_typesrBrrr r r r _mcp_utilsrrrrr TYPE_CHECKINGrrrr^r getLoggerrQ version_info UnionTyperArDr_UnionGenericAliastyping_extensionsmcprr mcp.typesrrTyper,rmetric_name_sdk_api_maprmetric_name_api_sdk_map BaseApiClientrr/r9rDrrJrrrVrXrsrxFunctionResponseOrDictr|rrBlobImageUnionDictrrr BlobOrDictrrrrrrrr ContentDict ContentTyperrContentListUnionContentListUnionDictrrrrrr r2r7SchemaUnionDictr:rESpeechConfigUnionDictrJrMSpeechConfigOrDictrQr[rdrgBatchJobSourceUnionDictrvr|EmbeddingsBatchJobSourceOrDictr~rBatchJobDestinationOrDictrrrirrrrrrrrrrrrrr ContentOrDictrrrrrrMetricSubclassrrr"r(r&rs#  )(( --------   EEEEEEEEEEEEEEEE******......$$$$$$   7 8 8w$., 78,0,,))))))  333333'''''''"%FK%%%'6;555555))))))) GGG $  ML,C,I,I,K,KLLL'( ZZZ  %ZZ Z !$ Z  ZZZZzK-cc4 ) 8<       ' >)25 c]";s;uz;;;;& 3       5 $-. 0 5  % ! 5 5 5 5 )40H+II J %*)ej4Au/AEJAAAAAu'AEJAAAAC%-.C5:CCCC<   d5&')NNO   %*    (738#456 d5'()&EM5#4e6IIJ * ; 147 8* ]****8  % D'e.?)@+MN  49d5=) )*    4i e$e&@%-OPi %- iiiiX8w18d8888v9s9t9999 M  M!+";< M  M M M M"*.l( "-1 l(l(l(  l( [. /l( 7% &l(  l( !S* l( l(l(l(l(\2 2$[%>?2 \2222: %', -w!"-9 [. /-9 %', --9el-9-9-9-9`E %-s2 3E e !EEEE&   $  e !       %/2 eEJO$%*  %/3Cy %*:N+";N3N3NNNN81  %81  &81 81818181v>  %>  -> #>>>>.9 U44 59 99992tCH~$sCx."9[69c9c9999$ss&%(!$(!#)3:3E< 5ej%+u7KKL MD  enc.A(B     e1em(HU0153FFG( %-((((     hu*+U-@@A          *  $-. 0   !     (C %& 'C $sCx.CCCCL34s4 D D  D