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

test: Add memory reset tests for different embedding providers

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-05 10:45:21 +00:00
parent 5006161d31
commit a038b751ef
1 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/memory/test_memory_reset.py Normal file
View File

@@ -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