d VSrSSKJr SSKJr SSKJr SSKrSSKJr "SS\5r g) zTA library for streaming prediction results from the Vertex AI PredictionService API.)absolute_import)division)unicode_literalsN)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)rrrrrequest curr_inputrs rStreamDirectPredict&PredictionStreamer.StreamDirectPredict s"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!)rrr'r(r"rs rStreamDirectRawPredict)PredictionStreamer.StreamDirectRawPredict?s]"kk==%>Gkk,,FF gY  sA A")r N) __name__ __module__ __qualname____firstlineno____doc__rr$r+__static_attributes__rrrrs0E>rr) r1 __future__rrrrgooglecloudsdk.api_lib.utilrobjectrr3rrr7s&[&' ,==r