From 33e532a491e1f2e9aa45175992c2a7f6ccf3b327 Mon Sep 17 00:00:00 2001 From: Ed_ Date: Thu, 14 May 2026 20:24:23 -0400 Subject: [PATCH] test(rag): Fix test_rag_engine.py mock assertion Updated the test to correctly mock _get_sentence_transformers instead of SentenceTransformer to account for lazy loading imports. --- tests/test_rag_engine.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_rag_engine.py b/tests/test_rag_engine.py index b5a1f34..34f28f7 100644 --- a/tests/test_rag_engine.py +++ b/tests/test_rag_engine.py @@ -31,7 +31,8 @@ def test_rag_engine_chroma(mock_chroma, mock_embed): vs_config = models.VectorStoreConfig(provider='chroma', collection_name='test') config = models.RAGConfig(enabled=True, vector_store=vs_config, embedding_provider='local') - with patch('src.rag_engine.SentenceTransformer') as mock_st: + with patch('src.rag_engine._get_sentence_transformers') as mock_st: + mock_st.return_value = MagicMock() engine = RAGEngine(config) assert engine.collection == mock_collection