g RdZddlmZddlmZddlmZdZdZ dZ dZ d d Z y ) z1A library that is used to support trace commands.)encoding)apis)logc|dk(rdSdS)zXReturns the correct API version. Args: release_track: base.ReleaseTrack object GAv1v1beta1 release_tracks 3lib/googlecloudsdk/api_lib/colab_enterprise/util.pyApiVersionSelectorrs $&5I5c@tjdt|S)z%Returns the client for the trace API. aiplatform) core_apisGetClientInstancerr s r GetClientr  $ $&}5 rc@tjdt|S)z'Returns the messages for the trace API.r)rGetMessagesModulerr s r GetMessagesr&rrc|jS)z2Returns whether the user specified the async flag.)async_)argss r GetAsyncConfigr-s rNc6tjtjtjd}|r|||j|d|S|j |||j } | tj| } d| vr| d}||||d| S)a>Waits for an operation if asynchronous flag is off. Args: operations_client: api_lib.ai.operations.OperationsClient, the client via which to poll. op: Cloud AI Platform operation, the operation to poll. op_ref: The operation reference to the operation resource. It's the result by calling resources.REGISTRY.Parse log_method: Logging method used for operation. kind: str, the resource kind (eg runtime template), which will be passed to logging function. asynchronous: bool, whether to wait for the operation or return immediately message: str, the message to display while waiting for the operation. resource: str, name of the resource the operation is acting on Returns: The result of the operation if asynchronous is true, or the Operation message otherwise. )createdeleteupdateT)resourcekindis_async)messagenameF) rCreatedResourceDeletedResourceUpdatedResourcer%WaitForOperationresponserMessageToPyValue) operations_clientopop_ref log_methodr" asynchronousr$r!logging_function response_msgr*s r WaitForOpMayber32s<######  Z "''tL I#33&'4 H((6H &!h:dU r)FNN) __doc__apitools.base.pyrgooglecloudsdk.api_lib.utilrrgooglecloudsdk.corerrrrrr3r rr r8s78%9#6  1r