Merge branch 'main' into bugfix-python-3-10

Python 3.10+ natively supports Self type annotation without explicit imports

.github/workflows/tests.yml

@@ -11,11 +11,10 @@ env:
 jobs:
   tests:
     runs-on: ubuntu-latest
+    timeout-minutes: 15
     strategy:
       matrix:
-        python-version: ["3.10", "3.11", "3.12"]
-      fail-fast: false # Allow all matrix jobs to finish even if one fails
-    timeout-minutes: 15
+        python-version: ['3.10', '3.11', '3.12']
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
@@ -25,7 +24,6 @@ jobs:
         with:
           enable-cache: true
 
-
       - name: Set up Python ${{ matrix.python-version }}
         run: uv python install ${{ matrix.python-version }}
 

pyproject.toml

@@ -38,7 +38,6 @@ dependencies = [
     "blinker>=1.9.0",
     "json5>=0.10.0",
 ]
-    "typing-extensions>=4.7.0", # For Self type backport
 
 [project.urls]
 Homepage = "https://crewai.com"

src/crewai/flow/flow.py

@@ -1043,6 +1043,7 @@ class Flow(Generic[T], metaclass=FlowMeta):
             import traceback
 
             traceback.print_exc()
+            raise
 
     def _log_flow_event(
         self, message: str, color: str = "yellow", level: str = "info"

src/crewai/memory/external/external_memory.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Dict, Optional, Self
+from typing import TYPE_CHECKING, Any, Dict, Optional
 
 from crewai.memory.external.external_memory_item import ExternalMemoryItem
 from crewai.memory.memory import Memory
@@ -52,7 +52,7 @@ class ExternalMemory(Memory):
     def reset(self) -> None:
         self.storage.reset()
 
-    def set_crew(self, crew: Any) -> Self:
+    def set_crew(self, crew: Any) -> "ExternalMemory":
         super().set_crew(crew)
 
         if not self.storage:

src/crewai/memory/memory.py

@@ -1,11 +1,5 @@
-import sys
 from typing import Any, Dict, List, Optional
 
-if sys.version_info >= (3, 11):
-    from typing import Self
-else:
-    from typing_extensions import Self
-
 from pydantic import BaseModel
 
 
@@ -44,6 +38,6 @@ class Memory(BaseModel):
             query=query, limit=limit, score_threshold=score_threshold
         )
 
-    def set_crew(self, crew: Any) -> Self:
+    def set_crew(self, crew: Any) -> "Memory":
         self.crew = crew
         return self

tests/test_compatibility.py

@@ -1,15 +0,0 @@
-import pytest
-
-
-def test_basic_import():
-    """
-    Tests that the crewai package can be imported without raising exceptions.
-    This helps catch basic installation and dependency issues, including import
-    errors related to Python version compatibility.
-    """
-    try:
-        import crewai
-    except ImportError as e:
-        pytest.fail(f"Failed to import crewai package: {e}")
-    except Exception as e:
-        pytest.fail(f"An unexpected error occurred during crewai import: {e}")