code: def get_prompt_tokens(self, _):
          
          if self.shell_ctx.default_command:
              prompt = 'az {}>> '.format(self.shell_ctx.default_command)
          else:
              prompt = 'az>> '
          return [(Token.Az, prompt)]
docs: returns prompt tokens