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)z6Vertex AI endpoints stream direct raw predict command.)absolute_import)division)unicode_literalsN)prediction_streamer)base) constants) endpoint_util)endpoints_util)flags) region_utilcztj|dtjtj|y)Nz)to do online stream direct raw prediction) prompt_func)r AddEndpointResourceArgr PromptForOpRegionAddDirectRawPredictInputArgparsers 5lib/surface/ai/endpoints/stream_direct_raw_predict.py_AddArgsr s/ 1// ##F+c#K|jjj}|jd|_t j ||jd5tj|j}|tjk(rtjd}ntjd}|jdsd|_|j!|j#|d|d  D]*}t%j&|j(|_|, d d d y #1swYy xYww) z2Run Vertex AI online stream direct raw prediction. locationsIdT)region is_predictionv1v1beta1formatjson method_nameinput)endpointrr N)CONCEPTSr!ParseAsDictrr AiplatformEndpointOverridesr ReadInputFromArgs json_requestr GA_VERSIONrPredictionStreamer IsSpecifiedrStreamDirectRawPredict RelativeNamebase64 b64decodeoutput)argsversion endpoint_ref input_jsonstreamerresps r_Runr6)s ''--/,##%m4$+00 dkk 11$2C2CDJ)&&&$77=h$77 Bh   H %dk//**,}-!0 $$T[[1dk j sAE!C D6- E6D?;Ec&eZdZdZedZdZy)StreamDirectRawPredictGaaRun Vertex AI online stream direct raw prediction. `{command}` sends a stream direct raw prediction request to Vertex AI endpoint for the given input. The request limit is 10MB. ## EXAMPLES To stream 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 rArgszStreamDirectRawPredictGa.ArgsSs  Vrc6t|tjSr:)r6rr(selfr0s rRunzStreamDirectRawPredictGa.RunWs i** ++rN)__name__ __module__ __qualname____doc__ staticmethodr;r?rrr8r8Bs  ,rr8ceZdZdZdZy)StreamDirectRawPredictBetaaRun Vertex AI online stream direct raw prediction. `{command}` sends a stream direct raw prediction request to Vertex AI endpoint for the given input. The request limit is 10MB. ## EXAMPLES To stream 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|tjSr:)r6r BETA_VERSIONr=s rr?zStreamDirectRawPredictBeta.Runls i,, --rN)r@rArBrCr?rErrrGrG[s  .rrG)rC __future__rrrr-#googlecloudsdk.api_lib.ai.endpointsrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airr r r r rr6DefaultUniverseOnly ReleaseTracks ReleaseTrackGACommandr8BETAALPHArGrErrrUs=&' C(378/5,2D%%((),t||,*,.D%%**D,=,=,C,CD.!9.E.r