3SrSSKrSSKJr SSKJr SSKJr SSKJ r SSK J r \Rr Sr S rS rSS jrS rS rSrSrSrSrSrSrSrSSjrSS\ S\R6S\4SjjrSSjrSrSr Sr!g)z(Notebook-executor executions api helper.N)runtime_templates) exceptions)parser_extensions) resources) console_iocSU;a[RRUSS9$[RRUSS9$![Ra N4f=f)zParse operation relative resource name to the operation reference object. Args: operation_name: The execution operation resource name Returns: The operation reference object z/notebookExecutionJobs/z>aiplatform.projects.locations.notebookExecutionJobs.operations) collectionz(aiplatform.projects.locations.operations)rREGISTRYParseRelativeName WrongResourceCollectionException)operation_names :lib/googlecloudsdk/api_lib/notebook_executor/executions.pyParseExecutionOperationrsu.0    1 1 N2    - -!K .   5 5   sAAAcfURRR5R5$)zGet the parent Location resource name for the execution or schedule resource. Args: args: Argparse object from Command.Run Returns: The resource name in the form projects/{project}/locations/{location}. )CONCEPTSregionParse RelativeNameargss rGetParentForExecutionOrScheduler5s&    # # % 2 2 44cfURRR5R5$)zGet the resource name for the execution. Args: args: Argparse object from Command.Run Returns: The resource name in the form projects/{project}/locations/{location}/notebookExecutionJobs/{execution_job_id}. )r executionrrrs rGetExecutionResourceNamerAs&  & & ( 5 5 77rcUR[X55n[U5(dU(d[R"SS5eU$)aChecks that the execution is a Workbench execution and returns it if so. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. service: The service to use for the API call. skip_workbench_check: Whether to skip validation of the execution type. Returns: The execution if it is a Workbench execution. Raises: InvalidArgumentException: If the execution is not a Workbench execution. EXECUTIONzeExecution is not of Workbench type. To manage Colab Enterprise executions use `gcloud colab` instead.GetCreateExecutionGetRequestIsWorkbenchExecutionrInvalidArgumentException)rmessagesserviceskip_workbench_checkrs r ValidateAndGetWorkbenchExecutionr&NsM"kk/) i ( (1E  - - 2  rcUR[X55n[U5(a[R"SS5eU$)aChecks that the execution is of Colab Enterprise type and returns it if so. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. service: The service to use for the API call. Returns: The execution if it is a Colab Enterprise execution. Raises: InvalidArgumentException: If the execution is a Workbench execution. rznExecution is not of Colab Enterprise type. To manage Workbench executions use `gcloud beta workbench` instead.r)rr#r$rs rValidateAndGetColabExecutionr(ksJkk/))$$  - - ;  rcURSL$)zFilter for Workbench executions. Args: execution: The execution item to check. Returns: True if the execution is a Workbench execution. N) kernelName)rs rr!r!s   T ))rcvSnURS5(a URU"U5URS9$g)zGet the dataform repository source from the args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: DataformRepositorySource message for the execution. cfURRR5R5$)N)rdataform_repository_namerrrs r!GetDataformRepositoryResourceNameNGetDataformRepositorySourceFromArgs..GetDataformRepositoryResourceNames$ == 1 1 7 7 9 F F HHrr-)dataformRepositoryResourceName commitShaN) IsSpecifiedHGoogleCloudAiplatformV1beta1NotebookExecutionJobDataformRepositorySource commit_sha)rr#r.s r#GetDataformRepositorySourceFromArgsr5sII 011  \ \'H'N// ]  rc~URnURS5(aU"URURS9$g)zGet the GCS notebook source from the args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: GcsNotebookSource message for the execution. gcs_notebook_uri)uri generationN)AGoogleCloudAiplatformV1beta1NotebookExecutionJobGcsNotebookSourcer2r7r9)rr#gcs_notebook_sources rGetGcsNotebookSourceFromArgsr<sHPP ())   ! !??  rcURnURS5(a#U"[R"URSS9S9$g)zCreate direct notebook source message from the args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: DirectNotebookSource message for the execution. direct_contentT)binary)contentN)DGoogleCloudAiplatformV1beta1NotebookExecutionJobDirectNotebookSourcer2rReadFromFileOrStdinr>)rr#notebook_sources rGetDirectNotebookSourceFromArgsrDsNaa/ &'' ..t/B/B6:<  rc2[UR5S-$)zGet the execution timeout from the args. Args: args: Argparse object from Command.Run Returns: Serialized Duration message for the execution timeout. s)strexecution_timeoutrs rGetExecutionTimeoutFromArgsrIs T # # $s **rcfURRR5R5$)zGet the runtime template resource name from the args. Args: args: Argparse object from Command.Run Returns: The notebook runtime template resource name. )rnotebook_runtime_templaterrrs rGetRuntimeTemplateResourceNamerLs&  0 0 6 6 8 E E GGrcURnU"[R"X5[R"X5[R"X5S9$)aGet the custom environment spec from the args for a Workbench execution. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: CustomEnvironmentSpec message for the execution. ) machineSpec networkSpecpersistentDiskSpec)EGoogleCloudAiplatformV1beta1NotebookExecutionJobCustomEnvironmentSpecruntime_templates_utilGetMachineSpecFromArgsGetNetworkSpecFromArgsGetPersistentDiskSpecFromArgs)rr#custom_environment_specs rGetCustomEnvironmentSpecrWsNTT !(??O(??O/MM  rcr[RRURSS9nUR 5$)z&Get the URL for an execution resource.z3aiplatform.projects.locations.notebookExecutionJobs) relative_namer )rr r nameSelfLink)resourcers rGetExecutionUrir]s8  22MMF3)    rcU(aDSn[X5nUR5nSnSn[R"X5n URn O*[ X5nSnSnUR n[U5nSn Sn URUU(aSO [X5U(a URO UR[U5U[X5URUUURU UU S9 $)aCreates the NotebookExecutionJob message for the create request. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the AIPlatform API. workbench_execution: Whether this execution is for a Workbench notebook. for_schedule: Whether this execution is used to create a schedule. Returns: Instance of the NotebookExecutionJob message. N) dataformRepositorySourcedirectNotebookSource displayNameexecutionTimeout executionUsergcsNotebookSource gcsOutputUri#notebookRuntimeTemplateResourceNamecustomEnvironmentSpecserviceAccountencryptionSpecworkbenchRuntimer*)rW@GoogleCloudAiplatformV1beta1NotebookExecutionJobWorkbenchRuntimerRCreateEncryptionSpecConfig kernel_namer5 user_emailrL0GoogleCloudAiplatformV1beta1NotebookExecutionJobrDexecution_display_name display_namerIr<gcs_output_uriservice_account) rr#workbench_execution for_scheduledataform_repository_sourcerVworkbench_runtimeexecution_userruntime_template_nameencryption_specrms rCreateNotebookExecutionJobr{ s!%6tFQQSN ,GG O""K!D "#__N:4@OK  B B9  *4 : --   248"4TD&&*?3))$(# C rrr# for_workbenchcN[U5n[XUSS9nURUUS9$)aUBuilds a NotebookExecutionJobsCreateRequest message for a CreateSchedule request. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. for_workbench: Indicates whether this is a Workbench execution. Returns: Instance of the NotebookExecutionJobsCreateRequest message. T)rtru)notebookExecutionJobparent)rr{=GoogleCloudAiplatformV1beta1CreateNotebookExecutionJobRequestrr#r|rnotebook_execution_jobs r'CreateExecutionCreateRequestForSchedulerBsA +4 0&5 -d  O O1  P rcb[U5n[XUS9nURUURUS9$)a8Builds a NotebookExecutionJobsCreateRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. for_workbench: Indicates whether this is a Workbench execution. Returns: Instance of the NotebookExecutionJobsCreateRequest message. )rt)0googleCloudAiplatformV1beta1NotebookExecutionJobnotebookExecutionJobIdr)rr{=AiplatformProjectsLocationsNotebookExecutionJobsCreateRequestexecution_job_idrs rCreateExecutionCreateRequestr[sH +4 0&5 -  O O7M!22  P rc2UR[U5S9$)zBuilds a NotebookExecutionJobsDeleteRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the NotebookExecutionJobsDeleteRequest message. rZ)=AiplatformProjectsLocationsNotebookExecutionJobsDeleteRequestrrr#s rCreateExecutionDeleteRequestrqs%LL'-Mrc2UR[U5S9$)zBuilds a NotebookExecutionsJobGetRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the NotebookExecutionsJobGetRequest message. r):AiplatformProjectsLocationsNotebookExecutionJobsGetRequestrrs rr r s%II'-Jrc2UR[U5S9$)zBuilds a NotebookExecutionJobsListRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the NotebookExecutionJobsListRequest message. )r);AiplatformProjectsLocationsNotebookExecutionJobsListRequestrrs rCreateExecutionListRequestrs%  M M ,T 2 N r)F)"__doc__types'googlecloudsdk.api_lib.colab_enterpriserrRgooglecloudsdk.callioperrgooglecloudsdk.corergooglecloudsdk.core.consoler Namespacerrrr&r(r!r5r<rDrIrLrWr]r{ ModuleTypeboolrrrr rrrrs/ _.5)2  ' ' 0 5 838:4 **** + H,7<6x  2,$$ r