Preparing new version (#1845)

* Preparing new version

tests/agents/agent_builder/base_agent_test.py

@@ -7,7 +7,7 @@ from crewai.agents.agent_builder.base_agent import BaseAgent
 from crewai.tools.base_tool import BaseTool
 
 
-class TestAgent(BaseAgent):
+class MockAgent(BaseAgent):
     def execute_task(
         self,
         task: Any,
@@ -29,7 +29,7 @@ class TestAgent(BaseAgent):
 
 
 def test_key():
-    agent = TestAgent(
+    agent = MockAgent(
         role="test role",
         goal="test goal",
         backstory="test backstory",

tests/cli/deploy/test_deploy_main.py

@@ -177,12 +177,12 @@ class TestDeployCommand(unittest.TestCase):
     def test_get_crew_status(self):
         mock_response = MagicMock()
         mock_response.status_code = 200
-        mock_response.json.return_value = {"name": "TestCrew", "status": "active"}
+        mock_response.json.return_value = {"name": "InternalCrew", "status": "active"}
         self.mock_client.crew_status_by_name.return_value = mock_response
 
         with patch("sys.stdout", new=StringIO()) as fake_out:
             self.deploy_command.get_crew_status()
-            self.assertIn("TestCrew", fake_out.getvalue())
+            self.assertIn("InternalCrew", fake_out.getvalue())
             self.assertIn("active", fake_out.getvalue())
 
     def test_get_crew_logs(self):

tests/project_test.py

@@ -27,7 +27,7 @@ class SimpleCrew:
 
 
 @CrewBase
-class TestCrew:
+class InternalCrew:
     agents_config = "config/agents.yaml"
     tasks_config = "config/tasks.yaml"
 
@@ -84,7 +84,7 @@ def test_task_memoization():
 
 
 def test_crew_memoization():
-    crew = TestCrew()
+    crew = InternalCrew()
     first_call_result = crew.crew()
     second_call_result = crew.crew()
 
@@ -107,7 +107,7 @@ def test_task_name():
 
 @pytest.mark.vcr(filter_headers=["authorization"])
 def test_before_kickoff_modification():
-    crew = TestCrew()
+    crew = InternalCrew()
     inputs = {"topic": "LLMs"}
     result = crew.crew().kickoff(inputs=inputs)
     assert "bicycles" in result.raw, "Before kickoff function did not modify inputs"
@@ -115,7 +115,7 @@ def test_before_kickoff_modification():
 
 @pytest.mark.vcr(filter_headers=["authorization"])
 def test_after_kickoff_modification():
-    crew = TestCrew()
+    crew = InternalCrew()
     # Assuming the crew execution returns a dict
     result = crew.crew().kickoff({"topic": "LLMs"})
 
@@ -126,7 +126,7 @@ def test_after_kickoff_modification():
 
 @pytest.mark.vcr(filter_headers=["authorization"])
 def test_before_kickoff_with_none_input():
-    crew = TestCrew()
+    crew = InternalCrew()
     crew.crew().kickoff(None)
     # Test should pass without raising exceptions
 

tests/test_manager_llm_delegation.py

@@ -6,7 +6,7 @@ from crewai import Agent, Task
 from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool
 
 
-class TestAgentTool(BaseAgentTool):
+class InternalAgentTool(BaseAgentTool):
     """Concrete implementation of BaseAgentTool for testing."""
 
     def _run(self, *args, **kwargs):
@@ -39,7 +39,7 @@ def test_agent_tool_role_matching(role_name, should_match):
     )
 
     # Create test agent tool
-    agent_tool = TestAgentTool(
+    agent_tool = InternalAgentTool(
         name="test_tool", description="Test tool", agents=[test_agent]
     )
 

tests/tools/test_base_tool.py

@@ -15,7 +15,7 @@ def test_creating_a_tool_using_annotation():
         my_tool.description
         == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, your agent will need this information to use it."
     )
-    assert my_tool.args_schema.schema()["properties"] == {
+    assert my_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (
@@ -29,7 +29,7 @@ def test_creating_a_tool_using_annotation():
         converted_tool.description
         == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, your agent will need this information to use it."
     )
-    assert converted_tool.args_schema.schema()["properties"] == {
+    assert converted_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (
@@ -54,7 +54,7 @@ def test_creating_a_tool_using_baseclass():
         my_tool.description
         == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, your agent will need this information to use it."
     )
-    assert my_tool.args_schema.schema()["properties"] == {
+    assert my_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert my_tool.run("What is the meaning of life?") == "What is the meaning of life?"
@@ -66,7 +66,7 @@ def test_creating_a_tool_using_baseclass():
         converted_tool.description
         == "Tool Name: Name of my tool\nTool Arguments: {'question': {'description': None, 'type': 'str'}}\nTool Description: Clear description for what this tool is useful for, your agent will need this information to use it."
     )
-    assert converted_tool.args_schema.schema()["properties"] == {
+    assert converted_tool.args_schema.model_json_schema()["properties"] == {
         "question": {"title": "Question", "type": "string"}
     }
     assert (

tests/tools/test_structured_tool.py

@@ -25,7 +25,7 @@ def schema_class():
     return TestSchema
 
 
-class TestCrewStructuredTool:
+class InternalCrewStructuredTool:
     def test_initialization(self, basic_function, schema_class):
         """Test basic initialization of CrewStructuredTool"""
         tool = CrewStructuredTool(

tests/utilities/evaluators/test_crew_evaluator_handler.py

@@ -12,7 +12,7 @@ from crewai.utilities.evaluators.crew_evaluator_handler import (
 )
 
 
-class TestCrewEvaluator:
+class InternalCrewEvaluator:
     @pytest.fixture
     def crew_planner(self):
         agent = Agent(role="Agent 1", goal="Goal 1", backstory="Backstory 1")

tests/utilities/test_planning_handler.py

@@ -16,7 +16,7 @@ from crewai.utilities.planning_handler import (
 )
 
 
-class TestCrewPlanner:
+class InternalCrewPlanner:
     @pytest.fixture
     def crew_planner(self):
         tasks = [
@@ -115,13 +115,13 @@ class TestCrewPlanner:
             def __init__(self, name: str, description: str):
                 tool_data = {"name": name, "description": description}
                 super().__init__(**tool_data)
-            
+
             def __str__(self):
                 return self.name
-            
+
             def __repr__(self):
                 return self.name
-            
+
             def to_structured_tool(self):
                 return self
 
@@ -149,11 +149,11 @@ class TestCrewPlanner:
                 ]
             )
         )
-        
+
         # Create planner with the new task
         planner = CrewPlanner([task], None)
         tasks_summary = planner._create_tasks_summary()
-        
+
         # Verify task summary content
         assert isinstance(tasks_summary, str)
         assert task.description in tasks_summary

tests/utilities/test_training_handler.py

@@ -4,7 +4,7 @@ import unittest
 from crewai.utilities.training_handler import CrewTrainingHandler
 
 
-class TestCrewTrainingHandler(unittest.TestCase):
+class InternalCrewTrainingHandler(unittest.TestCase):
     def setUp(self):
         self.handler = CrewTrainingHandler("trained_data.pkl")