o "iW @sx ddlZddlZddlmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd l mZdd l mZdd lmZmZed Z  d[d eeeefefdeeeefdeeeeefefdeeeffddZ  d[d eeeefefdeeeefdeeeeefefdeeeffddZ  d[d eeeefefdeeeefdeeeeefefdeeeffddZ  d[d eeeefefdeeeefdeeeeefefdeeeffddZ  d[d eeeefefdeeeefdeeeeefefdeeeffddZ  d[d eeeefefdeeeefdeeeeefefdeeeffddZ  d[d eeeefefdeeeefdeeeeefefdeeeffddZ  d[d eeeefefdeeeefdeeeeefefdeeeffdd Z!  d[d eeeefefdeeeefdeeeeefefdeeeffd!d"Z"  d[d eeeefefdeeeefdeeeeefefdeeeffd#d$Z#  d[d eeeefefdeeeefdeeeeefefdeeeffd%d&Z$  d[d eeeefefdeeeefdeeeeefefdeeeffd'd(Z%  d[d eeeefefdeeeefdeeeeefefdeeeffd)d*Z&  d[d eeeefefdeeeefdeeeeefefdeeeffd+d,Z'  d[d eeeefefdeeeefdeeeeefefdeeeffd-d.Z(  d[d eeeefefdeeeefdeeeeefefdeeeffd/d0Z)  d[d eeeefefdeeeefdeeeeefefdeeeffd1d2Z*  d[d eeeefefdeeeefdeeeeefefdeeeffd3d4Z+  d[d eeeefefdeeeefdeeeeefefdeeeffd5d6Z,  d[d eeeefefdeeeefdeeeeefefdeeeffd7d8Z-  d[d eeeefefdeeeefdeeeeefefdeeeffd9d:Z.  d[d eeeefefdeeeefdeeeeefefdeeeffd;d<Z/  d[d eeeefefdeeeefdeeeeefefdeeeffd=d>Z0  d[d eeeefefdeeeefdeeeeefefdeeeffd?d@Z1  d[d eeeefefdeeeefdeeeeefefdeeeffdAdBZ2  d[d eeeefefdeeeefdeeeeefefdeeeffdCdDZ3  d[d eeeefefdeeeefdeeeeefefdeeeffdEdFZ4  d[d eeeefefdeeeefdeeeeefefdeeeffdGdHZ5  d[d eeeefefdeeeefdeeeeefefdeeeffdIdJZ6  d[d eeeefefdeeeefdeeeeefefdeeeffdKdLZ7  d[d eeeefefdeeeefdeeeeefefdeeeffdMdNZ8  d[d eeeefefdeeeefdeeeeefefdeeeffdOdPZ9  d[d eeeefefdeeeefdeeeeefefdeeeffdQdRZ:  d[d eeeefefdeeeefdeeeeefefdeeeffdSdTZ;GdUdVdVe j<Z=GdWdXdXe j<Z>GdYdZdZZ?dS)\N)AnyOptionalUnion) urlencode) _api_module)_common) _transformers)types)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.tunings from_object parent_object root_objectreturncCi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurRt|dgtt|dg|||S) N samplingCountsampling_count flipEnabled flip_enabledautoraterModelautorater_modelgenerationConfiggeneration_config)getvsetv_GenerationConfig_from_vertexrrr to_objectr!Y/var/www/html/crm_dreinet/venv_linux/lib/python3.10/site-packages/google/genai/tunings.py_AutoraterConfig_from_vertex# r#cCr) Nrrrrrrrr)rr_GenerationConfig_to_vertexrr!r!r"_AutoraterConfig_to_vertex>r$r&cC0i}t|dgdurt|ddgt|dg|SNname_urlrrrr!r!r"#_CancelTuningJobParameters_to_mldevYr,cCr'r(r+rr!r!r"$_CancelTuningJobParameters_to_vertexer-r.cC.i}t|dgdurt|dgt|dg|SNsdkHttpResponsesdk_http_responser+rr!r!r"#_CancelTuningJobResponse_from_mldevq r3cCr/r0r+rr!r!r"$_CancelTuningJobResponse_from_vertexr4r5cCs<i}t|dgdurtdt|dgdur!t|dgt|dgt|dgdur-tdt|dgdurAt|gdt|dgt|d gdurUt|gd t|d gt|d gduratd t|d gdurmtdt|dgdurytdt|dgdurtdt|dgdurtdt|dgdurt|gdt|dgt|dgdurt|gdt|dgt|dgdurtdt|dgdurtdt|dgdurtdt|dgdurtd t|d!gdurtd"t|d#gdurtd$t|d%gdurtd&t|d'gdurtd(|S))Nvalidation_datasetz)rWr]r?)rYr]r?)r[r]r?r@exportLastCheckpointOnlyrB)rWr] adapterSize)rYr]r_)r[r]r_rC tuningModerDcustomBaseModelrE)rWr]rF)r[r]rFrG)rWr]rH)r[r]rHrIevaluationConfigrJrK)rYr]rKrLbaseTeacherModelrMtunedTeacherModelSourcerN)r[r]sftLossWeightMultiplierrO outputUrirPencryptionSpec)rr"_TuningValidationDataset_to_vertex_EvaluationConfig_to_vertexrrrr discriminatorr!r!r" _CreateTuningJobConfig_to_vertexs                                                  rlcCi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur:tt|dg||t|dgdurLtt|dg|||SN base_model baseModelpre_tuned_model preTunedModeltraining_datasetrT)rr_TuningDataset_to_mldevrSrr!r!r"*_CreateTuningJobParametersPrivate_to_mldevrucCrmrn)rr_TuningDataset_to_vertexrlrr!r!r"+_CreateTuningJobParametersPrivate_to_vertex3rvrxcCi}t|dgdurt|dgtt|dgt|dgdur+t|dgt|dgt|dgdurBt|dgtt|dg||t|dgdurYt|dgtt|dg|||S)Nmetrics outputConfig output_configautoraterConfigautorater_configinferenceGenerationConfiginference_generation_config)rrt t_metricsr#rrr!r!r"_EvaluationConfig_from_vertexL0  rcCry)Nrzr|r{r~r}rr)rrrrr&r%rr!r!r"riorricCs^i}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurNt|dgt|dgt|d gdurat|d gt|d gt|d gdurtt|d gt|d gt|d gdurt|d gt|d gt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdur t|dgt|dgt|dgdur!t|dgt|dgt|dgdur5t|dgt|dgt|dgdurIt|d gt|dgt|d!gdur]t|d"gt|d!gt|d#gdurqt|d#gt|d#gt|d$gdurt|d%gt|d$gt|d&gdurt|d'gt|d&gt|d(gdurt|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+rr!r!r"rs       rcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurNt|dgt|dgt|d gdurat|d gt|d gt|d gdurtt|d gt|d gt|d gdurt|d gt|d gt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdur t|dgt|dgt|dgdur!t|dgt|dgt|dgdur5t|dgt|dgt|dgdurMt|d gtt|dg||t|d!gdurat|d"gt|d!gt|d#gdurut|d#gt|d#gt|d$gdurt|d%gt|d$gt|d&gdurt|d'gt|d&gt|d(gdurt|d)gt|d(gt|d*gdurtd+|S),Nrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrenable_enhanced_civic_answerszFenable_enhanced_civic_answers parameter is not supported in Vertex AI.)rr_SpeechConfig_to_vertexrRrr!r!r"r%s       r%cCr'r(r+rr!r!r" _GetTuningJobParameters_to_mldevwr-rcCr'r(r+rr!r!r"!_GetTuningJobParameters_to_vertexr-rcCi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dgt|dgdur>t|ddgt|dg|SN page_size_querypageSize page_token pageTokenfilterr+rr!r!r"_ListTuningJobsConfig_to_mldev rcCrrr+rr!r!r"_ListTuningJobsConfig_to_vertexrrcC,i}t|dgdurtt|dg|||SNrT)rrrr!r!r""_ListTuningJobsParameters_to_mldev rcCrr)rrrr!r!r"#_ListTuningJobsParameters_to_vertexrrcit|dgdurtdgt|dgt|dgdur(tdgt|dgt|dgdurCtdgfddt|dgDS) Nr1r2 nextPageTokennext_page_token tunedModels tuning_jobscg|]}t|qSr!)_TuningJob_from_mldev.0itemrr r!r"  z6_ListTuningJobsResponse_from_mldev..r+rrrr!rr""_ListTuningJobsResponse_from_mldev    rcr) Nr1r2rr tuningJobsrcrr!)_TuningJob_from_vertexrrr!r"r rz7_ListTuningJobsResponse_from_vertex..r+rr!rr"#_ListTuningJobsResponse_from_vertexrrcs>it|dgdurtdgfddt|dgDS)Nspeaker_voice_configsspeakerVoiceConfigscrr!)_SpeakerVoiceConfig_to_vertexrrr!r"rrz6_MultiSpeakerVoiceConfig_to_vertex..r+rr!rr""_MultiSpeakerVoiceConfig_to_vertexs   rcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur4tdt|dgdur@td|S) N mime_typemimeTypevoice_sample_audiovoiceSampleAudio consent_audioz6consent_audio parameter is not supported in Vertex AI.voice_consent_signaturez@voice_consent_signature parameter is not supported in Vertex AI.rrrRrr!r!r" _ReplicatedVoiceConfig_to_vertex&s  rcCs\i}t|dgdurt|dgt|dgt|dgdur,t|dgtt|dg|||S)Nspeaker voice_config voiceConfig)rr_VoiceConfig_to_vertexrr!r!r"rAsrcCsi}t|dgdurt|dgtt|dg||t|dgdur,t|dgt|dgt|dgdurCt|dgtt|dg|||S)Nrr language_code languageCodemulti_speaker_voice_configmultiSpeakerVoiceConfig)rrrrrr!r!r"rVs,  rcCsTi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dg|S)Nr)modelendpointr+rr!r!r"_TunedModel_from_mldevvs rcCsji}t|dgdurtdt|dgdurtdt|dgdur3t|ddgddt|dgD|S)Ngcs_uriz1gcs_uri parameter is not supported in Gemini API.vertex_dataset_resourcezAvertex_dataset_resource parameter is not supported in Gemini API.examplescSg|]}|qSr!r!rr!r!r"rz+_TuningDataset_to_mldev..rQrr!r!r"rtsrtcCs|i}t|ddg}|durd}|dkr(t|dgdur't|ddgt|dgn1|dkrAt|dgdur@t|ddgt|dgn|d krYt|dgdurYt|d d gt|dgt|ddg}|durfd}|dkrt|d gdur~t|ddgt|d gn1|dkrt|d gdurt|ddgt|d gn|d krt|d gdurt|d d gt|d gt|d gdurtd|S)NrTrUrVrrWtrainingDatasetUrirXrYrZr[promptDatasetUrirrz1examples parameter is not supported in Vertex AI.rrjr!r!r"rwsp      rwcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur>t|dgtt|dgt|dgdurQt|dgt|dgt|ddgdurft|d gt|ddgt|dd gdur{t|d gt|dd gt|d gdurt|d gt|d gt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgtt|dg|||S)Nr1r2r)state createTime create_timer; startTime start_time completeTimeend_time updateTime update_timer9rpro_self tuned_model)rrrt_tuning_job_statusrrr!r!r"rsT  rcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur>t|dgtt|dgt|dgdurQt|dgt|dgt|dgdurdt|dgt|dgt|d gdurwt|d gt|d gt|d gdurt|d gt|d gt|d gdurt|d gt|d gt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdur$t|dgt|dgt|dgdur8t|dgt|dgt|dgdurLt|dgt|dgt|dgdur`t|d gt|dgt|d!gdurxt|d"gtt|d!g||t|d#gdurt|d$gt|d#gt|d%gdurt|d&gd'd(t|d%gDt|d)gdurt|d)gt|d)gt|d*gdurt|d+gt|d*gt|d,gdurt|d,gt|d,gt|d-gdurt|d.gt|d-gt|d/gdur t|d0gt|d/gt|d1gdurt|d2gt|d1gt|d3gdur1t|d4gt|d3gt|d5gdurEt|d6gt|d5gt|d7gdurYt|d8gt|d7gt|d9gdurmt|d:gt|d9gt|d;gdurt|d. experimentfullFineTuningSpecfull_fine_tuning_specrJrfrO pipelineJob pipeline_jobserviceAccountservice_accountr\r7tuningJobStatetuning_job_state veoTuningSpecveo_tuning_specdistillationSamplingSpecdistillation_sampling_spectuningJobMetadatatuning_job_metadata)rrrrrrr!r!r"rs        rcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurNt|dgt|dgt|dgdurat|dgt|dg|S)Nr1r2r)metadatadonerr+rr!r!r"_TuningOperation_from_mldevsrcCsTi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dg|S)NrvalidationDatasetUrirr+rr!r!r"rhs rhcCs\i}t|dgdurt|dgtt|dg||t|dgdur,t|dgt|dg|S)Nreplicated_voice_configreplicatedVoiceConfigprebuilt_voice_configprebuiltVoiceConfig)rrrrr!r!r"rs$   rc @heZdZdddedeejdejfddZdddeej dej fdd Z dddedeej dej fd d Zdddd d eedeejdejdeejdejf ddZdddd d eedeejdejdeejdejf ddZdddedeejdejfddZedddd edejdeejdejfddZdddeej deejfddZdS)TuningsNrTr)rTrc CRtj||d}|jjr!t|d|}|d}|rd|}nd}nt|d|}|d}|r4d|}nd}|d}|rF|dt|}| ddd}|j dur]|j j dur]|j j }t |}t |}|jd|||} | jsuint| j} |jjrt| d|} |jjst| d|} tjj| |d } tj| jd | _|j| | S zzGets a TuningJob. Args: name: The resource name of the tuning job. Returns: A TuningJob object. r)rTNr*z{name}r?rTgetresponsekwargsheaders)r _GetTuningJobParameters _api_clientvertexairr) format_maprrpoprT http_optionsrconvert_to_dictencode_unserializable_typesrequestbodyjsonloadsrr TuningJob_from_response model_dump HttpResponser.r2_verify_response selfr)rTparameter_model request_dictrequest_url_dictpath query_paramsr4r+ response_dict return_valuer!r!r"_gets^           z Tunings._getc CsPtj|d}|jjr t|d|}|d}|rd|}nd}nt|d|}|d}|r3d|}nd}|d}|rE|dt|}| ddd}|j dur\|j j dur\|j j }t |}t |}|jd|||}|jstint|j} |jjrt| d|} |jjst| d|} tjj| |d } tj|jd | _|j| | S Nr$r*rrrr(rTr)r*r-)r _ListTuningJobsParametersr0r1rr)r2rrr3rTr4rr5r6r7r8r9r:rrListTuningJobsResponser<r=r>r.r2r? rArTrBrCrDrErFr4r+rGrHr!r!r"_listMs\           z Tunings._listc Cr%) zPCancels a tuning job. Args: name (str): TuningJob resource name. r'Nr* {name}:cancelrr(rTpostr*r-)r _CancelTuningJobParametersr0r1r.r)r2r,rr3rTr4rr5r6r7r8r9r:r5r3CancelTuningJobResponser<r=r>r.r2r?r@r!r!r"cancelsb            zTunings.cancelrorqrTrorqrscCstj||||d}|jjstdt|d|}|d}|r$d|}nd}|d} | r6|dt| }| ddd} |j durM|j j durM|j j } t |}t |}|jd ||| } | jseint| j} |jjrut| d|} tjj| |d } tj| jd | _|j| | S aCreates 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. rorqrsrTz6This method is only supported in the Vertex AI client.Nr*rrr(rTrPr*r-)r !_CreateTuningJobParametersPrivater0r1rRrxr)r2rr3rTr4rr5r6r7r8r9r:rr;r<r=r>r.r2r?rArorqrsrTrBrCrDrErFr4r+rGrHr!r!r"_tunesR         z Tunings._tunecCstj||||d}|jjrtdt|d|}|d}|r$d|}nd}|d} | r6|dt| }| ddd} |j durM|j j durM|j j } t |}t |}|jd ||| } | jseint| j} |jjsut| d|} tjj| |d } tj| jd | _|j| | S aCreates 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. rVz=This method is only supported in the Gemini Developer client.Nr*rrr(rTrPr*r-)r rWr0r1rRrur)r2rr3rTr4rr5r6r7r8r9r:rTuningOperationr<r=r>r.r2r?rXr!r!r" _tune_mldev(sV         zTunings._tune_mldevcCs@|j||d}|jr|jjr|jjdurtj|j|jjd|SNr')r projectrIr r0r1r^ _IpythonUtilsdisplay_experiment_buttonrAr)rTjobr!r!r"r)us z Tunings.getSThe SDK's tuning implementation is experimental, and may change in future versions.c Cs|jjr|dr#d}|rt|dd}tj||d}|j|||d}nd}|durzt|dddurzt|d}t|trCtj di|}|j rI|j sMt dt t|j D]} t|j | trjtj|j | |j | <qTt|tru||d<n||_|}|j|||d}|durt|dddur||_n2|j|||d} | jdurd| jvr| jd} n| jdurt d | jd d } tj| tjjd }|jr|jjrtj|jd |SNz projects/rA)tuned_model_name checkpoint_id)rqrsrTrIzEEvaluation config must have at least one metric and an output config.)rorsrTrzOperation name is required.z /operations/r)r)r)tuning_job_resourcer!r0r1 startswithgetattrr PreTunedModelrY isinstancedictEvaluationConfigrzr|rRrangelenMetricmodel_validaterIr\rr) partitionr;JobStateJOB_STATE_QUEUEDr`display_model_tuning_button) rArorsrTrgrq tuning_jobvalidated_evaluation_configrIi operationrfr!r!r"tunes         z Tunings.tunecCs|j}td||j|d|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$)rNrrArT list_requestr!r!r"lists z Tunings.list)__name__ __module__ __qualname__strrr GetTuningJobConfigOrDictr;rIListTuningJobsConfigOrDictrLrNCancelTuningJobConfigOrDictrRrSPreTunedModelOrDictTuningDatasetOrDictCreateTuningJobConfigOrDictrYr[r\r)rexperimental_warningr|rrr!r!r!r"r#s P F Q N Q   Pr#c @r") AsyncTuningsNr$r)rTrc Ztj||d}|jjr"t|d|}|d}|rd|}nd}nt|d|}|d}|r5d|}nd}|d}|rG|dt|}| ddd}|j dur^|j j dur^|j j }t |}t |}|jd|||IdH} | jsyint| j} |jjrt| d|} |jjst| d|} tjj| |d } tj| jd | _|j| | Sr&)r r/r0r1rr)r2rrr3rTr4rr5r6 async_requestr8r9r:rrr;r<r=r>r.r2r?r@r!r!r"rIsd            zAsyncTunings._getc sXtj|d}|jjr!t|d|}|d}|rd|}nd}nt|d|}|d}|r4d|}nd}|d}|rF|dt|}| ddd}|j dur]|j j dur]|j j }t |}t |}|jd|||IdH}|jsxint|j} |jjrt| d|} |jjst| d|} tjj| |d } tj|jd | _|j| | SrJ)r rKr0r1rr)r2rrr3rTr4rr5r6rr8r9r:rrrLr<r=r>r.r2r?rMr!r!r"rNIsb            zAsyncTunings._listc r) zaCancels a tuning job asynchronously. Args: name (str): A TuningJob resource name. r'Nr*rOrr(rTrPr*r-)r rQr0r1r.r)r2r,rr3rTr4rr5r6rr8r9r:r5r3rRr<r=r>r.r2r?r@r!r!r"rSsd             zAsyncTunings.cancelrTrorqrscs&tj||||d}|jjstdt|d|}|d}|r%d|}nd}|d} | r7|dt| }| ddd} |j durN|j j durN|j j } t |}t |}|jd ||| IdH} | jsiint| j} |jjryt| d|} tjj| |d } tj| jd | _|j| | SrU)r rWr0r1rRrxr)r2rr3rTr4rr5r6rr8r9r:rr;r<r=r>r.r2r?rXr!r!r"rYsT          zAsyncTunings._tunecs&tj||||d}|jjrtdt|d|}|d}|r%d|}nd}|d} | r7|dt| }| ddd} |j durN|j j durN|j j } t |}t |}|jd ||| IdH} | jsiint| j} |jjsyt| d|} tjj| |d } tj| jd | _|j| | SrZ)r rWr0r1rRrur)r2rr3rTr4rr5r6rr8r9r:rr[r<r=r>r.r2r?rXr!r!r"r\& sX          zAsyncTunings._tune_mldevcsH|j||dIdH}|jr"|jjr"|jjdur"tj|j|jjd|Sr]r_rbr!r!r"r)s s zAsyncTunings.getrdc s|jjr|dr'd}|rt|dd}tj||d}|j|||dIdH}n|durzt|dddurzt|d}t|trEtj di|}|j rK|j sOt dt t|j D]}t|j |trltj|j ||j |<qVt|trw||d<n||_|j|||dIdH}n5|j|||dIdH} | jdurd| jvr| jd} n| jdurt d | jd d } tj| tjjd }|jr|jjrtj|jd |Sreri) rArorsrTrgrqrxrIrzr{rfr!r!r"r| sv          zAsyncTunings.tunecs$|j}td||j|dIdH|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) rr$N)rNr r}r!r!r"r szAsyncTunings.list)rrrrrr rr;rIrrLrNrrRrSrrrrYr[r\r)rrr|r rr!r!r!r"rs R H Q N Q   Jrc @seZdZUdZeZeeed<ede ee ffddZ ede fddZ edefdd Zed ed eedefd d Ze ddededeeddfddZedededdfddZededdfddZdS)r`z6Temporary class to hold the IPython related functions.displayed_experimentsrcCs,ddl}d|jvrddlm}|jjSdS)NrIPython) get_ipython)sysmodulesrr __class__r)rrr!r!r"_get_ipython_shell_name s    z%_IpythonUtils._get_ipython_shell_namecCs ttS)N)boolr`rr!r!r!r"is_ipython_available s z"_IpythonUtils.is_ipython_availablecCsdS)z8Returns the HTML style markup to support custom buttons.a r!r!r!r!r" _get_styles sz_IpythonUtils._get_stylesmarkerresource_partscCsBtt|D]}|||kr|dt|kr||dSqdS)z,Returns the part after the marker text part.rr)rprq)rrrzr!r!r"_parse_resource_name7 s z"_IpythonUtils._parse_resource_name open_in_newtexturliconNc Csd}||std|dddl}dt|}dtd|d |d |d |d }|d |d|d|d7}ddlm}ddlm }|||dS)a&Creates and displays the link to open the Vertex 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) rjrRuuidruuid4r`rIPython.displayrr) rrr CLOUD_UI_URLr button_idhtmlrrr!r!r" _display_link? s4    z_IpythonUtils._display_linkr r^cCsltr |tjvr dS|d}|d}|d}d|d|dd|}td|d tj|dS) aFunction to generate a link bound to the Vertex experiment. Args: experiment: The Vertex 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`rrsplitradd)r r^rlocationexperiment_nameurir!r!r"ras s   z'_IpythonUtils.display_experiment_buttonrhcCs^tsdS|d}|d}|d}|d}dd|d|d |}td |d dS) zFunction to generate a link bound to the Vertex model tuning job. Args: tuning_job_resource: The Vertex 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|)r`rrr)rhrr^r tuning_job_idrr!r!r"rw s z)_IpythonUtils.display_model_tuning_button)r)rrr__doc__setrr__annotations__ staticmethodrrrrrrrrrrrarwr!r!r!r"r` s4  5 3r`)NN)@r9loggingtypingrrr urllib.parserrrrr rr r rr rpagersr r getLoggerloggerrnrobjectr#r&r,r.r3r5rSrlrurxrrirr%rrrrrrrrrrrrrrtrwrrrrhr BaseModuler#rr`r!r!r!r"s                      o  $      %  %  o  z                        "      A  =         }{