fix: replace questionary with rich-toolkit

Signed-off-by: Frost Ming <[email protected]>

Author: frostming

pdm.lock

@@ -62,9 +62,9 @@ dependencies = [
     "tomli-w",
     "httpx-ws>=0.6.0",
     "aiosqlite>=0.20.0",
-    "questionary>=2.0.1",
     "a2wsgi>=1.10.7",
     "fsspec>=2025.7.0",
+    "rich-toolkit>=0.15.1",
 ]
 dynamic = ["version"]
 [project.urls]

pyproject.toml

@@ -49,13 +49,14 @@ def cloud_command():
 )
 def login(endpoint: str, api_token: str) -> None:  # type: ignore (not accessed)
     """Login to BentoCloud."""
-    import questionary
     from rich.prompt import Confirm
+    from rich.prompt import Prompt
+    from rich_toolkit.menu import Menu
 
     if not api_token:
-        choice = questionary.select(
-            message="How would you like to authenticate BentoML CLI? [Use arrows to move]",
-            choices=[
+        choice = Menu(
+            label="How would you like to authenticate BentoML CLI?",
+            options=[
                 {
                     "name": "Create a new API token with a web browser",
                     "value": "create",
@@ -104,9 +105,7 @@ def login(endpoint: str, api_token: str) -> None:  # type: ignore (not accessed)
                 )
                 time.sleep(1)
         elif choice == "paste":
-            api_token = click.prompt(
-                "? Paste your API token", type=str, hide_input=True
-            )
+            api_token = Prompt.ask(":key: Paste your API token", password=True)
     try:
         cloud_rest_client = RestApiClient(endpoint, api_token)
         user = cloud_rest_client.v1.get_current_user()

src/bentoml_cli/cloud.py

@@ -261,7 +261,8 @@ def codespace(
 
         $ bentoml code --attach <codespace-name>
     """
-    import questionary
+    from rich.prompt import Prompt
+    from rich_toolkit.menu import Menu
 
     if attach and (env or secret):
         raise CLIException("Cannot specify both --attach and --env or --secret")
@@ -288,16 +289,16 @@ def codespace(
                 ]
             ]
 
-        chosen = questionary.select(
-            message="Select a codespace to attach to or create a new one",
-            choices=[{"name": d.name, "value": d} for d in deployments]
+        choice = Menu["t.Literal['new'] | Deployment"](
+            label="Select a codespace to attach to or create a new one",
+            options=[{"name": d.name, "value": d} for d in deployments]
             + [{"name": "Create a new codespace", "value": "new"}],
         ).ask()
 
-        if chosen == "new":
-            name = questionary.text(
+        if choice == "new":
+            name = Prompt.ask(
                 "Enter a name for the new codespace, or leave it blank to get a random name derived from the service"
-            ).ask()
+            )
             deployment = create_deployment(
                 bento=bento_dir,
                 name=name or None,
@@ -307,14 +308,12 @@ def codespace(
                 env=env,
                 secret=secret,
             )
-        elif chosen is None:
-            return
         else:
             if env or secret:
                 rich.print(
                     "[yellow]Warning:[/] --env and --secret are ignored when attaching to an existing codespace"
                 )
-            deployment = t.cast(Deployment, chosen)
+            deployment = choice
     deployment.watch(bento_dir)