
    
                     v   S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSK	J
r
  SSK	Jr  SS	K	Jr  \R                  " \R                  R                  5       " S
 S\R                   5      5       r\R                  " \R                  R$                  \R                  R&                  5       " S S\5      5       rg)z,Vertex AI indexes upsert datapoints command.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)flagsc                   4    \ rS rSrSr\S 5       rS rS rSr	g)UpsertDatapointsV1   zUpsert data points into the specified index.

## EXAMPLES

To upsert datapoints into an index '123', run:

  $ {command} 123 --datapoints-from-file=example.json
  --project=example --region=us-central1
c                     [         R                  " U S5        [         R                  " SSS9R                  U 5        [         R                  " SS9R                  U 5        g )Nzto upsert data points fromindexT)requiredF)r
   AddIndexResourceArgGetDatapointsFilePathArgAddToParserGetDynamicMetadataUpdateMaskArg)parsers    +lib/surface/ai/indexes/upsert_datapoints.pyArgsUpsertDatapointsV1.Args(   sG    	f&BC	""7T:FFvN	))59EEfM    c                 h   UR                   R                  R                  5       nUR                  5       S   n[        R
                  " X$S9   [        R                  " US9nU[        R                  :X  a  UR                  X15      nOUR                  X15      nUsS S S 5        $ ! , (       d  f       g = f)NlocationsId)region)version)CONCEPTSr   ParseAsDictr	   AiplatformEndpointOverridesr   IndexesClientr   
GA_VERSIONUpsertDatapointsUpsertDatapointsBeta)selfargsr   	index_refr   index_client	operations          r   _RunUpsertDatapointsV1._Run.   s    ##))+I.F		2	27	J))':l	I((	( 11)B	 55iF	 
K	J	Js   AB##
B1c                 B    U R                  U[        R                  5      $ N)r+   r   r#   r&   r'   s     r   RunUpsertDatapointsV1.Run9   s    99T9//00r    N)
__name__
__module____qualname____firstlineno____doc__staticmethodr   r+   r0   __static_attributes__r2   r   r   r   r      s'     N N
	1r   r   c                       \ rS rSrSrS rSrg)UpsertDatapointsV1Beta1=   zUpsert data points into the specified index.

## EXAMPLES

To upsert datapoints into an index `123`, run:

  $ {command} 123 --datapoints-from-file=example.json
  --project=example --region=us-central1
c                 B    U R                  U[        R                  5      $ r.   )r+   r   BETA_VERSIONr/   s     r   r0   UpsertDatapointsV1Beta1.RunI   s    99T91122r   r2   N)r3   r4   r5   r6   r7   r0   r9   r2   r   r   r;   r;   =   s    3r   r;   N)r7   
__future__r   r   r   !googlecloudsdk.api_lib.ai.indexesr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   r	   r
   ReleaseTracksReleaseTrackGACreateCommandr   ALPHABETAr;   r2   r   r   <module>rJ      s    3 &  ' 4 ( 3 7 / D%%(()1++ 1 *1@ D%%++T->->-C-CD30 3 E3r   