ci: >ddlZddlmZmZmZmZddlmZddlmZddlm Z ddl m Z ddl mZddlmZdd lmZej&eZeej,e j.fZeej,ej2fZeej,e j6fZd d d d ddididddddddddididddddddddididdddZGdd ej<ZGd!d"Z d#e jBd$eej,e"ffd%Z#d$ej,fd&Z$y)'N)ListOptionalSequenceUnion)base) initializer)utils)types)_generative_models)_utils) struct_pb2zCode InterpreterzDThis extension generates and executes code in the specified languagecode_interpreter_toolz!Google Code Interpreter Extensionopen_api_gcs_uriz2gs://vertex-extension-public/code_interpreter.yamlGOOGLE_SERVICE_ACCOUNT_AUTH) auth_typegoogle_service_account_config)name descriptionapi_spec auth_config) display_namermanifestzVertex AI Searchz4This extension generates and executes search queriesvertex_ai_searchzVertex AI Search Extensionz2gs://vertex-extension-public/vertex_ai_search.yamlzWebpage Browserz/This extension fetches the content of a webpagewebpage_browserz Vertex Webpage Browser Extensionz1gs://vertex-extension-public/webpage_browser.yaml)code_interpreterrrc eZdZdZej ZdZdZdZ dZ dZ dZ de ffd Zed d d d d d eej$ej(fdee d ee dee deef dZede fdZdej$fdZdeej$fdZ dde deedeedeej$e ffdZde jBddfdZ"ed dde deefdZ#xZ$S) Extensionz*Represents a Vertex AI Extension resource. extension get_extensionlist_extensionsdelete_extensionparse_extension_pathextension_pathextension_namect||tjj t j |_|j||_ d|_ d|_ y)a4Retrieves an extension resource. Args: extension_name (str): Required. A fully-qualified resource name or ID such as "projects/123/locations/us-central1/extensions/456" or "456" when project and location are initialized or passed.  resource_name client_classN) super__init__r global_config create_client aip_utils$ExtensionExecutionClientWithOverrideexecution_api_client_get_gca_resource _gca_resource _api_spec_operation_schemas)selfr$ __class__s F/tmp/pip-target-z3e9_cxr/lib/python/vertexai/extensions/_extensions.pyr+zExtension.__init__fsd ~6$/$=$=$K$K"GG%L% !"33.3Q"&N)r$rrruntime_configrrrr9c 2|j|}tjj||t j ||xs|j |tj|t j}|r-tj|t j|_ |jjtjj!|}t"j%|||j'} t"j)|| |j*d|j-| j.|_tjj3t4j6|_d|_d|_|S)a:Creates a new Extension. Args: manifest (Union[dict[str, Any], ExtensionManifest]): Required. The manifest for the Extension to be created. extension_name (str): Optional. A fully-qualified extension resource name or extension ID such as "projects/123/locations/us-central1/extensions/456" or "456" when project and location are initialized or passed. If specifying the extension ID, it should be 4-63 characters, valid characters are lowercase letters, numbers and hyphens ("-"), and it should start with a number or a lower-case letter. If not provided, Vertex AI will generate a value for this ID. display_name (str): Optional. The user-defined name of the Extension. The name can be up to 128 characters long and can comprise any UTF-8 character. description (str): Optional. The description of the Extension. runtime_config (Union[dict[str, Any], RuntimeConfig]): Optional. Runtime config controlling the runtime behavior of this Extension. Defaults to None. Returns: Extension: The extension that was created. r&)rrrr)parentrzvertexai.preview.extensions) module_namer(N)__new__r%VertexAiResourceNounWithFutureManagerr+r r_generate_display_namer to_protoExtensionManifest RuntimeConfigr9 api_clientimport_extensionrr,common_location_path_LOGGERlog_create_with_lroresultlog_create_complete_resource_nounr1rr2r-r.r/r0r3r4) clsrr$rrr9 sdk_resourceroperation_futurecreated_extensions r7createzExtension.createwsvH{{3'  22;; ( < OO%E)C)C)E#__Xu/F/F/HI  '-##%(I $(22CC,,AACD  ##C)9:,335##     5 $ &2%C%C+00&D& "-8,E,E,S,S"GG-T- )"& *. 'r8returnc.|jjS)z/Full qualified resource name for the extension.)r2rr5s r7r'zExtension.resource_names!!&&&r8c|j.t|jjj|_|jS)z,Returns the (Open)API Spec of the extension.)r3_load_api_specr2rrrRs r7rzExtension.api_specs4 >> !+D,>,>,G,G,P,PQDN~~r8c|jF|jjDcgc]!}tj|j #c}|_|jScc}w)zBReturns the (Open)API schemas for each operation of the extension.)r4r2extension_operationsr to_dictfunction_declaration)r5ops r7operation_schemaszExtension.operation_schemassY  " " *,,AA'r667'D #&&& 's&A operation_idoperation_paramsruntime_auth_configctj|j||}|r-tj|tj |_|jj|}t|S)a2Executes an operation of the extension with the specified params. Args: operation_id (str): Required. The ID of the operation to be executed. operation_params (Union[dict[str, Any], Struct]): Optional. Parameters used for executing the operation. It should be in a form of map with param name as the key and actual param value as the value. E.g. if this operation requires a param "name" to be set to "abc", you can set this to {"name": "abc"}. Defaults to an empty dictionary. runtime_auth_config (Union[dict[str, Any], AuthConfig]): Optional. The Auth configuration to execute the operation. Returns: The result of executing the extension operation. )rr[r\) r ExecuteExtensionRequestr'r r@ AuthConfigr]r0execute_extension_try_parse_execution_response)r5r[r\r]requestresponses r7executezExtension.executesk.//##%-  *0//#  "+G ',,>>wG,X66r8contentsQueryExtensionResponsec tj|jtj|}|j j |}|jrt|jtj|S)a.Queries an extension with the specified contents. Args: contents (ContentsType): Required. The content of the current conversation with the model. For single-turn queries, this is a single instance. For multi-turn queries, this is a repeated field that contains conversation history + latest request. Returns: The result of querying the extension. Raises: RuntimeError: If the response contains an error. )rrf) r QueryExtensionRequestr'r _content_types_to_gapic_contentsr0query_extensionfailure_message RuntimeErrorrg _from_gapic)r5rfrcrds r7queryzExtension.querysn*--##'HHR ,,<tfEF F.t4zz'7&}5#J/)   r8)NN)%__name__ __module__ __qualname____doc__r.#ExtensionRegistryClientWithOverrider)rJ_getter_method _list_method_delete_method_parse_resource_name_method_format_resource_name_methodstrr+ classmethodrr JsonDictr rAr_RuntimeConfigOrJsonrOpropertyr'rrrZ _StructOrJson_AuthConfigOrJsonrer ContentsTyperorw __classcell__)r6s@r7rr[s4@@L N$N$L'N"8#3 's'" )-&*%)9=I)@)@@AI! I sm I c] I!!56IIV's''&// '8FOO#<'59;? "7"7#=1"7&&78 "7 v# $ "7H<$11< "<< :> A A !!56 A A r8rcdeZdZdZdeej fdZede jddfdZ y)rgz=A class representing the response from querying an extension.stepsc||_y)z>qAS  Ts"A= &A8A= 8A= N) rxryrzr{rr rr+rr rgrnr8r7rgrg^sGGd#5#=#=>  33  !    r8rgrdrPc~|j} tj|}|S#tj$rY|SwxYw)N)contentjsonloadsJSONDecodeError)rdrs r7rbrbusH##G **W% N     N s %<<c|jr9tjj}|j |jS|j r)tjj |j SiS)zBLoads the (Open)API Spec of the extension and converts it to JSON.) open_api_yamlr. yaml_utils_maybe_import_yaml safe_loadr load_yaml)ryamls r7rTrTs_##668~~h4455  " "##--h.G.GHH Ir8)%rtypingrrrrgoogle.cloud.aiplatformrrr r.google.cloud.aiplatform_v1beta1r vertexai.generative_modelsr vertexai.reasoning_enginesr google.protobufr LoggerrxrFrr`rStructrrBrrur>rrgExecuteExtensionResponserrbrTrr8r7rsv  22(/619-& $++h &//5+;+;;<fooz'8'889 V__e.A.AAB + R,>"H ;13  (+M&7"H ;13  $*H%="G ;13  K6r@ ::@ F  .,, 6??C  r8