diff --git a/tests/memory/test_memory_reset.py b/tests/memory/test_memory_reset.py new file mode 100644 index 000000000..964bd664b --- /dev/null +++ b/tests/memory/test_memory_reset.py @@ -0,0 +1,22 @@ +import os +import pytest +from crewai.memory import ShortTermMemory, LongTermMemory, EntityMemory +from crewai.utilities import EmbeddingConfigurator + +def test_memory_reset_with_ollama(): + os.environ["CREWAI_EMBEDDING_PROVIDER"] = "ollama" + os.environ["CREWAI_EMBEDDING_MODEL"] = "llama2" + + # Test each memory type + memories = [ShortTermMemory(), LongTermMemory(), EntityMemory()] + for memory in memories: + memory.reset() # Should not raise any OpenAI-related errors + +def test_memory_reset_with_openai(): + os.environ["CREWAI_EMBEDDING_PROVIDER"] = "openai" + os.environ["CREWAI_EMBEDDING_MODEL"] = "text-embedding-3-small" + + # Test each memory type + memories = [ShortTermMemory(), LongTermMemory(), EntityMemory()] + for memory in memories: + memory.reset() # Should work with OpenAI key