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 #2464: Fix embedding dimension mismatch with reset-memories command

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-25 13:57:20 +00:00
parent 35cb7fcf4d
commit 74e018ae43
2 changed files with 77 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
tests/knowledge/knowledge_storage_test.py Normal file
View File

@@ -0,0 +1,55 @@
import os
import pytest
from unittest.mock import MagicMock, patch
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
class TestKnowledgeStorage:
@patch("crewai.knowledge.storage.knowledge_storage.chromadb")
@patch("crewai.knowledge.storage.knowledge_storage.shutil")
def test_reset_with_default_collection(self, mock_shutil, mock_chromadb):
# Setup
mock_app = MagicMock()
mock_chromadb.PersistentClient.return_value = mock_app
# Execute
storage = KnowledgeStorage()
storage.reset()
# Verify
mock_app.reset.assert_called_once()
mock_shutil.rmtree.assert_called_once()
@patch("crewai.knowledge.storage.knowledge_storage.chromadb")
@patch("crewai.knowledge.storage.knowledge_storage.shutil")
def test_reset_with_custom_collection(self, mock_shutil, mock_chromadb):
# Setup
mock_app = MagicMock()
mock_chromadb.PersistentClient.return_value = mock_app
# Execute
storage = KnowledgeStorage(collection_name="custom_collection")
storage.reset()
# Verify
mock_app.delete_collection.assert_called_once_with(name="knowledge_custom_collection")
mock_app.reset.assert_not_called()
mock_shutil.rmtree.assert_not_called()
@patch("crewai.knowledge.storage.knowledge_storage.chromadb")
@patch("crewai.knowledge.storage.knowledge_storage.shutil")
def test_reset_with_custom_collection_fallback(self, mock_shutil, mock_chromadb):
# Setup
mock_app = MagicMock()
mock_app.delete_collection.side_effect = Exception("Collection not found")
mock_chromadb.PersistentClient.return_value = mock_app
# Execute
storage = KnowledgeStorage(collection_name="custom_collection")
storage.reset()
# Verify
mock_app.delete_collection.assert_called_once_with(name="knowledge_custom_collection")
mock_app.reset.assert_called_once()
mock_shutil.rmtree.assert_called_once()