FURTHER edits + versatility

pull/1080/head
Aksh Parekh 4 weeks ago
parent bb7fd5b79f
commit abeaf2fc69

@ -387,7 +387,7 @@ class AgentRearrange:
for agent_name in agent_names: for agent_name in agent_names:
agent = self.agents[agent_name] agent = self.agents[agent_name]
# Set agent.streaming_on if no streaming_callback # Set agent.streaming_on if no streaming_callback
if self.streaming_callback is None: if self.streaming_callback is not None:
agent.streaming_on = True agent.streaming_on = True
result = agent.run( result = agent.run(
task=self.conversation.get_str(), task=self.conversation.get_str(),
@ -437,7 +437,7 @@ class AgentRearrange:
) )
# Set agent.streaming_on if no streaming_callback # Set agent.streaming_on if no streaming_callback
if self.streaming_callback is None: if self.streaming_callback is not None:
agent.streaming_on = True agent.streaming_on = True
current_task = agent.run( current_task = agent.run(
task=self.conversation.get_str(), task=self.conversation.get_str(),

@ -95,6 +95,7 @@ class SequentialWorkflow:
max_loops=self.max_loops, max_loops=self.max_loops,
output_type=self.output_type, output_type=self.output_type,
team_awareness=self.team_awareness, team_awareness=self.team_awareness,
streaming_callback=self.streaming_callback,
*args, *args,
**kwargs, **kwargs,
) )

Loading…
Cancel
Save