Li 4UdZddlZddlmZmZddlmZmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZmZmZddlmZddlmZej8rddlZddlZdd l m!Z!dd l mZdd l mZe jDd Z#e jHd k\r+ejJZ&ej,ejJfZ'ddlm(Z(nejRZ&ej,fZ'ddl*m(Z(ej8r ddl+m,Z-ddl.m/Z0n/eZ-ejbe2d<eZ0ejbe2d< ddl+m,Z-ddl.m/Z0ddddZ4e4jkDcic]\}}|| c}}Z6dede7ejpde9fdZ:ddde!jvdede!jvd"eeeBejeBejeYfdeeBe<eBejffd?Z[d>eeejejejfdeBejfd@Z^dAejddfdBZ_dCeddfdDZ`dAedee!jvddfdEZa dpdFdGdAejdee!jvdHeejdIe9ddf dJZbdKedee!jvdejfdLZddCeejefde(ejfdMZfdee!jvdCeejefdeejfdNZgdCeejefdeejfdOZjdCejdeejfdPZlde!jvdCedeeej^effdQZmde!jvdCeBedeBej^fdRZnde!jvdSeeeejejfdeBejfdhZ dqdieeeejejfdje9dejfdkZdleejeejfdejfdmZdneBejdeBeyez#_is_duck_type_of..^sFEGC'Fc36K|]}t|ywr&r')r)f obj_privates r+r,z#_is_duck_type_of..dsPA Q/Pr-) isinstancedictr(allr$pydantic BaseModeltype ValueError)r r! all_matchedr0s` @r+_is_duck_type_ofr9KsT'#~"> FS5E5EFF+ C););<EkPc9O9OPPk   s.B BB)collection_hierarchy_depthclient resource_namecollection_identifierr;c|j|d xr|d|jddz|k(}|jr|jdr|S|jdrd|jd|S|j|drd|jd|jd|S|r"d|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_namerIjs!J  " "&;% ??$# #1]O 4 : :3 ?! C # $ ' __ ,   ! !, /(- 99  ! !%:$;1"= >( FOO3DAm_ UU -( FOO3DAF[E\\]^k]l mm +%&a  77 modelcn|s tdd|vsd|vsd|vr td|jr[|jds"|jds|jdr|Sd |vr|jd d \}}d|d |Sd |S|jdr|S|jd r|Sd|S)Nzmodel is required.z..?&zinvalid model parameter.rAmodels/ publishers/r@r z/models/zpublishers/google/models/z tunedModels/)r7rErCsplit)r<rK publishermodel_ids r+t_modelrTs  ) ** U]cUlcUl / 00 __ %   I &   M * l !KKQ/i9+XhZ 88( 00  " l   . ) lug rJ api_client base_modelsc*|jr|ryy|ryy)Nzpublishers/google/modelsmodels tunedModels)rE)rUrVs r+ t_models_urlrZs '   rJresponsec*|sgS|jd}||S|jd}||S|jd}||S|jd|jdgStjdtjd|gS)NrXrYpublisherModels httpHeaders jsonPayloadz!Cannot determine the models type.z1Cannot determine the models type for response: %s)getloggerwarningdebug)r[rX tuned_modelspublisher_modelss r+t_extract_modelsrfs  I/7||H/E&  M5=\\-5P, 9A:! ll=!- ,,} % - I NN67 LLDhO IrJct||}|sy|jdr+|jrd|jd|jd|S|jdr+|jrd|jd|jd|S|S)NrPrArBr@rOz/publishers/google/)rTrCrErFrG)rUrKs r+t_caches_modelrhs *e $%   m$)<)< J&&'{:3F3F2GqP  "z':': z))*+j6I6I5JJ]^c]d ee LrJimgc ddl}|j}tj}|t ||j s|jdk(r|j|dd}n|j|dd}|jd|j}tj||S#t$rd}YwxYw) NrRGBAPNG)formatz image/pngJPEGz image/jpeg) mime_typedata) PIL.PngImagePluginPngImagePlugin ImportErrorioBytesIOr1 PngImageFilemodesaveseekreadrBlob)riPILrrbytesiororps r+ pil_to_blobr~ s''N JJL' S.55 6 V HHWUH#IHHWVH$I ,,q/ $ id 33! NsB00 B>=B>function_responsec|s tdt|trtjj |St |tjr|Stdt|)Nzfunction_response is required.zOCould not parse input as FunctionResponse. Unsupported function_response type: ) r7r1r2rFunctionResponsemodel_validater9 TypeErrorr6)rs r+t_function_responser9sp  5 66!4(  ! ! 0 01B CC)5+A+AB   $$():$;#< > rJfunction_responsesc|s tdt|tr|Dcgc] }t|c}St|gScc}w)Nz function_responses are required.)r7r1r r)rr[s r+t_function_responsesrIsJ  7 88"H-:L Mh  ) MM  2 3 44 NsAblobscvt|tr|Dcgc] }t|c}St|gScc}wr&)r1listt_blob)rblobs r+t_blobsrWs4t%* +TF4L ++ 5M? ,6rc|s tdt|tjr|St |t rtjj |Sd|jjjvr4 ddl }|jj}|t ||r t|Stdt|#t$rd}Y=wxYw)Nzblob is required.imagerz6Could not parse input as Blob. Unsupported blob type: )r7r9rr{r1r2r __class____name__lower PIL.ImageImagersr~rr6)rr| PIL_Images r+rr`s ( ))dEJJ' Kd :: $ $T ** ''--//))//iD)!<  >tDzlK  isC CCct|}|jr|jjdr|Std|j)Nzimage/Unsupported mime type: rrorCr7rs r+ t_image_blobrz@ $ ^^11(; K,T^^,>?@@rJct|}|jr|jjdr|Std|j)Nzaudio/rrrs r+ t_audio_blobrrrJpartc| tdt|trtj|St |tj rX|jr |js tdtjj|j|jSt|tr tjj|St |tjr|Sd|jj j#vrH ddl}|j&j&}|+t||rtjt+|Std t-|#tj$r6tjtjj|cYSwxYw#t($rd}YwxYw) Nzcontent part is required.)textz$file uri and mime_type are required.)file_uriro) file_datarr) inline_datazUnsupported content part type: )r7r1strrPartr9Fileurirofrom_urir2rr4ValidationErrorFileDatarrrrrrsr~r6rr|rs r+t_partrsg \ 0 11c ::4  dEJJ' 884>> = >> ::  DNN  KKdG ZZ & &t ,,dEJJ' K ''--//))//iD)!< ZZK$5 664T$ZLABB  # #G ZZ%.."?"?"E FFG is%7E7G7AG?G GGpartsc|t|tr |s tdt|tr|Dcgc] }t|c}St|gScc}w)Nzcontent parts are required.)r1rr7r)rrs r+t_partsrsP  ]z%.u 2 33t%* +TF4L ++ 5M? ,sA predictionsc |syg}|D]Z}|jds|jtjtj|dd|dd\|S)NrgcsUri imageBytes)gcs_uri image_bytes)r)r`appendrGeneratedImager)rimages predictions r+t_image_predictionsrsq   & j~~g mm   KK$W-h7(1,?   -rJcontentc| tdt|tjr|St |t r tjj |St|tjr tjt|gSt|tjr:|jrtj|gStj|gStj|S#tj$rJt|}|jrtj|gcYStj|gcYSwxYw)Nzcontent is required.r)r7r9rContentr1r2rr4rr function_call ModelContent UserContentrr)r possible_parts r+ t_contentrs' _ + ,,gu}}- N ]] ) )' 22guzz*   F7O#4 55guzz*   '+  gY /    ))  # #Wom ( (   M? 3  7sC>>AEEEcontentsct|tr|Dcgc] }t|}}n t|g}|jrg}|D]}|t|trt j j|}|jB|jD]?}|jr|j|j+tjdA|S|Scc}w)Nz/Non-text part found, only returning text parts.) r1rrrEr2rrrrrrrarb)r<rrtransformed_contents text_partsrs r+t_contents_for_embedrs$>FG7Ig.GG%h/0 __J' Q   gt $MM009' == $mm Qdyy *nnNP  Q Q  %HsCcV|t|tr |s tdt|ts t|gSg}g}dtt j tfdtt j fd}dt jdtfddtt jdtffd dtt jd tt jddffd dtt jd tt jd t j ddffd }|D]}t|t jst|trQ||t|tr&|jt j| l|j|~||r ||||t|tr/|jt jj!|tdt#||||S)Nzcontents are required.rr"c^t|ts4t|tjst|tj ryt|t r$|sy tj j|yd|jjjvr* ddl}|jj}| t||ryy#tj$r= tjj|Yy#tj$rYYywxYwwxYw#t $rd}YpwxYw)NTFrr)r1rr9rrrr2rr4rrrrrrrrsrs r+_is_partzt_contents.._is_part s 4 D%** - D%** - $  !!$'$..))//11IIOO   :dI#> % % %  .. ' ' -''    s<C !D DDDDDD D,+D,c|j Sr&)r)rs r+ _is_user_partz!t_contents.._is_user_part2s!! !!rJrc,tfd|DS)Nc3.K|] }|ywr&)r)rrs r+r,z6t_contents.._are_user_parts..6s5t}T"5s)r3)rrs r+_are_user_partsz#t_contents.._are_user_parts5s 5u5 55rJresultaccumulated_partsc|sy|j|rtj|ntj|g|ddy)Nr)rrrr)rrrs r+$_append_accumulated_parts_as_contentz8t_contents.._append_accumulated_parts_as_content8sM   MM , -  12   &7 8 arJ current_partct|}||k(r|j|y|||g|ddyr&)rr)rrrrrrs r+_handle_current_partz(t_contents.._handle_current_partEsH ,'L\"o6G&HH|,*63DE*^rJrzUnsupported content type: )r1rr7rr r PartUnionDictrrrboolrr9rrr2rr6) rrrrrrrrrs @@@r+ t_contentsrs *Xt4X - .. Hd # h   "&(*# %%%s* +#$$%#J"%**""6T%**-6$6 5==! ejj)    ,5==! ,ejj) ,'' , ,& Eg/:gt3L*63DE GT " e''g67 g ' 6#4g> GT " mmEMM009: 3DM?C DD E'v/@A -rJschemac |jdddk(r d|d<|d=yd|vrl|dD]c}d|vs|ddk(sd|d<|djddit|ddk(s>|ddjD] \}}|||< |d=eyy) 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" } } r6NnullTnullableanyOfr r)r`removelenitems)ritemkeyvals r+handle_null_fieldsrlsX ZZ'F:v&w 4DLF2!zw/0 vg 1 $ /!,224 hc3F3K WorJoriginctd|)z2Raises an error if the schema type is unsupported.Unsupported schema type: )r7rs r+"_raise_for_unsupported_schema_typers.vh788rJc~|r;|js.|jds|jdr tdyyy)NadditionalPropertiesadditional_propertiesz8additionalProperties is not supported in the Gemini API.)rEr`r7)rr<s r+'_raise_for_unsupported_mldev_propertiesrsE oo **+ , ZZ/ 0 O PP1 rJTorder_propertiesdefsrcR|jddk(r|d=t|dD]\}}|j|dx}|||<!9|jdijD]\}}t |t ||jddx} &|j | jdd d tjd tjffd } |jd x} | Dcgc] }| | c}|d <y|jd} t| tr | j} t| tr| j} |jd} | | dk(r | g|d<|d=n td| dk(r|jdx}jt!|jD]\}}| |||<t#|jdkDr"r d|vrt!|j%|d<|jdx}t|t&r | ||d<yyy| dk(rL|jdx} | ||d<|jdx}|Dcgc] }| | c}|d<yyycc}wcc}w)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' } titlePlaceholderLiteralEnum))rr)any_ofr) prefix_items prefixItems)property_orderingpropertyOrderingNz$defsr$refdefs/ sub_schemar"c||jddx}|jdd}t||S)z@Returns the processed `sub_schema`, resolving its '$ref' if any.rNrrr)poprQprocess_schema)rrefr<rrs r+_recursez process_schema.._recurse sF~~fd++8 '*2./j:vt>NO rJrr6constSTRINGenumzLiteral values must be strings.OBJECT propertiesr rrrARRAYrr)r`rrrrrupdaterQr StringDictr1rvaluerupperr7rrkeysr2)rr<rr from_nameto_namer_rrrr schema_typerr name additionalrprefixesprefixs ``` r+rrsF ZZ44w)&&9 i It,,9fWo \ ::gr "D : fd5E V ZZ %%c2 MM$syy)"-./7--'2D2D 7##f0>DE x +EF7O  6"+ T"##K S!##%K **W % hwfVn / 8 99Hjj.. ;":#3#3#560 $ #J/ 40 j !A % .&*:??+<&="#jj!788 E J %)1*)=%& &F gG$$1 fWoJJ}--:>FGFx/Gf];EFLHs JJ$rc .d}|D]g}t|jtrd} t|jtr;t d|j dt |j||r@|Dcgc]}t|j}}t|j|t}|Gfddtj}|j}t|||dd }tjj|Scc}w) NFTz Enum member z( value must be a string or integer, got )r6ceZdZUWed<y)"_process_enum..Placeholder placeholderNr __module__ __qualname____annotations__)enum_to_processsr+ Placeholderres   rJr"r r)r1rintrrrr6rrr4r5model_json_schemarrSchemar) rr<is_integer_enummember str_membersstr_enumr" enum_schemar!s @r+ _process_enumr+Qs/f&,,$o  c *   &6<< ! #  /3783v||$8K8DMM;S9HO!H&&!--/+f%L)-8+  $ $[ 119s8DcJt|txrtd|DS)zEVerifies the schema is of type dict[str, Any] for mypy type checking.c3<K|]}t|tywr&)r1r)r)rs r+r,z(_is_type_dict_str_any..rs*"jc*s)r1r2r3rs r+_is_type_dict_str_anyr.ns, FD ! c*&,*'rJc<syttr6tr+t|tj j Sttr t|Sttj rqttt j k(r tjd}t||tj j |SttsettrUttj r;j#}t||tj j |Stts;tts+tt$st'j(t*vr^Gfddtj }|j#}t|||dd}tj j |St-d)NT) exclude_unsetceZdZUWed<y)t_schema..PlaceholderrNrrsr+r"r2s rJr"r rr)r1r2r.rrr%rrr+r9r model_dumprr6 issubclassr4r5r$VersionedUnionTypetyping get_origin _UNION_TYPESr7)r<rrr"s ` r+t_schemar9ws  "7"?66" << & &v ..!  ((fell+ F|tELLN++)0   T  2F66" << & &v .. V\ * VT " VX// 0  % % 'F66" << & &v ..& FD ! F. /   6 "l 2h(( * * ,F66" L !- 0F << & &v ...vh788rJch|syt|tjr|St|tr>tjtj tj |St|trtjj|Stdt|)N) voice_name)prebuilt_voice_config) voice_configzUnsupported speechConfig type: ) r9r SpeechConfigr1r VoiceConfigPrebuiltVoiceConfigr2rr7r6rs r+t_speech_configrAs  fe001 M   &&"'";";v"N       , ,V 444T&\NCDDrJct|tjr|}t|trtjj |}j td|S)Nzr1r2rmulti_speaker_voice_configr7)r speech_configs r+t_live_speech_configrEs]fe001M&&55f=M--9 F  rJcl|sytj|stj|r6tjtj j ||gStt|tr t|St|trtjj|S|S)N)r<callablefunction_declarations) inspect isfunctionismethodrrFunctionDeclaration from_callablerr9rr1r2r)r<rs r+t_toolrOs   7#3#3F#; ::  % % 3 3 4   /@ f %%&$ :: $ $V ,, MrJc6|sgStjg}g}|D]Z}t||}||jr,|j |xj|jz c_J|j |\|jr|j ||S)NrH)rrrOrIr)r<r function_tooltoolstooltransformed_tools r+t_toolsrUs  I**26- % 'dfd+#  0 011=++  2 2 +  %& '(( LL ,rJrct||dS)NcachedContents)r>)rI)r<rs r+t_cached_content_namerXs ?N D( T!M .A- 6   J =A- J  J#t    55#s ~~g  ! !%   ! !  !  ! ! )#/00rJct|trtjdi|}t |tjr7t |j du|jdug}|dk7r td|Stdt|)Nr r[zUnsupported source type: r) r1r2rEmbeddingsBatchJobSourcer9rgr\rer7r6)r<rYris r+t_embedding_batch_job_sourcerm.sT  ( ( /3 /Cc599: D( T!M  H  J 0c < ==rJdestct|trtjdi|}|St|tr^|j drtjd|S|j drtjd|St d|t|tjr|St d|) Nr]r^r_r`rarbzUnsupported destination: r)r1r2rBatchJobDestinationrrCr7r9)rns r+t_batch_job_destinationrqEsd  $ $ ,t ,D K$ w  & &  !  & & 24&9 ::e778 K 07 88rJc|jdijdg}|s|S|D]2}|jdi}|sd|vs|jd|d<|S|S)NinlinedResponsesr[ embeddinginlinedEmbedContentResponses)r`r)rninline_responsesr[inner_responses r+t_recv_batch_job_destinationrx^sXX0"599"  K" h\\*b1N n$-1XX6H-Id )* +  +rJc*|js;d}tj||r|jddSt d|dd}tj||r|jddS|j r|St d|d)Nzbatches/[^/]+$r@rzInvalid batch job name: .z:^projects/[^/]+/locations/[^/]+/batchPredictionJobs/[^/]+$)rErematchrQr7isdigit)r<r mldev_patternvertex_patterns r+t_batch_job_nameros %M xx t$ ZZ_R  1$q9 ::P.XXnd# ::c?2  ||~ K /vQ7 88rJstatecZ|dk(ry|dk(ry|dk(ry|dk(ry|d k(ry |d k(ry |d k(ry|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[ '' " %%  %%  ''  $$  ''  %%  LrJg?g4@g @g?structc|jdx}rd|vr|}d}t}|jddk7rt|tkDrtd|d||j d|i }t j |||z }t|tzt}|jddk7rt|jd x}rtd|d |d ||jd S|S)Nrz /operations/gdoneTz Operation z timed out. GET) http_methodpath request_dicterrorz failed with error: z. r[) r`!LRO_POLLING_INITIAL_DELAY_SECONDSLRO_POLLING_TIMEOUT_SECONDS RuntimeErrorrequesttimesleepminLRO_POLLING_MULTIPLIER!LRO_POLLING_MAXIMUM_DELAY_SECONDS)rUrr operation total_seconds delay_secondsrs r+t_resolve_operationrs jj  d n&<$*IM5M -- 4 ' 4 4Zv]9+FGG$$$R%i jj}$m 0 0 +m -- 4 ' g&&u&  tf0s9+ F  == $$ MrJct|tjr |j}nft|tjr |j }n?t|tj r%|j|jj }nd}| tdt|tstdt|d|jdrM|jdd}tjd|}|td||j!d }|S|jdr|jdd}|S) NzFile name is required.z"Could not convert object of type `z` to a file name.zhttps://rdr z [a-z0-9]+z&Could not extract file name from URI: r)r9rrrVideorGeneratedVideovideor7r1rr6rCrQr{r|group)rsuffixr|s r+ t_file_namersdEJJ' 99Dekk* 88De223 zz ZZ^^d d \ - .. D#   ,T$ZL8IJ  __Z ZZ !! $F HH[& )E } ?vF GG ;;q>D + x ::h  "D +rJstatusc`|dk(rtjjS|dk(rtjjS|dk(rtjjS|dk(rtjj StjD]}t |j|k(s|cS|S)NSTATE_UNSPECIFIEDCREATINGACTIVEFAILED)rJobStaterrrrrr)rrs r+t_tuning_job_statusrs "" >> / //  >> + ++  >> - --  >> * ** U[[ V #  MrJct|trtjj |St |tjr|St dt|d)NCould not convert input (type "z") to `types.Content`)r1r2rrrr9r7r6)rs r+t_content_strictrsR == ' ' 00/ N  )$w-8MN rJcvt|tr|Dcgc] }t|c}St|gScc}wr&)r1r r)rrs r+t_contents_strictrs8(#5= >' W % >> X & '' ?rturns turn_completec|tj|S tjt||S#t$r}t dt |d|d}~wwxYw)N)r)r)rrrz") to `types.LiveClientContent`)rLiveClientContentr Exceptionr7r6)rres r+t_client_contentrsv  ]  " " ??   " "/#    )$u+7$ $    s ; A"AA"inputc|std| tjt|S#t$r}tdt |d|d}~wwxYw)Nz#A tool response is required, got: )rrz$") to `types.LiveClientToolResponse`)r7rLiveClientToolResponserrr6)rrs r+t_tool_responserss  ;E7C DD  ' '/5I    )$u+7) )    s2 AAAmetricscg}|D]}i}ddg|d<t|dgj}|dk(ri|d<n|dk(ri|d<n|jd r|jd d }d |i|d <ndt |drJ|j r>d|j i}t|dg}|r||d<t|dg}|rd|i|d<||d<nt d||j||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_metricsrrexact_match_specr bleu_specrouger rouge_typerprompt_templatemetric_prompt_templatejudge_model_system_instructionsystem_instructionreturn_raw_outputcustom_output_format_configpointwise_metric_specz0Unsupported metric type or invalid metric name: )getvrrCreplacer(rr7r) rmetrics_payloadmetricmetric_payload_item metric_namerpointwise_specrrs r+ t_metricsr'sVO$2,.  4/0 &*002k  %24./ & +-K(  ! !' * ((b1 -9:,FL) 6, -&2H2H2F4J4JK! 56  1C.- . ()  !#4; .6 78F34 A+ O  01I$2J rJr&)NT)__doc__base64collections.abcrrrrrrJrtloggingr{sysrr builtin_typesr6rrr r r r _mcp_utilsrrrr TYPE_CHECKINGrr|r4rr getLoggerra version_info UnionTyper5r8r_UnionGenericAliastyping_extensionsmcprr mcp.typesrrTyper rsmetric_name_sdk_api_maprmetric_name_api_sdk_mapr6r5rr9 BaseApiClientrr#rIrTrZr rrfrhr{r~FunctionResponseOrDictrrrBlobImageUnionDictrrr BlobOrDictrrrrrrrr ContentDict ContentTyperrContentListUnionContentListUnionDictrrrrrr%r+SchemaUnionDictr.r9SpeechConfigUnionDictr>rASpeechConfigOrDictrErOrUrXBatchJobSourceUnionDictrfrjEmbeddingsBatchJobSourceOrDictrlrmBatchJobDestinationOrDictrprqr2rxrrrrrrrrrrrrr ContentOrDictrrrrrrMetricSubclassr)kvs00r+rs  ) -   EE*.    7 8w$..,, 7 78,00,,,) 3'"%FKK%'6;;5)$  -D,I,I,KLDAq1a4L#D););$<H'( Z  % %ZZ Z !$ Z  ZzK--cc4 )) 8<     '  >))25 c]"4S4UZZ42 33    5 $$--. 0 5  % ! 5 ))40H0H+II J %**))ejj4Au//AEJJAAu''AEJJAC%--.C5::C<   d5&&')<)NNO   %** (738#456 d5''()&EMM5#4#4e6I6IIJ * ; 1 147 8* ]]*8  % % D'e.?.?)@+MN  49d5==) )* 4i e$$e&@&@%--OPi %-- iX8w118d8v9s9t9 Q  Q!+";";< Q  Q"*.TH " TH   TH [.. /TH 7%% &TH  TH  THn2 2$[%>%>?2 \\2: %'', -w!!"-9 [.. /-9 %'', --9ell-9`E %--s2 3E e  !E&   $ $  e  !    % %/2 eEJJO$%*  % %/3Cy %**2N+";";N3N3N/1  % %/1  & &/1 /1d>  % %>  - -> ##>.9 U444 59 92tCH~$sCx."9[669c9c9$ss&%(!$(!#))3:3E3E< 5ejj%++u7K7KKL MD  ennc.A(B  e11emm(HU00153F3FFG( %--(    hu**+U-@-@@A       *  $$--. 0   !!  (3 %&& '3 $sCx.3S' GMs! ^ ^ ^^