
                            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  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.                  " \R0                  R2                  5       " S S\R4                  5      5       r\R.                  " \R0                  R8                  \R0                  R:                  5       " S S\5      5       rg)z!Vertex AI indexes update command.    )absolute_import)division)unicode_literals)encoding)
operations)client)base)	constants)endpoint_util)flags)indexes_util)operations_util)labels_util)logc                   4    \ rS rSrSr\S 5       rS rS rSr	g)UpdateV1"   Update an Vertex AI index.

## EXAMPLES

To update index `123` under project `example` in region `us-central1`, run:

  $ {command} --display-name=new-name
  --metadata-file=path/to/your/metadata.json --project=example
  --region=us-central1
c                 :   [         R                  " U S5        [         R                  " SSS9R                  U 5        [         R                  " S5      R                  U 5        [         R
                  " S5      R                  U 5        [        R                  " U 5        g )Nz	to updateindexF)required)r   AddIndexResourceArgGetDisplayNameArgAddToParserGetDescriptionArgGetMetadataFilePathArgr   AddUpdateLabelsFlags)parsers     lib/surface/ai/indexes/update.pyArgsUpdateV1.Args/   sm    	fk2	Ge4@@H	G$008	  )55f=$$V,    c                    UR                   R                  R                  5       nUR                  5       S   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R                  b  [        R                  " UR                  5      n[         R"                  R%                  [        R&                  R)                  UR                  SUR+                  5       SR)                  UR+                  5       XE5      S95        UsS S S 5        $ [,        R.                  " [0        R2                  " US9U[        R                  " UR                  5      SS	9n	U	b4  [4        R6                  " U	5      n
S
U
;   a  [         R8                  " U
S
   SS9  U	sS S S 5        $ ! , (       d  f       g = f)NlocationsId
projectsId)region)versionzupdate indexz%--index={} --region={} [--project={}])nameverbidsub_commandsupdate)operations_clientopop_ref
log_methodr(   zVertex AI index)kind)CONCEPTSr   ParseAsDictr   AiplatformEndpointOverridesr   IndexesClientr
   
GA_VERSIONPatch	PatchBetametadata_filer   ParseIndexOperationr(   r   statusPrint"OPERATION_CREATION_DISPLAY_MESSAGEformatNamer   WaitForOpMayber   OperationsClientr   MessageToPyValueUpdatedResource)selfargsr'   	index_refr&   
project_idindex_client	operationr/   response_msgresponses              r   _RunUpdateV1._Run7   s   ##))+I.F!!#L1J		2	27	J))':l	I((	( &&y7	 **9;					'11)..A

88??^^#;;=DKKNN$f:	 @ ;	< # 
K	J( %33&77H11)..A	l
 
	!,,\:X


hv.5F
G; 
K	J	Js   C-GA?G
G-c                 B    U R                  U[        R                  5      $ N)rM   r
   r7   rE   rF   s     r   RunUpdateV1.RunZ   s    99T9//00r"    N)
__name__
__module____qualname____firstlineno____doc__staticmethodr    rM   rR   __static_attributes__rT   r"   r   r   r   "   s&    	 - -!F1r"   r   c                       \ rS rSrSrS rSrg)UpdateV1Beta1^   r   c                 B    U R                  U[        R                  5      $ rP   )rM   r
   BETA_VERSIONrQ   s     r   rR   UpdateV1Beta1.Runk   s    99T91122r"   rT   N)rU   rV   rW   rX   rY   rR   r[   rT   r"   r   r]   r]   ^   s    	3r"   r]   N)rY   
__future__r   r   r   apitools.base.pyr   googlecloudsdk.api_lib.air   !googlecloudsdk.api_lib.ai.indexesr   googlecloudsdk.callioper	   googlecloudsdk.command_lib.air
   r   r   r   r   $googlecloudsdk.command_lib.util.argsr   googlecloudsdk.corer   ReleaseTracksReleaseTrackGAUpdateCommandr   BETAALPHAr]   rT   r"   r   <module>rp      s    ( &  ' % 0 4 ( 3 7 / 6 9 < # D%%(()81t!! 81 *81v D%%**D,=,=,C,CD3H 3 E3r"   