Convert swarm run to be sequential.

Author: tcdent

agentstack/frameworks/openai_swarm.py

@@ -80,35 +80,6 @@ def {task.name}(self, messages: list[str] = []) -> Agent:
         existing_agent_methods = self.get_agent_methods()
         if not len(existing_agent_methods):
             return  # no agents to update
-        
-        # add a call to `self._handoff(task_name)` to the front of the update_method's
-        # `function` argument which is a list of functions
-        update_method = existing_agent_methods[-1]
-        try:
-            agent_instance = asttools.find_method_calls(update_method, 'Agent')[0]
-        except IndexError:
-            raise ValidationError(f"Agent method `{update_method.name}` does not instantiate `Agent` in {ENTRYPOINT}")
-
-        existing_agent_tools = asttools.find_kwarg_in_method_call(agent_instance, 'functions')
-        if not existing_agent_tools:
-            raise ValidationError(
-                f"`@agent` method `{update_method.name}` does not have a keyword argument `functions` in {ENTRYPOINT}"
-            )
-        
-        assert isinstance(existing_agent_tools.value, ast.List)
-        existing_elts = existing_agent_tools.value.elts
-        existing_elts.insert(0, ast.Call(
-            func=ast.Attribute(
-                value=ast.Name(id='self', ctx=ast.Load()),
-                attr='_handoff',
-                ctx=ast.Load(),
-            ),
-            args=[ast.Constant(value=task.name)],
-            keywords=[],
-        ))
-        new_node = ast.List(elts=existing_elts, ctx=ast.Load())
-        start, end = self.get_node_range(existing_agent_tools.value)
-        self.edit_node_range(start, end, new_node)
 
     def get_agent_methods(self) -> list[ast.FunctionDef]:
         """An `agent` method is a method decorated with `@agent`."""

agentstack/templates/openai_swarm/{{cookiecutter.project_metadata.project_slug}}/src/stack.py

@@ -4,34 +4,22 @@
 
 class {{ cookiecutter.project_metadata.project_name|replace('-', '')|replace('_', '')|capitalize }}Stack:
 
-    def _handoff(self, task_name: str):
-        """Return a task formatted as a tool for handing off to another agent."""
-        task = getattr(self, task_name)
-        def func(messages: list[str] = []) -> Agent:
-            return task(messages=messages)
-        func.__name__ = task_name
-        return func
-
-    def _get_first_task(self) -> Agent:
-        """Get the first task."""
-        task_name = agentstack.get_all_task_names()[0]
-        return getattr(self, task_name)()
-
     def run(self, inputs: list[str]):
         app = Swarm()
-        response = app.run(
-            agent=self._get_first_task(), 
-            messages=[], 
-            context_variables=inputs, 
-            debug=agentstack.conf.DEBUG, 
-        )
-
-        for message in response.messages:
-            if message.get('tool_calls'):
-                for tool_call in message['tool_calls']:
-                    agentstack.log.notify(f"Calling tool `{tool_call['function']['name']}`")
-                    agentstack.log.debug(tool_call['function']['arguments'])
-            elif message.get('role') != 'tool':
-                agentstack.log.notify(f"{message['role']}:")
+        history = []
+        for task in agentstack.get_all_tasks():
+            agentstack.log.notify(f"Running task `{task.name}`")
+            response = app.run(
+                agent=getattr(self, task.name)(), 
+                messages=history, 
+                context_variables=inputs, 
+                debug=agentstack.conf.DEBUG, 
+            )
+            for message in response.messages:
+                if message.get('role') not in ['user', 'assistant']:
+                    continue
+                if message.get('tool_calls'):
+                    continue
+                history.append(message)
                 agentstack.log.info(message['content'])
 

agentstack/templates/proj_templates/hello_alex.json

@@ -7,7 +7,7 @@
         "name": "alex",
         "role": "You are a friendly assistant.",
         "goal": "Help the user with any of their requests.",
-        "backstory": "After years travelling the world, you've decided to get back into tech, just in time for the AI boom!. You're working on AgentStack, the fastest way to get started with AI agents. You have a README file available at: ./README.md",
+        "backstory": "After years traveling the world, you've decided to get back into tech, just in time for the AI boom!. You're working on AgentStack, the fastest way to get started with AI agents. You have a README file available at: ./README.md",
         "model": "openai/gpt-4o"
     }],
     "tasks": [{

agentstack/templates/proj_templates/hola_alex.json

@@ -0,0 +1,37 @@
+{
+    "name": "hola_alex",
+    "description": "This is the start of your AgentStack project.",
+    "template_version": 4,
+    "framework": "openai_swarm",
+    "agents": [{
+        "name": "alex",
+        "role": "You are a friendly assistant.",
+        "goal": "Help the user with any of their requests.",
+        "backstory": "After years traveling the world, you've decided to get back into tech, just in time for the AI boom!. You're working on AgentStack, the fastest way to get started with AI agents. You have a README file available at: ./README.md",
+        "llm": "openai/gpt-4o"
+    }, {
+        "name": "translator",
+        "role": "You are a professional translator.",
+        "goal": "Translate content passed to you into the correct language.",
+        "backstory": "You have an ivy league education in linguistics.",
+        "llm": "openai/gpt-4o"
+    }],
+    "tasks": [{
+        "name": "hello_world",
+        "description": "As is tradition in software, let's start by saying 'Hello, World!'. Then, pick one or two tasks that they should try to do next with AgentStack.",
+        "expected_output": "The sentence Hello, World! followed by two things the user should try to customize their agent further.",
+        "agent": "alex"
+    }, {
+        "name": "translate_to_spanish",
+        "description": "Translate the content to Spanish.",
+        "expected_output": "The content translated to Spanish.",
+        "agent": "translator"
+    }],
+    "tools": [{
+        "name": "file_read", 
+        "agents": ["alex"]
+    }],
+    "method": "sequential",
+    "graph": [], 
+    "inputs": {}
+}