diff --git a/tests/crew_test.py b/tests/crew_test.py index 62b934883..f74d9f61d 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -2384,6 +2384,16 @@ def test_multiple_conditional_tasks(researcher, writer): @pytest.mark.vcr(filter_headers=["authorization"]) def test_using_contextual_memory(): from unittest.mock import patch + + # Check if ChromaDB is available + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") math_researcher = Agent( role="Researcher", @@ -2412,6 +2422,16 @@ def test_using_contextual_memory(): @pytest.mark.vcr(filter_headers=["authorization"]) def test_using_contextual_memory_with_long_term_memory(): from unittest.mock import patch + + # Check if ChromaDB is available + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") math_researcher = Agent( role="Researcher", @@ -2441,6 +2461,16 @@ def test_using_contextual_memory_with_long_term_memory(): @pytest.mark.vcr(filter_headers=["authorization"]) def test_warning_long_term_memory_without_entity_memory(): from unittest.mock import patch + + # Check if ChromaDB is available + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") math_researcher = Agent( role="Researcher", @@ -2478,6 +2508,16 @@ def test_warning_long_term_memory_without_entity_memory(): @pytest.mark.vcr(filter_headers=["authorization"]) def test_long_term_memory_with_memory_flag(): from unittest.mock import patch + + # Check if ChromaDB is available + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") math_researcher = Agent( role="Researcher", @@ -2513,6 +2553,16 @@ def test_long_term_memory_with_memory_flag(): @pytest.mark.vcr(filter_headers=["authorization"]) def test_using_contextual_memory_with_short_term_memory(): from unittest.mock import patch + + # Check if ChromaDB is available + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") math_researcher = Agent( role="Researcher", @@ -2542,6 +2592,16 @@ def test_using_contextual_memory_with_short_term_memory(): @pytest.mark.vcr(filter_headers=["authorization"]) def test_disabled_memory_using_contextual_memory(): from unittest.mock import patch + + # Check if ChromaDB is available + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") math_researcher = Agent( role="Researcher", diff --git a/tests/memory/long_term_memory_test.py b/tests/memory/long_term_memory_test.py index 3639054e3..ad576d45e 100644 --- a/tests/memory/long_term_memory_test.py +++ b/tests/memory/long_term_memory_test.py @@ -7,10 +7,28 @@ from crewai.memory.long_term.long_term_memory_item import LongTermMemoryItem @pytest.fixture def long_term_memory(): """Fixture to create a LongTermMemory instance""" + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") + return LongTermMemory() def test_save_and_search(long_term_memory): + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") + memory = LongTermMemoryItem( agent="test_agent", task="test_task", diff --git a/tests/memory/short_term_memory_test.py b/tests/memory/short_term_memory_test.py index 6cde2a044..d017b2cea 100644 --- a/tests/memory/short_term_memory_test.py +++ b/tests/memory/short_term_memory_test.py @@ -12,6 +12,15 @@ from crewai.task import Task @pytest.fixture def short_term_memory(): """Fixture to create a ShortTermMemory instance""" + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") + agent = Agent( role="Researcher", goal="Search relevant data and provide results", @@ -29,6 +38,15 @@ def short_term_memory(): def test_save_and_search(short_term_memory): + try: + import chromadb + HAS_CHROMADB = True + except ImportError: + HAS_CHROMADB = False + + if not HAS_CHROMADB: + pytest.skip("ChromaDB is required for this test") + memory = ShortTermMemoryItem( data="""test value test value test value test value test value test value test value test value test value test value test value test value