From c7c1ea16f5bb51f0318b7ddbf631ec24b2e783af Mon Sep 17 00:00:00 2001 From: Kye Gomez Date: Fri, 4 Jul 2025 01:38:00 -0700 Subject: [PATCH] removed is_last parameter which messed up sequential workflow --- sequential_workflow_example.py | 7 +++++-- swarms/structs/agent.py | 4 ++++ swarms/structs/rearrange.py | 3 --- 3 files changed, 9 insertions(+), 5 deletions(-) diff --git a/sequential_workflow_example.py b/sequential_workflow_example.py index a2242602..bb96f1c4 100644 --- a/sequential_workflow_example.py +++ b/sequential_workflow_example.py @@ -1,7 +1,7 @@ from swarms import Agent, SequentialWorkflow -import litellm +# import litellm -litellm._turn_on_debug() +# litellm._turn_on_debug() # Initialize market research agent @@ -16,6 +16,7 @@ market_researcher = Agent( model_name="claude-3-sonnet-20240229", max_loops=1, temperature=0.7, + streaming_on=True, ) # Initialize financial analyst agent @@ -29,6 +30,7 @@ financial_analyst = Agent( 5. Recommending financial strategies""", model_name="claude-3-sonnet-20240229", max_loops=1, + streaming_on=True, temperature=0.7, ) @@ -44,6 +46,7 @@ technical_analyst = Agent( model_name="claude-3-sonnet-20240229", max_loops=1, temperature=0.7, + streaming_on=True, ) # Create list of agents diff --git a/swarms/structs/agent.py b/swarms/structs/agent.py index b4366258..f275f095 100644 --- a/swarms/structs/agent.py +++ b/swarms/structs/agent.py @@ -2528,6 +2528,10 @@ class Agent: ValueError: If task is empty. """ + # Filter out is_last from kwargs if present + if 'is_last' in kwargs: + del kwargs['is_last'] + try: # Set streaming parameter in LLM if streaming is enabled if self.streaming_on and hasattr(self.llm, "stream"): diff --git a/swarms/structs/rearrange.py b/swarms/structs/rearrange.py index c0e75c7d..dd9a65c7 100644 --- a/swarms/structs/rearrange.py +++ b/swarms/structs/rearrange.py @@ -281,7 +281,6 @@ class AgentRearrange(BaseSwarm): ) for task_idx, task in enumerate(tasks): - is_last = task == tasks[-1] agent_names = [ name.strip() for name in task.split(",") ] @@ -297,7 +296,6 @@ class AgentRearrange(BaseSwarm): result = agent.run( task=self.conversation.get_str(), img=img, - is_last=is_last, *args, **kwargs, ) @@ -326,7 +324,6 @@ class AgentRearrange(BaseSwarm): current_task = agent.run( task=self.conversation.get_str(), img=img, - is_last=is_last, *args, **kwargs, )