andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Update external memory tests to handle optional ChromaDB dependency

Browse Source 
Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-30 09:38:50 +00:00
parent ea5f6b592c
commit 5566f4716b
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/memory/external_memory_test.py
View File

@@ -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( def test_crew_external_memory_save_with_memory_flag(
mem_method, crew_with_external_memory 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( with patch(
f"crewai.memory.external.external_memory.ExternalMemory.{mem_method}" f"crewai.memory.external.external_memory.ExternalMemory.{mem_method}"
) as mock_method: ) as mock_method: