From d200d00bb5092e8f534a6661b2344a75f0c58537 Mon Sep 17 00:00:00 2001 From: Lucas Gomide Date: Wed, 9 Apr 2025 10:40:12 -0300 Subject: [PATCH] refactor: remove explicit Self import from typing Python 3.10+ natively supports Self type annotation without explicit imports --- src/crewai/memory/external/external_memory.py | 4 ++-- src/crewai/memory/memory.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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