
    l
                     |   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Z
ej                   ej                  ej                  j                         G d d	ej                                      Zej                   ej                  ej                  j                          G d
 dej                                      Zy)z4Lists supported models for GKE Inference Quickstart.    )util)base)commands)
exceptions)logz1
To list all supported models, run:

$ {command}
c                       e Zd ZdZd Zd Zy)ListList supported models.c                    t        j                  t        j                  j                        }t        j
                  t        j                  j                        }	 |j                  j                  |j                               }|j                  r|j                  S g S # t        j                  $ rF}t        j                  d|        t        j                  j                  d|        g cY d }~S d }~ww xY w)NAn error has occurred: )r   GetClientInstancer   ReleaseTrackGAGetMessagesModulemodelsFetch GkerecommenderModelsFetchRequestr   Errorr   errorstatusPrintself_clientmessagesresponsees         0lib/surface/container/ai/profiles/models/list.pyRunzList.Run#   s    ##D$5$5$8$89F%%d&7&7&:&:;H$$

3
3
5h 
	 	ii)!-.	jj045i%   A B B C82;C3-C83C8c                     |rGt         j                  j                  d       |D ]"  }t         j                  j                  d|       $ y t         j                  j                  d       y NzSupported models:z- zNo supported models found.r   outr   r   r   	resources
model_names       r   DisplayzList.Display4   E    	ggmm'(!*dJ' " 
ggmm01    N__name__
__module____qualname____doc__r    r)    r+   r   r	   r	      s     "2r+   r	   c                       e Zd ZdZd Zd Zy)	ListAlphar
   c                    t        j                  t        j                  j                        }t        j
                  t        j                  j                        }	 |j                  j                  |j                               }|j                  r|j                  S g S # t        j                  $ rF}t        j                  d|        t        j                  j                  d|        g cY d }~S d }~ww xY w)Nr   zAn error has occured: )r   r   r   r   ALPHAr   r   r	   GkerecommenderModelsListRequest
modelNamesr   r   r   r   r   r   r   s         r   r    zListAlpha.RunB   s    ##D$5$5$;$;<F%%d&7&7&=&=>H	##H$L$L$NOh			"""	 	ii)!-.	jj/s34ir!   c                     |rGt         j                  j                  d       |D ]"  }t         j                  j                  d|       $ y t         j                  j                  d       y r#   r$   r&   s       r   r)   zListAlpha.DisplayQ   r*   r+   Nr,   r1   r+   r   r3   r3   =   s     2r+   r3   N)r0   %googlecloudsdk.api_lib.ai.recommenderr   googlecloudsdk.callioper   googlecloudsdk.command_lib.runr   googlecloudsdk.corer   r   	_EXAMPLESDefaultUniverseOnlyReleaseTracksr   r   r	   r5   r3   r1   r+   r   <module>r@      s    ; 6 ( 3 * #	 D%%(()28== 2 * 2: D%%++,2 2 - 2r+   