SrSSKJr SSKJr SSKJr SSKJr SSKJr SSKJ r SS jr "S S \R5r "S S \ 5rg)z@Utilities for dealing with long-running operations (simple uri).)absolute_import)division)unicode_literals)apis)waiter) constantsNcH[R"[RXS9$)N)no_http)rGetClientInstancerAI_PLATFORM_API_NAME) api_versionr s +lib/googlecloudsdk/api_lib/ai/operations.pyr r s"  $$k DDc8^\rSrSrSrU4SjrSrSrSrU=r $)AiPlatformOperationPollerzPoller for AI Platform operations API. This is necessary because the core operations library doesn't directly support simple_uri. c>Xl[[U]URRRURRR5 gN)clientsuperr__init__projects_locations_operations)selfr __class__s rr"AiPlatformOperationPoller.__init__&s>K #T3 88 88:rc8URRU5$r)rGetr operation_refs rPollAiPlatformOperationPoller.Poll,s ;;??= ))rcU$r)r operations r GetResult#AiPlatformOperationPoller.GetResult/s r)r) __name__ __module__ __qualname____firstlineno____doc__rr r%__static_attributes__ __classcell__)rs@rrrs : *rrcN\rSrSrSrSS\R 4SjrSrSSjr Sr g) OperationsClient3z9High-level client for the AI Platform operations surface.NcU=(d [[RU5UlU=(d URRUlgr)r rAI_PLATFORM_API_VERSIONrMESSAGES_MODULEmessages)rrr4versions rrOperationsClient.__init__6s<4-))'24DK; ; ;DMrcURRRURR UR 5S95$)N)name)rrrr4/AiplatformProjectsLocationsOperationsGetRequest RelativeNamers rrOperationsClient.Get>sC ;; 4 4 8 8 EE++- F / 00rc[U5nURU5(aU$UcSRUR55n[R "XRX4S9$)aWait until the operation is complete or times out. Args: operation: The operation resource to wait on operation_ref: The operation reference to the operation resource. It's the result by calling resources.REGISTRY.Parse message: str, the message to print while waiting. max_wait_ms: int, number of ms to wait before raising WaitException. Returns: The operation resource when it has completed Raises: OperationTimeoutError: when the operation polling times out OperationError: when the operation completed with an error zWaiting for operation [{}]) max_wait_ms)rIsDoneformatNamerWaitFor)rr$rmessager=pollers rWaitForOperation!OperationsClient.WaitForOperationCsW&'t ,F }}Y ,33M4F4F4HIg >>w r)rr4)Ni@w) r'r(r)r*r+r BETA_VERSIONrrrDr,r#rrr/r/3s,A --<0 AHrr/)NF)r+ __future__rrrgooglecloudsdk.api_lib.utilrrgooglecloudsdk.command_lib.airr CloudOperationPollerrobjectr/r#rrrLsAG&',.3D  ; ;(+v+r