o
    pkj=V                     @   sv  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eeeef ef d	eeeef  d
eeef fddZ	d.deeeef ef d	eeeef  d
eeef fddZ	d.deeeef ef d	eeeef  d
eeef fddZ	d.dedeeeef ef d	eeeef  d
eeef fddZ	d.dedeeeef ef d	eeeef  d
eeef fddZ	d.deeeef ef d	eeeef  d
eeef fddZ	d.deeeef ef d	eeeef  d
eeef fddZ	d.deeeef ef d	eeeef  d
eeef fddZ	d.deeeef ef d	eeeef  d
eeef fddZ	d.deeeef ef d	eeeef  d
eeef fddZ	d.dedeeeef ef d	eeeef  d
eeef fd d!Z	d.dedeeeef ef d	eeeef  d
eeef fd"d#Z	d.deeeef ef d	eeeef  d
eeef fd$d%Z	d.deeeef ef d	eeeef  d
eeef fd&d'Z	d.deeeef ef d	eeeef  d
eeef fd(d)Z	d.deeeef ef d	eeeef  d
eeef fd*d+Z 	d.deeeef ef d	eeeef  d
eeef fd,d-Z!dS )/    )AnyOptionalUnion   )_transformers)BaseApiClient)get_value_by_path)set_value_by_pathNfrom_objectparent_objectreturnc                 C   s   i }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r-tdt | dgd ur9tdt | d	gd urEtd
t | dgd urQtdt | dgd ur]td|S )Napi_keyapiKeyapi_key_configzvapi_key_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.	auth_typezqauth_type parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.google_service_account_configzgoogle_service_account_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.http_basic_auth_configz~http_basic_auth_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.oauth_configztoauth_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.oidc_configzsoidc_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.getvsetv
ValueErrorr
   r   	to_object r   ]/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/_tokens_converters.py_AuthConfig_to_mldev   s8   r   c                 C   sl   i }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r4t|dgt | dg |S )Ndatadisplay_nametdisplay_name parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.	mime_typemimeTyper   r   r   r   r   _Blob_to_mldevI      r#   c                    sb   i  t | dgd urt dg fddt | dgD  t | dgd ur/t dgt | dg  S )Npartsc                       g | ]}t | qS r   )_Part_to_mldev.0itemr   r   r   
<listcomp>f       z%_Content_to_mldev.<locals>.<listcomp>roler   r   )r
   r   r   r+   r   _Content_to_mldev]   s   

	r0   
api_clientc                 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rRt|dgt| t |dg| t |dgd uret|d	gt |dg |S )
Nexpire_time
expireTimenew_session_expire_timenewSessionExpireTimeuseslive_connect_constraintsbidiGenerateContentSetuplock_additional_fields	fieldMask)r   r    _LiveConnectConstraints_to_mldevr1   r
   r   r   r   r   r   _CreateAuthTokenConfig_to_mldevr   s8   



r=   c                 C   s6   i }t |dgd urt|dgt| t |dg| |S )Nconfig)r   r   r=   r<   r   r   r   #_CreateAuthTokenParameters_to_mldev   s   r?   c                 C   s    i }t | dgd urtd|S )Nr>   znconfig parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.)r   r   r   r   r   r   $_CreateAuthTokenParameters_to_vertex   s   r@   c                 C   sl   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r4t|dgt | dg |S )Nr   r    file_urifileUrir!   r"   )r   r   r   r   r   r   r   _FileData_to_mldev   s   rC   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rGtdt | dgd urStd|S )Nidargsnamepartial_argsztpartial_args parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.will_continuezuwill_continue parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.r   r   r   r   r   _FunctionCall_to_mldev   s    rI   c                 C   sr   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r7td|S )Nauth_config
authConfigenable_widgetenableWidgetgrounding_typeszwgrounding_types parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.)r   r   r   r   r   r   r   r   _GoogleMaps_to_mldev   s   rO   c                 C   s   i }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r-tdt | dgd ur@t|dgt | dg |S )	Nsearch_typessearchTypesblocking_confidencez{blocking_confidence parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.exclude_domainszwexclude_domains parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.time_range_filtertimeRangeFilterr   r   r   r   r   _GoogleSearch_to_mldev  s    rV   c              
      s  i t |dgd urt|ddgt |dg t |dgd ur*t|g dt |dg t |dgd ur>t|g dt |dg t |dgd urRt|g d	t |dg t |d
gd urft|g dt |d
g t |dgd urzt|g dt |dg 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|g dtt |dg 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dgttt |dg t |dgd urt|ddg fddt t |dgD  t |dgd ur4t|ddgtt |dg t |dgd urIt|dd gt |dg t |d!gd ur^t|dd"gt |d!g t |d#gd urst|dd$gt |d#g t |d%gd urt|dd&gt |d%g t |d'gd urt|dd'gt |d'g t |d(gd urtd)t |d*gd urt|dd+gt |d*g t |d,gd urt|dd-gt |d,g t |d.gd urt|dd/gfd0dt |d.gD  t |d1gd urt|g d2t	t |d1g S )3Ngeneration_configsetupgenerationConfigresponse_modalities)rX   rY   responseModalitiestemperature)rX   rY   r\   top_p)rX   rY   topPtop_k)rX   rY   topKmax_output_tokens)rX   rY   maxOutputTokensmedia_resolution)rX   rY   mediaResolutionseed)rX   rY   re   speech_config)rX   rY   speechConfigthinking_config)rX   rY   thinkingConfigenable_affective_dialog)rX   rY   enableAffectiveDialogsystem_instructionsystemInstructiontoolsc                    s   g | ]}t t |qS r   )_Tool_to_mldevtt_toolr(   r1   r   r   r   r,     s    z/_LiveConnectConfig_to_mldev.<locals>.<listcomp>session_resumptionsessionResumptioninput_audio_transcriptioninputAudioTranscriptionoutput_audio_transcriptionoutputAudioTranscriptionrealtime_input_configrealtimeInputConfigcontext_window_compressioncontextWindowCompressionproactivityexplicit_vad_signalz{explicit_vad_signal parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.history_confighistoryConfigavatar_configavatarConfigsafety_settingssafetySettingsc                    r&   r   )_SafetySetting_to_mldevr(   r+   r   r   r,     r-   translation_config)rX   rY   translationConfig)
r   r   rp   t_live_speech_configr0   	t_contentt_tools!_SessionResumptionConfig_to_mldevr   _TranslationConfig_to_mldev)r1   r
   r   r   rr   r   _LiveConnectConfig_to_mldev   s4  









	








	r   c              	   C   sf   i }t |dgd urt|ddgt| t |dg t |dgd ur1t|dgt| t |dg| |S )NmodelrX   r>   )r   r   rp   t_modelr   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rQt|dgtt | dg| t | d	gd urgt|d
gtt | d	g| t | dgd urzt|dgt | dg t | dgd urt|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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r?t|dgt | dg |S )Nrc   rd   code_execution_resultcodeExecutionResultexecutable_codeexecutableCode	file_datafileDatafunction_callfunctionCallfunction_responsefunctionResponseinline_data
inlineDatatextthoughtthought_signaturethoughtSignaturevideo_metadatavideoMetadata	tool_calltoolCalltool_responsetoolResponsepart_metadatapartMetadataaudio_transcriptionaudioTranscriptionmedia_processingmediaProcessing)r   r   rC   rI   r#   r   r   r   r   r'     s   



r'   c                 C   sl   i }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r4t|dgt | dg |S )Ncategorymethodznmethod parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.	thresholdr   r   r   r   r   r   N  r$   r   c                 C   sF   i }t | dgd urt|dgt | dg t | dgd ur!td|S )Nhandletransparentzstransparent parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.r   r   r   r   r   r   b  s   r   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r4t|dgt | dg t | dgd urJt|dgtt | dg| t | d	gd ur`t|d
gtt | d	g| t | dgd urst|dgt | dg t | dgd urtdt | dgd urt|dgdd t | dgD  t | dgd urt|dgt | dg t | dgd urtdt | dgd urt|dgt | dg t | dgd urt|dgdd t | dgD  t | dgd urtd|S )N	retrievalzqretrieval parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.computer_usecomputerUsefile_search
fileSearchgoogle_searchgoogleSearchgoogle_maps
googleMapscode_executioncodeExecutionenterprise_web_searchz}enterprise_web_search parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.function_declarationsfunctionDeclarationsc                 S      g | ]}|qS r   r   r(   r   r   r   r,         z"_Tool_to_mldev.<locals>.<listcomp>google_search_retrievalgoogleSearchRetrievalparallel_ai_searchzzparallel_ai_search parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.url_context
urlContextmcp_servers
mcpServersc                 S   r   r   r   r(   r   r   r   r,     r   exa_ai_searchzuexa_ai_search parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.)r   r   r   rV   rO   r   r   r   r   ro   s  sp   
ro   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 )Necho_target_languageechoTargetLanguagetarget_language_codetargetLanguageCoder/   r   r   r   r   r     s   

r   )N)"typingr   r   r    r   rp   _api_clientr   _commonr   r   r	   r   dictstrobjectr   r#   r0   r=   r?   r@   rC   rI   rO   rV   r   r;   r'   r   r   ro   r   r   r   r   r   <module>   s,  

1





,













 A



Z





Q
