diff --git a/src/crewai/crew.py b/src/crewai/crew.py index 60f7c5677..f8abc817a 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -1384,3 +1384,6 @@ class Crew(BaseModel): memory_system.reset() except Exception as e: raise RuntimeError(f"Failed to reset {name} memory") from e + + +Crewai = Crew diff --git a/tests/test_crewai_alias.py b/tests/test_crewai_alias.py new file mode 100644 index 000000000..989d84502 --- /dev/null +++ b/tests/test_crewai_alias.py @@ -0,0 +1,17 @@ +import unittest + +class TestCrewaiAlias(unittest.TestCase): + """Test the Crewai alias for backward compatibility.""" + + def test_crewai_alias_import(self): + """Test that Crewai can be imported from crewai.crew.""" + try: + from crewai.crew import Crewai + from crewai.crew import Crew + + self.assertEqual(Crewai, Crew) + except ImportError: + self.fail("Failed to import Crewai from crewai.crew") + +if __name__ == "__main__": + unittest.main()