mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
Improve handling of optional configurations in memory and storage
- Initialize contextual_memory in src/crewai/agent.py and src/crewai/crew.py - Make UserMemory optional and add checks in src/crewai/memory/contextual/contextual_memory.py - Add crew checks in src/crewai/memory/entity/entity_memory.py and src/crewai/memory/short_term/short_term_memory.py - Allow optional storage_path in src/crewai/memory/storage/base_rag_storage.py - Update storage classes to accept optional db_path in: src/crewai/memory/storage/kickoff_task_outputs_storage.py, src/crewai/memory/storage/ltm_sqlite_storage.py, and src/crewai/memory/storage/mem0_storage.py - Modify src/crewai/memory/storage/rag_storage.py to use storage_path - Enhance src/crewai/utilities/embedding_configurator.py to handle missing providers
This commit is contained in:
@@ -19,7 +19,7 @@ class Mem0Storage(Storage):
|
||||
|
||||
self.memory_type = type
|
||||
self.crew = crew
|
||||
self.memory_config = crew.memory_config
|
||||
self.memory_config = crew.memory_config if crew else None
|
||||
|
||||
# User ID is required for user memory type "user" since it's used as a unique identifier for the user.
|
||||
user_id = self._get_user_id()
|
||||
@@ -27,9 +27,10 @@ class Mem0Storage(Storage):
|
||||
raise ValueError("User ID is required for user memory type")
|
||||
|
||||
# API key in memory config overrides the environment variable
|
||||
mem0_api_key = self.memory_config.get("config", {}).get("api_key") or os.getenv(
|
||||
"MEM0_API_KEY"
|
||||
)
|
||||
if self.memory_config and self.memory_config.get("config"):
|
||||
mem0_api_key = self.memory_config.get("config").get("api_key")
|
||||
else:
|
||||
mem0_api_key = os.getenv("MEM0_API_KEY")
|
||||
self.memory = MemoryClient(api_key=mem0_api_key)
|
||||
|
||||
def _sanitize_role(self, role: str) -> str:
|
||||
|
||||
Reference in New Issue
Block a user