o pkjb @s ddlZddlZddlmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd l mZdd l mZdd lmZmZed Z d`deeeefefdeeeefdeeeffddZ d`deeeefefdeeeefdeeeffddZ d`deeeefefdeeeefdeeeffddZ d`deeeefefdeeeefdeeeffddZ d`deeeefefdeeeefdeeeffddZ d`deeeefefdeeeefdeeeffddZ! d`deeeefefdeeeefdeeeffddZ" d`dedeeeefefdeeeefdeeeffd d!Z# d`dedeeeefefdeeeefdeeeffd"d#Z$ d`dedeeeefefdeeeefdeeeffd$d%Z% d`dedeeeefefdeeeefdeeeffd&d'Z& d`deeeefefdeeeefdeeeffd(d)Z' d`deeeefefdeeeefdeeeffd*d+Z( d`deeeefefdeeeefdeeeffd,d-Z) d`deeeefefdeeeefdeeeffd.d/Z* d`deeeefefdeeeefdeeeffd0d1Z+ d`dedeeeefefdeeeefdeeeffd2d3Z, d`dedeeeefefdeeeefdeeeffd4d5Z- d`deeeefefdeeeefdeeeffd6d7Z. d`deeeefefdeeeefdeeeffd8d9Z/ d`deeeefefdeeeefdeeeffd:d;Z0 d`deeeefefdeeeefdeeeffdd?Z2 d`deeeefefdeeeefdeeeffd@dAZ3 d`deeeefefdeeeefdeeeffdBdCZ4 d`deeeefefdeeeefdeeeffdDdEZ5 d`deeeefefdeeeefdeeeffdFdGZ6 d`deeeefefdeeeefdeeeffdHdIZ7 d`deeeefefdeeeefdeeeffdJdKZ8 d`deeeefefdeeeefdeeeffdLdMZ9 d`deeeefefdeeeefdeeeffdNdOZ: d`deeeefefdeeeefdeeeffdPdQZ; d`deeeefefdeeeefdeeeffdRdSZ< d`deeeefefdeeeefdeeeffdTdUZ= d`deeeefefdeeeefdeeeffdVdWZ> d`dedeeeefefdeeeefdeeeffdXdYZ? d`dedeeeefefdeeeefdeeeffdZd[Z@Gd\d]d]e jAZBGd^d_d_e jAZCdS)aN)AnyOptionalUnion) urlencode) _api_module)_common) _transformers)types) BaseApiClient)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.caches from_object parent_objectreturncCsi}t|dgdurt|dgt|dgt|dgdur!tdt|dgdur-tdt|dgdur9tdt|d gdurEtd t|d gdurQtd t|d gdur]td|S)Napi_keyapiKeyapi_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.getvsetv ValueErrorrr to_objectr!Q/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/caches.py_AuthConfig_to_mldev#s8r#cCsli}t|dgdurt|dgt|dgt|dgdur!tdt|dgdur4t|dgt|dg|S)Ndata display_nametdisplay_name parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. mime_typemimeTyperrr!r!r"_Blob_to_mldevRsr)cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurGtd|S)N environmentexcluded_predefined_functionsexcludedPredefinedFunctions!enable_prompt_injection_detectionenablePromptInjectionDetectiondisabled_safety_policieszdisabled_safety_policies parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rrr!r!r"_ComputerUse_to_vertexfs(  r0cbit|dgdurtdgfddt|dgDt|dgdur/tdgt|dgS)Npartscg|]}t|qSr!)_Part_to_mldev.0itemr r!r" z%_Content_to_mldev..rolerrrrr!r8r"_Content_to_mldev   r>cr1)Nr2cr3r!)_Part_to_vertexr5r8r!r"r9r:z&_Content_to_vertex..r;r<r=r!r8r"_Content_to_vertexr?rAc s^it|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurXt|dgfddtt|dgDt|d gdurqt|d gttt|d gt|d gdurt|d gfd dt|d gDt|d gdurt|dgtt|d gt|dgdurtdS)Nttl expire_time expireTimer% displayNamecontentscr3r!)r>r5r8r!r"r9r:z7_CreateCachedContentConfig_to_mldev..system_instructionsystemInstructiontoolscr3r!)_Tool_to_mldevr5r8r!r"r9r: tool_config toolConfig kms_key_nameztkms_key_name parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.)rrt t_contentsr> t_content_ToolConfig_to_mldevrr=r!r8r"#_CreateCachedContentConfig_to_mldevsT     rRc snit|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurXt|dgfddtt|dgDt|d gdurqt|d gttt|d gt|d gdurt|d gfd dt|d gDt|d gdurt|dgtt|d gt|dgdurt|ddgt|dgS)NrBrCrDr%rErFcr3r!)rAr5r8r!r"r9r:z8_CreateCachedContentConfig_to_vertex..rGrHrIcr3r!)_Tool_to_vertexr5r8r!r"r9r:rKrLrMencryption_spec kmsKeyName)rrrNrOrArP_ToolConfig_to_vertexr=r!r8r"$_CreateCachedContentConfig_to_vertexsX      rW api_clientc CXi}t|dgdurt|dgt|t|dgt|dgdur*tt|dg||SNmodelconfig)rrrNt_caches_modelrRrXrrr r!r!r"'_CreateCachedContentParameters_to_mldev( r_c CrYrZ)rrrNr]rWr^r!r!r"(_CreateCachedContentParameters_to_vertex=r`rac C8i}t|dgdurt|ddgt|t|dg|SNname_urlrrrNt_cached_content_namer^r!r!r"'_DeleteCachedContentParameters_to_mldevRrhc Crbrcrfr^r!r!r"(_DeleteCachedContentParameters_to_vertexbrirjcC.i}t|dgdurt|dgt|dg|SNsdkHttpResponsesdk_http_responser<rr!r!r"'_DeleteCachedContentResponse_from_mldevr rocCrkrlr<rr!r!r"(_DeleteCachedContentResponse_from_vertexrprqcCsli}t|dgdurtdt|dgdur!t|dgt|dgt|dgdur4t|dgt|dg|S)Nr%r&file_urifileUrir'r()rrrrr!r!r"_FileData_to_mldevsrtcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurGtdt|dgdurStd|S)Nidargsrd 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.rrr!r!r"_FunctionCall_to_mldevs rycCsli}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur4td|S)Nallowed_function_namesallowedFunctionNamesmodestream_function_call_argumentszstream_function_call_arguments parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.rrr!r!r"_FunctionCallingConfig_to_mldevs r~c Crbrcrfr^r!r!r"$_GetCachedContentParameters_to_mldevrirc Crbrcrfr^r!r!r"%_GetCachedContentParameters_to_vertexrircCsri}t|dgdurt|dgtt|dg|t|dgdur+t|dgt|dgt|dgdur7td|S)N auth_config authConfig enable_widget enableWidgetgrounding_typeszwgrounding_types parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.)rrr#rrr!r!r"_GoogleMaps_to_mldevsrcCsi}t|dgdurt|dgt|dgt|dgdur!tdt|dgdur-tdt|dgdur@t|dgt|dg|S) N search_types searchTypesblocking_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_filtertimeRangeFilterrrr!r!r"_GoogleSearch_to_mldev s rcCXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|SN page_size_querypageSize page_token pageTokenr<rr!r!r""_ListCachedContentsConfig_to_mldev) rcCrrr<rr!r!r"#_ListCachedContentsConfig_to_vertex>rrcC*i}t|dgdurtt|dg||SNr\)rrrr!r!r"&_ListCachedContentsParameters_to_mldevSsrcCrr)rrrr!r!r"'_ListCachedContentsParameters_to_vertex^s  rcCi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S) Nrmrn nextPageTokennext_page_tokencachedContentscached_contentscSg|]}|qSr!r!r5r!r!r"r9|z:_ListCachedContentsResponse_from_mldev..r<rr!r!r"&_ListCachedContentsResponse_from_mldevkrcCr) NrmrnrrrrcSrr!r!r5r!r!r"r9rz;_ListCachedContentsResponse_from_vertex..r<rr!r!r"'_ListCachedContentsResponse_from_vertexrrcCs8i}t|dgdurtdt|dgdurtd|S)Nrdzlname parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.streamable_http_transportzstreamable_http_transport parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.)rrrr!r!r"_McpServer_to_vertexsrcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurQt|dgtt|dg|t|d gdurgt|d gtt|d g|t|d gdurzt|d gt|d gt|d gdurt|dgtt|d g|t|dgdurt|dgt|dgt|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|dg|S)Nmedia_resolutionmediaResolutioncode_execution_resultcodeExecutionResultexecutable_codeexecutableCode file_datafileData function_call functionCallfunction_responsefunctionResponse inline_data inlineDatatextthoughtthought_signaturethoughtSignaturevideo_metadata videoMetadata tool_calltoolCall tool_response toolResponse part_metadata partMetadataaudio_transcriptionaudioTranscriptionmedia_processingmediaProcessing)rrrtryr)rr!r!r"r4s    r4cCsBi}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|dgt|d gt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurtdt|dgdurtdt|dgdurtdt|dgdur t|dgt|dgt|dgdurt|dgt|dg|S)Nrrrrrrrrrrrrrrrrrrrrrzqtool_call parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rzutool_response parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rzupart_metadata parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rrrrrrr!r!r"r@sx    r@cCsi}t|dgdurt|dgt|dgt|dgdur+t|dgtt|dg|t|dgdur>t|dgt|dg|S)Nretrieval_configretrievalConfigfunction_calling_configfunctionCallingConfig$include_server_side_tool_invocations includeServerSideToolInvocations)rrr~rr!r!r"rQZs(  rQcCsli}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur4td|S)Nrrrrrzinclude_server_side_tool_invocations parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rrr!r!r"rVws  rVcCsi}t|dgdurtdt|dgdur!t|dgt|dgt|dgdur4t|dgt|dgt|dgdurJt|dgtt|dg|t|d gdur`t|d gtt|d g|t|d gdurst|d gt|d gt|d gdurtdt|dgdurt|dgddt|dgDt|dgdurt|dgt|dgt|dgdurtdt|dgdurt|dgt|dgt|dgdurt|dgddt|dgDt|dgdurtd|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 googleMapscode_execution codeExecutionenterprise_web_searchz}enterprise_web_search parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.function_declarationsfunctionDeclarationscSrr!r!r5r!r!r"r9rz"_Tool_to_mldev..google_search_retrievalgoogleSearchRetrievalparallel_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 mcpServerscSrr!r!r5r!r!r"r9r exa_ai_searchzuexa_ai_search parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.)rrrrrrr!r!r"rJsp rJcsit|dgdurtdgt|dgt|dgdur+tdgtt|dgt|dgdur7tdt|dgdurJtdgt|dgt|dgdur]td gt|dgt|d gdurptd gt|d gt|d gdurtd gt|d gt|dgdurtdgddt|dgDt|dgdurtdgt|dgt|dgdurtdgt|dgt|dgdurtdgt|dgt|dgdurtdgfddt|dgDt|dgdurtdgt|dgS)Nrrrrzsfile_search parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rrrrrrrenterpriseWebSearchrrcSrr!r!r5r!r!r"r9 rz#_Tool_to_vertex..rrrparallelAiSearchrrrrcr3r!)rr5r8r!r"r9"r:r exaAiSearch)rrr0rr=r!r8r"rSsp      rScCTi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dg|SNrBrCrDr<rr!r!r"#_UpdateCachedContentConfig_to_mldev. rcCrrr<rr!r!r"$_UpdateCachedContentConfig_to_vertex=rrc CZi}t|dgdurt|ddgt|t|dgt|dgdur+tt|dg||SNrdrer\)rrrNrgrr^r!r!r"'_UpdateCachedContentParameters_to_mldevL rc Crr)rrrNrgrr^r!r!r"(_UpdateCachedContentParameters_to_vertexarrc@eZdZdddedeejdejfddZdddedeej dejfd d Z dddedeej dej fd d Z dddedeejdejfd dZdddeejdejfddZdddeejdeejfddZdS)CachesNr\r[r\rc CPtj||d}|jjr!t|j|}|d}|rd|}nd}nt|j|}|d}|r4d|}nd}|d}|rF|dt|}| ddd}|j dur]|j j dur]|j j }t |}t |}|jd|||} | jsuint| j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid } |j| | S)aCreates a cached contents resource. Usage: .. code-block:: python contents = ... // Initialize the content to cache. response = client.caches.create( model= ... // The publisher model id contents=contents, config={ 'display_name': 'test cache', 'system_instruction': 'What is the sum of the two pdfs?', 'ttl': '86400s', }, ) r[r\rerr?r\Npostresponse_schemaresponse_json_schemainclude_all_fieldsrrrresponsekwargs)r _CreateCachedContentParameters _api_clientvertexairaget format_mapr_rpopr\ http_optionsrconvert_to_dictencode_unserializable_typesrequestbodyjsonloads CachedContent_from_responsegetattr_verify_response selfr[r\parameter_model request_dictrequest_url_dictpath query_paramsrr response_dict return_valuer!r!r"createxsj            z Caches.createrdc Cr)zGets cached content configurations. .. code-block:: python client.caches.get(name= ... ) // The server-generated resource name. rdr\re{name}rrr\Nrrrrrr)r _GetCachedContentParametersrrrrrrrrr\rrrrrrrr r r r r  rrdr\rrrrrrrrrr!r!r"rsf             z Caches.getc Cstj||d}|jjr!t|j|}|d}|rd|}nd}nt|j|}|d}|r4d|}nd}|d}|rF|dt|}| ddd}|j dur]|j j dur]|j j }t |}t |}|jd|||} | jsuint| j} |jjrt| } |jjst| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid } tj| jd| _|j| | S)zDeletes cached content. Usage: .. code-block:: python client.caches.delete(name= ... ) // The server-generated resource name. rrerrrr\Ndeleterrrrrheaders)r _DeleteCachedContentParametersrrrjrrrhrrr\rrrrrrrr rqroDeleteCachedContentResponser r  HttpResponserrnr rr!r!r"r%sx            z Caches.deletec Cr)zUpdates cached content configurations. .. code-block:: python response = client.caches.update( name= ... // The server-generated resource name. config={ 'ttl': '7600s', }, ) rrerrrr\Npatchrrrrr)r _UpdateCachedContentParametersrrrrrrrrr\rrrrrrrr r r r r rr!r!r"updatesj            z Caches.updatec Csvtj|d}|jjrt|}|d}|rd|}nd}nt|}|d}|r/d|}nd}|d}|rA|dt|}| ddd}|j durX|j j durX|j j }t |}t |}|jd|||}|jspint|j} |jjr~t| } |jjst| } tjj| t|ddrdt|j ddt|j d dt|j d dd inid } tj|jd | _|j| | SNrrerrrr\rrrrrrr)r _ListCachedContentsParametersrrrrrrrrr\rrrrrrrr rrListCachedContentsResponser r r!rrnr  rr\rrrrrrrrrr!r!r"_listsj            z Caches._listcCs|j}td||j|d|S)aLists cached contents. Args: config (ListCachedContentsConfig): Optional configuration for the list request. Returns: A Pager object that contains one page of cached contents. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python for cached_content in client.caches.list(): print(cached_content.name) rr)r)rrr\ list_requestr!r!r"list!s z Caches.list)__name__ __module__ __qualname__strrr CreateCachedContentConfigOrDictr rGetCachedContentConfigOrDictrDeleteCachedContentConfigOrDictr rUpdateCachedContentConfigOrDictr$ListCachedContentsConfigOrDictr'r)rr,r!r!r!r"rvsZ a T ` X Jrc@r) AsyncCachesNrr[r\rc Xtj||d}|jjr"t|j|}|d}|rd|}nd}nt|j|}|d}|r5d|}nd}|d}|rG|dt|}| ddd}|j dur^|j j dur^|j j }t |}t |}|jd|||IdH} | jsyint| j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid } |j| | S)aCreates a cached contents resource. Usage: .. code-block:: python contents = ... // Initialize the content to cache. response = await client.aio.caches.create( model= ... // The publisher model id contents=contents, config={ 'display_name': 'test cache', 'system_instruction': 'What is the sum of the two pdfs?', 'ttl': '86400s', }, ) rrerrrr\Nrrrrrr)r rrrrarrr_rrr\rrrr async_requestrrr r r r r rr!r!r"rAsl             zAsyncCaches.createrdc r7)zGets cached content configurations. .. code-block:: python await client.aio.caches.get(name= ... ) // The server-generated resource name. rrerrrr\Nrrrrrr)r rrrrrrrrrr\rrrrr8rrr r r r r rr!r!r"rsl             zAsyncCaches.getc stj||d}|jjr"t|j|}|d}|rd|}nd}nt|j|}|d}|r5d|}nd}|d}|rG|dt|}| ddd}|j dur^|j j dur^|j j }t |}t |}|jd|||IdH} | jsyint| j} |jjrt| } |jjst| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid } tj| jd| _|j| | S)zDeletes cached content. Usage: .. code-block:: python await client.aio.caches.delete(name= ... ) // The server-generated resource name. rrerrrr\Nrrrrrrr)r rrrrjrrrhrrr\rrrrr8rrr rqror r r r!rrnr rr!r!r"rsz             zAsyncCaches.deletec r7)zUpdates cached content configurations. .. code-block:: python response = await client.aio.caches.update( name= ... // The server-generated resource name. config={ 'ttl': '7600s', }, ) rrerrrr\Nr"rrrrr)r r#rrrrrrrrr\rrrrr8rrr r r r r rr!r!r"r$Nsl             zAsyncCaches.updatec s~tj|d}|jjrt|}|d}|rd|}nd}nt|}|d}|r0d|}nd}|d}|rB|dt|}| ddd}|j durY|j j durY|j j }t |}t |}|jd|||IdH}|jstint|j} |jjrt| } |jjst| } tjj| t|ddrdt|j ddt|j d dt|j d dd inid } tj|jd | _|j| | Sr%)r r&rrrrrrrrr\rrrrr8rrr rrr'r r r!rrnr r(r!r!r"r)sp             zAsyncCaches._listcs$|j}td||j|dIdH|S)aLists cached contents asynchronously. Args: config (ListCachedContentsConfig): Optional configuration for the list request. Returns: A Pager object that contains one page of cached contents. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python async for cached_content in await client.aio.caches.list(): print(cached_content.name) rrN)r)rr*r!r!r"r,szAsyncCaches.list)r-r.r/r0rr r1r rr2rr3r rr4r$r5r'r)rr,r!r!r!r"r6?sZ a W a X Lr6)N)Drloggingtypingrrr urllib.parserrrr rNr rr r rr rpagersrr getLoggerloggerdictr0objectr#r)r0r>rArRrWr_rarhrjrorqrtryr~rrrrrrrrrrrr4r@rQrVrJrSrrrr BaseModulerr6r!r!r!r"s           1    !      >  @                                           Z  W      Q  O        L