Merge branch 'main' into tool-suggestions

Author: tcdent

TODO.md

@@ -0,0 +1,35 @@
+## Wizard
+
+
+## LangGraph
+
+[ ] Bug: empty `bind_tools` list on a new Agent throws an error
+
+[ ] Reliably handle adding two Tasks and two Agent via CLI
+
+[ ] Create a new template demonstrating graph usage
+
+
+"Top of Funnel"
+
+[√] Hello World!
+
+[√] Catch common exception types from the user's first `agentstack run` and provide a helpful error message.
+    - In order to avoid importing all the sub-libraries, use string matching to determine what the exception is. 
+
+[√] Allow the user to select from a list of tools if a <tool_name> is not provided to `agenstack add tools`
+
+[ ] Handle all setup tasks that `agentstack init` needs. 
+    - Wrap the output in a `curses` window to show progress.
+    - Prevent the user from having to scroll back up to find the help text from the `init` command.
+    - Use `uv` for all package management. 
+
+[√] Modify agentops init to indicate which tools are being used.
+    - Possibly with an `agentstack.get_tags()` method that automatically sets relevant tags.
+
+[√] If there is only one agent in the project, treat it as a default. 
+    - New tasks should be assigned to it. 
+
+[√] Wrap tools in the framework's protocol before including them in the project.
+    - Access tools with `agentstack.tools['tool_name']` which dynamically generates the wrapped tool. 
+    - This will be a breaking change with the 0.3 release.

agentstack/cli/init.py

@@ -124,6 +124,10 @@ 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:
         raise Exception(f"Framework '{framework}' is not supported.")
     log.info(f"Using framework: {framework}")

agentstack/frameworks/__init__.py

@@ -27,6 +27,9 @@
     OPENAI_SWARM,
     LLAMAINDEX,
 ]
+ALIASED_FRAMEWORKS = {
+    'crew': CREWAI,
+}
 DEFAULT_FRAMEWORK = CREWAI
 
 

tests/test_cli_init.py

@@ -4,10 +4,14 @@
 from pathlib import Path
 import shutil
 from cli_test_utils import run_cli
+from agentstack import conf
+from agentstack import frameworks
+from agentstack.cli import init_project
 from agentstack.templates import get_all_templates
 
 BASE_PATH = Path(__file__).parent
 
+
 class CLIInitTest(unittest.TestCase):
     def setUp(self):
         self.framework = os.getenv('TEST_FRAMEWORK')
@@ -21,9 +25,20 @@ def setUp(self):
     def tearDown(self):
         shutil.rmtree(self.project_dir, ignore_errors=True)
 
-    @parameterized.expand([(template.name, ) for template in get_all_templates()])
+    @parameterized.expand([(template.name,) for template in get_all_templates()])
     def test_init_command(self, template_name: str):
         """Test the 'init' command to create a project directory."""
         result = run_cli('init', 'test_project', '--template', template_name)
         self.assertEqual(result.returncode, 0)
         self.assertTrue((self.project_dir / 'test_project').exists())
+
+    @parameterized.expand([(k, v) for k, v in frameworks.ALIASED_FRAMEWORKS.items()])
+    def test_init_command_aliased_framework_empty_project(self, alias: str, framework: str):
+        """Test the 'init' command with an aliased framework."""
+        if framework != self.framework:
+            self.skipTest(f"{alias} is not related to this framework")
+
+        init_project(slug_name='test_project', template='empty', framework=alias)
+        conf.set_path(self.project_dir / 'test_project')
+        config = conf.ConfigFile()
+        assert config.framework == framework