dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd l m Z dd l mZd Zd Zej"ej$j&Gddej(Zej"ej$j,ej$j.GddeZy)z$Vertex AI endpoints predict command.)absolute_import)division)unicode_literals)client)base) constants) endpoint_util)endpoints_util)flags) region_utilcztj|dtjtj|y)Nzto do online prediction) prompt_func)r AddEndpointResourceArgr PromptForOpRegionAddPredictInstanceArgparsers #lib/surface/ai/endpoints/predict.py_AddArgsrs/ //1f%c2|jjj}|jd|_t j ||jd5tj|}tj|j}|tjk(r|j||}n|j||}|j!ds$tj"|j$|_|cdddS#1swYyxYw)z Run Vertex AI online prediction. locationsIdT)region is_prediction)versionformatN)CONCEPTSendpointParseAsDictrr AiplatformEndpointOverridesrEndpointsClientr ReadInstancesFromArgs json_requestr GA_VERSIONPredict PredictBeta IsSpecifiedGetDefaultFormat predictionsr)argsr endpoint_refendpoints_clientinstances_jsonresultss r_Runr0&s''--/,##%m4$+00 dkk7--g>#99$:K:KLN)&&& ((~Fg ,,\>Jg   H %"33G4G4GHdk 777s B$D  Dc&eZdZdZedZdZy) PredictGaRun Vertex AI online prediction. `{command}` sends a prediction request to Vertex AI endpoint for the given instances. This command will read up to 100 instances, though the service itself will accept instances up to the payload limit size (currently, 1.5MB). ## EXAMPLES To predict against an endpoint ``123'' under project ``example'' in region ``us-central1'', run: $ {command} 123 --project=example --region=us-central1 --json-request=input.json ct|yN)rrs rArgszPredictGa.ArgsLs  Vrc6t|tjSr5)r0rr%selfr+s rRunz PredictGa.RunPs i** ++rN)__name__ __module__ __qualname____doc__ staticmethodr6r:rrr2r2:s  ,rr2ceZdZdZdZy)r'r3c6t|tjSr5)r0r BETA_VERSIONr8s rr:zPredictBeta.Runfs i,, --rN)r;r<r=r>r:r@rrr'r'Ts  .rr'N)r> __future__rrr#googlecloudsdk.api_lib.ai.endpointsrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airr r r r rr0 ReleaseTracks ReleaseTrackGACommandr2BETAALPHAr'r@rrrNs+&'6(378/5&(D%%((), ,*,2D%%**D,=,=,C,CD.).E.r