From d1fd44f477c44c5751c4864e5b71b6c73f2f9493 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Mon, 31 Mar 2025 12:41:06 +0000 Subject: [PATCH] Fix #2500: Add Crewai alias for backward compatibility Co-Authored-By: Joe Moura --- src/crewai/crew.py | 3 +++ tests/test_crewai_alias.py | 17 +++++++++++++++++ 2 files changed, 20 insertions(+) create mode 100644 tests/test_crewai_alias.py 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()