2SrSSKrSSKJr "SS\5rg)zTA library for streaming prediction results from the Vertex AI PredictionService API.N)apisc*\rSrSrSrSrSrSrSrg)PredictionStreamerz(Streams prediction responses using gRPC.c<[R"SU5Ulg)N aiplatform)rGetGapicClientInstanceclient)selfversions >lib/googlecloudsdk/api_lib/ai/endpoints/prediction_streamer.py__init__PredictionStreamer.__init__s--lGDDKc# # URRRUS9nUH_nURR URRR R [R"U555 Ma URRR R [R"U55Ul URRR[U/55HnUv M g7f)aStreams prediction results from the Cloud Vertex AI PredictionService API. Args: endpoint: The name of the endpoint to stream predictions from. inputs: The inputs to send to the endpoint. parameters: The parameters to send to the endpoint. Yields: Streamed prediction results. )endpointN) r typesStreamDirectPredictRequestinputsappendTensor from_jsonjsondumps parameters predictionstream_direct_predictiter)r rrrrequest curr_inputrs r StreamDirectPredict&PredictionStreamer.StreamDirectPredicts"kk::H:MG  nn ++   " " , ,TZZ -C D**11;; :Gkk,,BB gY  sD Dc## URRRXUS9nURRR [ U/55HnUv M g7f)aStreams prediction results from the Cloud Vertex AI PredictionService API. Args: endpoint: The name of the endpoint to stream predictions from. method_name: The name of the method to call. input: The input bytes to send to the endpoint. Yields: Streamed prediction results. )r method_nameinputN)r rStreamDirectRawPredictRequestrstream_direct_raw_predictr)r rr$r%rrs r StreamDirectRawPredict)PredictionStreamer.StreamDirectRawPredict<s]"kk==%>Gkk,,FF gY  sA A")r N) __name__ __module__ __qualname____firstlineno____doc__rr!r(__static_attributes__rr rrs0E>rr)r.rgooglecloudsdk.api_lib.utilrobjectrr0rr r3s[ ,==r