
    	                     D   d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
 ddl	mZ dd	l	mZ dd
l	mZ d Zd Zej                    ej"                  ej$                  j&                  ej$                  j(                         G d dej*                                      Zy)z2Vertex AI deployment resource pools query command.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)flags)region_utilc                 P    t        j                  | dt        j                         y )Nzto query)prompt_func)r
   AddDeploymentResourcePoolArgr   .PromptForDeploymentResourcePoolSupportedRegionparsers    Alib/surface/ai/deployment_resource_pools/query_deployed_models.py	_ArgsBetar      s     $$LLN    c                 Z   t         j                  }| j                  j                  j	                         }|j                         d   | _        t        j                  || j                        5  t        j                  |      }|j                  |      }|cddd       S # 1 sw Y   yxY w)zOQueries Vertex AI deployed models sharing a specified deployment resource pool.locationsId)region)versionN)r   BETA_VERSIONCONCEPTSdeployment_resource_poolParseAsDictr   r	   AiplatformEndpointOverridesr   DeploymentResourcePoolsClientQueryDeployedModelsBeta)argsr   deployment_resource_pool_ref deployment_resource_pools_clientquery_responses        r   _RunBetar%   $   s    ""'!%!G!G!M!M!O,335mD$+00M'-'K'K($ 6MM$&N  NMMs   .)B!!B*c                   &    e Zd ZdZed        Zd Zy)QueryDeployedModelsV1Beta1aH  Queries Vertex AI deployed models sharing a specified deployment resource pool.

  This command queries deployed models sharing the specified deployment resource
  pool.

  ## EXAMPLES

  To query a deployment resource pool with name ''example'' in region
  ''us-central1'', run:

    $ {command} example --region=us-central1
  c                     t        |       S N)r   r   s    r   ArgszQueryDeployedModelsV1Beta1.ArgsC   s    Vr   c                     t        |      S r)   )r%   )selfr!   s     r   RunzQueryDeployedModelsV1Beta1.RunG   s    D>r   N)__name__
__module____qualname____doc__staticmethodr*   r-    r   r   r'   r'   3   s       r   r'   N)r1   
__future__r   r   r   3googlecloudsdk.api_lib.ai.deployment_resource_poolsr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   r	   r
   r   r   r%   HiddenReleaseTracksReleaseTrackALPHABETAListCommandr'   r3   r   r   <module>r>      s    9 &  ' F ( 3 7 / 5N D%%++T->->-C-CD!1!1  E r   