dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z d Z d Zd Zd ZdZddZy)z-Utilities for AI Platform endpoints commands.)absolute_import)division)unicode_literalsN)errors) resources)yaml) console_iocd|vr" tjj|dStjj|dS#tj$rY6wxYw)zParse operation resource to the operation reference object. Args: operation_name: The operation resource to wait on Returns: The operation reference object z /endpoints/z2aiplatform.projects.locations.endpoints.operations) collectionz(aiplatform.projects.locations.operations)rREGISTRYParseRelativeName WrongResourceCollectionException)operation_names 3lib/googlecloudsdk/command_lib/ai/endpoints_util.pyParseOperationrsun$    1 1 I2    - -!K .   5 5   s AAActj|d}tj|5} t j |cdddS#t $rtjd|dwxYw#1swYyxYw)zLoads a YAML file.T)binaryNz6Input is not in JSON format. See `gcloud ai endpoints z --help` for details.) r ReadFromFileOrStdinioBytesIOrload ValueErrorrInvalidInstancesFileError) file_path sdk_methoddatafs r _LoadYamlr3sv  ' ' $ ?$ zz$1 YYq\   , , B:, ! " sA3A  #A00A33A<ct|d}t|tstjdd|vrtjdt|dt stjd|S)aReads the instances from the given file path ('-' for stdin). Args: json_request: str or None, a path to a file ('-' for stdin) containing the JSON body of a prediction request. Returns: A list of instances. Raises: InvalidInstancesFileError: If the input file is invalid (invalid format or contains too many/zero instances), or an improper combination of input files was given. predictrz]Input instances are not in JSON format. See `gcloud ai endpoints predict --help` for details. instancesz3Invalid JSON request: missing "instances" attributez0Invalid JSON request: "instances" must be a list)r isinstancedictrrlist json_requestrequests rReadInstancesFromArgsr)@s ly 9' GT "  * * @    * *=  GK($ /  * *:  .cRt|d}d|vrtjd|S)z7Validates and reads json request for Direct Prediction.zdirect-predictr!inputsz Input json must contain "inputs"rrrr&s rReadInputsFromArgsr.ds. l/? @' W  * *+M NN .r*ct|d}d|vrtjdd|vrd|vrtjd|S)z;Validates and reads json request for Direct Raw Prediction.zdirect-raw-predictr!inputzInput json must contain "input" method_name methodNamez5Input json must contain "method_name" or "methodName"r-r&s rReadInputFromArgsr3lsU l/C D' G  * *+L MM'!l'&A  * *?  .r*ct|tsy|syt|dtr=djt |dj }dj ||Sdj |S)z1Get default output format for prediction results.jsonNrz, zc table( {}:format="table( {} )" )ztable[no-heading]({}))r#r%r$joinsortedkeysformat) predictionskey_namer8s rGetDefaultFormatr<xsr K &   +a.$' 99VKN//12 3D  vh% & # ) )( 33r*)r:)__doc__ __future__rrrrgooglecloudsdk.command_lib.airgooglecloudsdk.corerrgooglecloudsdk.core.consoler rrr)r.r3r<r*rrCsA4&' 0)$2, !H 4r*