From a038b751ef84d9cee048fcec95b57f5fcde5739f Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 5 Feb 2025 10:45:21 +0000 Subject: [PATCH] test: Add memory reset tests for different embedding providers Co-Authored-By: Joe Moura --- tests/memory/test_memory_reset.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 tests/memory/test_memory_reset.py 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