
    
                        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r\	R                  \	R                   " \	R"                  R$                  5       " S S	\R&                  5      5       5       r\	R                  \	R                   " \	R"                  R(                  5       " S
 S\R&                  5      5       5       rg)z4Lists supported models for GKE Inference Quickstart.    )
exceptions)util)base)commands)logz1
To list all supported models, run:

$ {command}
c                   $    \ rS rSrSrS rS rSrg)List!   List supported models.c                    [         R                  " [        R                  R                  5      n[         R
                  " [        R                  R                  5      n UR                  R                  UR                  5       5      nUR                  (       a  UR                  $ / $ ! [        R                   a*  n[        R                  " U[         R                  5      eS nAff = f)N)r   GetClientInstancer   ReleaseTrackGAGetMessagesModulemodelsFetch GkerecommenderModelsFetchRequestapitools_exceptions	HttpErrorapi_lib_exceptionsHttpExceptionHTTP_ERROR_FORMAT)self_clientmessagesresponseerrors         0lib/surface/container/ai/profiles/models/list.pyRunList.Run&   s    ##D$5$5$8$89F%%d&7&7&:&:;H	L$$

3
3
5h 
	(( L,,UD4J4JKKLs   AB& $B& &C$:%CC$c                     U(       aI  [         R                  R                  S5        U H#  n[         R                  R                  SU5        M%     g [         R                  R                  S5        g NzSupported models:z- zNo supported models found.r   outPrintr   r   	resources
model_names       r   DisplayList.Display5   E    	ggmm'(!*dJ' " 
ggmm01     N__name__
__module____qualname____firstlineno____doc__r    r*   __static_attributes__r.   r-   r   r	   r	   !   s     L2r-   r	   c                   $    \ rS rSrSrS rS rSrg)	ListAlpha>   r   c                    [         R                  " [        R                  R                  5      n[         R
                  " [        R                  R                  5      n UR                  R                  UR                  5       5      nUR                  (       a  UR                  $ / $ ! [        R                   aG  n[        R                  " SU 35        [        R                  R                  SU 35        / s S nA$ S nAff = f)NzAn error has occurred: zAn error has occured: )r   r   r   r   ALPHAr   r   r	   GkerecommenderModelsListRequest
modelNamesr   Errorr   r   statusr&   )r   r   r   r   r   es         r   r    ListAlpha.RunC   s    ##D$5$5$;$;<F%%d&7&7&=&=>H	##H$L$L$NOh			"""	 	ii)!-.	jj/s34is%   AB& $B& &D:<C<6D<Dc                     U(       aI  [         R                  R                  S5        U H#  n[         R                  R                  SU5        M%     g [         R                  R                  S5        g r#   r$   r'   s       r   r*   ListAlpha.DisplayR   r,   r-   r.   Nr/   r.   r-   r   r7   r7   >   s     2r-   r7   N)r4   apitools.base.pyr   r   %googlecloudsdk.api_lib.ai.recommenderr   googlecloudsdk.api_lib.utilr   googlecloudsdk.callioper   googlecloudsdk.command_lib.runr   googlecloudsdk.corer   	_EXAMPLESDefaultUniverseOnlyReleaseTracksr   r   r	   r:   r7   r.   r-   r   <module>rL      s    ; > 6 H ( 3 * #	 D%%(()28== 2 * 26 D%%++,2 2 - 2r-   