Merge 3c9391405b into a88127b820

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:

swarms/structs/majority_voting.py

@@ -45,6 +45,11 @@ def parse_code_completion(agent_response, question):
     Returns:
         tuple: A tuple containing the parsed Python code and a boolean indicating success.
     """
+    # Input validation
+    if not isinstance(agent_response, str) or not isinstance(question, str):
+        logger.error("Invalid inputs provided to parse_code_completion.")
+        return "", False
+
     python_code = extract_last_python_code_block(agent_response)
     if python_code is None:
         if agent_response.count("impl]") == 0:
@@ -181,6 +186,11 @@ class MajorityVoting:
             List[Any]: The majority vote.
 
         """
+        # Input validation
+        if not isinstance(task, str) or not task.strip():
+            logger.error("Invalid task provided to run.")
+            return []
+
         # Route to each agent
         with concurrent.futures.ThreadPoolExecutor() as executor:
             logger.info("Running agents concurrently")