Fix: Format imports to pass lint checks

Co-Authored-By: Joe Moura <joao@crewai.com>

pyproject.toml

@@ -13,6 +13,7 @@ dependencies = [
     "openai>=1.13.3",
     "litellm==1.60.2",
     "instructor>=1.3.3",
+    "typing_extensions>=4.0.0",
     # Text Processing
     "pdfplumber>=0.11.4",
     "regex>=2024.9.11",

src/crewai/memory/external/external_memory.py

@@ -1,9 +1,16 @@
-from typing import TYPE_CHECKING, Any, Dict, Optional, Self
+import sys
 
 from crewai.memory.external.external_memory_item import ExternalMemoryItem
 from crewai.memory.memory import Memory
 from crewai.memory.storage.interface import Storage
 
+if sys.version_info < (3, 11):
+    from typing import TYPE_CHECKING, Any, Dict, Optional
+
+    from typing_extensions import Self
+else:
+    from typing import TYPE_CHECKING, Any, Dict, Optional, Self
+
 if TYPE_CHECKING:
     from crewai.memory.storage.mem0_storage import Mem0Storage
 

src/crewai/memory/memory.py

@@ -1,7 +1,14 @@
-from typing import Any, Dict, List, Optional, Self
+import sys
 
 from pydantic import BaseModel
 
+if sys.version_info < (3, 11):
+    from typing import Any, Dict, List, Optional
+
+    from typing_extensions import Self
+else:
+    from typing import Any, Dict, List, Optional, Self
+
 
 class Memory(BaseModel):
     """

tests/imports_test.py

@@ -13,3 +13,14 @@ def test_crew_output_import():
     from crewai import CrewOutput
     
     assert CrewOutput is not None
+
+
+def test_memory_imports():
+    """Test that memory imports work correctly across Python versions."""
+    import importlib
+    importlib.import_module("crewai.memory.memory")
+    importlib.import_module("crewai.memory.external.external_memory")
+    
+    from crewai.memory.memory import Memory
+    
+    assert hasattr(Memory, "set_crew")