fix: type check warning

crewAIInc · Sep 22, 2024 · de73905 · de73905
1 parent a11ffb4
commit de73905
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/crewai/memory/entity/entity_memory.py b/src/crewai/memory/entity/entity_memory.py
@@ -11,7 +11,7 @@ class EntityMemory(Memory):
     Inherits from the Memory class.
     """
 
-    def __init__(self, crew=None, embedder_config=None, storage: Storage=None):
+    def __init__(self, crew=None, embedder_config=None, storage: Storage | None=None):
         storage = (
             storage
             if storage

diff --git a/src/crewai/memory/long_term/long_term_memory.py b/src/crewai/memory/long_term/long_term_memory.py
@@ -15,7 +15,7 @@ class LongTermMemory(Memory):
     LongTermMemoryItem instances.
     """
 
-    def __init__(self, storage: Storage=None):
+    def __init__(self, storage: Storage: Storage | None=None):
         storage = storage if storage else LTMSQLiteStorage()
         super().__init__(storage)
 

diff --git a/src/crewai/memory/short_term/short_term_memory.py b/src/crewai/memory/short_term/short_term_memory.py
@@ -14,7 +14,7 @@ class ShortTermMemory(Memory):
     MemoryItem instances.
     """
 
-    def __init__(self, crew=None, embedder_config=None, storage: Storage=None):
+    def __init__(self, crew=None, embedder_config=None, storage: Storage | None=None):
         storage = (
             storage
             if storage