Deploy command

MANIFEST.in

@@ -1,4 +1,5 @@
 recursive-include agentstack/templates *
 recursive-include agentstack/frameworks/templates *
 recursive-include agentstack/_tools *
+recursive-include agentstack/serve/serve.py
 include agentstack.json .env .env.example

agentstack/cli/cli.py

@@ -1,12 +1,28 @@
 from typing import Optional
 import os, sys
+import importlib
+from typing import Optional
+import os
+import time
+from datetime import datetime
+
+import json
+import shutil
 from art import text2art
 import inquirer
 from agentstack import conf, log
+from agentstack.cli.agentstack_data import CookiecutterData, ProjectStructure, ProjectMetadata, FrameworkData
 from agentstack.conf import ConfigFile
+from agentstack.generation import InsertionPoint, ProjectFile
+from agentstack import frameworks
+from agentstack import inputs
+from agentstack.agents import get_all_agents
+from agentstack.tasks import get_all_tasks
+from agentstack.utils import get_package_path, open_json_file, term_color, is_snake_case, get_framework, \
+    validator_not_empty, verify_agentstack_project
+from agentstack.templates import TemplateConfig
 from agentstack.exceptions import ValidationError
 from agentstack.utils import validator_not_empty, is_snake_case
-from agentstack.generation import InsertionPoint
 
 
 PREFERRED_MODELS = [
@@ -81,6 +97,305 @@ def get_validated_input(
         return value
 
 
+def ask_agent_details():
+    agent = {}
+
+    agent['name'] = get_validated_input(
+        "What's the name of this agent? (snake_case)", min_length=3, snake_case=True
+    )
+
+    agent['role'] = get_validated_input("What role does this agent have?", min_length=3)
+
+    agent['goal'] = get_validated_input("What is the goal of the agent?", min_length=10)
+
+    agent['backstory'] = get_validated_input("Give your agent a backstory", min_length=10)
+
+    agent['model'] = inquirer.list_input(
+        message="What LLM should this agent use?", choices=PREFERRED_MODELS, default=PREFERRED_MODELS[0]
+    )
+
+    return agent
+
+
+def ask_task_details(agents: list[dict]) -> dict:
+    task = {}
+
+    task['name'] = get_validated_input(
+        "What's the name of this task? (snake_case)", min_length=3, snake_case=True
+    )
+
+    task['description'] = get_validated_input("Describe the task in more detail", min_length=10)
+
+    task['expected_output'] = get_validated_input(
+        "What do you expect the result to look like? (ex: A 5 bullet point summary of the email)",
+        min_length=10,
+    )
+
+    task['agent'] = inquirer.list_input(
+        message="Which agent should be assigned this task?",
+        choices=[a['name'] for a in agents],
+    )
+
+    return task
+
+
+def ask_design() -> dict:
+    use_wizard = inquirer.confirm(
+        message="Would you like to use the CLI wizard to set up agents and tasks?",
+    )
+
+    if not use_wizard:
+        return {'agents': [], 'tasks': []}
+
+    os.system("cls" if os.name == "nt" else "clear")
+
+    title = text2art("AgentWizard", font="shimrod")
+
+    print(title)
+
+    print("""
+🪄 welcome to the agent builder wizard!! 🪄
+
+First we need to create the agents that will work together to accomplish tasks:
+    """)
+    make_agent = True
+    agents = []
+    while make_agent:
+        print('---')
+        print(f"Agent #{len(agents)+1}")
+        agent = None
+        agent = ask_agent_details()
+        agents.append(agent)
+        make_agent = inquirer.confirm(message="Create another agent?")
+
+    print('')
+    for x in range(3):
+        time.sleep(0.3)
+        print('.')
+    print('Boom! We made some agents (ﾉ>ω<)ﾉ :｡･:*:･ﾟ’★,｡･:*:･ﾟ’☆')
+    time.sleep(0.5)
+    print('')
+    print('Now lets make some tasks for the agents to accomplish!')
+    print('')
+
+    make_task = True
+    tasks = []
+    while make_task:
+        print('---')
+        print(f"Task #{len(tasks) + 1}")
+        task = ask_task_details(agents)
+        tasks.append(task)
+        make_task = inquirer.confirm(message="Create another task?")
+
+    print('')
+    for x in range(3):
+        time.sleep(0.3)
+        print('.')
+    print('Let there be tasks (ノ ˘_˘)ノ　ζ|||ζ　ζ|||ζ　ζ|||ζ')
+
+    return {'tasks': tasks, 'agents': agents}
+
+
+def ask_tools() -> list:
+    use_tools = inquirer.confirm(
+        message="Do you want to add agent tools now? (you can do this later with `agentstack tools add <tool_name>`)",
+    )
+
+    if not use_tools:
+        return []
+
+    tools_to_add = []
+
+    adding_tools = True
+    script_dir = os.path.dirname(os.path.abspath(__file__))
+    tools_json_path = os.path.join(script_dir, '..', 'tools', 'tools.json')
+
+    # Load the JSON data
+    tools_data = open_json_file(tools_json_path)
+
+    while adding_tools:
+        tool_type = inquirer.list_input(
+            message="What category tool do you want to add?",
+            choices=list(tools_data.keys()) + ["~~ Stop adding tools ~~"],
+        )
+
+        tools_in_cat = [f"{t['name']} - {t['url']}" for t in tools_data[tool_type] if t not in tools_to_add]
+        tool_selection = inquirer.list_input(message="Select your tool", choices=tools_in_cat)
+
+        tools_to_add.append(tool_selection.split(' - ')[0])
+
+        log.info("Adding tools:")
+        for t in tools_to_add:
+            log.info(f'  - {t}')
+        log.info('')
+        adding_tools = inquirer.confirm("Add another tool?")
+
+    return tools_to_add
+
+
+def ask_project_details(slug_name: Optional[str] = None) -> dict:
+    name = inquirer.text(message="What's the name of your project (snake_case)", default=slug_name or '')
+
+    if not is_snake_case(name):
+        log.error("Project name must be snake case")
+        return ask_project_details(slug_name)
+
+    questions = inquirer.prompt(
+        [
+            inquirer.Text("version", message="What's the initial version", default="0.1.0"),
+            inquirer.Text("description", message="Enter a description for your project"),
+            inquirer.Text("author", message="Who's the author (your name)?"),
+        ]
+    )
+
+    questions['name'] = name
+
+    return questions
+
+
+def insert_template(
+    project_details: dict,
+    framework_name: str,
+    design: dict,
+    template_data: Optional[TemplateConfig] = None,
+):
+    framework = FrameworkData(
+        name=framework_name.lower(),
+    )
+    project_metadata = ProjectMetadata(
+        project_name=project_details["name"],
+        description=project_details["description"],
+        author_name=project_details["author"],
+        version="0.0.1",
+        license="MIT",
+        year=datetime.now().year,
+        template=template_data.name if template_data else 'none',
+        template_version=template_data.template_version if template_data else 0,
+    )
+
+    project_structure = ProjectStructure(
+        method=template_data.method if template_data else "sequential",
+        manager_agent=template_data.manager_agent if template_data else None,
+    )
+    project_structure.agents = design["agents"]
+    project_structure.tasks = design["tasks"]
+    project_structure.inputs = design["inputs"]
+
+    cookiecutter_data = CookiecutterData(
+        project_metadata=project_metadata,
+        structure=project_structure,
+        framework=framework_name.lower(),
+    )
+
+    template_path = get_package_path() / f'templates/{framework.name}'
+    with open(f"{template_path}/cookiecutter.json", "w") as json_file:
+        json.dump(cookiecutter_data.to_dict(), json_file)
+        # TODO this should not be written to the package directory
+
+    # copy .env.example to .env
+    shutil.copy(
+        f'{template_path}/{"{{cookiecutter.project_metadata.project_slug}}"}/.env.example',
+        f'{template_path}/{"{{cookiecutter.project_metadata.project_slug}}"}/.env',
+    )
+
+    cookiecutter(str(template_path), no_input=True, extra_context=None)
+
+    # TODO: inits a git repo in the directory the command was run in
+    # TODO: not where the project is generated. Fix this
+    # TODO: also check if git is installed or if there are any git repos above the current dir
+    try:
+        pass
+        # subprocess.check_output(["git", "init"])
+        # subprocess.check_output(["git", "add", "."])
+    except:
+        print("Failed to initialize git repository. Maybe you're already in one? Do this with: git init")
+
+
+def export_template(output_filename: str):
+    """
+    Export the current project as a template.
+    """
+    try:
+        metadata = ProjectFile()
+    except Exception as e:
+        raise Exception(f"Failed to load project metadata: {e}")
+
+    # Read all the agents from the project's agents.yaml file
+    agents: list[TemplateConfig.Agent] = []
+    for agent in get_all_agents():
+        agents.append(
+            TemplateConfig.Agent(
+                name=agent.name,
+                role=agent.role,
+                goal=agent.goal,
+                backstory=agent.backstory,
+                allow_delegation=False,  # TODO
+                model=agent.llm,  # TODO consistent naming (llm -> model)
+            )
+        )
+
+    # Read all the tasks from the project's tasks.yaml file
+    tasks: list[TemplateConfig.Task] = []
+    for task in get_all_tasks():
+        tasks.append(
+            TemplateConfig.Task(
+                name=task.name,
+                description=task.description,
+                expected_output=task.expected_output,
+                agent=task.agent,
+            )
+        )
+
+    # Export all of the configured tools from the project
+    tools_agents: dict[str, list[str]] = {}
+    for agent_name in frameworks.get_agent_names():
+        for tool_name in frameworks.get_agent_tool_names(agent_name):
+            if not tool_name:
+                continue
+            if tool_name not in tools_agents:
+                tools_agents[tool_name] = []
+            tools_agents[tool_name].append(agent_name)
+
+    tools: list[TemplateConfig.Tool] = []
+    for tool_name, agent_names in tools_agents.items():
+        tools.append(
+            TemplateConfig.Tool(
+                name=tool_name,
+                agents=agent_names,
+            )
+        )
+
+    template = TemplateConfig(
+        template_version=3,
+        name=metadata.project_name,
+        description=metadata.project_description,
+        framework=get_framework(),
+        method="sequential",  # TODO this needs to be stored in the project somewhere
+        manager_agent=None,  # TODO
+        agents=agents,
+        tasks=tasks,
+        tools=tools,
+        inputs=inputs.get_inputs(),
+    )
+
+    try:
+        template.write_to_file(conf.PATH / output_filename)
+        log.success(f"Template saved to: {conf.PATH / output_filename}")
+    except Exception as e:
+        raise Exception(f"Failed to write template to file: {e}")
+
+
+def serve_project():
+    verify_agentstack_project()
+
+    # TODO: only silence output conditionally - maybe a debug or verbose option
+    os.system("docker stop agentstack-local > /dev/null 2>&1")
+    os.system("docker rm agentstack-local > /dev/null 2>&1")
+    with importlib.resources.path('agentstack.serve', 'Dockerfile') as path:
+        os.system(f"docker build -t agent-service -f {path} . --progress=plain")
+    os.system("docker run --name agentstack-local -p 6969:6969 agent-service")
+
+
 def parse_insertion_point(position: Optional[str] = None) -> Optional[InsertionPoint]:
     """
     Parse an insertion point CLI argument into an InsertionPoint enum.

agentstack/cli/run.py

@@ -1,4 +1,4 @@
-from typing import Optional, List
+from typing import Optional, List, Dict
 import sys
 import asyncio
 import traceback
@@ -16,7 +16,7 @@
 MAIN_MODULE_NAME = "main"
 
 
-def _format_friendly_error_message(exception: Exception):
+def format_friendly_error_message(exception: Exception):
     """
     Projects will throw various errors, especially on first runs, so we catch
     them here and print a more helpful message.
@@ -133,4 +133,4 @@ def run_project(command: str = 'run', cli_args: Optional[List[str]] = None):
     except ImportError as e:
         raise ValidationError(f"Failed to import AgentStack project at: {conf.PATH.absolute()}\n{e}")
     except Exception as e:
-        raise Exception(_format_friendly_error_message(e))
+        raise Exception(format_friendly_error_message(e))