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

Fix import error handling for optional chromadb dependency

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-26 19:06:19 +00:00
parent 9f57e266b8
commit 6435a419d7
3 changed files with 10 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/storage/test_optional_chromadb.py
View File

@@ -12,7 +12,6 @@ class TestOptionalChromadb(unittest.TestCase):
with patch.dict(sys.modules, {"chromadb": None}):
with pytest.raises(ImportError) as excinfo:
from crewai.memory.storage.rag_storage import RAGStorage
storage = RAGStorage(type="test")
assert "ChromaDB is not installed" in str(excinfo.value)
@@ -21,6 +20,5 @@ class TestOptionalChromadb(unittest.TestCase):
with patch.dict(sys.modules, {"chromadb": None}):
with pytest.raises(ImportError) as excinfo:
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
storage = KnowledgeStorage()
assert "ChromaDB is not installed" in str(excinfo.value)