From ef0b8e69136c0bd514f8ef3db4f7d1c5743b6f7e Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Thu, 20 Mar 2025 13:29:13 +0000
Subject: [PATCH] Fix security issue: Use environment variables for API keys in
 examples

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 src/crewai/memory/storage/mem0_storage.py | 4 ++--
 src/crewai/memory/user/user_memory.py     | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/crewai/memory/storage/mem0_storage.py b/src/crewai/memory/storage/mem0_storage.py
index 4a65e12ed..ab7eeaf03 100644
--- a/src/crewai/memory/storage/mem0_storage.py
+++ b/src/crewai/memory/storage/mem0_storage.py
@@ -18,8 +18,8 @@ class Mem0Storage(Storage):
         memory_config={
             "provider": "mem0",
             "config": {
-                "user_id": "test-user",
-                "api_key": "mem0-api-key",
+                "user_id": "your-user-id",
+                "api_key": os.getenv("MEM0_API_KEY"),  # Use environment variable
                 "vector_store": {
                     "provider": "redis",
                     "config": {
diff --git a/src/crewai/memory/user/user_memory.py b/src/crewai/memory/user/user_memory.py
index 051199ad3..070d915eb 100644
--- a/src/crewai/memory/user/user_memory.py
+++ b/src/crewai/memory/user/user_memory.py
@@ -1,3 +1,4 @@
+import os
 from typing import Any, Dict, Optional
 
 from crewai.memory.memory import Memory
@@ -18,8 +19,8 @@ class UserMemory(Memory):
         memory_config={
             "provider": "mem0",
             "config": {
-                "user_id": "test-user",
-                "api_key": "mem0-api-key",
+                "user_id": "your-user-id",
+                "api_key": os.getenv("MEM0_API_KEY"),  # Use environment variable
                 "vector_store": {
                     "provider": "redis",
                     "config": {