Add command to update user's password

Author: adrysn

src/ai/backend/client/auth.py

@@ -105,3 +105,19 @@ async def logout(cls) -> None:
         rqst = Request(cls.session, 'POST', '/server/logout')
         async with rqst.fetch() as resp:
             resp.raw_response.raise_for_status()
+
+    @api_function
+    @classmethod
+    async def update_password(cls, old_password: str, new_password: str, new_password2: str) -> dict:
+        """
+        Update user's password. This API works only for account owner.
+        """
+        from .request import Request
+        rqst = Request(cls.session, 'POST', '/auth/update-password')
+        rqst.set_json({
+            'old_password': old_password,
+            'new_password': new_password,
+            'new_password2': new_password2,
+        })
+        async with rqst.fetch() as resp:
+            return await resp.json()

src/ai/backend/client/cli/config.py

@@ -97,3 +97,24 @@ def logout():
                 pass
         except Exception as e:
             print_error(e)
+
+
+@main.command()
+@click.argument('old_password', metavar='OLD_PASSWORD')
+@click.argument('new_password', metavar='NEW_PASSWORD')
+@click.argument('new_password2', metavar='NEW_PASSWORD2')
+def update_password(old_password, new_password, new_password2):
+    '''
+    Update user's password.
+    '''
+    config = get_config()
+    if config.endpoint_type != 'session':
+        print_warn('To update password, your endpoint type must be "session".')
+        raise click.Abort()
+
+    with Session() as session:
+        try:
+            session.Auth.update_password(old_password, new_password, new_password2)
+            print_done('Password updated.')
+        except Exception as e:
+            print_error(e)