From 5566f4716b94acecb6fede15bc534bcc8e1ef293 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Fri, 30 May 2025 09:38:50 +0000
Subject: [PATCH] Update external memory tests to handle optional ChromaDB
 dependency
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-Authored-By: João <joao@crewai.com>
---
 tests/memory/external_memory_test.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/memory/external_memory_test.py b/tests/memory/external_memory_test.py
index eae69a5af..cf973b1a5 100644
--- a/tests/memory/external_memory_test.py
+++ b/tests/memory/external_memory_test.py
@@ -169,6 +169,15 @@ def test_crew_external_memory_reset(mem_type, crew_with_external_memory):
 def test_crew_external_memory_save_with_memory_flag(
     mem_method, crew_with_external_memory
 ):
+    try:
+        import chromadb
+        HAS_CHROMADB = True
+    except ImportError:
+        HAS_CHROMADB = False
+        
+    if not HAS_CHROMADB:
+        pytest.skip("ChromaDB is required for this test")
+        
     with patch(
         f"crewai.memory.external.external_memory.ExternalMemory.{mem_method}"
     ) as mock_method: