o
    vyic;                    @   s
  d dl Z d dlZd dlmZmZmZ d dlmZ ddlm	Z	 ddlm
Z
 ddlmZ ddlmZ dd	l
mZ dd
l
mZ ddlmZmZ edZ		dQdeeeef ef deeeef  deeeeef ef  deeef fddZ		dQdeeeef ef deeeef  deeeeef ef  deeef fddZ		dQdeeeef ef deeeef  deeeeef ef  deeef fddZ		dQdeeeef ef deeeef  deeeeef ef  deeef fddZ		dQdeeeef ef deeeef  deeeeef ef  deeef fddZ		dQdeeeef ef deeeef  deeeeef ef  deeef fddZ		dQdeeeef ef deeeef  deeeeef ef  deeef fddZ 		dQdeeeef ef deeeef  deeeeef ef  deeef fdd Z!		dQdeeeef ef deeeef  deeeeef ef  deeef fd!d"Z"		dQdeeeef ef deeeef  deeeeef ef  deeef fd#d$Z#		dQdeeeef ef deeeef  deeeeef ef  deeef fd%d&Z$		dQdeeeef ef deeeef  deeeeef ef  deeef fd'd(Z%		dQdeeeef ef deeeef  deeeeef ef  deeef fd)d*Z&		dQdeeeef ef deeeef  deeeeef ef  deeef fd+d,Z'		dQdeeeef ef deeeef  deeeeef ef  deeef fd-d.Z(		dQdeeeef ef deeeef  deeeeef ef  deeef fd/d0Z)		dQdeeeef ef deeeef  deeeeef ef  deeef fd1d2Z*		dQdeeeef ef deeeef  deeeeef ef  deeef fd3d4Z+		dQdeeeef ef deeeef  deeeeef ef  deeef fd5d6Z,		dQdeeeef ef deeeef  deeeeef ef  deeef fd7d8Z-		dQdeeeef ef deeeef  deeeeef ef  deeef fd9d:Z.		dQdeeeef ef deeeef  deeeeef ef  deeef fd;d<Z/		dQdeeeef ef deeeef  deeeeef ef  deeef fd=d>Z0		dQdeeeef ef deeeef  deeeeef ef  deeef fd?d@Z1		dQdeeeef ef deeeef  deeeeef ef  deeef fdAdBZ2		dQdeeeef ef deeeef  deeeeef ef  deeef fdCdDZ3		dQdeeeef ef deeeef  deeeeef ef  deeef fdEdFZ4		dQdeeeef ef deeeef  deeeeef ef  deeef fdGdHZ5		dQdeeeef ef deeeef  deeeeef ef  deeef fdIdJZ6G dKdL dLe	j7Z8G dMdN dNe	j7Z9G dOdP dPZ:dS )R    N)AnyOptionalUnion)	urlencode   )_api_module)_common)_transformers)types)get_value_by_path)set_value_by_path)
AsyncPagerPagerzgoogle_genai.tuningsfrom_objectparent_objectroot_objectreturnc                 C      i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg t | dgd ur;t|dgt | dg t | dgd urRt|dgtt | dg|| |S )	NsamplingCountsampling_countflipEnabledflip_enabledautoraterModelautorater_modelgenerationConfiggeneration_config)getvsetv_GenerationConfig_from_vertexr   r   r   	to_object r!   W/home/ubuntu/maya3_transcribe/venv/lib/python3.10/site-packages/google/genai/tunings.py_AutoraterConfig_from_vertex#       r#   c                 C   r   )	Nr   r   r   r   r   r   r   r   )r   r   _GenerationConfig_to_vertexr   r!   r!   r"   _AutoraterConfig_to_vertex>   r$   r&   c                 C   0   i }t | dgd urt|ddgt | dg |S Nname_urlr   r   r   r!   r!   r"   #_CancelTuningJobParameters_to_mldevY      r,   c                 C   r'   r(   r+   r   r!   r!   r"   $_CancelTuningJobParameters_to_vertexe   r-   r.   c                 C   .   i }t | dgd urt|dgt | dg |S NsdkHttpResponsesdk_http_responser+   r   r!   r!   r"   #_CancelTuningJobResponse_from_mldevq      r3   c                 C   r/   r0   r+   r   r!   r!   r"   $_CancelTuningJobResponse_from_vertex   r4   r5   c                 C   s"  i }t | dgd urtdt | dgd ur!t|dgt | dg t | dgd ur-tdt | dgd urAt|g dt | dg t | d	gd urUt|g d
t | d	g t | dgd uratdt | dgd urmtdt | dgd urytdt | dgd urtdt | dgd urtdt | dgd urt|g dt | dg t | dgd urt|g dt | dg t | dgd urtdt | dgd urtdt | dgd urtdt | dgd urtd t | d!gd urtd"t | d#gd urtd$t | d%gd urtd&|S )'Nvalidation_datasetz<validation_dataset parameter is not supported in Gemini API.tuned_model_display_namedisplayNamedescriptionz5description parameter is not supported in Gemini API.epoch_count)
tuningTaskhyperparameters
epochCountlearning_rate_multiplier)r;   r<   learningRateMultiplierexport_last_checkpoint_onlyzEexport_last_checkpoint_only parameter is not supported in Gemini API.pre_tuned_model_checkpoint_idzGpre_tuned_model_checkpoint_id parameter is not supported in Gemini API.adapter_sizez6adapter_size parameter is not supported in Gemini API.tuning_modez5tuning_mode parameter is not supported in Gemini API.custom_base_modelz;custom_base_model parameter is not supported in Gemini API.
batch_size)r;   r<   	batchSizelearning_rate)r;   r<   learningRateevaluation_configz;evaluation_config parameter is not supported in Gemini API.labelsz0labels parameter is not supported in Gemini API.betaz.beta parameter is not supported in Gemini API.base_teacher_modelz<base_teacher_model parameter is not supported in Gemini API.tuned_teacher_model_sourcezDtuned_teacher_model_source parameter is not supported in Gemini API.sft_loss_weight_multiplierzDsft_loss_weight_multiplier parameter is not supported in Gemini API.
output_uriz4output_uri parameter is not supported in Gemini API.r   
ValueErrorr   r   r!   r!   r"   _CreateTuningJobConfig_to_mldev   s   




rR   c                 C   s  i }t |ddg}|d u rd}|dkr+t | dgd ur*t|dgtt | dg|| n7|dkrGt | dgd urFt|dgtt | dg|| n|dkrbt | dgd urbt|d	gtt | dg|| t | d
gd urut|dgt | d
g t | dgd urt|dgt | dg t |ddg}|d u rd}|dkrt | dgd urt|g dt | dg n1|dkrt | dgd urt|g dt | dg n|dkrt | dgd urt|g dt | dg t |ddg}|d u rd}|dkrt | dgd urt|g dt | dg n5|dkr"t | dgd ur!t|g dt | dg n|dkr<t | dgd ur<t|g dt | dg t |ddg}|d u rJd}|dkret | dgd urdt|ddgt | dg n5|dkrt | dgd urt|ddgt | dg n|dkrt | dgd urt|d	dgt | dg t |ddg}|d u rd}|dkrt | dgd urt|g dt | dg n5|dkrt | dgd urt|g dt | dg n|dkrt | dgd urt|g dt | dg t |ddg}|d u rd}|dkr t | dgd ur t|ddgt | dg t | dgd ur4t|dgt | dg t |ddg}|d u rBd}|dkr\t | dgd ur\t|g d t | dg t |ddg}|d u rjd}|dkrt | d!gd urt|g d"t | d!g t |ddg}|d u rd}|dkrt | d#gd urt|dd$gtt | d#g|| n=|dkrt | d#gd urt|dd$gtt | d#g|| n|dkrt | d#gd urt|d	d$gtt | d#g|| t | d%gd urt|d%gt | d%g t | d&gd urt|g d't | d&g t | d(gd ur,t|d	d)gt | d(g t | d*gd urAt|d	d+gt | d*g t | d,gd urVt|g d-t | d,g t | d.gd urjt|d/gt | d.g |S )0NconfigmethodSUPERVISED_FINE_TUNINGr6   supervisedTuningSpecPREFERENCE_TUNINGpreferenceOptimizationSpecDISTILLATIONdistillationSpecr7   tunedModelDisplayNamer9   r:   )rV   hyperParametersr=   )rX   r\   r=   )rZ   r\   r=   r>   )rV   r\   r?   )rX   r\   r?   )rZ   r\   r?   r@   exportLastCheckpointOnlyrB   )rV   r\   adapterSize)rX   r\   r^   )rZ   r\   r^   rC   
tuningModerD   customBaseModelrE   )rV   r\   rF   rG   )rV   r\   rH   rI   evaluationConfigrJ   rK   )rX   r\   rK   rL   baseTeacherModelrM   tunedTeacherModelSourcerN   )rZ   r\   sftLossWeightMultiplierrO   	outputUri)r   r   "_TuningValidationDataset_to_vertex_EvaluationConfig_to_vertexr   r   r   r    discriminatorr!   r!   r"    _CreateTuningJobConfig_to_vertex   s  










































rj   c                 C      i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg t | dgd ur:tt | dg|| t | dgd urLtt | dg|| |S N
base_model	baseModelpre_tuned_modelpreTunedModeltraining_datasetrS   )r   r   _TuningDataset_to_mldevrR   r   r!   r!   r"   *_CreateTuningJobParametersPrivate_to_mldev     rs   c                 C   rk   rl   )r   r   _TuningDataset_to_vertexrj   r   r!   r!   r"   +_CreateTuningJobParametersPrivate_to_vertex  rt   rv   c                 C      i }t | dgd urt|dgtt | dg t | dgd ur+t|dgt | dg t | dgd urBt|dgtt | dg|| |S )NmetricsoutputConfigoutput_configautoraterConfigautorater_config)r   r   t	t_metricsr#   r   r!   r!   r"   _EvaluationConfig_from_vertex7     r   c                 C   rw   )Nrx   rz   ry   r|   r{   )r   r   r}   r~   r&   r   r!   r!   r"   rg   O  r   rg   c                 C   s^  i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg t | dgd ur;t|dgt | dg t | dgd urNt|dgt | dg t | d	gd urat|d
gt | d	g t | dgd urtt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd ur!t|dgt | dg t | dgd ur5t|dgt | dg t | dgd urIt|d gt | dg t | d!gd ur]t|d"gt | d!g t | d#gd urqt|d#gt | d#g t | d$gd urt|d%gt | d$g t | d&gd urt|d'gt | d&g t | d(gd urt|d)gt | d(g |S )*NmodelConfigmodel_selection_configresponseJsonSchemaresponse_json_schemaaudioTimestampaudio_timestampcandidateCountcandidate_countenableAffectiveDialogenable_affective_dialogfrequencyPenaltyfrequency_penaltylogprobsmaxOutputTokensmax_output_tokensmediaResolutionmedia_resolutionpresencePenaltypresence_penaltyresponseLogprobsresponse_logprobsresponseMimeTyperesponse_mime_typeresponseModalitiesresponse_modalitiesresponseSchemaresponse_schemaroutingConfigrouting_configseedspeechConfigspeech_configstopSequencesstop_sequencestemperaturethinkingConfigthinking_configtopKtop_ktopPtop_pr+   r   r!   r!   r"   r   g  s   






r   c                 C   sx  i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg t | dgd ur;t|dgt | dg t | dgd urNt|dgt | dg t | d	gd urat|d
gt | d	g t | dgd urtt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd ur!t|dgt | dg t | dgd ur5t|dgt | dg t | dgd urIt|d gt | dg t | d!gd ur]t|d"gt | d!g t | d#gd urqt|d#gt | d#g t | d$gd urt|d%gt | d$g t | d&gd urt|d'gt | d&g t | d(gd urt|d)gt | d(g t | d*gd urtd+|S ),Nr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   enable_enhanced_civic_answerszFenable_enhanced_civic_answers parameter is not supported in Vertex AI.r   r   rQ   r   r!   r!   r"   r%     s   






r%   c                 C   r'   r(   r+   r   r!   r!   r"    _GetTuningJobParameters_to_mldevF  r-   r   c                 C   r'   r(   r+   r   r!   r!   r"   !_GetTuningJobParameters_to_vertexR  r-   r   c                 C      i }t | dgd urt|ddgt | dg t | dgd ur*t|ddgt | dg t | dgd ur>t|ddgt | dg |S N	page_size_querypageSize
page_token	pageTokenfilterr+   r   r!   r!   r"   _ListTuningJobsConfig_to_mldev^     
r   c                 C   r   r   r+   r   r!   r!   r"   _ListTuningJobsConfig_to_vertexw  r   r   c                 C   ,   i }t | dgd urtt | dg|| |S NrS   )r   r   r   r!   r!   r"   "_ListTuningJobsParameters_to_mldev     r   c                 C   r   r   )r   r   r   r!   r!   r"   #_ListTuningJobsParameters_to_vertex  r   r   c                       i t | dgd urtdgt | dg t | dgd ur(tdgt | dg t | dgd urCtdg fddt | dgD  S )	Nr1   r2   nextPageTokennext_page_tokentunedModelstuning_jobsc                       g | ]}t | qS r!   )_TuningJob_from_mldev.0itemr   r    r!   r"   
<listcomp>      
z6_ListTuningJobsResponse_from_mldev.<locals>.<listcomp>r+   r   r   r   r!   r   r"   "_ListTuningJobsResponse_from_mldev      
	r   c                    r   )	Nr1   r2   r   r   
tuningJobsr   c                    r   r!   )_TuningJob_from_vertexr   r   r!   r"   r     r   z7_ListTuningJobsResponse_from_vertex.<locals>.<listcomp>r+   r   r!   r   r"   #_ListTuningJobsResponse_from_vertex  r   r   c                 C   sT   i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg |S )Nr)   modelendpointr+   r   r!   r!   r"   _TunedModel_from_mldev  s   r   c                 C   sj   i }t | dgd urtdt | dgd urtdt | dgd ur3t|ddgdd t | dgD  |S )Ngcs_uriz1gcs_uri parameter is not supported in Gemini API.vertex_dataset_resourcezAvertex_dataset_resource parameter is not supported in Gemini API.examplesc                 S   s   g | ]}|qS r!   r!   r   r!   r!   r"   r     s    z+_TuningDataset_to_mldev.<locals>.<listcomp>rP   r   r!   r!   r"   rr     s   rr   c                 C   s|  i }t |ddg}|d u rd}|dkr(t | dgd ur't|ddgt | dg n1|dkrAt | dgd ur@t|ddgt | dg n|d	krYt | dgd urYt|d
dgt | dg t |ddg}|d u rfd}|dkrt | dgd ur~t|ddgt | dg n1|dkrt | dgd urt|ddgt | dg n|d	krt | dgd urt|d
dgt | dg t | dgd urtd|S )NrS   rT   rU   r   rV   trainingDatasetUrirW   rX   rY   rZ   promptDatasetUrir   r   z1examples parameter is not supported in Vertex AI.r   rh   r!   r!   r"   ru   	  sp   





ru   c                 C   s  i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg t | dgd ur>t|dgtt | dg t | dgd urQt|dgt | dg t | ddgd urft|d	gt | ddg t | dd
gd ur{t|dgt | dd
g t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgtt | dg|| |S )Nr1   r2   r)   state
createTimecreate_timer;   	startTime
start_timecompleteTimeend_time
updateTimeupdate_timer9   rn   rm   _selftuned_model)r   r   r}   t_tuning_job_statusr   r   r!   r!   r"   r   L  sT   r   c                 C   s4  i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg t | dgd ur>t|dgtt | dg t | dgd urQt|dgt | dg t | dgd urdt|dgt | dg t | d	gd urwt|d
gt | d	g t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd urt|dgt | dg t | dgd ur$t|dgt | dg t | dgd ur8t|dgt | dg t | dgd urLt|dgt | dg t | dgd ur`t|d gt | dg t | d!gd urxt|d"gtt | d!g|| t | d#gd urt|d$gt | d#g t | d%gd urt|d%gt | d%g t | d&gd urt|d&gt | d&g t | d'gd urt|d(gt | d'g t | d)gd urt|d*gt | d)g t | d+gd urt|d,gt | d+g t | d-gd urt|d.gt | d-g t | d/gd urt|d0gt | d/g |S )1Nr1   r2   r)   r   r   r   r   r   endTimer   r   r   errorr9   rn   rm   
tunedModelr   rp   ro   rV   supervised_tuning_specrX   preference_optimization_specrZ   distillation_spectuningDataStatstuning_data_statsencryptionSpecencryption_specpartnerModelTuningSpecpartner_model_tuning_specra   rI   r`   rD   
experimentrJ   re   rO   pipelineJobpipeline_jobserviceAccountservice_accountr[   r7   veoTuningSpecveo_tuning_spec)r   r   r}   r   r   r   r!   r!   r"   r     s   




r   c                 C   s   i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg t | dgd ur;t|dgt | dg t | dgd urNt|dgt | dg t | dgd urat|dgt | dg |S )Nr1   r2   r)   metadatadoner   r+   r   r!   r!   r"   _TuningOperation_from_mldev  s   r   c                 C   sT   i }t | dgd urt|dgt | dg t | dgd ur(t|dgt | dg |S )Nr   validationDatasetUrir   r+   r   r!   r!   r"   rf     s   
rf   c                   @   h  e Zd Zdddedeej dejfddZdddeej	 dej
fdd	Zdddedeej dejfd
dZdddddee deej dejdeej dejf
ddZdddddee deej dejdeej dejf
ddZdddedeej dejfddZeddddedejdeej dejfddZdddeej	 deej fddZdS )TuningsNrS   r)   rS   r   c                C   J  t j||d}| jjr!t|d|}|d}|rd|}nd}nt|d|}|d}|r4d|}nd}|d}|rF| dt| }|	dd d}|j
dur]|j
jdur]|j
j}t|}t|}| jd|||}	|	jsui nt|	j}
| jjrt|
}
| jjst|
}
t jj|
| d	}t j|	jd
|_| j| |S zzGets a TuningJob.

    Args:
      name: The resource name of the tuning job.

    Returns:
      A TuningJob object.
    r)   rS   Nr*   z{name}r   ?rS   getresponsekwargsheaders)r
   _GetTuningJobParameters_api_clientvertexair   r  
format_mapr   r   poprS   http_optionsr   convert_to_dictencode_unserializable_typesrequestbodyjsonloadsr   r   	TuningJob_from_response
model_dumpHttpResponser  r2   _verify_responseselfr)   rS   parameter_modelrequest_dictrequest_url_dictpathquery_paramsr  r  response_dictreturn_valuer!   r!   r"   _get4  sV   





zTunings._getc                C   sH  t j|d}| jjr t|d |}|d}|rd|}nd}nt|d |}|d}|r3d|}nd}|d}|rE| dt| }|	dd  d }|j
d ur\|j
jd ur\|j
j}t|}t|}| jd|||}|jsti nt|j}	| jjrt|	}	| jjst|	}	t jj|	| d	}
t j|jd
|
_| j|
 |
S Nr   r*   r   r   r   r  rS   r  r  r  )r
   _ListTuningJobsParametersr
  r  r   r  r  r   r   r  rS   r  r   r  r  r  r  r  r  r   r   ListTuningJobsResponser  r  r  r  r2   r  r  rS   r  r  r  r  r   r  r  r!  r"  r!   r!   r"   _list  sT   





zTunings._listc                C   r   )zPCancels a tuning job.

    Args:
      name (str): TuningJob resource name.
    r  Nr*   {name}:cancelr   r  rS   postr  r  )r
   _CancelTuningJobParametersr
  r  r.   r  r  r,   r   r  rS   r  r   r  r  r  r  r  r  r5   r3   CancelTuningJobResponser  r  r  r  r2   r  r  r!   r!   r"   cancel  sZ   





zTunings.cancelrm   ro   rS   rm   ro   rq   c                C   s  t j||||d}| jjstdt|d|}|d}|r$d|}nd}|d}	|	r6| dt|	 }|	dd d}
|j
durM|j
jdurM|j
j}
t|}t|}| jd	|||
}|jsei nt|j}| jjrst|}t jj|| d
}t j|jd|_| j| |S a  Creates 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.
    rm   ro   rq   rS   z6This method is only supported in the Vertex AI client.Nr*   r   r   r  rS   r*  r  r  )r
   !_CreateTuningJobParametersPrivater
  r  rQ   rv   r  r  r   r  rS   r  r   r  r  r  r  r  r  r   r  r  r  r  r  r2   r  r  rm   ro   rq   rS   r  r  r  r  r   r  r  r!  r"  r!   r!   r"   _tune  sN   




zTunings._tunec                C   s  t j||||d}| jjrtdt|d|}|d}|r$d|}nd}|d}	|	r6| dt|	 }|	dd d}
|j
durM|j
jdurM|j
j}
t|}t|}| jd	|||
}|jsei nt|j}| jjsst|}t jj|| d
}t j|jd|_| j| |S a  Creates 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.
    r0  z=This method is only supported in the Gemini Developer client.Nr*   r   r   r  rS   r*  r  r  )r
   r1  r
  r  rQ   rs   r  r  r   r  rS   r  r   r  r  r  r  r  r  r   TuningOperationr  r  r  r  r2   r  r2  r!   r!   r"   _tune_mldevP  sR   




zTunings._tune_mldevc                C   s@   | j ||d}|jr| jjr| jjd urtj|j| jjd |S Nr  )r   projectr#  r   r
  r  r8  _IpythonUtilsdisplay_experiment_buttonr  r)   rS   jobr!   r!   r"   r    s   zTunings.getSThe SDK's tuning implementation is experimental, and may change in future versions.c                C   s  | j jr|dr#d }|rt|dd }tj||d}| j|||d}nd }|d urzt|dd d urzt|d}t|trCtj	di |}|j
rI|jsMtdtt|j
D ]}	t|j
|	 trjtj|j
|	 |j
|	< qTt|tru||d< n||_|}| j|||d}|d urt|dd d ur||_n2| j|||d}
|
jd urd|
jv r|
jd }n|
jd u rtd	|
jd
d }tj|tjjd}|jr| j jrtj|jd |S Nz	projects/rA   )tuned_model_namecheckpoint_id)ro   rq   rS   rI   zEEvaluation config must have at least one metric and an output config.)rm   rq   rS   r   zOperation name is required.z/operations/r   )r)   r   )tuning_job_resourcer!   r
  r  
startswithgetattrr
   PreTunedModelr3  
isinstancedictEvaluationConfigrx   rz   rQ   rangelenMetricmodel_validaterI   r6  r   r)   	partitionr  JobStateJOB_STATE_QUEUEDr:  display_model_tuning_button)r  rm   rq   rS   rA  ro   
tuning_jobvalidated_evaluation_configrI   i	operationr@  r!   r!   r"   tune  s   






zTunings.tunec                C   s   | j }td|| j |d|S )a  Lists `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)
    r   r   )r(  r   r  rS   list_requestr!   r!   r"   list   s   
zTunings.list)__name__
__module____qualname__strr   r
   GetTuningJobConfigOrDictr  r#  ListTuningJobsConfigOrDictr&  r(  CancelTuningJobConfigOrDictr,  r-  PreTunedModelOrDictTuningDatasetOrDictCreateTuningJobConfigOrDictr3  r5  r6  r  r   experimental_warningrV  r   rY  r!   r!   r!   r"   r   2  s    
L
B
M
L
O
	Pr   c                   @   r   )AsyncTuningsNr   r)   rS   r   c                   R  t j||d}| jjr"t|d|}|d}|rd|}nd}nt|d|}|d}|r5d|}nd}|d}|rG| dt| }|	dd d}|j
dur^|j
jdur^|j
j}t|}t|}| jd|||I dH }	|	jsyi nt|	j}
| jjrt|
}
| jjst|
}
t jj|
| d	}t j|	jd
|_| j| |S r   )r
   r	  r
  r  r   r  r  r   r   r  rS   r  r   r  r  async_requestr  r  r  r   r   r  r  r  r  r  r2   r  r  r!   r!   r"   r#    s\   






zAsyncTunings._getc                   sP  t j|d}| jjr!t|d |}|d}|rd|}nd}nt|d |}|d}|r4d|}nd}|d}|rF| dt| }|	dd  d }|j
d ur]|j
jd ur]|j
j}t|}t|}| jd|||I d H }|jsxi nt|j}	| jjrt|	}	| jjst|	}	t jj|	| d	}
t j|jd
|
_| j|
 |
S r$  )r
   r%  r
  r  r   r  r  r   r   r  rS   r  r   r  r  rg  r  r  r  r   r   r&  r  r  r  r  r2   r  r'  r!   r!   r"   r(  k  sZ   






zAsyncTunings._listc                   rf  )zaCancels a tuning job asynchronously.

    Args:
      name (str): A TuningJob resource name.
    r  Nr*   r)  r   r  rS   r*  r  r  )r
   r+  r
  r  r.   r  r  r,   r   r  rS   r  r   r  r  rg  r  r  r  r5   r3   r,  r  r  r  r  r2   r  r  r!   r!   r"   r-    s\   






zAsyncTunings.cancelr.  rm   ro   rq   c                   s"  t j||||d}| jjstdt|d|}|d}|r%d|}nd}|d}	|	r7| dt|	 }|	dd d}
|j
durN|j
jdurN|j
j}
t|}t|}| jd	|||
I dH }|jsii nt|j}| jjrwt|}t jj|| d
}t j|jd|_| j| |S r/  )r
   r1  r
  r  rQ   rv   r  r  r   r  rS   r  r   r  r  rg  r  r  r  r   r  r  r  r  r  r2   r  r2  r!   r!   r"   r3    sP   





zAsyncTunings._tunec                   s"  t j||||d}| jjrtdt|d|}|d}|r%d|}nd}|d}	|	r7| dt|	 }|	dd d}
|j
durN|j
jdurN|j
j}
t|}t|}| jd	|||
I dH }|jsii nt|j}| jjswt|}t jj|| d
}t j|jd|_| j| |S r4  )r
   r1  r
  r  rQ   rs   r  r  r   r  rS   r  r   r  r  rg  r  r  r  r   r5  r  r  r  r  r2   r  r2  r!   r!   r"   r6  >  sT   





zAsyncTunings._tune_mldevc                   sH   | j ||dI d H }|jr"| jjr"| jjd ur"tj|j| jjd |S r7  r9  r<  r!   r!   r"   r    s   zAsyncTunings.getr>  c                   s  | j jr|dr'd }|rt|dd }tj||d}| j|||dI d H }n|d urzt|dd d urzt|d}t|trEtj	di |}|j
rK|jsOtdtt|j
D ]}t|j
| trltj|j
| |j
|< qVt|trw||d< n||_| j|||dI d H }n5| j|||dI d H }	|	jd urd|	jv r|	jd }
n|	jd u rtd	|	jd
d }
tj|
tjjd}|jr| j jrtj|jd |S r?  rC  )r  rm   rq   rS   rA  ro   rR  rI   rT  rU  r@  r!   r!   r"   rV    sv   






zAsyncTunings.tunec                   s$   | j }td|| j |dI dH |S )a  Lists `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)
    r   r   N)r(  r   rW  r!   r!   r"   rY    s   zAsyncTunings.list)rZ  r[  r\  r]  r   r
   r^  r  r#  r_  r&  r(  r`  r,  r-  ra  rb  rc  r3  r5  r6  r  r   rd  rV  r   rY  r!   r!   r!   r"   re    s    
N
D
M
L
O
	Jre  c                
   @   s   e Zd ZU dZe Zee ed< ede	ee
f fddZedefddZedefdd	Zed
edee defddZe	ddededee ddfddZedededdfddZededdfddZdS )r:  z6Temporary class to hold the IPython related functions.displayed_experimentsr   c                  C   s,   dd l } d| jv rddlm} | jjS dS )Nr   IPython)get_ipython )sysmodulesri  rj  	__class__rZ  )rl  rj  r!   r!   r"   _get_ipython_shell_name		  s
   

z%_IpythonUtils._get_ipython_shell_namec                   C   s   t t S )N)boolr:  ro  r!   r!   r!   r"   is_ipython_available	  s   z"_IpythonUtils.is_ipython_availablec                   C   s   dS )z8Returns the HTML style markup to support custom buttons.a  
    <link rel="stylesheet" href="https://fonts.googleapis.com/icon?family=Material+Icons">
    <style>
      .view-vertex-resource,
      .view-vertex-resource:hover,
      .view-vertex-resource:visited {
        position: relative;
        display: inline-flex;
        flex-direction: row;
        height: 32px;
        padding: 0 12px;
          margin: 4px 18px;
        gap: 4px;
        border-radius: 4px;

        align-items: center;
        justify-content: center;
        background-color: rgb(255, 255, 255);
        color: rgb(51, 103, 214);

        font-family: Roboto,"Helvetica Neue",sans-serif;
        font-size: 13px;
        font-weight: 500;
        text-transform: uppercase;
        text-decoration: none !important;

        transition: box-shadow 280ms cubic-bezier(0.4, 0, 0.2, 1) 0s;
        box-shadow: 0px 3px 1px -2px rgba(0,0,0,0.2), 0px 2px 2px 0px rgba(0,0,0,0.14), 0px 1px 5px 0px rgba(0,0,0,0.12);
      }
      .view-vertex-resource:active {
        box-shadow: 0px 5px 5px -3px rgba(0,0,0,0.2),0px 8px 10px 1px rgba(0,0,0,0.14),0px 3px 14px 2px rgba(0,0,0,0.12);
      }
      .view-vertex-resource:active .view-vertex-ripple::before {
        position: absolute;
        top: 0;
        bottom: 0;
        left: 0;
        right: 0;
        border-radius: 4px;
        pointer-events: none;

        content: '';
        background-color: rgb(51, 103, 214);
        opacity: 0.12;
      }
      .view-vertex-icon {
        font-size: 18px;
      }
    </style>
  r!   r!   r!   r!   r"   _get_styles	  s   z_IpythonUtils._get_stylesmarkerresource_partsc                 C   sB   t t|D ]}|| | kr|d t|k r||d    S qdS )z,Returns the part after the marker text part.r   rk  )rJ  rK  )rs  rt  rT  r!   r!   r"   _parse_resource_nameM	  s
   z"_IpythonUtils._parse_resource_nameopen_in_newtexturliconNc           	      C   s   d}| |std| dddl}dt|  }dt  d| d	| d
| d|  d}|d| d| d| d7 }ddlm} ddlm	} ||| dS )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.comzOnly urls starting with z are allowed.r   Nzview-vertex-resource-z	
        z-
        <a class="view-vertex-resource" id="z" href="#view-z;">
          <span class="material-icons view-vertex-icon">z</span>
          <span>z</span>
        </a>
        z]
        <script>
          (function () {
            const link = document.getElementById('z');
            link.addEventListener('click', (e) => {
              if (window.google?.colab?.openUrl) {
                window.google.colab.openUrl('z8');
              } else {
                window.open('z', '_blank');
              }
              e.stopPropagation();
              e.preventDefault();
            });
          })();
        </script>
    )display)HTML)
rD  rQ   uuidr]  uuid4r:  rr  IPython.displayrz  r{  )	rw  rx  ry  CLOUD_UI_URLr|  	button_idhtmlrz  r{  r!   r!   r"   _display_linkU	  s4   
	
z_IpythonUtils._display_linkr   r8  c                 C   sl   t  r	| t jv rdS | d}|d }|d }d| d| d d|  }t d|d	 t j|  dS )
a  Function 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=zView Experimentscience)r:  rq  rh  splitr  add)r   r8  rt  locationexperiment_nameurir!   r!   r"   r;  	  s   


z'_IpythonUtils.display_experiment_buttonrB  c                 C   s^   t  sdS | d}|d }|d }|d }dd| d|  d	|  }t d
|d dS )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}
    Nr  r   r  r  z?https://console.cloud.google.com/vertex-ai/generative/language/z
locations/z/tuning/tuningJob/z	?project=zView Tuning JobrV  )r:  rq  r  r  )rB  rt  r8  r  tuning_job_idr  r!   r!   r"   rQ  	  s   
z)_IpythonUtils.display_model_tuning_button)rv  )rZ  r[  r\  __doc__setrh  r]  __annotations__staticmethodr   r   ro  rp  rq  rr  rY  ru  r   r  r;  rQ  r!   r!   r!   r"   r:  	  s4   
 	53r:  )NN);r  loggingtypingr   r   r   urllib.parser   rk  r   r   r	   r}   r
   r   r   r   r   pagersr   r   	getLoggerloggerrH  r]  objectr#   r&   r,   r.   r3   r5   rR   rj   rs   rv   r   rg   r   r%   r   r   r   r   r   r   r   r   r   rr   ru   r   r   r   rf   
BaseModuler   re  r:  r!   r!   r!   r"   <module>   s  














j

  









o

t





















E

=

 



   m   k