crewAI/tests/test_crewbase_agent_method.py

import unittest
from crewai import Agent, Task


class TestTaskInitFix(unittest.TestCase):
    """Test the fix for issue #2219 where agent methods are not handled correctly in tasks."""

    def test_task_init_handles_callable_agent(self):
        """Test that the Task.__init__ method correctly handles callable agents."""
        
        # Create an agent instance
        agent_instance = Agent(
            role="Test Agent",
            goal="Test Goal",
            backstory="Test Backstory"
        )
        
        # Create a callable that returns the agent instance
        def callable_agent():
            return agent_instance
        
        # Create a task with the callable agent
        task = Task(
            description="Test Task",
            expected_output="Test Output",
            agent=callable_agent
        )
        
        # Verify that the agent in the task is an instance, not a callable
        self.assertIsInstance(task.agent, Agent)
        self.assertEqual(task.agent.role, "Test Agent")
        self.assertIs(task.agent, agent_instance)