replace inquirer with a better lib (mostly just to understand the cli/wiz flows, but also for searchable models)

use_shortcuts is a bummer with search, so bye

make the filters behave

use a real list of the providers

we dont need preferred models honestly, it all lives in the lite llm package anyway

cleanup linting/comment changes, move all models list to the other option

cleanup cli/wizard implementation and move a util to utils

get tests passing and do a little more pr noise cleanup

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

format manually something is weird.

missing double-quote

get tests passing and do a little more pr noise cleanup

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

formatting noise

format manually something is weird.

Author: adubatl

agentstack/auth.py

@@ -7,7 +7,7 @@
 import socket
 from pathlib import Path
 
-import inquirer
+import questionary
 from appdirs import user_data_dir
 from agentstack import log
 
@@ -73,6 +73,7 @@ def do_GET(self):
             self.end_headers()
             self.wfile.write(f'Error: {str(e)}'.encode())
 
+
 def find_free_port():
     """Find a free port on localhost"""
     with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
@@ -81,6 +82,7 @@ def find_free_port():
         port = s.getsockname()[1]
     return port
 
+
 def start_auth_server():
     """Start the local authentication server"""
     port = find_free_port()
@@ -96,7 +98,7 @@ def login():
         token = get_stored_token()
         if token:
             log.success("You are already authenticated!")
-            if not inquirer.confirm('Would you like to log in with a different account?'):
+            if not questionary.confirm('Would you like to log in with a different account?').ask():
                 return
 
         # Start the local server
@@ -139,4 +141,4 @@ def get_stored_token():
             config = json.load(f)
             return config.get('bearer_token')
     except Exception:
-        return None
+        return None

agentstack/cli/cli.py

@@ -1,13 +1,63 @@
 from typing import Optional
 from art import text2art
-import inquirer
+import questionary
 from agentstack import conf, log
 from agentstack.conf import ConfigFile
-from agentstack.exceptions import ValidationError
-from agentstack.utils import validator_not_empty, is_snake_case
 from agentstack.generation import InsertionPoint
 from agentstack import repo
-from agentstack.providers import get_available_models, get_all_available_models
+from agentstack.providers import get_available_models
+from agentstack.utils import is_snake_case
+
+PREFERRED_MODELS = [
+    'groq/deepseek-r1-distill-llama-70b',
+    'deepseek/deepseek-chat',
+    'deepseek/deepseek-coder',
+    'deepseek/deepseek-reasoner',
+    'openai/gpt-4o',
+    'anthropic/claude-3-5-sonnet',
+    'openai/o1-preview',
+    'openai/gpt-4-turbo',
+    'anthropic/claude-3-opus',
+]
+
+
+def get_validated_input(
+    message: str,
+    validate_func=None,
+    min_length: int = 0,
+    snake_case: bool = False,
+) -> str:
+    """Helper function to get validated input from user.
+
+    Args:
+        message: The prompt message to display
+        validate_func: Optional custom validation function that returns (bool, str)
+        min_length: Minimum length requirement (0 for no requirement)
+        snake_case: Whether to enforce snake_case naming
+    """
+    while True:
+
+        def validate(text: str) -> str | bool:
+            if min_length and len(text) < min_length:
+                return f"Input must be at least {min_length} characters long"
+
+            if snake_case and not is_snake_case(text):
+                return "Input must be in snake_case format (lowercase with underscores)"
+
+            if validate_func:
+                is_valid, error_msg = validate_func(text)
+                if not is_valid:
+                    return error_msg
+
+            return True
+
+        value = questionary.text(
+            message,
+            validate=validate if validate_func or min_length or snake_case else None,
+        ).ask()
+
+        if value:
+            return value
 
 
 def welcome_message():
@@ -31,10 +81,10 @@ def undo() -> None:
         log.warning("There are uncommitted changes that may be overwritten.")
         for changed in changed_files:
             log.info(f" - {changed}")
-        should_continue = inquirer.confirm(
-            message="Do you want to continue?",
+        should_continue = questionary.confirm(
+            "Do you want to continue?",
             default=False,
-        )
+        ).ask()
         if not should_continue:
             return
 
@@ -50,67 +100,38 @@ def configure_default_model():
 
     log.info("Project does not have a default model configured.")
 
-    while True:
-        # Get models from litellm
-        preferred_models = get_available_models()
-        all_models = get_all_available_models()
-
-        other_msg = "Other (enter a model name)"
-        advanced_msg = f"Select from {len(all_models)} models for advanced use cases"
-
-        model = inquirer.list_input(
-            message="Which model would you like to use?",
-            choices=preferred_models + [advanced_msg, other_msg],
-        )
-
-        if model == other_msg:
-            log.info('A list of available models is available at: "https://docs.litellm.ai/docs/providers"')
-            model = inquirer.text(message="Enter the model name")
-            break
-
-        elif model == advanced_msg:
-            return_msg = "↩ Return to preferred models"
-            advanced_model = inquirer.list_input(
-                message="Select from all available models",
-                choices=[return_msg] + all_models,
-            )
-
-            if advanced_model == return_msg:
-                continue  # Go back to preferred models list
-
-            model = advanced_model
-            break
-
-        else:
-            break  # Selected from preferred models
+    # First question - show preferred models + "Other" option
+    other_msg = "Other (see all available models)"
+    model_choice = questionary.select(
+        "Which model would you like to use?",
+        choices=PREFERRED_MODELS + [other_msg],
+        use_indicator=True,
+        use_shortcuts=False,
+        use_jk_keys=False,
+        use_emacs_keys=False,
+        use_arrow_keys=True,
+        use_search_filter=True,
+    ).ask()
+
+    # If they choose "Other", show searchable all available models
+    if model_choice == other_msg:
+        log.info('\nA complete list of models is available at: "https://docs.litellm.ai/docs/providers"')
+        available_models = get_available_models()
+
+        model_choice = questionary.select(
+            "Select from all available models:",
+            choices=available_models,
+            use_indicator=True,
+            use_shortcuts=False,
+            use_jk_keys=False,
+            use_emacs_keys=False,
+            use_arrow_keys=True,
+            use_search_filter=True,
+        ).ask()
 
     log.debug("Writing default model to project config.")
     with ConfigFile() as agentstack_config:
-        agentstack_config.default_model = model
-
-
-def get_validated_input(
-    message: str,
-    validate_func=None,
-    min_length: int = 0,
-    snake_case: bool = False,
-) -> str:
-    """Helper function to get validated input from user.
-
-    Args:
-        message: The prompt message to display
-        validate_func: Optional custom validation function
-        min_length: Minimum length requirement (0 for no requirement)
-        snake_case: Whether to enforce snake_case naming
-    """
-    while True:
-        value = inquirer.text(
-            message=message,
-            validate=validate_func or validator_not_empty(min_length) if min_length else None,
-        )
-        if snake_case and not is_snake_case(value):
-            raise ValidationError("Input must be in snake_case")
-        return value
+        agentstack_config.default_model = model_choice
 
 
 def parse_insertion_point(position: Optional[str] = None) -> Optional[InsertionPoint]:

agentstack/cli/init.py

@@ -1,7 +1,7 @@
-import os, sys
+import os
+import sys
 from typing import Optional
-from pathlib import Path
-import inquirer
+import questionary
 from textwrap import shorten
 
 from agentstack import conf, log
@@ -38,12 +38,12 @@ def require_uv():
 
 def prompt_slug_name() -> str:
     """Prompt the user for a project name."""
-    
+
     def _validate(slug_name: Optional[str]) -> bool:
         if not slug_name:
             log.error("Project name cannot be empty")
             return False
-            
+
         if not is_snake_case(slug_name):
             log.error("Project name must be snake_case")
             return False
@@ -53,22 +53,12 @@ def _validate(slug_name: Optional[str]) -> bool:
             return False
 
         return True
-    
-    def _prompt() -> str:
-        return inquirer.text(
-            message="Project name (snake_case)",
-        )
-    
+
     log.info(
         "Provide a project name. This will be used to create a new directory in the "
         "current path and will be used as the project name. 🐍 Must be snake_case."
     )
-    slug_name = None
-    while not _validate(slug_name):
-        slug_name = _prompt()
-    
-    assert slug_name  # appease type checker
-    return slug_name
+    return questionary.text("Project name (snake_case)", validate=_validate).ask()
 
 
 def select_template(slug_name: str, framework: Optional[str] = None) -> TemplateConfig:
@@ -77,16 +67,23 @@ def select_template(slug_name: str, framework: Optional[str] = None) -> Template
 
     EMPTY = 'empty'
     choices = [
-        (EMPTY, "🆕 Empty Project"),
+        questionary.Choice('🆕 Empty Project', EMPTY),
     ]
     for template in templates:
-        choices.append((template.name, shorten(f"⚡️ {template.name} - {template.description}", 80)))
+        choices.append(
+            questionary.Choice(f"⚡️ {template.name} - {shorten(template.description, 80)}", template.name)
+        )
 
-    choice = inquirer.list_input(
-        message="Do you want to start with a template?",
-        choices=[c[1] for c in choices],
-    )
-    template_name = next(c[0] for c in choices if c[1] == choice)
+    template_name = questionary.select(
+        "Do you want to start with a template?",
+        choices=choices,
+        use_indicator=True,
+        use_shortcuts=False,
+        use_jk_keys=False,
+        use_emacs_keys=False,
+        use_arrow_keys=True,
+        use_search_filter=True,
+    ).ask()
 
     if template_name == EMPTY:
         return TemplateConfig(
@@ -148,11 +145,11 @@ def init_project(
 
     if framework is None:
         framework = template_data.framework
-    
+
     if framework in frameworks.ALIASED_FRAMEWORKS:
         framework = frameworks.ALIASED_FRAMEWORKS[framework]
-    
-    if not framework in frameworks.SUPPORTED_FRAMEWORKS:
+
+    if framework not in frameworks.SUPPORTED_FRAMEWORKS:
         raise Exception(f"Framework '{framework}' is not supported.")
     log.info(f"Using framework: {framework}")
 
@@ -163,7 +160,7 @@ def init_project(
     packaging.create_venv()
     log.info("Installing dependencies...")
     packaging.install_project()
-    
+
     if repo.find_parent_repo(conf.PATH):
         # if a repo already exists, we don't want to initialize a new one
         log.info("Found existing git repository; disabling tracking.")

agentstack/cli/run.py

@@ -10,7 +10,7 @@
 from agentstack.exceptions import ValidationError
 from agentstack import inputs
 from agentstack import frameworks
-from agentstack.utils import get_framework, verify_agentstack_project
+from agentstack.utils import verify_agentstack_project
 
 MAIN_FILENAME: Path = Path("src/main.py")
 MAIN_MODULE_NAME = "main"
@@ -98,7 +98,7 @@ def run_project(command: str = 'run', cli_args: Optional[List[str]] = None):
     """Validate that the project is ready to run and then run it."""
     conf.assert_project()
     verify_agentstack_project()
-    
+
     if conf.get_framework() not in frameworks.SUPPORTED_FRAMEWORKS:
         raise ValidationError(f"Framework {conf.get_framework()} is not supported by agentstack.")
 
@@ -124,7 +124,7 @@ def run_project(command: str = 'run', cli_args: Optional[List[str]] = None):
         log.notify("Running your agent...")
         project_main = _import_project_module(conf.PATH)
         main = getattr(project_main, command)
-        
+
         # handle both async and sync entrypoints
         if asyncio.iscoroutinefunction(main):
             asyncio.run(main())