Fix #2500: Add Crewai alias for backward compatibility

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-31 12:41:06 +00:00
parent 356d4d9729
commit d1fd44f477
2 changed files with 20 additions and 0 deletions

View File

@@ -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

View File

@@ -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()