From 13115c68bd2df231f5140493d776590a574a7c95 Mon Sep 17 00:00:00 2001 From: lucasgomide Date: Mon, 31 Mar 2025 11:44:45 -0300 Subject: [PATCH] fix: surfacing properly supported types by Mem0Storage --- src/crewai/memory/storage/mem0_storage.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/crewai/memory/storage/mem0_storage.py b/src/crewai/memory/storage/mem0_storage.py index 6c9ffc682..782b17152 100644 --- a/src/crewai/memory/storage/mem0_storage.py +++ b/src/crewai/memory/storage/mem0_storage.py @@ -13,9 +13,12 @@ class Mem0Storage(Storage): def __init__(self, type, crew=None): super().__init__() - - if type not in ["user", "short_term", "long_term", "entities"]: - raise ValueError("Invalid type for Mem0Storage. Must be 'user' or 'agent'.") + supported_types = ["user", "short_term", "long_term", "entities"] + if type not in supported_types: + raise ValueError( + "Invalid type for Mem0Storage. Must be one of: " + + ", ".join(supported_types) + ) self.memory_type = type self.crew = crew