|dZddlmZddlmZddlmZddlmZddlm Z dZ ejejejjGdd ejZejejejj Gd d ejZy ) z;Lists supported model servers for GKE Inference Quickstart.)util)base)commands) exceptions)logzt To list all supported model servers for a model, run: $ {command} --model=deepseek-ai/DeepSeek-R1-Distill-Qwen-7B c,eZdZdZedZdZdZy)ListzList supported model servers for a given model. To get supported models, run `gcloud container ai profiles models list`. c,|jdddyNz--modelTz The model.)requiredhelp add_argumentparsers 7lib/surface/container/ai/profiles/model_servers/list.pyArgsz List.Args(  ctjtjj}tj tjj} |j |j}|jj|}|jr |jSgS#tj$rF}tjd|tjjd|gcYd}~Sd}~wwxYw)N)modelAn error has occurred: )rGetClientInstancer ReleaseTrackGAGetMessagesModule&GkerecommenderModelServersFetchRequestr modelServersFetchrErrorrerrorstatusPrintselfargsclientmessagesrequestresponsees rRunzList.Run0s  # #D$5$5$8$8 9F%%d&7&7&:&:;H ?? @g$$**73h   $$$    ii)!-. jj045 i%AB-+B--D;D;DDc|rGtjjd|D]"}tjjd|$ytjjdyNzSupported model servers:z- z!No supported model servers found.routr#r%_ resourcesmodel_server_names rDisplayz List.DisplayBG ggmm./(   d-. ) ggmm78rN__name__ __module__ __qualname____doc__ staticmethodrr,r6rrr r s% $9rr c,eZdZdZedZdZdZy) ListAlphaa List supported model servers for a given model. To get supported models, run `gcloud alpha container ai profiles models list` or to get all supported model and server combinations, run `gcloud alpha container ai profiles model-and-server-combinations list`. c,|jdddyr rrs rrzListAlpha.ArgsVrrctjtjj}tj tjj} |j |j}|jj|}|jr |jSgS#tj$rF}tjd|tjj!d|gcYd}~Sd}~wwxYw)N) modelNamer)rrrrALPHAr%GkerecommenderModelServersListRequestrrr modelServerNamesrr rr!r"r#r$s rr,z ListAlpha.Run^s  # #D$5$5$;$; H >>JJ?g$$))'2h  " "(((    ii)!-. jj045 ir-c|rGtjjd|D]"}tjjd|$ytjjdyr/r0r2s rr6zListAlpha.Displaypr7rNr8r>rrr@r@Ks%$9rr@N)r<%googlecloudsdk.api_lib.ai.recommenderrgooglecloudsdk.calliopergooglecloudsdk.command_lib.runrgooglecloudsdk.corerr_EXAMPLEDefaultUniverseOnly ReleaseTracksrrr rDr@r>rrrOsB6(3*# D%%(()'98=='9*'9TD%%++,)9 )9-)9r