diff --git a/src/crewai/memory/external/external_memory.py b/src/crewai/memory/external/external_memory.py index 4ecf3d065..be35f513b 100644 --- a/src/crewai/memory/external/external_memory.py +++ b/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: diff --git a/src/crewai/memory/memory.py b/src/crewai/memory/memory.py index ba8c10a29..20538a186 100644 --- a/src/crewai/memory/memory.py +++ b/src/crewai/memory/memory.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, List, Optional, Self +from typing import Any, Dict, List, Optional from pydantic import BaseModel @@ -38,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