[WIP][feature enhancement] Add all supported litellm models to "other" option in cli/wiz

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,14 +1,12 @@
 from typing import Optional
-import os, sys
 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
+from agentstack.utils import is_snake_case
 
 PREFERRED_MODELS = [
     'groq/deepseek-r1-distill-llama-70b',
@@ -23,6 +21,48 @@
 ]
 
 
+def get_validated_input(
+    message: str,
+    default: Optional[str] = None,
+    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
+        if no validators are provided the default validation applies
+    """
+    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,
+            default=default,
+            validate=validate if validate_func or min_length or snake_case else None,
+        ).ask()
+
+        if value:
+            return value
+
+
 def welcome_message():
     title = text2art("AgentStack", font="smisome1")
     tagline = "The easiest way to build a robust agent application!"
@@ -38,16 +78,16 @@ def welcome_message():
 def undo() -> None:
     """Undo the last committed changes."""
     conf.assert_project()
-    
+
     changed_files = repo.get_uncommitted_files()
     if changed_files:
         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
 
@@ -62,43 +102,39 @@ def configure_default_model():
         return  # Default model already set
 
     log.info("Project does not have a default model configured.")
-    other_msg = "Other (enter a model name)"
-    model = inquirer.list_input(
-        message="Which model would you like to use?",
-        choices=PREFERRED_MODELS + [other_msg],
-    )
 
-    if model == other_msg:  # If the user selects "Other", prompt for a model name
-        log.info('A list of available models is available at: "https://docs.litellm.ai/docs/providers"')
-        model = inquirer.text(message="Enter the model name")
+    # 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 (some may not yet be supported by AgentStack):",
+            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]:
@@ -113,4 +149,3 @@ def parse_insertion_point(position: Optional[str] = None) -> Optional[InsertionP
         raise ValueError(f"Position must be one of {','.join(valid_positions)}.")
 
     return next(x for x in InsertionPoint if x.value == position)
-

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/tools.py

@@ -1,16 +1,13 @@
 from typing import Optional
-import itertools
 from difflib import get_close_matches
-import inquirer
+import questionary
 from agentstack import conf, log
 from agentstack.utils import term_color, is_snake_case
 from agentstack import generation
 from agentstack import repo
 from agentstack._tools import get_all_tools, get_all_tool_names
 from agentstack.agents import get_all_agents
 from pathlib import Path
-import sys
-import json
 
 
 def list_tools():
@@ -29,7 +26,7 @@ def list_tools():
             if tool.category not in categories:
                 categories[tool.category] = []
             categories[tool.category].append(tool)
-    
+
     print("\n\nAvailable AgentStack Tools:")
     # Display tools by category
     for category in sorted(categories.keys()):
@@ -38,7 +35,7 @@ def list_tools():
             print("  - ", end='')
             print(term_color(f"{tool.name}", 'blue'), end='')
             print(f": {tool.url if tool.url else 'AgentStack default tool'}")
-
+    
     # Display custom tools if any exist
     if custom_tools:
         print("\nCustom Tools:")
@@ -77,33 +74,23 @@ def add_tool(tool_name: Optional[str], agents=Optional[list[str]]):
     if not tool_name:
         # Get all available tools including custom ones
         available_tools = [t for t in get_all_tools() if t is not None]
-        tool_names = [t.name for t in available_tools]
-
-        # ask the user for the tool name
-        tools_list = [
-            inquirer.List(
-                "tool_name",
-                message="Select a tool to add to your project",
-                choices=tool_names,
-            )
-        ]
-        try:
-            tool_name = inquirer.prompt(tools_list)['tool_name']
-        except TypeError:
-            return  # user cancelled the prompt
+
+        tool_name = questionary.select(
+            "Select a tool to add to your project",
+            choices=[t.name for t in available_tools],
+            use_indicator=True,
+            use_shortcuts=False,
+            use_jk_keys=False,
+            use_emacs_keys=False,
+            use_arrow_keys=True,
+            use_search_filter=True,
+        ).ask()
 
         # ask the user for the agents to add the tool to
-        agents_list = [
-            inquirer.Checkbox(
-                "agents",
-                message="Select which agents to make the tool available to",
-                choices=[agent.name for agent in get_all_agents()],
-            )
-        ]
-        try:
-            agents = inquirer.prompt(agents_list)['agents']
-        except TypeError:
-            return  # user cancelled the prompt
+        agents = questionary.checkbox(
+            "Select which agents to make the tool available to",
+            choices=[agent.name for agent in get_all_agents()],
+        ).ask()
 
     assert tool_name  # appease type checker