쎒jN <ddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z ddl m Z ddl mZdd l mZdd l mZdd lmZmZejd Z dGd eeeefefdeeeefdeeeeefefdeeeffdZ dGd eeeefefdeeeefdeeeeefefdeeeffdZ dGd eeeefefdeeeefdeeeeefefdeeeffdZ dGd eeeefefdeeeefdeeeeefefdeeeffdZ dGd eeeefefdeeeefdeeeeefefdeeeffdZ dGd eeeefefdeeeefdeeeeefefdeeeffdZ dGd eeeefefdeeeefdeeeeefefdeeeffdZ! dGd eeeefefdeeeefdeeeeefefdeeeffdZ" dGd eeeefefdeeeefdeeeeefefdeeeffdZ# dGd eeeefefdeeeefdeeeeefefdeeeffdZ$ dGd eeeefefdeeeefdeeeeefefdeeeffdZ% dGd eeeefefdeeeefdeeeeefefdeeeffdZ& dGd eeeefefdeeeefdeeeeefefdeeeffdZ' dGd eeeefefdeeeefdeeeeefefdeeeffdZ( dGd eeeefefdeeeefdeeeeefefdeeeffdZ) dGd eeeefefdeeeefdeeeeefefdeeeffd Z* dGd eeeefefdeeeefdeeeeefefdeeeffd!Z+ dGd eeeefefdeeeefdeeeeefefdeeeffd"Z, dGd eeeefefdeeeefdeeeeefefdeeeffd#Z- dGd eeeefefdeeeefdeeeeefefdeeeffd$Z. dGd eeeefefdeeeefdeeeeefefdeeeffd%Z/ dGd eeeefefdeeeefdeeeeefefdeeeffd&Z0 dGd eeeefefdeeeefdeeeeefefdeeeffd'Z1 dGd eeeefefdeeeefdeeeeefefdeeeffd(Z2 dGd eeeefefdeeeefdeeeeefefdeeeffd)Z3 dGd eeeefefdeeeefdeeeeefefdeeeffd*Z4 dGd eeeefefdeeeefdeeeeefefdeeeffd+Z5 dGd eeeefefdeeeefdeeeeefefdeeeffd,Z6 dGd eeeefefdeeeefdeeeeefefdeeeffd-Z7 dGd eeeefefdeeeefdeeeeefefdeeeffd.Z8 dGd eeeefefdeeeefdeeeeefefdeeeffd/Z9 dGd eeeefefdeeeefdeeeeefefdeeeffd0Z: dGd eeeefefdeeeefdeeeeefefdeeeffd1Z; dGd eeeefefdeeeefdeeeeefefdeeeffd2Z< dGd eeeefefdeeeefdeeeeefefdeeeffd3Z= dGd eeeefefdeeeefdeeeeefefdeeeffd4Z> dGd eeeefefdeeeefdeeeeefefdeeeffd5Z? dGd eeeefefdeeeefdeeeeefefdeeeffd6Z@ dGd eeeefefdeeeefdeeeeefefdeeeffd7ZA dGd eeeefefdeeeefdeeeeefefdeeeffd8ZB dGd eeeefefdeeeefdeeeeefefdeeeffd9ZC dGd eeeefefdeeeefdeeeeefefdeeeffd:ZD dGd eeeefefdeeeefdeeeeefefdeeeffd;ZE dGd eeeefefdeeeefdeeeeefefdeeeffd<ZF dGd eeeefefdeeeefdeeeeefefdeeeffd=ZG dGd eeeefefdeeeefdeeeeefefdeeeffd>ZH dGd eeeefefdeeeefdeeeeefefdeeeffd?ZI dGd eeeefefdeeeefdeeeeefefdeeeffd@ZJGdAdBe jKZLGdCdDe jKZMGdEdFZNdS)HN)AnyOptionalUnion) urlencode) _api_module)_common) _transformers)types)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.tunings from_object parent_object root_objectreturnc i}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg0t|dgtt|dg|||S) N samplingCountsampling_count flipEnabled flip_enabledautoraterModelautorater_modelgenerationConfiggeneration_configgetvsetv_GenerationConfig_from_vertexrrr to_objects BC:\Projects\DSCons\.venv\Lib\site-packages\google\genai\tunings.py_AutoraterConfig_from_vertexr$#s !) +())5%&[?:K(L(LMMM + ''3^$d;&H&HIII +()**6&'ks !) +()**6_%tK:J9K'L'LMMM +'((4]OT+7G%H%HIII +)*++7%&[;L:M(N(NOOO ++,--9 # 23 4 4i   r%cpi}t|dg"t|ddgt|dg|SNname_urlrrr!s r##_CancelTuningJobParameters_to_mldevr.YD !) +x  ,VV$d;&A&ABBB r%cpi}t|dg"t|ddgt|dg|Sr*r-r!s r#$_CancelTuningJobParameters_to_vertexr1er/r%cni}t|dg!t|dgt|dg|SNsdkHttpResponsesdk_http_responser-r!s r##_CancelTuningJobResponse_from_mldevr6qP !) +)*++7'(${=N>> r%c i}t|dg0t|dgtt|dg||t|dg!t|dgt|dg|S)Nr<r;r=)rr0_SingleReinforcementTuningRewardConfig_to_vertexr!s r#G_CompositeReinforcementTuningRewardConfigWeightedRewardConfig_to_vertexrBs !) +())5 8 / 0 0)[   +z"".XJ[8* = =>>> r%c it|dg.tdgfdt|dgDS)NweightedRewardConfigsweighted_reward_configsc2g|]}t|S)r?.0itemrr"s r# zI_CompositeReinforcementTuningRewardConfig_from_vertex..s=    Vi     r%r-r!s `@r#5_CompositeReinforcementTuningRewardConfig_from_vertexrLs} !) +/011= "#     [+B*CDD        r%c it|dg.tdgfdt|dgDS)NrErDc2g|]}t|SrG)rBrHs r#rKzG_CompositeReinforcementTuningRewardConfig_to_vertex..s=    Ti     r%r-r!s `@r#3_CompositeReinforcementTuningRewardConfig_to_vertexrOs} !) +1233? !     [+D*EFF        r%c it|dg.tdgfdt|dgDt|dg!tdgt|dgS)Npartsc2g|]}t|SrG)_Part_to_vertexrHs r#rKz&_Content_to_vertex..s5    D)[ 9 9   r%roler-r!s `@r#_Content_to_vertexrUs !) +y!!-       [7)44    +x  ,VHd;99::: r%ci}t|dgtdt|dg!t|dgt|dgt|dgtdt|dg"t|gdt|dgt|d g"t|gd t|d gt|d gtd t|d gtdt|dgtdt|dgtdt|dgtdt|dg"t|gdt|dgt|dg"t|gdt|dgt|dgtdt|dgtdt|dgtdt|dgtd t|d!gtd"t|d#gtd$t|d%gtd&t|d'gtd(t|d)gtd*t|d+gtd,t|d-gtd.t|d/gtd0t|d1gtd2t|d3gtd4t|d5gtd6t|d7gtd8|S)9Nvalidation_datasetzzvalidation_dataset parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.tuned_model_display_name displayName descriptionzsdescription parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. epoch_count) tuningTaskhyperparameters epochCountlearning_rate_multiplier)r\r]learningRateMultiplierexport_last_checkpoint_onlyzexport_last_checkpoint_only parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.pre_tuned_model_checkpoint_idzpre_tuned_model_checkpoint_id parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. adapter_sizeztadapter_size parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. tuning_modezstuning_mode parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.custom_base_modelzycustom_base_model parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. batch_size)r\r] batchSize learning_rate)r\r] learningRateevaluation_configzyevaluation_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.labelsznlabels parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.betazlbeta parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.base_teacher_modelzzbase_teacher_model parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.tuned_teacher_model_sourceztuned_teacher_model_source parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.sft_loss_weight_multiplierzsft_loss_weight_multiplier parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. output_urizroutput_uri parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.encryption_speczwencryption_spec parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.r<zureward_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.composite_reward_configzcomposite_reward_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.samples_per_promptzzsamples_per_prompt parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.evaluate_intervalzyevaluate_interval parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.checkpoint_intervalz{checkpoint_interval parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.max_output_tokenszymax_output_tokens parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.thinking_levelzvthinking_level parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.validation_dataset_uriz~validation_dataset_uri parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.r ValueErrorrr!s r#_CreateTuningJobConfig_to_mldevr{sg !) +,-..:  B    +2344@  [5677  + ''3  <    + ''3777 [=/**  +2344@CCC [5677  +5677C  M    +7899E  M    +'((4  <    + ''3  <    ++,--9  B    + ~&&2666 [<.))  +())5999 [?+,,  ++,--9  B    +z"".  3    +x  ,  3    +,-..:  B    +4566B  M    +4566B  M    + ~&&2  <    +)*++7  <    +())5  <    +1233?  M    +,-..:  B    ++,--9  B    +-.//;  B    ++,--9  B    +()**6  <    +0122>  M   r%c i}t|ddg}|d}|dkrBt|dg0t|dgtt|dg||n|dkrBt|dg0t|dgtt|dg||n|dkrBt|dg0t|d gtt|dg||nG|d krAt|dg0t|d gtt|dg||t|d g!t|d gt|d gt|dg!t|dgt|dgt|ddg}|d}|dkr4t|dg"t|gdt|dgn|dkr4t|dg"t|gdt|dgns|dkr4t|dg"t|gdt|dgn9|d kr3t|dg"t|gdt|dgt|ddg}|d}|dkr4t|dg"t|gdt|dgn|dkr4t|dg"t|gdt|dgns|dkr4t|dg"t|gdt|dgn9|d kr3t|dg"t|gdt|dgt|ddg}|d}|dkr4t|dg"t|ddgt|dgns|dkr4t|dg"t|ddgt|dgn9|dkr3t|dg"t|d dgt|dgt|ddg}|d}|dkr4t|dg"t|gdt|dgn|dkr4t|dg"t|gdt|dgns|dkr4t|dg"t|gdt|dgn9|d kr3t|dg"t|gdt|dgt|ddg}|d}|dkr4t|d g"t|dd!gt|d gn9|dkr3t|d g"t|d d!gt|d gt|d"g!t|d#gt|d"gt|ddg}|d}|dkr4t|d$g"t|gd%t|d$gns|dkr4t|d$g"t|gd&t|d$gn9|d kr3t|d$g"t|gd't|d$gt|ddg}|d}|dkr4t|d(g"t|gd)t|d(gn9|dkr3t|d(g"t|gd*t|d(gt|ddg}|d}|dkrCt|d+g1t|dd,gtt|d+g||n|dkrCt|d+g1t|dd,gtt|d+g||nH|dkrBt|d+g1t|d d,gtt|d+g||t|d-g!t|d-gt|d-gt|d.g"t|gd/t|d.gt|d0g"t|d d1gt|d0gt|d2g"t|d d3gt|d2gt|d4g"t|gd5t|d4gt|d6g!t|d7gt|d6gt|d8g!t|d9gt|d8gt|d:g1t|d d;gt t|d:g||t|dg"t|gd?t|d>gt|d@g"t|gdAt|d@gt|dBg"t|gdCt|dBgt|dDg"t|gdEt|dDgt|dFg"t|gdGt|dFgt|dHg"t|d dIgt|dHg|S)JNconfigmethodSUPERVISED_FINE_TUNINGrWsupervisedTuningSpecPREFERENCE_TUNINGpreferenceOptimizationSpec DISTILLATIONdistillationSpecREINFORCEMENT_TUNINGreinforcementTuningSpecrXtunedModelDisplayNamerZr[)rhyperParametersr^)rrr^)rrr^)rrr^r_)rrr`)rrr`)rrr`)rrr`raexportLastCheckpointOnlyrc)rr adapterSize)rrr)rrr)rrrrd tuningModerecustomBaseModelrf)rrrg)rrrg)rrrgrh)rrri)rrrirjevaluationConfigrkrl)rrrlrmbaseTeacherModelrntunedTeacherModelSourcero)rrsftLossWeightMultiplierrp outputUrirqencryptionSpecr<singleRewardConfigrrcompositeRewardConfigrs)rrsamplesPerPromptrt)rrevaluateIntervalru)rrcheckpointIntervalrv)rrmaxOutputTokensrw)rr thinkingLevelrxvalidationDatasetUri)rr"_TuningValidationDataset_to_vertex_EvaluationConfig_to_vertexrArOrrrr" discriminators r# _CreateTuningJobConfig_to_vertexrs] !){Xx$899-,M... K./00<  ! " ,;!5 677K  +++ 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 , , ... K-))5  F F F {]O , , {Xx$899-,M... K4566B  O O O {78 9 9 +++ K4566B    {78 9 9&& K4566B  K K K {78 9 9 ... K4566B    {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  @ @ @ {^, - - ... K.)**6  G G G {^, - - {Xx$899-,M... K-))5  !< 0 {]O , , && K-))5  | , {]O , ,  ++,--9  [./00 {Xx$899-,M... K,((4  B B B {\N + + && K,((4  > > > {\N + + ... K,((4  E E E {\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  [,-..  +())5 "$898 / 0 0)[   +1233? "$;<; 89 : :     +,-..:JJJ [/011  ++,--9JJJ [./00  +-.//;LLL [0122  ++,--9III [./00  +()**6GGG [+,--  +0122> "$:; [3455 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_datasetr})rr_TuningDataset_to_mldevr{r!s r#*_CreateTuningJobParametersPrivate_to_mldevr5s !) + ~&&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|||Sr)rr_TuningDataset_to_vertexrr!s r#+_CreateTuningJobParametersPrivate_to_vertexrNs !) + ~&&2[M4 l^#D#DEEE +)*++7_%tK:K9L'M'MNNN +*+,,8 [-.//K +z"".$ [8*%%y+ r%c i}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg0t|dgtt|dg||t|d g!t|d gt|d gt|d g!t|d gt|d g|S) Nrrcr^r[r`r_rrrirhrgrfrr!s r#(_DistillationHyperParameters_from_vertexrgsp !) + ''3^$d;&H&HIII + ~&&2]OT+ ~%F%FGGG +0122> #$ [3455  +*+,,8 % 12 3 3Y    +'((4_%tK.9I'J'JKKK + }%%1\ND{m$D$DEEE r%c i}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|d g0t|d gtt|d g|||S) NrrmrrnrrxpromptDatasetUriprompt_dataset_urir]rrrr!s r#%_DistillationSamplingSpec_from_vertexrsj !) +*+,,8  [-.//  +1233? %& [4566  +./00< !" [1233  +*+,,8  [-.//  +)*++7 0 01 2 2I{   r%c i}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg0t|dgtt|dg||t|dg!t|dgt|dgt|d g!t|d gt|d gt|d g!t|d gt|d gt|d g!t|dgt|d gt|dg!t|dgt|dgt|dg!t|dgt|dg|S)Nrrrrmrhyper_parameterspipelineRootDirectorypipeline_root_directory studentModel student_modeltrainingDatasetUritraining_dataset_urirrnrrxrrdrr!s r#_DistillationSpec_from_vertexrsU !) +*+,,8  [-.//  +*+,,8  [-.//  +)*++7 0 01 2 2I{   +/011= "# [2344  +'((4_%tK.9I'J'JKKK +,-..:   [/011  +1233? %& [4566  +./00< !" [1233  + ~&&2]OT+ ~%F%FGGG 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$r r!s r#_EvaluationConfig_from_vertexrs( !) + {##/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#rrs( !) + {##/YKT+ {-K-K!L!LMMM +())5^$d;8I&J&JKKK +*+,,8 " 12 3 3Y    +5677C $%# <= > >     r%c  i}t|dg!t|dgt|dgt|dg3t|dgtjt|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(gt|d*g+t|d+gd,t|d*gDt|d-g!t|d.gt|d-g|S)/N modelConfigmodel_selection_configresponseJsonSchemaresponse_json_schemaaudioTimestampaudio_timestampcandidateCountcandidate_countenableAffectiveDialogenable_affective_dialogfrequencyPenaltyfrequency_penaltylogprobsrrvmediaResolutionmedia_resolutionpresencePenaltypresence_penaltyresponseLogprobsresponse_logprobsresponseMimeTyperesponse_mime_typeresponseModalitiesresponse_modalitiesresponseSchemaresponse_schema routingConfigrouting_configseed speechConfig speech_config stopSequencesstop_sequences temperaturethinkingConfigthinking_configtopKtop_ktopPtop_presponseFormatresponse_formatcg|]}|SrGrGrIrJs r#rKz1_GenerationConfig_from_vertex..s@@@$@@@r%audioTranscriptionConfigaudio_transcription_config)rrr t_json_schemar!s r#r r Bs !) + ''3 !" [=/**  +,-..:   [+?*@AABB  +()**6&'k??@@@  +2344@ %& [5677 r%c @ i}t|dg!t|dgt|dgt|dg3t|dgtjt|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|dg0t|d gt t|dg||t|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(gt|d*gt d+t|d,g+t|d-gd.t|d,gDt|d/gt d0t|d1g!t|d2gt|d1g|S)3Nrrrrrrrrrrrrrrvrrrrrrrrrrrrrrrrrrrrrrrrrrrenable_enhanced_civic_answerszenable_enhanced_civic_answers parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rrcg|]}|SrGrGrs r#rKz/_GenerationConfig_to_vertex..7sAAA$AAAr%translation_configzztranslation_config parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rr)rrrr_SpeechConfig_to_vertexrzr!s r#r'r's/ !) +0122>  [3455  +./00<  [+A*BCCDD  +)*++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  M    +)*++7 AA${->,?@@AAA  +,-..:  ?    +4566B #$ [7899 r%cpi}t|dg"t|ddgt|dg|Sr*r-r!s r# _GetTuningJobParameters_to_mldevrJr/r%cpi}t|dg"t|ddgt|dg|Sr*r-r!s r#!_GetTuningJobParameters_to_vertexrVr/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_vertexrbs !) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"".8,d; .K.KLLL r%cli}t|dg tt|dg|||S)Nr})rrr!s r##_ListTuningJobsParameters_to_vertexr {sH !) +z"".# [8*%%y+ r%c Tit|dg!tdgt|dgt|dg!tdgt|dgt|dg.tdgfdt|dgDS)Nr4r5 nextPageTokennext_page_token tuningJobs tuning_jobsc2g|]}t|SrG)_TuningJob_from_vertexrHs r#rKz7_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%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|dgt|d gt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dgtdt|dgtdt|dgtdt|dg!t|dgt|dgt|dg!t|dgt|dg|S)Nrrcode_execution_resultcodeExecutionResultexecutable_codeexecutableCode file_datafileData function_call functionCallfunction_responsefunctionResponse inline_data inlineDatatextthoughtthought_signaturethoughtSignaturevideo_metadata videoMetadata tool_callzqtool_call parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode. tool_responsezutool_response parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode. part_metadatazupart_metadata parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.audio_transcriptionaudioTranscriptionmedia_processingmediaProcessingrrrzr!s r#rSrSs !) +*+,,8%&[;M:N(O(O +/011=  [2344  +)*++7%&[;L:M(N(NOOO + }%%1ZL${[M"B"BCCC +())5^$d;8I&J&JKKK ++,--9  [./00  + ''3\ND}o$F$FGGG +x  ,VHd;99::: + {##/YKkI;!?!?@@@ ++,--9  [./00  +()**6_%tK:J9K'L'LMMM + }%%1  9    +())5  ?    +())5  ?    +-.//;  [0122  +*+,,8%&[;M:N(O(O r%c i}t|dg0t|dgtt|dg||t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|d g!t|d gt|d g|S) NrrautoraterPromptautorater_promptautoraterResponseParseConfigautorater_response_parse_configparsedResponseConversionScorer!parsed_response_conversion_scorerexactMatchScorerexact_match_scorer)rrr$r!s r#/_ReinforcementTuningAutoraterScorer_from_vertexr< si !) +)*++7 $ 01 2 2I{   +)*++7&'k9si !) +*+,,8 " 12 3 3Y    +*+,,8%&[;M:N(O(O +9:;;G '( [<=>>  +;<==I )* [>?@@  +,-..:  [/011 r%c rit|dg.tdgfdt|dgDt|dg!tdgt|dgt|dg0tdgtt|dgS)Ncontentsc2g|]}t|SrG)rUrHs r#rKz9_ReinforcementTuningExample_to_vertex..os5    tY < <   r% referencessystem_instructionsystemInstruction)rrrUr!s `@r#%_ReinforcementTuningExample_to_vertexrEes !) + |$$0       [:,77    + ~&&2\ND|n$E$EFFF +,-..:  34 5 5y+   r%c :i}t|dg0t|dgtt|dg||t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|d g0t|d gtt|d g|||S) Nrrrrrrrxrrrsingle_reward_config)rrrLr>r!s r#$_ReinforcementTuningSpec_from_vertexrHsz !) +/011= "#= 67 8 8)[   +,-..:   [/011  +./00< !" [1233  +)*++7&'kr>s !) +)*++7 7 01 2 2I{   + ~&&2]OT+ ~%F%FGGG +-.//; ! [0122  +3455A '( [6788  +1233? %& [4566  +./00< "# [1233 r%c i}t|dg0t|dgtt|dg||t|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 g|S) NrSrRrUrTrWrVrYrXr[rZr]r\)rrr>r!s r#rArAs !) +*+,,8 5 12 3 3Y    + ''3\ND}o$F$FGGG +/011=  [2344  +6788D $% [9:;;  +4566B "# [7899  +1233?   [4566 r%c i}t|dg!t|dgt|dgt|dg0t|dgtt|dg|||S)Nspeaker voice_config voiceConfig)rr_VoiceConfig_to_vertexr!s r#rr3s !) + {##/YKkI;!?!?@@@ +'((4  ~. / /K   r%c ri}t|dg0t|dgtt|dg||t|dg!t|dgt|dgt|dg0t|dgtt|dg|||S)Nrarb language_code languageCodemulti_speaker_voice_configmultiSpeakerVoiceConfig)rrrcrr!s r#rrHs !) +'((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_mldevrlhss !) +x  ,WItK&::;;; +x  ,ZL${VH"="=>>> r%c i}t|dgtdt|dgtdt|dg,t|ddgdt|dgD|S)Ngcs_urizogcs_uri parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.vertex_dataset_resourcezvertex_dataset_resource parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.examplescg|]}|SrGrGrs r#rKz+_TuningDataset_to_mldev..s:::$:::r%ryr!s r#rrws !) + {##/  <    +1233?  M    + |$$0 Z ::${ZL99::: r%c>i}t|ddg}|d}|dkr4t|dg"t|ddgt|dgn|dkr4t|dg"t|ddgt|dgns|d kr4t|dg"t|d d gt|dgn9|d kr3t|dg"t|d dgt|dgt|ddg}|d}|dkr4t|dg"t|ddgt|dgn|dkr4t|dg"t|ddgt|dgns|d kr4t|dg"t|d d gt|dgn9|d kr3t|dg"t|d dgt|dgt|dgtd|S)Nr}r~rrnrrrrrrrrrrorpzpexamples parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.r2rs r#rrs !){Xx$899-,M... K)%%1  !#7 8 {YK ( ( +++ K)%%1  ')= > {YK ( ( && K)%%1  1 2 {YK ( ( ... K)%%1  $&: ; {YK ( ( {Xx$899-,M... K3455A  !#7 8 {67 8 8 +++ K3455A  ')= > {67 8 8 && K3455A  1 2 {67 8 8 ... K3455A  $&: ; {67 8 8  + |$$0  5   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)Nr4r5r+state createTime create_timer\ startTime start_time completeTimeend_time updateTime update_timerZrr_self tuned_model)rrrt_tuning_job_statusrlr!s r#_TuningJob_from_mldevrsf !) +)*++7'(${=Ng0t|d?gtt|d>g|||S)@Nr4r5r+rtrurvrwrxendTimerzr{r|errorrZrr tunedModelr~rrrsupervised_tuning_specrpreference_optimization_specrdistillation_specrreinforcement_tuning_spectuningDataStatstuning_data_statsrrqpartnerModelTuningSpecpartner_model_tuning_specrrjrreevaluateDatasetRunsevaluate_dataset_runscg|]}|SrGrGrs r#rKz*_TuningJob_from_vertex..sEEE$EEEr% experimentfullFineTuningSpecfull_fine_tuning_specrkrrp pipelineJob pipeline_jobserviceAccountservice_accountrrXtuningJobStatetuning_job_state veoTuningSpecveo_tuning_spectuningJobMetadatatuning_job_metadataveoLoraTuningSpecveo_lora_tuning_specdistillationSamplingSpecdistillation_sampling_spec)rrrrrrHrrr!s r#rrs' !) +)*++7'(${=N $% [3455  +*+,,8 % 12 3 3Y    +)*++7'(${=N>> + }%%1\ND{m$D$DEEE + ''3^$d;&H&HIII +()**6&'kz7_ValidateRewardResponse_from_vertex..G sKKK$!QAKKKr%)rritemsr!s r##_ValidateRewardResponse_from_vertexr2 s !) +)*++7'(${=N> ?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. rNr, {name}:cancelrrr}postrrrrrr)r _CancelTuningJobParametersrrr1rrr.rrr}rr rrrrrrr9r6CancelTuningJobResponserrrrr5rrs r#cancelzTunings.cancel s6 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%rrr}rrrc 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. rrrr}rNr,r rrr}rrrrrrr)r !_CreateTuningJobParametersPrivaterrrzrrrrrr}rr rrrrrrrrrrrrr5rrrrrr}rrrrrrrrrs r#_tunez Tunings._tunem st&=') O   $   5   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. rziThis method is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.Nr, tunedModelsrrr}rrrrrrr)r rrrrzrrrrrr}rr rrrrrrrTuningOperationrrrrr5rrs r# _tune_mldevzTunings._tune_mldev su&=') O    4   @ 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%rGrrr}rrrrGrrc 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||SaValidates a reinforcement tuning reward configuration. Allows users to validate a reinforcement tuning reward configuration against a sample response and example before creating a reinforcement tuning job, so that they can iterate on the reward configuration without having to create a tuning job each time. Args: parent: The resource name of the Location to validate the reward in, e.g. `projects/{project}/locations/{location}`. sample_response: The sample response for validating the reward configuration. example: The example to validate the reward configuration. single_reward_config: Single reward function configuration for reinforcement tuning. Mutually exclusive with composite_reward_config. composite_reward_config: Composite reward function configuration for reinforcement tuning. Mutually exclusive with single_reward_config. config: Optional parameters for the request. Returns: A ValidateRewardResponse with the computed reward(s). )rrrrGrrr}rNr,z5{parent}/tuningJobs:validateReinforcementTuningRewardrrr}rrrrrrr)r _ValidateRewardParametersrrrzrrrrrr}rr rrrrrrrValidateRewardResponserrrrr5rrrrrrGrrr}rrrrrrrrrs r#validate_rewardzTunings.validate_reward' sJ5'1 7 O   $G  5   9 4l&))&11 G C N N     G##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM  9 m/>> ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !?L$&+%7 &&&L" %%l333 r%c|||}|jrC|jjr7|jj+t |j|jj|SNr)rprojectrrrrr _IpythonUtilsdisplay_experiment_buttonrr+r}jobs r#rz Tunings.get sq ))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/rb)tuned_model_name checkpoint_id)rrr}rjzEEvaluation config must have at least one metric and an output config.)rrr}rzOperation name is required.z /operations/r)r+rt)tuning_job_resourcerGrr startswithrr PreTunedModelr isinstancedictEvaluationConfigrrrzrangelenMetricmodel_validaterjrrr+ partitionrJobStateJOB_STATE_QUEUEDr display_model_tuning_button) rrrr}rr tuning_jobvalidated_evaluation_configrji operationrs r#tunez Tunings.tune s  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) rr)rrrr} list_requests r#listz Tunings.list s7&:L  & !!   r%)"__name__ __module__ __qualname__strrr GetTuningJobConfigOrDictrrListTuningJobsConfigOrDictrrCancelTuningJobConfigOrDictrrPreTunedModelOrDictTuningDatasetOrDictCreateTuningJobConfigOrDictrrr ContentOrDict ReinforcementTuningExampleOrDict+SingleReinforcementTuningRewardConfigOrDict.CompositeReinforcementTuningRewardConfigOrDictValidateRewardConfigOrDictrrrr experimental_warningr&rr*rGr%r#rrh s :> \\\ \u56 \   \\\\~EIFFF @AF #FFFFX=A [[[ [u89 [  $ [[[[@#'=A<@ [[[3-[  9: [ 1 [ u89 [ [[[[@#'=A<@ [[[3-[  9: [ 1 [ u89 [ [[[[J  ;?ssss* s 5 s %  ; s (  > su78s #ssssr:>  u56    $ 7+=A MMMM1 M u89 M   MMM M`EI @A U_r%rceZdZdddedeejdejfdZdddeej dej fdZ dddedeej dej fdZdddd d eed eejd ejdeejdejf d Zdddd d eed eejd ejdeejdejf dZdddddedejdejdeejdeejdeejdejfdZdddedeejdejfdZejdddd ed ejdeejdejfdZdddeej de ejfdZ!dS) AsyncTuningsNrr+r}rc 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| | Sr)r rrrrrrrrrr}rr rr async_requestrrrrrrrrrrr5rrs r#rzAsyncTunings._get s3 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| | Sr)r rrrrzr rrrrr}rr rrr>rrrrrrrrrr5rrs r#rzAsyncTunings._list s5O   $   5   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. rNr,rrrr}rrrrrrr)r rrrr1rrr.rrr}rr rrr>rrrr9r6rrrrrr5rrs r#rzAsyncTunings.cancel s6 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%rrrrc &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| | Sr)r rrrrzrrrrrr}rr rrr>rrrrrrrrrr5rrs r#rzAsyncTunings._tune& s&=') O   $   5   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| | Sr)r rrrrzrrrrrr}rr rrr>rrrrrrrrrr5rrs r#rzAsyncTunings._tune_mldev s&=') O    4   @ 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%rrrrrGrrc *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||Sr)r rrrrzrrrrrr}rr rrr>rrrrrrrrrr5rrs r#rzAsyncTunings.validate_reward sJ5'1 7 O   $G  5   9 4l&))&11 G C N N     G##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM  9 m/>> ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !?L$&+%7 &&&L" %%l333 r%cK|||d{V}|jrC|jjr7|jj+t |j|jj|Srr r s r#rzAsyncTunings.getUs  tF 33 3 3 3 3 3 3C    %   $ 0--^"*. Jr%rcK|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 |Srr) rrrr}rrr"rjr$r%rs r#r&zAsyncTunings.tunegs   ;   { + +*   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) rrN)rrr(s r#r*zAsyncTunings.listsM&:L jjj''''''''   r%)"r+r,r-r.rr r/rrr0rrr1rrr2r3r4rrrr5r6r7r8r9rrrr r:r&rr*rGr%r#r<r< s :> ^^^ ^u56 ^   ^^^^BEIHHH @AH #HHHH\=A [[[ [u89 [  $ [[[[@#'=A<@ [[[3-[  9: [ 1 [ u89 [ [[[[@#'=A<@ [[[3-[  9: [ 1 [ u89 [ [[[[J  ;?ssss* s 5 s %  ; s (  > su78s #ssssr:>  u56    $ 7+=A GGGG1 G u89 G   GGG GTEI @A%/"r%r<c XeZdZUdZeZeeed<ede ee ffdZ ede fdZ edefdZededejedefd Ze dd ed ed eeddfdZedededdfdZededdfdZdS)r z6Temporary class to hold the IPython related functions.displayed_experimentsrcTddl}d|jvrddlm}|jjSdS)NrIPython) get_ipython)sysmodulesrJrK __class__r+)rMrKs r#_get_ipython_shell_namez%_IpythonUtils._get_ipython_shell_namesAJJJCK%%%%%% []] $ -- 2r%cNttS)N)boolr rPrGr%r#is_ipython_availablez"_IpythonUtils.is_ipython_availables  5577 8 88r%cdS)z8Returns the HTML style markup to support custom buttons.a rGrGr%r# _get_stylesz_IpythonUtils._get_styless 1 1 r%markerresource_partsctt|D]1}|||kr#|dzt|kr ||dzcS2dS)z,Returns the part after the marker text part.rrL)rr)rVrWr$s r#_parse_resource_namez"_IpythonUtils._parse_resource_namesd 3~&& ' '%%  f $ $Q^1D1D)D)Da!e$$$$ 2r% open_in_newr%urliconNc 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) rrzuuidr.uuid4r rUIPython.displayr^r_) r%r[r\ CLOUD_UI_URLr` button_idhtmlr^r_s r# _display_linkz_IpythonUtils._display_link#sB6L >>, ' 'O M,MMM N NNKKK;DJJLL(9(9;;I  " " $ $  -6  FO  :>      D  4=   /2   "   D"('''''$$$$$$ GDDJJr%rrc: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)r rSrHsplitrfadd)rrrWlocationexperiment_nameuris r#r z'_IpythonUtils.display_experiment_buttonWs  . . 0 0 < < < f %%c**Na H$R(O L  6 6O 6 6 6 7 #' # # $  13 BBB'++J77777r%rctsdS|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} Nrhrrirjz?https://console.cloud.google.com/vertex-ai/generative/language/z locations/z/tuning/tuningJob/z ?project=zView Tuning Jobr&)r rSrlrf)rrWrrn tuning_job_idrps r#r!z)_IpythonUtils.display_model_tuning_buttonvs  - - / /  f(..s33NQGa H"2&M J Bx B B= B B C g     13?????r%)rZ)r+r,r-__doc__setrHr.__annotations__ staticmethodrrrPrRrSrUbuiltinsr*rYrrfr r!rGr%r#r r s>>$'CEES)))5c?<9999<93s333<3j#+=#5 <1>11 11!)#1 111<1f8C8#8$888<8<@s@t@@@<@@@r%r )NN)Orwrloggingtypingrrr urllib.parserrLrr r rr r rr rpagersrr getLoggerloggerrr.objectr$r(r.r1r6r9r?rBrLrOrUr{rrrrrrrrr r'rrrr rrrSr<r>rErHrPr>rArrrlrrrrrrrrrc BaseModulerr<r rGr%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;?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(^ 0/3;?tttCH~v-.tDcN+t%S#X 678t #s(^ ttttr/3;?DDtCH~v-.DDcN+D%S#X 678D #s(^ DDDDR /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(^ """"N/3;?++tCH~v-.+DcN++%S#X 678+ #s(^ ++++`/3;???tCH~v-.?DcN+?%S#X 678? #s(^ ????H/3;?  tCH~v-. DcN+ %S#X 678  #s(^    J/3;?  tCH~v-. DcN+ %S#X 678  #s(^    J/3;?xxtCH~v-.xDcN+x%S#X 678x #s(^ xxxxz/3;?JJtCH~v-.JDcN+J%S#X 678J #s(^ JJJJ^/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;?SStCH~v-.SDcN+S%S#X 678S #s(^ SSSSp/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(^ B/3;?..tCH~v-..DcN+.%S#X 678. #s(^ ....f/3;?..tCH~v-..DcN+.%S#X 678. #s(^ ....f/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(^ </3;?MMtCH~v-.MDcN+M%S#X 678M #s(^ MMMMd/3;?88tCH~v-.8DcN+8%S#X 6788 #s(^ 8888z/3;?rrtCH~v-.rDcN+r%S#X 678r #s(^ rrrrn/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(^ ////h/3;?tCH~v-.DcN+%S#X 678 #s(^ :/3;?tCH~v-.DcN+%S#X 678 #s(^ 6r r r r r k$r r r jp p p p p ;)p p p f|@|@|@|@|@|@|@|@|@|@r%