diff --git a/tests/chunkers/test_sentence_chunker.py b/tests/chunkers/test_sentence_chunker.py index 3fcc256cf..28aaeabb9 100644 --- a/tests/chunkers/test_sentence_chunker.py +++ b/tests/chunkers/test_sentence_chunker.py @@ -9,8 +9,8 @@ class TestSentenceChunker(unittest.TestCase): def test_sentence_chunker(self): """Test SentenceChunker functionality with mocked backend.""" - with patch("memos.chunkers.sentence_chunker.ChonkieSentenceChunker") as mock_chunker_cls: - # Set up the mock for ChonkieSentenceChunker + with patch("chonkie.SentenceChunker") as mock_chunker_cls: + # Set up the mock for SentenceChunker mock_chunker = MagicMock() mock_chunks = [ MagicMock( diff --git a/tests/vec_dbs/test_qdrant.py b/tests/vec_dbs/test_qdrant.py index c2eabd218..828240ae1 100644 --- a/tests/vec_dbs/test_qdrant.py +++ b/tests/vec_dbs/test_qdrant.py @@ -28,7 +28,7 @@ def config(): @pytest.fixture def mock_qdrant_client(): - with patch("memos.vec_dbs.qdrant.QdrantClient") as mockclient: + with patch("qdrant_client.QdrantClient") as mockclient: yield mockclient