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

fix: Make model_name optional for custom embedders

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-10 00:01:24 +00:00
parent 5566c587a4
commit a2e1b3896e
2 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/memory/test_memory_reset.py
View File

@@ -74,7 +74,11 @@ def test_memory_reset_with_custom_provider(temp_db_dir):
return [[0.5] * 10] * len(input)
memory = ShortTermMemory(
path=str(temp_db_dir), embedder_config={"provider": CustomEmbedder()}
path=str(temp_db_dir),
embedder_config={
"provider": "custom",
"config": {"embedder": CustomEmbedder()}
}
)
memory.reset() # Should work with custom embedder
@@ -132,7 +136,11 @@ def test_memory_reset_cleans_up_files(temp_db_dir):
if "CREWAI_EMBEDDING_PROVIDER" in os.environ:
del os.environ["CREWAI_EMBEDDING_PROVIDER"]
memory = ShortTermMemory(
path=str(temp_db_dir), embedder_config={"provider": TestEmbedder()}
path=str(temp_db_dir),
embedder_config={
"provider": "custom",
"config": {"embedder": TestEmbedder()}
}
)
memory.save("test memory", {"test": "metadata"})
assert any(temp_db_dir.iterdir()) # Directory should have files