Interactive Wizard

agentstack/_tools/__init__.py

@@ -13,6 +13,12 @@
 TOOLS_CONFIG_FILENAME: str = 'config.json'
 
 
+class ToolCategory(pydantic.BaseModel):
+    name: str
+    title: str  # human readable title
+    description: str
+
+
 class ToolConfig(pydantic.BaseModel):
     """
     This represents the configuration data for a tool.
@@ -100,6 +106,19 @@ def module(self) -> ModuleType:
             )
 
 
+def get_all_tool_categories() -> list[ToolCategory]:
+    categories = []
+    filename = TOOLS_DIR / 'categories.json'
+    data = open_json_file(filename)
+    for name, category in data.items():
+        categories.append(ToolCategory(name=name, **category))
+    return categories
+
+
+def get_all_tool_category_names() -> list[str]:
+    return [category.name for category in get_all_tool_categories()]
+
+
 def get_all_tool_paths() -> list[Path]:
     """
     Get all the paths to the tool configuration files.
@@ -121,3 +140,8 @@ def get_all_tool_names() -> list[str]:
 
 def get_all_tools() -> list[ToolConfig]:
     return [ToolConfig.from_tool_name(path) for path in get_all_tool_names()]
+
+
+def get_tool(name: str) -> ToolConfig:
+    return ToolConfig.from_tool_name(name)
+

agentstack/_tools/categories.json

@@ -0,0 +1,46 @@
+{
+    "browsing": {
+        "title": "Browsing",
+        "description": "Tools that are used to browse the web."
+    },
+    "code-execution": {
+        "title": "Code Execution",
+        "description": "Tools that are used to execute code."
+    },
+    "computer-control": {
+        "title": "Computer Control",
+        "description": "Tools that are used to control a computer."
+    },
+    "database": {
+        "title": "Database",
+        "description": "Tools that are used to interact with databases."
+    },
+    "finance": {
+        "title": "Finance",
+        "description": "Tools that are used to interact with financial services."
+    },
+    "image-analysis": {
+        "title": "Image Analysis",
+        "description": "Tools that are used to analyze images."
+    },
+    "network-protocols": {
+        "title": "Network Protocols",
+        "description": "Tools that are used to interact with network protocols."
+    },
+    "search": {
+        "title": "Search",
+        "description": "Tools that are used to search for information."
+    },
+    "storage": {
+        "title": "Storage",
+        "description": "Tools that are used to interact with storage."
+    },
+    "unified-apis": {
+        "title": "Unified APIs",
+        "description": "Tools that provide a unified API for interacting with multiple services."
+    },
+    "web-retrieval": {
+        "title": "Web Retrieval",
+        "description": "Tools that are used to retrieve information from the web."
+    }
+}

agentstack/_tools/payman/config.json

@@ -1,6 +1,6 @@
 {
   "name": "payman",
-  "category": "financial-infra",
+  "category": "finance",
   "tools": [
     "send_payment",
     "search_available_payees",

agentstack/_tools/stripe/config.json

@@ -1,7 +1,7 @@
 {
   "name": "stripe",
   "url": "https://github.com/stripe/agent-toolkit",
-  "category": "application-specific",
+  "category": "finance",
   "env": {
     "STRIPE_SECRET_KEY": null
   },

agentstack/_tools/weaviate/config.json

@@ -1,7 +1,7 @@
 {
   "name": "weaviate",
   "url": "https://github.com/weaviate/weaviate-python-client",
-  "category": "vector-store",
+  "category": "database",
   "env": {
     "WEAVIATE_URL": null,
     "WEAVIATE_API_KEY": null,

agentstack/cli/__init__.py

@@ -1,9 +1,7 @@
-from .cli import configure_default_model, welcome_message, get_validated_input, parse_insertion_point
+from .cli import LOGO, configure_default_model, welcome_message, get_validated_input, parse_insertion_point
 from .init import init_project
-from .wizard import run_wizard
 from .run import run_project
 from .tools import list_tools, add_tool, remove_tool
 from .tasks import add_task
 from .agents import add_agent
 from .templates import insert_template, export_template
-

agentstack/cli/cli.py

@@ -1,34 +1,31 @@
 from typing import Optional
 import os, sys
-from art import text2art
 import inquirer
 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 import providers
 from agentstack.generation import InsertionPoint
 
 
-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',
-]
+LOGO = """\
+    ___       ___       ___       ___       ___       ___       ___       ___       ___       ___   
+   /\  \     /\  \     /\  \     /\__\     /\  \     /\  \     /\  \     /\  \     /\  \     /\__\  
+  /::\  \   /::\  \   /::\  \   /:| _|_    \:\  \   /::\  \    \:\  \   /::\  \   /::\  \   /:/ _/_ 
+ /::\:\__\ /:/\:\__\ /::\:\__\ /::|/\__\   /::\__\ /\:\:\__\   /::\__\ /::\:\__\ /:/\:\__\ /::-"\__\\
+ \/\::/  / \:\:\/__/ \:\:\/  / \/|::/  /  /:/\/__/ \:\:\/__/  /:/\/__/ \/\::/  / \:\ \/__/ \;:;-",-"
+   /:/  /   \::/  /   \:\/  /    |:/  /   \/__/     \::/  /   \/__/      /:/  /   \:\__\    |:|  |  
+   \/__/     \/__/     \/__/     \/__/               \/__/               \/__/     \/__/     \|__|  
+"""
 
 
 def welcome_message():
-    title = text2art("AgentStack", font="smisome1")
     tagline = "The easiest way to build a robust agent application!"
     border = "-" * len(tagline)
 
     # Print the welcome message with ASCII art
-    log.info(title)
+    log.info(LOGO)
     log.info(border)
     log.info(tagline)
     log.info(border)
@@ -45,7 +42,7 @@ def configure_default_model():
     other_msg = "Other (enter a model name)"
     model = inquirer.list_input(
         message="Which model would you like to use?",
-        choices=PREFERRED_MODELS + [other_msg],
+        choices=providers.get_preferred_model_ids() + [other_msg],
     )
 
     if model == other_msg:  # If the user selects "Other", prompt for a model name

agentstack/cli/init.py

@@ -14,7 +14,6 @@
 from agentstack.templates import get_all_templates, TemplateConfig
 
 from agentstack.cli import welcome_message
-from agentstack.cli.wizard import run_wizard
 from agentstack.cli.templates import insert_template
 
 
@@ -67,7 +66,7 @@ def init_project(
     slug_name: Optional[str] = None,
     template: Optional[str] = None,
     framework: Optional[str] = None,
-    use_wizard: bool = False,
+    template_data: Optional[TemplateConfig] = None,
 ):
     """
     Initialize a new project in the current directory.
@@ -78,9 +77,6 @@ def init_project(
     - insert Tasks, Agents and Tools
     """
     # TODO prevent the user from passing the --path argument to init
-    if template and use_wizard:
-        raise Exception("Template and wizard flags cannot be used together")
-
     require_uv()
     welcome_message()
 
@@ -102,16 +98,14 @@ def init_project(
     if os.path.exists(conf.PATH):  # cookiecutter requires the directory to not exist
         raise Exception(f"Directory already exists: {conf.PATH}")
 
-    if use_wizard:
-        log.debug("Initializing new project with wizard.")
-        template_data = run_wizard(slug_name)
-    elif template:
+    if not template_data and template:
         log.debug(f"Initializing new project with template: {template}")
         template_data = TemplateConfig.from_user_input(template)
-    else:
+    elif not template_data:
         log.debug("Initializing new project with template selection.")
         template_data = select_template(slug_name, framework)
 
+    assert template_data  # appease type checker
     log.notify("🦾 Creating a new AgentStack project...")
     log.info(f"Using project directory: {conf.PATH.absolute()}")