add back in manager delegation tests

tests/test_manager_llm_delegation.py

@@ -1,56 +1,56 @@
-# from unittest.mock import MagicMock
+from unittest.mock import MagicMock
 
-# import pytest
+import pytest
 
-# from crewai import Agent, Task
+from crewai import Agent, Task
-# from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool
+from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool
 
 
-# class InternalAgentTool(BaseAgentTool):
+class InternalAgentTool(BaseAgentTool):
-#     """Concrete implementation of BaseAgentTool for testing."""
+    """Concrete implementation of BaseAgentTool for testing."""
 
-#     def _run(self, *args, **kwargs):
+    def _run(self, *args, **kwargs):
-#         """Implement required _run method."""
+        """Implement required _run method."""
-#         return "Test response"
+        return "Test response"
 
 
-# @pytest.mark.parametrize(
+@pytest.mark.parametrize(
-#     "role_name,should_match",
+    "role_name,should_match",
-#     [
+    [
-#         ("Futel Official Infopoint", True),  # exact match
+        ("Futel Official Infopoint", True),  # exact match
-#         ('  "Futel Official Infopoint"  ', True),  # extra quotes and spaces
+        ('  "Futel Official Infopoint"  ', True),  # extra quotes and spaces
-#         ("Futel Official Infopoint\n", True),  # trailing newline
+        ("Futel Official Infopoint\n", True),  # trailing newline
-#         ('"Futel Official Infopoint"', True),  # embedded quotes
+        ('"Futel Official Infopoint"', True),  # embedded quotes
-#         (" FUTEL\nOFFICIAL   INFOPOINT ", True),  # multiple whitespace and newline
+        (" FUTEL\nOFFICIAL   INFOPOINT ", True),  # multiple whitespace and newline
-#         ("futel official infopoint", True),  # lowercase
+        ("futel official infopoint", True),  # lowercase
-#         ("FUTEL OFFICIAL INFOPOINT", True),  # uppercase
+        ("FUTEL OFFICIAL INFOPOINT", True),  # uppercase
-#         ("Non Existent Agent", False),  # non-existent agent
+        ("Non Existent Agent", False),  # non-existent agent
-#         (None, False),  # None agent name
+        (None, False),  # None agent name
-#     ],
+    ],
-# )
+)
-# def test_agent_tool_role_matching(role_name, should_match):
+def test_agent_tool_role_matching(role_name, should_match):
-#     """Test that agent tools can match roles regardless of case, whitespace, and special characters."""
+    """Test that agent tools can match roles regardless of case, whitespace, and special characters."""
-#     # Create test agent
+    # Create test agent
-#     test_agent = Agent(
+    test_agent = Agent(
-#         role="Futel Official Infopoint",
+        role="Futel Official Infopoint",
-#         goal="Answer questions about Futel",
+        goal="Answer questions about Futel",
-#         backstory="Futel Football Club info",
+        backstory="Futel Football Club info",
-#         allow_delegation=False,
+        allow_delegation=False,
-#     )
+    )
 
-#     # Create test agent tool
+    # Create test agent tool
-#     agent_tool = InternalAgentTool(
+    agent_tool = InternalAgentTool(
-#         name="test_tool", description="Test tool", agents=[test_agent]
+        name="test_tool", description="Test tool", agents=[test_agent]
-#     )
+    )
 
-#     # Test role matching
+    # Test role matching
-#     result = agent_tool._execute(agent_name=role_name, task="Test task", context=None)
+    result = agent_tool._execute(agent_name=role_name, task="Test task", context=None)
 
-#     if should_match:
+    if should_match:
-#         assert (
+        assert (
-#             "coworker mentioned not found" not in result.lower()
+            "coworker mentioned not found" not in result.lower()
-#         ), f"Should find agent with role name: {role_name}"
+        ), f"Should find agent with role name: {role_name}"
-#     else:
+    else:
-#         assert (
+        assert (
-#             "coworker mentioned not found" in result.lower()
+            "coworker mentioned not found" in result.lower()
-#         ), f"Should not find agent with role name: {role_name}"
+        ), f"Should not find agent with role name: {role_name}"