You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
swarms/examples/models/reasoning_duo_batched.py

19 lines
605 B

from swarms.agents.reasoning_duo import ReasoningDuo
if __name__ == "__main__":
# Initialize the ReasoningDuo with two lightweight models
duo = ReasoningDuo(
model_names=["gpt-4o-mini", "gpt-4o-mini"],
# max_loops=1, # Remove this line
)
# Batched tasks to process
tasks = [
"Summarize the benefits of solar energy.",
"List three uses of robotics in healthcare.",
]
# Run the batch once and print each result
results = duo.batched_run(tasks)
for task, output in zip(tasks, results):
print(f"Task: {task}\nResult: {output}\n")