From 28e5191fa35cd688fa13c2e97a2a30fbb0b78029 Mon Sep 17 00:00:00 2001 From: nathanogaga118 <106948598+nathanogaga118@users.noreply.github.com> Date: Tue, 28 Jan 2025 14:41:20 -0800 Subject: [PATCH] Update agent_registry.py --- swarms/structs/agent_registry.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/swarms/structs/agent_registry.py b/swarms/structs/agent_registry.py index 09348622..3bdaa42f 100644 --- a/swarms/structs/agent_registry.py +++ b/swarms/structs/agent_registry.py @@ -97,11 +97,16 @@ class AgentRegistry: agent (Agent): The agent to add. Raises: - ValueError: If the agent_name already exists in the registry. + ValueError: If the agent_name already exists in the registry or is invalid. ValidationError: If the input data is invalid. """ name = agent.agent_name + # Input validation for agent_name + if not name or not isinstance(name, str): + logger.error("Invalid agent name provided.") + raise ValueError("Invalid agent name provided.") + self.agent_to_py_model(agent) with self.lock: @@ -242,7 +247,7 @@ class AgentRegistry: with self.lock: agents = list(self.agents.values()) logger.info("Returning all agents.") - return agents + returnagents except Exception as e: logger.error(f"Error: {e}") raise e