jb >ddlZddlZddlmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd l mZdd l mZdd lmZmZejd Z d6d eeeefefdeeeefdeeeeefefdeeeffdZ d6d eeeefefdeeeefdeeeeefefdeeeffdZ d6d eeeefefdeeeefdeeeeefefdeeeffdZ d6d eeeefefdeeeefdeeeeefefdeeeffdZ d6d eeeefefdeeeefdeeeeefefdeeeffdZ d6d eeeefefdeeeefdeeeeefefdeeeffdZ d6d eeeefefdeeeefdeeeeefefdeeeffdZ d6d eeeefefdeeeefdeeeeefefdeeeffdZ! d6d eeeefefdeeeefdeeeeefefdeeeffdZ" d6d eeeefefdeeeefdeeeeefefdeeeffdZ# d6d eeeefefdeeeefdeeeeefefdeeeffdZ$ d6d eeeefefdeeeefdeeeeefefdeeeffdZ% d6d eeeefefdeeeefdeeeeefefdeeeffdZ& d6d eeeefefdeeeefdeeeeefefdeeeffdZ' d6d eeeefefdeeeefdeeeeefefdeeeffdZ( d6d eeeefefdeeeefdeeeeefefdeeeffd Z) d6d eeeefefdeeeefdeeeeefefdeeeffd!Z* d6d eeeefefdeeeefdeeeeefefdeeeffd"Z+ d6d eeeefefdeeeefdeeeeefefdeeeffd#Z, d6d eeeefefdeeeefdeeeeefefdeeeffd$Z- d6d eeeefefdeeeefdeeeeefefdeeeffd%Z. d6d eeeefefdeeeefdeeeeefefdeeeffd&Z/ d6d eeeefefdeeeefdeeeeefefdeeeffd'Z0 d6d eeeefefdeeeefdeeeeefefdeeeffd(Z1 d6d eeeefefdeeeefdeeeeefefdeeeffd)Z2 d6d eeeefefdeeeefdeeeeefefdeeeffd*Z3 d6d eeeefefdeeeefdeeeeefefdeeeffd+Z4 d6d eeeefefdeeeefdeeeeefefdeeeffd,Z5 d6d eeeefefdeeeefdeeeeefefdeeeffd-Z6 d6d eeeefefdeeeefdeeeeefefdeeeffd.Z7 d6d eeeefefdeeeefdeeeeefefdeeeffd/Z8Gd0d1e j9Z:Gd2d3e j9Z;Gd4d5Z)r\hyperParametersrA)r^rbrA)r`rbrArB)r\rbrC)r^rbrC)r`rbrCrDexportLastCheckpointOnlyrF)r\rb adapterSize)r^rbrd)r`rbrdrG tuningModerHcustomBaseModelrI)r\rbrJ)r`rbrJrK)r\rbrL)r`rbrLrMevaluationConfigrNrO)r^rbrOrPbaseTeacherModelrQtunedTeacherModelSourcerR)r`rbsftLossWeightMultiplierrS outputUrirTencryptionSpec)rr"_TuningValidationDataset_to_vertex_EvaluationConfig_to_vertexrrrr! discriminators r" _CreateTuningJobConfig_to_vertexrqsm !){Xx$899-,M... K./00<  ! " ,;!5 677K  +++ K./00<  ' ( ,;!5 677K  && K./00<    ,;!5 677K   +2344@ ! [5677  + ''3kM?)K)KLLL{Xx$899-,M... K-))5  C C C {]O , , +++ K-))5  I I I {]O , , && K-))5  ? ? ? {]O , , {Xx$899-,M... K4566B  O O O {78 9 9 +++ K4566B    {78 9 9&& K4566B  K K K {78 9 9 {Xx$899-,M... K7899E  !#= > {:; < < +++ K7899E  ')C D {:; < < && K7899E  9 : {:; < < {Xx$899-,M... K.)**6  D D D {^, - - +++ K.)**6  J J J {^, - - && K.)**6  @ @ @ {^, - - {Xx$899-,M... K-))5  !< 0 {]O , , && K-))5  | , {]O , ,  ++,--9  [./00 {Xx$899-,M... K,((4  B B B {\N + + && K,((4  > > > {\N + + {Xx$899-,M... K/*++7  E E E {_- . . && K/*++7  A A A {_- . . {Xx$899-,M... K-.//;  !#5 6 %;!4 566 ;  +++ K-.//;  '); < %;!4 566 ;  && K-.//;  1 2 %;!4 566 ;   +z"". Dxj$A$ABBB +x  ,AAA [6(##  +,-..: /0 [/011  +4566B 67 [7899  +4566BJJJ [7899  + ~&&2 tK,'H'HIII +)*++7  [,-.. r$ci}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg tt|dg||t|dg tt|dg|||SN base_model baseModelpre_tuned_model preTunedModeltraining_datasetrY)rr_TuningDataset_to_mldevrWr s r"*_CreateTuningJobParametersPrivate_to_mldevrzs !) + ~&&2[M4 l^#D#DEEE +)*++7_%tK:K9L'M'MNNN +*+,,8 [-.//K +z"".# [8*%%y+ r$ci}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg tt|dg||t|dg tt|dg|||Srs)rr_TuningDataset_to_vertexrqr s r"+_CreateTuningJobParametersPrivate_to_vertexr}2s !) + ~&&2[M4 l^#D#DEEE +)*++7_%tK:K9L'M'MNNN +*+,,8 [-.//K +z"".$ [8*%%y+ r$c i}t|dg3t|dgtjt|dgt|dg!t|dgt|dgt|dg0t|dgt t|dg||t|dg0t|dgt t|dg|||S)Nmetrics outputConfig output_configautoraterConfigautorater_configinferenceGenerationConfiginference_generation_config)rrt t_metricsr#rr s r"_EvaluationConfig_from_vertexrKs( !) + {##/YKT+ {-K-K!L!LMMM +'((4_%tK.9I'J'JKKK +)*++7 $ 01 2 2I{   +3455A &'% :; < <     r$c i}t|dg3t|dgtjt|dgt|dg!t|dgt|dgt|dg0t|dgt t|dg||t|dg0t|dgt t|dg|||S)Nrrrrrrr)rrrrr'r&r s r"rnrnns( !) + {##/YKT+ {-K-K!L!LMMM +())5^$d;8I&J&JKKK +*+,,8 " 12 3 3Y    +5677C $%# <= > >     r$ci}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|d g!t|d gt|d gt|d g!t|d gt|d gt|d g!t|d gt|d gt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|d gt|dgt|d!g!t|d"gt|d!gt|d#g!t|d#gt|d#gt|d$g!t|d%gt|d$gt|d&g!t|d'gt|d&gt|d(g!t|d)gt|d(g|S)*N modelConfigmodel_selection_configresponseJsonSchemaresponse_json_schemaaudioTimestampaudio_timestampcandidateCountcandidate_countenableAffectiveDialogenable_affective_dialogfrequencyPenaltyfrequency_penaltylogprobsmaxOutputTokensmax_output_tokensmediaResolutionmedia_resolutionpresencePenaltypresence_penaltyresponseLogprobsresponse_logprobsresponseMimeTyperesponse_mime_typeresponseModalitiesresponse_modalitiesresponseSchemaresponse_schema routingConfigrouting_configseed speechConfig speech_config stopSequencesstop_sequences temperaturethinkingConfigthinking_configtopKtop_ktopPtop_pr,r s r"rrs !) + ''3 !" [=/**  +,-..:   [/011  +()**6&'k  [3455  +./00<  [1233  +)*++7%&[;L:M(N(NOOO +)*++7%&[;L:M(N(NOOO +1233? ! [4566  ++,--9  [./00  + |$$0ZL${ZL"A"ABBB ++,--9%&[;N:O(P(P +*+,,8%&[;M:N(O(O +*+,,8%&[;M:N(O(O ++,--9  [./00  +,-..:  [/011  +-.//;  [0122  +)*++7%&[;L:M(N(NOOO +()**6_%tK:J9K'L'LMMM +x  ,VHd;99::: +())5  / 0 0)[   +()**6_%tK:J9K'L'LMMM + ''3]OT+ %G%GHHH +)*++7%&[;L:M(N(NOOO +y!!-VHd; ::;;; +y!!-VHd; ::;;; +7899E  &   r$cpi}t|dg"t|ddgt|dg|Sr)r,r s r" _GetTuningJobParameters_to_mldevrwr.r$cpi}t|dg"t|ddgt|dg|Sr)r,r s r"!_GetTuningJobParameters_to_vertexrr.r$c<i}t|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dg|S)N page_size_querypageSize page_token pageTokenfilterr,r s r"_ListTuningJobsConfig_to_vertexrs !) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"".8,d; .K.KLLL r$cli}t|dg tt|dg|||S)NrY)rrr s r"#_ListTuningJobsParameters_to_vertexrsH !) +z"".# [8*%%y+ r$c Tit|dg!tdgt|dgt|dg!tdgt|dgt|dg.tdgfdt|dgDS)Nr3r4 nextPageTokennext_page_token tuningJobs tuning_jobsc2g|]}t|S)_TuningJob_from_vertex.0itemrr!s r" z7_ListTuningJobsResponse_from_vertex..s5    #4K @ @   r$r,r s `@r"#_ListTuningJobsResponse_from_vertexrs !) +)*++7'(${=N.s5    *$ ; G G   r$r,r s `@r""_MultiSpeakerVoiceConfig_to_vertexrs{ !) +/011=      [+B*CDD    r$cRi}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dgtdt|dgtd|S) N mime_typemimeTypevoice_sample_audiovoiceSampleAudio consent_audiozMconsent_audio parameter is not supported in Gemini Enterprise Agent Platform.voice_consent_signaturezWvoice_consent_signature parameter is not supported in Gemini Enterprise Agent Platform.rrrVr s r" _ReplicatedVoiceConfig_to_vertexrs !) + }%%1ZL${[M"B"BCCC +,-..:  [/011  +())5      +1233?  &   r$c i}t|dg!t|dgt|dgt|dg0t|dgtt|dg|||S)Nspeaker voice_config voiceConfig)rr_VoiceConfig_to_vertexr s r"rrs !) + {##/YKkI;!?!?@@@ +'((4  ~. / /K   r$c ri}t|dg0t|dgtt|dg||t|dg!t|dgt|dgt|dg0t|dgtt|dg|||S)Nrr language_code languageCodemulti_speaker_voice_configmultiSpeakerVoiceConfig)rrrrr s r"rrs !) +'((4  ~. / /K   +())5^$d;8I&J&JKKK +4566B "#* ;< = =     r$ci}t|dg!t|dgt|dgt|dg!t|dgt|dg|S)Nr*modelendpointr,r s r"_TunedModel_from_mldevr8ss !) +x  ,WItK&::;;; +x  ,ZL${VH"="=>>> r$c i}t|dgtdt|dgtdt|dg,t|ddgdt|dgD|S)Ngcs_uriz1gcs_uri parameter is not supported in Gemini API.vertex_dataset_resourcezAvertex_dataset_resource parameter is not supported in Gemini API.examplescg|]}|Srrrrs r"rz+_TuningDataset_to_mldev..Ys:::$:::r$rUr s r"ryryGs !) + {##/ H I II +1233? K   + |$$0 Z ::${ZL99::: r$cVi}t|ddg}|d}|dkr4t|dg"t|ddgt|dgns|dkr4t|dg"t|ddgt|dgn9|d kr3t|dg"t|d d gt|dgt|ddg}|d}|dkr4t|d g"t|ddgt|d gns|dkr4t|d g"t|ddgt|d gn9|d kr3t|d g"t|d d gt|d gt|d gtd|S)NrYrZr[rr\trainingDatasetUrir]r^r_r`promptDatasetUrirrzHexamples parameter is not supported in Gemini Enterprise Agent Platform.rros r"r|r|_sT !){Xx$899-,M... K)%%1  !#7 8 {YK ( ( +++ K)%%1  ')= > {YK ( ( && K)%%1  1 2 {YK ( ( {Xx$899-,M... K3455A  !#7 8 {67 8 8 +++ K3455A  ')= > {67 8 8 && K3455A  1 2 {67 8 8  + |$$0     r$c <i}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg3t|dgtjt|dgt|dg!t|dgt|dgt|ddg"t|d gt|ddgt|dd g"t|d gt|dd gt|d g!t|d gt|d gt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg0t|dgt t|dg|||S)Nr3r4r*state createTime create_timer? startTime start_time completeTimeend_time updateTime update_timer=rurt_self tuned_model)rrrt_tuning_job_statusrr s r"_TuningJob_from_mldevrsf !) +)*++7'(${=N.BsEEE$EEEr$ experimentfullFineTuningSpecfull_fine_tuning_specrNrkrS pipelineJob pipeline_jobserviceAccountservice_accountrar;tuningJobStatetuning_job_state veoTuningSpecveo_tuning_specdistillationSamplingSpecdistillation_sampling_spectuningJobMetadatatuning_job_metadata)rrrrrr s r"rrsl !) +)*++7'(${=N $% [3455  +*+,,8 % 12 3 3Y    +)*++7'(${=N>> + }%%1\ND{m$D$DEEE + ''3^$d;&H&HIII +()**6&'kr?rVrr5r@rrArYrBr rCrDrErFrGrHrListTuningJobsResponserJrKrLr<r4rM rOrYrPrQrRrSrTrBr9rUrVs r"_listz Tunings._list!sd5O   $   >   9 4l&))&11 &&'788##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''t\<PPH&mJBBHM1J1JM  9 m/>> ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !?L$&+%7 &&&L" %%l333 r$c tj||}|jjrAt |d|}|d}|rd|}nCd}n@t|d|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||} | jsint#j| j} |jjrt'| d|} |jjst)| d|} tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini } tj| j| _|j| | S)zPCancels a tuning job. Args: name (str): TuningJob resource name. r3Nr+ {name}:cancelrr4rYpostrrr6r7r8r;)r _CancelTuningJobParametersr>r?r0r5r@r-rrArYrBr rCrDrErFrGrHr8r5CancelTuningJobResponserJrKrLr<r4rMrNs r"cancelzTunings.cancelis6 O  9 4l&))&11 ))*:;;8 4l&))&11 ))*:;;##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM  : m   $9 m0?? ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !@L$&+%7 &&&L" %%l333 r$rtrvrYrtrvrxc tj||||}|jjst dt |d|}|d}|rd|}nd}|d} | r|dt| }| ddd} |j |j j |j j } tj |}tj|}|jd ||| } | jsint#j| j} |jjrt'| d|} tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini} tj| j| _|j| | SaCreates a tuning job and returns the TuningJob object. Args: base_model: The name of the model to tune. training_dataset: The training dataset to use. config: The configuration to use for the tuning job. Returns: A TuningJob object. rtrvrxrYrZNr+rrr4rYrarrr6r7r8r;)r !_CreateTuningJobParametersPrivater>r?rVr}r5r@rrArYrBr rCrDrErFrGrHrrIrJrKrLr<r4rMrOrtrvrxrYrPrQrRrSrTrBr9rUrVs r"_tunez Tunings._tunest&=') O   $   >   A 4l&))&11 &&'788##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM  , m?11 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !2L$&+%7 &&&L" %%l333 r$c tj||||}|jjrt dt |d|}|d}|rd|}nd}|d} | r|dt| }| ddd} |j |j j |j j } tj |}tj|}|jd ||| } | jsint#j| j} |jjst'| d|} tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini} tj| j| _|j| | SaCreates a tuning job and returns the TuningJob object. Args: base_model: The name of the model to tune. training_dataset: The training dataset to use. config: The configuration to use for the tuning job. Returns: A TuningJob operation. rhz=This method is only supported in the Gemini Developer client.Nr+ tunedModelsrr4rYrarrr6r7r8r;)r rir>r?rVrzr5r@rrArYrBr rCrDrErFrGrHr&TuningOperationrJrKrLr<r4rMrjs r" _tune_mldevzTunings._tune_mldev#ss&=') O    I  @ 4l&))&11 ''(899##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM   $1 m(77 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !8L$&+%7 &&&L" %%l333 r$c|||}|jrC|jjr7|jj+t |j|jj|SNr3)rprojectrWrr>r?rs _IpythonUtilsdisplay_experiment_buttonrOr*rYjobs r"r5z Tunings.getsq ))f) - -C    %   $ 0--^"*. Jr$SThe SDK's tuning implementation is experimental, and may change in future versions.c|jjr|drEd}|rt|dd}t j||}||||}nd}|t|ddt|d}t|trt j di|}|j r|j stdtt|j D]T} t|j | tr2tj|j | |j | <Ut|tr||d<n||_|}||||}|t|dd||_n||||} | jd| jvr| jd} n6| jtd | jd d } t j| tjj }|jr,|jjr t2|j |SNz projects/rE)tuned_model_name checkpoint_id)rvrxrYrMzEEvaluation config must have at least one metric and an output config.)rtrxrYr zOperation name is required.z /operations/r)r*r)tuning_job_resourcerr>r? startswithrKr PreTunedModelrk isinstancedictEvaluationConfigrrrVrangelenMetricmodel_validaterMrpr$r* partitionrIJobStateJOB_STATE_QUEUEDrudisplay_model_tuning_button) rOrtrxrYr}rv tuning_jobvalidated_evaluation_configrMi operationr|s r"tunez Tunings.tunes  A   { + +0E  Q!&*I4PP--'}   ZZ+-   IM#   3T::F%f.ABB  )4 0 0 L % 6 K K9J K K #+ &4  .67788  a+3A6== -2\-H-H#+A...'* % % 9*;F& ' ''8F $(9 %ZZ!-     3T::F)D* &""+#i   'LI !899 9$>33NCCAF?/j4+4//(o0 r$c\|j}td||||S)aLists `TuningJob` objects. Args: config: The configuration for the list request. Returns: A Pager object that contains one page of tuning jobs. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python for tuning_job in client.tunings.list(): print(tuning_job.name) rr0)r^rrOrY list_requests r"listz Tunings.lists7&:L  & !!   r$)__name__ __module__ __qualname__strrr GetTuningJobConfigOrDictrIrWListTuningJobsConfigOrDictr\r^CancelTuningJobConfigOrDictrcrdPreTunedModelOrDictTuningDatasetOrDictCreateTuningJobConfigOrDictrkrorpr5r experimental_warningrrrrr$r"r/r/s  :> \\\ \u56 \   \\\\~EIFFF @AF #FFFFX=A [[[ [u89 [  $ [[[[@#'=A<@ [[[3-[  9: [ 1 [ u89 [ [[[[@#'=A<@ ZZZ3-Z  9: Z 1 Z u89 Z ZZZZ@:>  u56    $ 7+=A MMMM1 M u89 M   MMM M`EI @A U_r$r/c eZdZdddedeejdejfdZdddeej dej fdZ dddedeej dej fdZdddd d eed eejd ejdeejdejf d Zdddd d eed eejd ejdeejdejf dZdddedeejdejfdZejdddd ed ejdeejdejfdZdddeej deejfdZdS) AsyncTuningsNr0r*rYrc Ktj||}|jjrAt |d|}|d}|rd|}nCd}n@t|d|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||d{V} | jsint#j| j} |jjrt'| d|} |jjst)| d|} tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini } tj| j| _|j| | Sr2)r r=r>r?rr5r@rrrArYrBr rCrD async_requestrFrGrHrrrIrJrKrLr<r4rMrNs r"rWzAsyncTunings._gets3 O  6 4l&))&11 ""#3445 4l&))&11 ""#344##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33 t\<H'mJBBHM1J1JM  , m   $+ m?11 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !2L$&+%7 &&&L" %%l333 r$c Ktj|}|jjst dt |d|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||d{V}|jsint#j|j} |jjrt'| d|} tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini } tj|j| _|j| | SrY)r r[r>r?rVrr5r@rrArYrBr rCrDrrFrGrHrr\rJrKrLr<r4rMr]s r"r^zAsyncTunings._listbs5O   $   >   9 4l&))&11 &&'788##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33 t\<H'mJBBHM1J1JM  9 m/>> ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !?L$&+%7 &&&L" %%l333 r$c Ktj||}|jjrAt |d|}|d}|rd|}nCd}n@t|d|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||d{V} | jsint#j| j} |jjrt'| d|} |jjst)| d|} tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini } tj| j| _|j| | S)zaCancels a tuning job asynchronously. Args: name (str): A TuningJob resource name. r3Nr+r`rr4rYrarrr6r7r8r;)r rbr>r?r0r5r@r-rrArYrBr rCrDrrFrGrHr8r5rcrJrKrLr<r4rMrNs r"rdzAsyncTunings.cancels6 O  9 4l&))&11 ))*:;;8 4l&))&11 ))*:;;##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM  : m   $9 m0?? ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !@L$&+%7 &&&L" %%l333 r$rertrvrxc &Ktj||||}|jjst dt |d|}|d}|rd|}nd}|d} | r|dt| }| ddd} |j |j j |j j } tj |}tj|}|jd ||| d{V} | jsint#j| j} |jjrt'| d|} tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini} tj| j| _|j| | Srg)r rir>r?rVr}r5r@rrArYrBr rCrDrrFrGrHrrIrJrKrLr<r4rMrjs r"rkzAsyncTunings._tune s&=') O   $   >   A 4l&))&11 &&'788##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM  , m?11 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !2L$&+%7 &&&L" %%l333 r$c &Ktj||||}|jjrt dt |d|}|d}|rd|}nd}|d} | r|dt| }| ddd} |j |j j |j j } tj |}tj|}|jd ||| d{V} | jsint#j| j} |jjst'| d|} tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini} tj| j| _|j| | Srm)r rir>r?rVrzr5r@rrArYrBr rCrDrrFrGrHr&rorJrKrLr<r4rMrjs r"rpzAsyncTunings._tune_mldevf s&=') O    I  @ 4l&))&11 ''(899##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM   $1 m(77 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !8L$&+%7 &&&L" %%l333 r$cK|||d{V}|jrC|jjr7|jj+t |j|jj|Srrrtrws r"r5zAsyncTunings.get s  tF 33 3 3 3 3 3 3C    %   $ 0--^"*. Jr$rycK|jjr}|drKd}|rt|dd}t j||}||||d{V}n|t|ddt|d}t|trt j di|}|j r|j stdtt|j D]T}t|j |tr2tj|j ||j |<Ut|tr||d<n||_||||d{V}n||||d{V} | jd| jvr| jd} n6| jtd | jd d } t j| tjj }|jr,|jjr t2|j |Sr{r) rOrtrxrYr}rvrrMrrr|s r"rzAsyncTunings.tune s   ;   { + +*   Q!&*I4PP--'}    ::+-&           3T::F%f.ABB  )4 0 0 L % 6 K K9J K K #+ &4  .67788  a+3A6== -2\-H-H#+A...'* % % 9*;F& ' ''8F $::!-&        ((+)i   'LI !899 9$>33NCCAF?/j4+4//(o0 r$clK|j}td|||d{V|S)aLists `TuningJob` objects asynchronously. Args: config: The configuration for the list request. Returns: A Pager object that contains one page of tuning jobs. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python async for tuning_job in await client.aio.tunings.list(): print(tuning_job.name) rr0N)r^rrs r"rzAsyncTunings.list! sM&:L jjj''''''''   r$)rrrrrr rrIrWrr\r^rrcrdrrrrkrorpr5r rrrrrr$r"rrs  :> ^^^ ^u56 ^   ^^^^BEIHHH @AH #HHHH\=A [[[ [u89 [  $ [[[[@#'=A<@ [[[3-[  9: [ 1 [ u89 [ [[[[@#'=A<@ ZZZ3-Z  9: Z 1 Z u89 Z ZZZZ@:>  u56    $ 7+=A GGGG1 G u89 G   GGG GTEI @A%/"r$rc NeZdZUdZeZeeed<ede ee ffdZ ede fdZ edefdZededeedefd Ze dd ed ed eeddfdZedededdfdZededdfdZdS)ruz6Temporary class to hold the IPython related functions.displayed_experimentsrcTddl}d|jvrddlm}|jjSdS)NrIPython) get_ipython)sysmodulesrr __class__r)rrs r"_get_ipython_shell_namez%_IpythonUtils._get_ipython_shell_nameB sAJJJCK%%%%%% []] $ -- 2r$cNttS)N)boolrurrr$r"is_ipython_availablez"_IpythonUtils.is_ipython_availableL s  5577 8 88r$cdS)z8Returns the HTML style markup to support custom buttons.a rrr$r" _get_stylesz_IpythonUtils._get_stylesP s 1 1 r$markerresource_partsctt|D]1}|||kr#|dzt|kr ||dzcS2dS)z,Returns the part after the marker text part.rr)rr)rrrs r"_parse_resource_namez"_IpythonUtils._parse_resource_name sd3~&& ' '%%  f $ $Q^1D1D)D)Da!e$$$$ 2r$ open_in_newtexturliconNc Zd}||std|dddl}dt|}dt d|d |d |d |d }|d |d|d|dz }ddlm}ddlm }|||dS)a@Creates and displays the link to open the Gemini Enterprise Agent Platform resource. Args: text: The text displayed on the clickable button. url: The url that the button will lead to. Only cloud console URIs are allowed. icon: The icon name on the button (from material-icons library) z https://console.cloud.google.comzOnly urls starting with z are allowed.rNzview-vertex-resource-z z- z z z] )display)HTML) rrVuuidruuid4rurIPython.displayrr) rrr CLOUD_UI_URLr button_idhtmlrrs r" _display_linkz_IpythonUtils._display_link sB6L >>, ' 'O M,MMM N NNKKK;DJJLL(9(9;;I  " " $ $  -6  FO  :>      D  4=   /2   "   D"('''''$$$$$$ GDDJJr$rrsc:tr|tjvrdS|d}|d}|d}d|d|dzd|z}td|d tj|dS) a\Function to generate a link bound to the Gemini Enterprise Agent Platform experiment. Args: experiment: The Gemini Enterprise Agent Platform experiment name. Example format: projects/{project_id}/locations/{location}/metadataStores/default/contexts/{experiment_name} project: The project (alphanumeric) name. N/zAhttps://console.cloud.google.com/vertex-ai/experiments/locations/z /experiments/z runs?project=zView Experimentscience)rurrsplitradd)rrsrlocationexperiment_nameuris r"rvz'_IpythonUtils.display_experiment_button s  . . 0 0 < < < f %%c**Na H$R(O L  6 6O 6 6 6 7 #' # # $  13 BBB'++J77777r$r~ctsdS|d}|d}|d}|d}dd|d|zd |z}td |d dS) a"Function to generate a link bound to the Gemini Enterprise Agent Platform model tuning job. Args: tuning_job_resource: The Gemini Enterprise Agent Platform tuning job name. Example format: projects/{project_id}/locations/{location}/tuningJobs/{tuning_job_id} Nrrrrz?https://console.cloud.google.com/vertex-ai/generative/language/z locations/z/tuning/tuningJob/z ?project=zView Tuning Jobr)rurrr)r~rrsr tuning_job_idrs r"rz)_IpythonUtils.display_model_tuning_button s  - - / /  f(..s33NQGa H"2&M J Bx B B= B B C g     13?????r$)r)rrr__doc__setrr__annotations__ staticmethodrrrrrrrrrrrvrrr$r"ruru= s>>$'CEES)))5c?<9999<93s333<3j3S c<1>11 11!)#1 111<1f8C8#8$888<8<@s@t@@@<@@@r$ru)NN)=rGloggingtypingrrr urllib.parserrrr r rr r rr rpagersrr getLoggerloggerrrobjectr#r'r-r0r5r8rWrqrzr}rrnrr&rrrrrrrrrrryr|rrr&rmr BaseModuler/rrurr$r"rs$ ''''''''''"""""" ............%%%%%%%%  1 2 2 /3;?tCH~v-.DcN+%S#X 678 #s(^ :/3;?tCH~v-.DcN+%S#X 678 #s(^ :/3;?  tCH~v-. DcN+ %S#X 678  #s(^    /3;?  tCH~v-. DcN+ %S#X 678  #s(^    /3;?  tCH~v-. DcN+ %S#X 678  #s(^     /3;?  tCH~v-. DcN+ %S#X 678  #s(^     /3;?jjtCH~v-.jDcN+j%S#X 678j #s(^ jjjj^/3;?]]tCH~v-.]DcN+]%S#X 678] #s(^ ]]]]D /3;?tCH~v-.DcN+%S#X 678 #s(^ 6/3;?tCH~v-.DcN+%S#X 678 #s(^ 6/3;?  tCH~v-. DcN+ %S#X 678  #s(^    J/3;?  tCH~v-. DcN+ %S#X 678  #s(^    J/3;?jjtCH~v-.jDcN+j%S#X 678j #s(^ jjjj^/3;?vvtCH~v-.vDcN+v%S#X 678v #s(^ vvvvv/3;?  tCH~v-. DcN+ %S#X 678  #s(^    /3;?  tCH~v-. DcN+ %S#X 678  #s(^    /3;?tCH~v-.DcN+%S#X 678 #s(^ 6/3;?  tCH~v-. DcN+ %S#X 678  #s(^     /3;?tCH~v-.DcN+%S#X 678 #s(^ :/3;?tCH~v-.DcN+%S#X 678 #s(^ */3;?tCH~v-.DcN+%S#X 678 #s(^ B/3;?tCH~v-.DcN+%S#X 678 #s(^ ./3;?tCH~v-.DcN+%S#X 678 #s(^ D/3;?  tCH~v-. DcN+ %S#X 678  #s(^    "/3;?tCH~v-.DcN+%S#X 678 #s(^ 4/3;???tCH~v-.?DcN+?%S#X 678? #s(^ ????H/3;?88tCH~v-.8DcN+8%S#X 6788 #s(^ 8888z/3;?ZZtCH~v-.ZDcN+Z%S#X 678Z #s(^ ZZZZ~/3;?tCH~v-.DcN+%S#X 678 #s(^ 8/3;?tCH~v-.DcN+%S#X 678 #s(^ */3;?tCH~v-.DcN+%S#X 678 #s(^ 6|||||k$|||~zzzzz;)zzzzz@z@z@z@z@z@z@z@z@z@r$