SrSSKJr SSKJr SSKJr SSKrSSKJr SSKJ r SSKJ r SS K J r S r S rS rS rSrSSjrg)z-Utilities for AI Platform endpoints commands.)absolute_import)division)unicode_literalsN)errors) resources)yaml) console_iocSU;a[RRUSS9$[RRUSS9$![Ra N4f=f)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   sAAAc [R"USS9n[R"U5n[R "U5sSSS5 $![ a [R"SUS35ef=f!,(df  g=f)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:, ! " sA5A  %A22A55 Bc[USS9n[U[5(d[R"S5eSU;a[R"S5e[US[ 5(d[R"S5eU$)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($ / /  * *:  .cR[USS9nSU;a[R"S5eU$)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*c[USS9nSU;a[R"S5eSU;aSU;a[R"S5eU$)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*c[U[5(dgU(dg[US[5(a<SR[ USR 555nSR X5$SR U5$)z1Get default output format for prediction results.jsonNrz, zc table( {}:format="table( {} )" )ztable[no-heading]({}))r#r%r$joinsortedkeysformat) predictionskey_namer8s rGetDefaultFormatr<xsv 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*