ydZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z dd l m Z dd l mZdd l mZdd lmZdd lmZddlZdZdZe j0e j2e j4j6Gdde j8Ze j0e j2e j4j<e j4j>GddeZ y)z/Vertex AI endpoints stream-raw-predict command.)absolute_import)division)unicode_literalsN)client)base) constants) endpoint_util)flags) region_util) exceptions) console_ioctj|dtjtjj |tj j |y)Nz"to do online stream raw prediction) prompt_func)r AddEndpointResourceArgr PromptForOpRegionGetRawPredictHeadersArg AddToParserGetRawPredictRequestArgparsers .lib/surface/ai/endpoints/stream_raw_predict.py_AddArgsr#sO *// !--f5!--f5c #LK|jjj}|jd|_t j ||j5|jjdr%tj|jddd}n|jjd}tj| }|j||j|D]d}|j!d s*t"j$j&j)|> t+j,|j/df dddy#t0$r,t3j4d t7j8|zwxYw#1swYyxYww) z Run Vertex AI online prediction. locationsId)region@NT)binaryzutf-8)versionformatz=No JSON object could be decoded from the HTTP response body: )CONCEPTSendpointParseAsDictrr AiplatformEndpointOverridesrequest startswithr ReadFromFileOrStdinencoderEndpointsClientStreamRawPredict http_headers IsSpecifiedsysstdoutbufferwritejsonloadsdecode ValueErrorcore_exceptionsErrorsix text_type)argsr endpoint_refr'endpoints_clientresponses r_Runr?.sO''--/,##%m4$+00M ||s#..t||AB/?Mg ##G,g--g>$55d''   h ' ) jj122NM( ## LmmH% &   )NMs7AF$ C F.&E F F$ 5FFF!F$c&eZdZdZedZdZy)r,Run Vertex AI online stream raw prediction. `{command}` sends a stream raw prediction request to a Vertex AI endpoint. The request can be given on the command line or read from a file or stdin. ## EXAMPLES To predict against an endpoint ``123'' under project ``example'' in region ``us-central1'', reading the request from the command line, run: $ {command} 123 --project=example --region=us-central1 --request='{ "instances": [ { "values": [1, 2, 3, 4], "key": 1 }, { "values": [5, 6, 7, 8], "key": 2 } ] }' If the request body was in the file ``input.json'', run: $ {command} 123 --project=example --region=us-central1 --request=@input.json To send the image file ``image.jpeg'' and set the *content type*, run: $ {command} 123 --project=example --region=us-central1 --http-headers=Content-Type=image/jpeg --request=@image.jpeg ct|yN)rrs rArgszStreamRawPredict.Argsls  Vrc6t|tjSrC)r?r GA_VERSIONselfr;s rRunzStreamRawPredict.Runps i** ++rN)__name__ __module__ __qualname____doc__ staticmethodrDrIrrr,r,Ns 6,rr,ceZdZdZdZy)StreamRawPredictBetarAc6t|tjSrC)r?r BETA_VERSIONrGs rrIzStreamRawPredictBeta.Runs i,, --rN)rJrKrLrMrIrOrrrQrQts 6.rrQ)!rM __future__rrrr3r/#googlecloudsdk.api_lib.ai.endpointsrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airr r r googlecloudsdk.corer r7googlecloudsdk.core.consoler r9rr?DefaultUniverseOnly ReleaseTracks ReleaseTrackGACommandr,BETAALPHArQrOrrras6&' 6(37/5=2 6 @D%%(()!,t||!,*!,HD%%**D,=,=,C,CD.+.E.r