Update ma_utils.py

5 days ago · 22afe4b24b
parent 4baf617310
commit 22afe4b24b
1 changed files with 4 additions and 2 deletions
--- a/swarms/structs/ma_utils.py
+++ b/swarms/structs/ma_utils.py
@ -131,9 +131,11 @@ def set_random_models_for_agents(
        return random.choice(model_names)

    if isinstance(agents, list):
-        for agent in agents:
+        return [
            setattr(agent, "model_name", random.choice(model_names))
-        return agents
+            or agent
+            for agent in agents
+        ]
    else:
        setattr(agents, "model_name", random.choice(model_names))
        return agents