From 541fa13df7552c9831858058332f5d939bb38e3e Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 4 Mar 2025 19:53:35 +0000
Subject: [PATCH] Fix remaining type checking issues

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 src/crewai/memory/entity/entity_memory.py       | 4 ++--
 src/crewai/memory/long_term/long_term_memory.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/crewai/memory/entity/entity_memory.py b/src/crewai/memory/entity/entity_memory.py
index bba6a6774..1eecc237a 100644
--- a/src/crewai/memory/entity/entity_memory.py
+++ b/src/crewai/memory/entity/entity_memory.py
@@ -65,8 +65,8 @@ class EntityMemory(Memory):
     def save(
         self,
         value: Any,
-        metadata: Dict[str, Any] = None,
-        agent: str = None,
+        metadata: Optional[Dict[str, Any]] = None,
+        agent: Optional[str] = None,
     ) -> None:
         """Saves an entity item or value into the storage."""
         if isinstance(value, EntityMemoryItem):
diff --git a/src/crewai/memory/long_term/long_term_memory.py b/src/crewai/memory/long_term/long_term_memory.py
index 3a68b83b7..d5fb274b8 100644
--- a/src/crewai/memory/long_term/long_term_memory.py
+++ b/src/crewai/memory/long_term/long_term_memory.py
@@ -54,8 +54,8 @@ class LongTermMemory(Memory):
     def save(
         self,
         value: Any,
-        metadata: Dict[str, Any] = None,
-        agent: str = None,
+        metadata: Optional[Dict[str, Any]] = None,
+        agent: Optional[str] = None,
     ) -> None:
         """Saves a value into the memory."""
         if isinstance(value, LongTermMemoryItem):