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

test: Use ChromaDB in-memory mode for tests to avoid file system issues

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-09 23:47:35 +00:00
parent dbea3758eb
commit 318a3ad3e7
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/memory/test_memory_reset.py
View File

@@ -19,7 +19,16 @@ def temp_db_dir() -> Generator[Path, None, None]:
"""Create a temporary directory for test databases.""" """Create a temporary directory for test databases."""
with tempfile.TemporaryDirectory() as tmpdir: with tempfile.TemporaryDirectory() as tmpdir:
path = Path(tmpdir) path = Path(tmpdir)
yield path # Ensure directory exists and is writable
path.mkdir(parents=True, exist_ok=True)
# Set ChromaDB to use in-memory mode for tests
os.environ["CHROMA_IN_MEMORY"] = "true"
try:
yield path
finally:
# Clean up ChromaDB environment variable
if "CHROMA_IN_MEMORY" in os.environ:
del os.environ["CHROMA_IN_MEMORY"]
def test_memory_reset_with_openai(temp_db_dir): def test_memory_reset_with_openai(temp_db_dir):