
    	                     b   S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSK	J
r
  SSK	Jr  SS	K	Jr  SS
K	Jr  S rS r\R                   \R"                  " \R$                  R&                  \R$                  R(                  5      \R*                   " S S\R,                  5      5       5       5       rg)z2Vertex AI deployment resource pools query command.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)flags)region_utilc                 L    [         R                  " U S[        R                  S9  g )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   [         R                  nU R                  R                  R	                  5       nUR                  5       S   U l        [        R                  " XR                  S9   [        R                  " US9nUR                  U5      nUsSSS5        $ ! , (       d  f       g= f)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                   .    \ rS rSrSr\S 5       rS rSrg)QueryDeployedModelsV1Beta13   a:  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                     [        U 5      $ N)r   r   s    r   ArgsQueryDeployedModelsV1Beta1.ArgsD   s    Vr   c                     [        U5      $ r*   )r%   )selfr!   s     r   RunQueryDeployedModelsV1Beta1.RunH   s    D>r    N)	__name__
__module____qualname____firstlineno____doc__staticmethodr+   r/   __static_attributes__r1   r   r   r'   r'   3   s       r   r'   N)r6   
__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UniverseCompatibleListCommandr'   r1   r   r   <module>rD      s    9 &  ' F ( 3 7 / 5N D%%++T->->-C-CD!1!1   E r   