dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd l m Z dd l mZdd lmZd Zej$ej&j(Gddej*Zej$ej&j.ej&j0GddeZy)z$Vertex AI endpoints explain command.)absolute_import)division)unicode_literals)client)base) constants) endpoint_util)endpoints_util)flags) region_util)logc|jjj}|jd|_t j ||jd5tj|}tj|j}|tjk(r|j|||}n|j|||}t!|d8t"j$j'dj)|j*|j-dstj.|d |_|cdddS#1swYyxYw) z!Run Vertex AI online explanation. locationsIdT)region is_prediction)versiondeployedModelIdNz0Deployed model id to be used for explanation: {}format explanations)key_name)CONCEPTSendpointParseAsDictrr AiplatformEndpointOverridesrEndpointsClientr ReadInstancesFromArgs json_requestr GA_VERSIONExplain ExplainBetagetattrr statusPrintrr IsSpecifiedGetDefaultFormat)argsr endpoint_refendpoints_clientinstances_jsonresultss #lib/surface/ai/endpoints/explain.py_Runr-s''--/,##%m4$+00 dkk7--g>#99$:K:KLN)&&& ((~tLg ,,\>4Pgw)*6 jj < C C%% '(   H %"33 N,dk %777s C"E  Ec&eZdZdZedZdZy) ExplainGaRequest an online explanation from an Vertex AI endpoint. `{command}` sends an explanation request to the Vertex AI endpoint for the given instances. This command reads up to 100 instances, though the service itself accepts instances up to the payload limit size (currently, 1.5MB). ## EXAMPLES To send an explanation request to the endpoint for the json file, input.json, run: $ {command} ENDPOINT_ID --region=us-central1 --json-request=input.json ctj|dtjtj|tj dj |y)Nz to request an online explanation) prompt_funcF)required)r AddEndpointResourceArgr PromptForOpRegionAddPredictInstanceArgGetDeployedModelId AddToParser)parsers r,ArgszExplainGa.ArgsIsH   *113 ' e,88@c6t|tjSN)r-rrselfr's r,Runz ExplainGa.RunRs i** ++r;N)__name__ __module__ __qualname____doc__ staticmethodr:r@r;r,r/r/8s" AA,r;r/ceZdZdZdZy)r!r0c6t|tjSr=)r-r BETA_VERSIONr>s r,r@zExplainBeta.Rungs i,, --r;N)rArBrCrDr@rFr;r,r!r!Vs  .r;r!N)rD __future__rrr#googlecloudsdk.api_lib.ai.endpointsrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airr r r r googlecloudsdk.corer r- ReleaseTracks ReleaseTrackGACommandr/BETAALPHAr!rFr;r,rUs+&'6(378/5#2D%%((), ,*,:D%%**D,=,=,C,CD.).E.r;