쎒jN UdZddlZddlmZmZddlmZmZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZmZmZddlmZddlmZdd lmZdd lmZejrddlZ ddl!Z!dd lm"Z"dd lmZdd lmZe j#dZ$e j%dkrej&Z'ejej&fZ(ddlm)Z)nej*Z'ejfZ(ddl+m)Z)ejr ddl,m-Z.ddl/m0Z1neZ.ej2e3d<eZ1ej2e3d<ddddZ4de45DZ6ddde"j7de8de8de9d e8f d!Z:de"j7d"e8d e8fd#Z;d$e"j7d%ed e?ej>fd(Z@d$e"j7d"e8d ee8fd)ZAd*ed ejBfd+ZCd,ejDd ejEfd-ZFd.eejDeejDfd e?ejEfd/ZGd0eejHe?ejHfd e?ejBfd1ZId2ejHd ejBfd3ZJd2ejHd ejBfd4ZKd2ejLd ejBfd5ZMd6eejNd ejOfd7ZPd8eee?ejNejNe?ejOfd e?ejOfd9ZQd:eeee8efd ee?ejRfd;ZSeejTejUejNfZVdee?ejTe?ejUeVfd ee?e8e?ejTffd?ZXd>eeejYejZejTfd e?ejTfd@Z[dAej>d dfdBZ\dCed dfdDZ]dAedee"j7d dfdEZ^ dsdFddGdAej>dee"j7dHeej>dIefdNZddee"j7dCeejcefd eejafdOZedCeejfefd eejgfdPZhdCejid eejgfdQZjde"j7dCed eeej0effdRZkde"j7dCe?ed e?ej0fdSZlde"j7dTe8d e8fdUZmde"j7dVejnd ejofdWZpde"j7dVejqd ejrfdXZsdYee8ejtfd ejufdZZvdYewe8efd ewe8effd[Zxde"j7dTe8d e8fd\Zyd]e8d e8fd^Zzd_Z{d`Z|daZ}dbZ~d$e"j7dcej>d efddZdTeee8ejejejfd e8fdeZdfe8d eeje8ffdgZdeeejejfd e?ejTfdiZ dtdjeeeejejfdker)DsLLLDAq1aLLL)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_namer;GsOJ  " "&;#>#>#> ? ??$# 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.r3models/ publishers/r2r z/models/zpublishers/google/models/z tunedModels/) ValueErrorr7r5split)r-r< publishermodel_ids r(t_modelrFs + ) * ** 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)r7)rGrHs r( t_models_urlrLs4  ' ' X X ]r*responsecf|sgS|d}||S|d}||S|d}||S|d|dgStdtd|gS)NrJrKpublisherModels httpHeaders jsonPayloadz!Cannot determine the models type.z1Cannot determine the models type for response: %s)getloggerwarningdebug)rMrJ tuned_modelspublisher_modelss r(t_extract_modelsrXs  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)NrAr3r4r2r@z/publishers/google/)rFr5r7r8r9)rGr<s r(t_caches_modelrZs *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)r[ image_format save_paramsimage_io image_bytesrjs r( pil_to_blobrzs, $+ 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: ) rB isinstancerkrFunctionResponsemodel_validater TypeErrortype)r{s 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%rMs r( z(t_function_responses..)s! M M Mh  ) ) M M Mr*)rBr}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..2 + + +TF4LL + + +r*)r}listr)rs r(t_blobsr.s8t + +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.r[rz6Could not parse input as Blob. Unsupported blob type: )rBrrrur}rkr __class____name__rt PIL.ImageImage ImportErrorrzrr)rPIL PIL_Images r(rr7s * ( ) ))T5:&& Kd+ : $ $T * ** '--////)/ii iiiD)!? = > >> :  DN  K KKdGG Z & &t , ,,  #GGG Z%."?"?"E"E F F FFFFGT5:&& K '--////)/ii iiiD)!.rr*)r}rrBrrs r(t_partsr}s^  ]z%..]u] 2 3 33t + +U + + ++ 5MM?r* predictionsc |sdSg}|D]k}|drT|tjtj|dd|ddl|S)Nr[gcsUri imageBytes)gcs_uriry)r[)rRappendrGeneratedImager)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)rBrrContentr}rkrrrr 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}rrr7rkrrrrrrrSrT)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.rr0c\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)NTFr[r)r}rrrrrrkrrrrrrrtrrrrs 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_part s! !!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_parts s& 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_partsw ,''L}\""oo6G&H&HHH|,,,,,**63DEEE*^r*rzUnsupported content type: )r}rrBrr r PartUnionDictrrrboolrrrrrkrr) 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)rRremovelenitems)ritemkeyvals r(handle_null_fieldsrCsX 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: )rBrs r("_raise_for_unsupported_schema_typer~s7v77888r*c|r@|js;|ds|drtddSdSdS)NadditionalPropertiesadditional_propertiesztadditionalProperties is only supported in Gemini Enterprise Agent Platform mode, , not in Gemini Developer API mode.)r7rRrB)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_schemar0c|ddx}4|dd}t|vriSt||S)z@Returns the processed `sub_schema`, resolving its '$ref' if any.rNrrr)poprCidprocess_schema)rrefr-rrrs r(_recursez process_schema.._recurses|~~fd+++8 '**2./j J- - -  )-  r*rc&g|] }|Sr$r$)r%rrs r(rz"process_schema..s#EEE xx ++EEEr*rconstSTRINGenumzLiteral values must be strings.OBJECT propertiesr rrrARRAYrrc&g|] }|Sr$r$)r%prefixrs r(rz"process_schema..;s#GGGFxx//GGGr*)setr addrRrr rr rrorCr StringDictrr}rvaluerupperrBrrkeysrk)rr-rrr from_nameto_namer_rr r schema_typerrname additionalrprefixesrs ```` @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..Os"8883v|$$888r*)rceZdZUed<dS)"_process_enum..Placeholder placeholderNr __module__ __qualname____annotations__)enum_to_processsr( Placeholderr)Ss      r*r0rr*)r}rintrrr"rrrr BaseModelmodel_json_schemar rSchemar) rr-is_integer_enumr' str_membersstr_enumr0 enum_schemar/s @r( _process_enumr9?sE/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..`s=**"jc******r*)r}rkrrs r(_is_type_dict_str_anyr<\sC 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(r0r@sr*r0rr*r)r}rkr<r rr4rrr9rr model_dumprr issubclassrr2r3VersionedUnionTypetyping get_origin _UNION_TYPESrB)r-rrr0s ` r(t_schemarGesZ  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 VoiceConfigPrebuiltVoiceConfigrkrrBrrs r(t_speech_configrOs  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|r6tjtj||dgStjr7 ddl m}t||rt|Sn#t$rYnwxYwt|trtj|S|S)NT)r-callableuse_json_schemafunction_declarationsrr)inspect isfunctionismethodrrFunctionDeclaration from_callabler_is_mcp_loaded mcp.typesrrrr}rkr)r-r_McpTools r(t_toolras  4 7#3F#;#; :  % 3 3 4          ,,,,,,  * **!&)))*     d - : $ $V , ,, -s7$B B*)B*c|sgStjg}g}|D]}t||}||jrc|j\|xj|jz c_|}d|_|dr||~|||jr|||S)NrWT) exclude_none)rrrarX model_copyrAr)r-r function_tooltoolstooltransformed_tool tool_copys r(t_toolsrjs  I*2666- %''dfd++#  0 '1=++  2 ++%//11 *. '   T  2 2 " ,,y ! ! ! %&&&(  LL ,r*r"c&t||dS)NcachedContents)r/)r;)r-r"s r(t_cached_content_namerms  D( T!M  H   J <c<< = ==r*destct|trtjdi|}|St|trh|drtjd|S|drtjd|St d|t|tjr|St d|) NrrrsrtrurvrwzUnsupported destination: r$)r}rkrBatchJobDestinationrr5rBr)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)NinlinedResponsesrM embeddinginlinedEmbedContentResponses)rRr )rinline_responsesrMinner_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/[^/]+$r2rzInvalid batch job name: .z:^projects/[^/]+/locations/[^/]+/batchPredictionJobs/[^/]+$)r7r~rrCrBisdigit)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. rM) rR!LRO_POLLING_INITIAL_DELAY_SECONDSLRO_POLLING_TIMEOUT_SECONDS RuntimeErrorrequesttimesleepminLRO_POLLING_MULTIPLIER!LRO_POLLING_MAXIMUM_DELAY_SECONDS)rGrr" 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://rzr z [a-z0-9]+z&Could not extract file name from URI: r)rrrr"VideorGeneratedVideovideorBr}rrr5rCr~rgroup)r"suffixrs 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}rkrrrrrBr)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 ExceptionrBr)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`)rBrLiveClientToolResponserrr)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"r exact_match_specr! bleu_specrouger r` rouge_typer"prompt_templatemetric_prompt_templatejudge_model_system_instructionsystem_instructionreturn_raw_outputcustom_output_format_configpointwise_metric_specz0Unsupported metric type or invalid metric name: )r}rkr UnifiedMetricrrrrArgetvrtr5replacehasattrrrB) 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*c|Srr$rs r( t_json_schemarvs -r*r)NT)__doc__base64collections.abcrrrrrrYrploggingr~sysrr builtin_typesrDrrr r r r r`rrrrrr TYPE_CHECKINGrrrr getLoggerrS version_info UnionTyperCrFr_UnionGenericAliastyping_extensionsmcprrr_rrTyper.metric_name_sdk_api_maprmetric_name_api_sdk_map BaseApiClientrr1r;rFrrLrrrXrZrurzFunctionResponseOrDictr~rrBlobImageUnionDictrrr BlobOrDictrrrrrrrr ContentDict ContentTyperrContentListUnionContentListUnionDictrrrrrr r4r9SchemaUnionDictr<rGSpeechConfigUnionDictrLrOSpeechConfigOrDictrSrarjrmBatchJobSourceUnionDictr|rEmbeddingsBatchJobSourceOrDictrrBatchJobDestinationOrDictrrrkrrrrrrrrrrrrrr ContentOrDictrrrrrrMetricSubclassrrrr$r*r(rs  )(( --------   EEEEEEEEEEEEEEEE******......$$$$$$   7 8 8w$., 78,0,,)))))) 333333'''''''"%FK%%%'6;$  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$%4  %/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.CCCCL34##r*