dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l m Z dd l m Z dd l mZdd l mZd ZdZe j$e j&e j(j*Gdde j,Ze j$e j&e j(j0e j(j2GddeZy)z/Vertex AI endpoints direct raw predict command.)absolute_import)division)unicode_literalsN)client)base) constants) endpoint_util)endpoints_util)flags) region_utilcztj|dtjtj|y)Nz"to do online direct raw prediction) prompt_func)r AddEndpointResourceArgr PromptForOpRegionAddDirectRawPredictInputArgparsers .lib/surface/ai/endpoints/direct_raw_predict.py_AddArgsr s/ *// ##F+c|jjj}|jd|_t j ||jd5tj|}tj|j}tj|djd|d<|tj k(r|j#||}n|j%||}tj|j&|_|j)ds&tj*|j&d |_|cd d d S#1swYy xYw) z+Run Vertex AI online direct raw prediction. locationsIdT)region is_prediction)versioninputzutf-8formatoutput)key_nameN)CONCEPTSendpointParseAsDictrr AiplatformEndpointOverridesrEndpointsClientr ReadInputFromArgs json_requestbase64 b64decodedecoder GA_VERSIONDirectRawPredictDirectRawPredictBetar IsSpecifiedGetDefaultFormatr)argsr endpoint_refendpoints_client input_jsonresultss r_Runr5)s ''--/,##%m4$+00 dkk--g> 11$2C2CDJ **:g+>?FFwOJw)&&& 11, Kg 55lJOg%%gnn5GN   H %"33 ..8dk -s C4EE&c&eZdZdZedZdZy)DirectRawPredictGaaRun Vertex AI online direct raw prediction. `{command}` sends a direct raw prediction request to Vertex AI endpoint for the given input. The request limit is 10MB. ## EXAMPLES To direct 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 rArgszDirectRawPredictGa.ArgsWs  Vrc6t|tjSr9)r5rr+selfr0s rRunzDirectRawPredictGa.Run[s i** ++rN)__name__ __module__ __qualname____doc__ staticmethodr:r>rrr7r7Fs  ,rr7ceZdZdZdZy)r-aRun Vertex AI online direct raw prediction. `{command}` sends a direct raw prediction request to Vertex AI endpoint for the given input. The request limit is 10MB. ## EXAMPLES To direct raw predict against an endpoint ``123'' under project ``example'' in region ``us-central1'', run: $ {command} 123 --project=example --region=us-central1 --json-request=input.json c6t|tjSr9)r5r BETA_VERSIONr<s rr>zDirectRawPredictBeta.Runps i,, --rN)r?r@rArBr>rDrrr-r-_s  .rr-)rB __future__rrrr(#googlecloudsdk.api_lib.ai.endpointsrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airr r r r rr5DefaultUniverseOnly ReleaseTracks ReleaseTrackGACommandr7BETAALPHAr-rDrrrSs6&' 6(378/5,:D%%((),,*,.D%%**D,=,=,C,CD.-.E.r