from swarms.agents.base import agent
from swarms.structs.nonlinear_worfklow import NonLinearWorkflow, Task

prompt = "develop a feedforward network in pytorch"
prompt2 = "Develop a self attention using pytorch"

task1 = Task("task1", prompt)
task2 = Task("task2", prompt2, parents=[task1])

#add tasks to workflow
workflow = NonLinearWorkflow(agent)

#add tasks to tree
workflow.add(task1)
workflow.add(task2)

#run
workflow.run()