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):