fix sequential workflow outputs

pull/788/head
Kye Gomez 2 months ago
parent fe8386780f
commit 07aee705bc

@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
[tool.poetry] [tool.poetry]
name = "swarms" name = "swarms"
version = "7.4.5" version = "7.4.6"
description = "Swarms - TGSC" description = "Swarms - TGSC"
license = "MIT" license = "MIT"
authors = ["Kye Gomez <kye@apac.ai>"] authors = ["Kye Gomez <kye@apac.ai>"]

@ -22,6 +22,10 @@ agents = [
), ),
] ]
router = SwarmRouter(agents=agents, swarm_type="SequentialWorkflow") router = SwarmRouter(agents=agents, swarm_type="SequentialWorkflow", output_type="dict", return_entire_history=False)
print(router.run("How are you doing?")) output = router.run("How are you doing?")
print(output)
print(type(output))

@ -139,12 +139,6 @@ class SequentialWorkflow:
result = self.agent_rearrange.run( result = self.agent_rearrange.run(
task=task, task=task,
img=img, img=img,
device=device,
all_cores=all_cores,
device_id=device_id,
all_gpus=all_gpus,
no_use_clusterops=no_use_clusterops,
output_type=self.output_type,
*args, *args,
**kwargs, **kwargs,
) )
@ -156,7 +150,8 @@ class SequentialWorkflow:
result = self.agent_rearrange.conversation.return_messages_as_list() result = self.agent_rearrange.conversation.return_messages_as_list()
elif self.output_type == "str" or self.return_json: elif self.output_type == "str" or self.return_json:
result = self.agent_rearrange.conversation.get_str() result = self.agent_rearrange.conversation.get_str()
else:
return result return result
except Exception as e: except Exception as e:
logger.error( logger.error(

Loading…
Cancel
Save