andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 23:58:34 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix API Key Behavior and Entity Handling in Mem0 Integration (#1857)

Browse Source 
* docs: clarify how to specify org_id and project_id in Mem0 configuration

* Add org_id and project_id to mem0 config and fix mem0 entity '400 Bad Request'

* Remove ruff changes to docs

---------

Co-authored-by: Brandon Hancock (bhancock_ai) <109994880+bhancockio@users.noreply.github.com>
This commit is contained in:
Alessandro Romano
2025-01-07 18:46:10 +01:00
committed by GitHub
parent 0e94236735
commit 355bf3b48b
2 changed files with 30 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
docs/concepts/memory.mdx
View File

@@ -134,6 +134,23 @@ crew = Crew(
) )
``` ```
## Memory Configuration Options
If you want to access a specific organization and project, you can set the `org_id` and `project_id` parameters in the memory configuration.
```python Code
from crewai import Crew
crew = Crew(
agents=[...],
tasks=[...],
verbose=True,
memory=True,
memory_config={
"provider": "mem0",
"config": {"user_id": "john", "org_id": "my_org_id", "project_id": "my_project_id"},
},
)
```
## Additional Embedding Providers ## Additional Embedding Providers

18
src/crewai/memory/storage/mem0_storage.py
View File

@@ -27,10 +27,18 @@ class Mem0Storage(Storage):
raise ValueError("User ID is required for user memory type") raise ValueError("User ID is required for user memory type")
# API key in memory config overrides the environment variable # API key in memory config overrides the environment variable
mem0_api_key = self.memory_config.get("config", {}).get("api_key") or os.getenv( config = self.memory_config.get("config", {})
"MEM0_API_KEY" mem0_api_key = config.get("api_key") or os.getenv("MEM0_API_KEY")
) mem0_org_id = config.get("org_id")
self.memory = MemoryClient(api_key=mem0_api_key) mem0_project_id = config.get("project_id")
# Initialize MemoryClient with available parameters
if mem0_org_id and mem0_project_id:
self.memory = MemoryClient(
api_key=mem0_api_key, org_id=mem0_org_id, project_id=mem0_project_id
)
else:
self.memory = MemoryClient(api_key=mem0_api_key)
def _sanitize_role(self, role: str) -> str: def _sanitize_role(self, role: str) -> str:
""" """
@@ -57,7 +65,7 @@ class Mem0Storage(Storage):
metadata={"type": "long_term", **metadata}, metadata={"type": "long_term", **metadata},
) )
elif self.memory_type == "entities": elif self.memory_type == "entities":
entity_name = None entity_name = self._get_agent_name()
self.memory.add( self.memory.add(
value, user_id=entity_name, metadata={"type": "entity", **metadata} value, user_id=entity_name, metadata={"type": "entity", **metadata}
) )