From 318a3ad3e746662a4d4d44875b1a56a0eaa253cd Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sun, 9 Feb 2025 23:47:35 +0000 Subject: [PATCH] test: Use ChromaDB in-memory mode for tests to avoid file system issues Co-Authored-By: Joe Moura --- tests/memory/test_memory_reset.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/tests/memory/test_memory_reset.py b/tests/memory/test_memory_reset.py index b1c8591df..0855ec351 100644 --- a/tests/memory/test_memory_reset.py +++ b/tests/memory/test_memory_reset.py @@ -19,7 +19,16 @@ def temp_db_dir() -> Generator[Path, None, None]: """Create a temporary directory for test databases.""" with tempfile.TemporaryDirectory() as 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):