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.
2026-05-14 20:24:23 -04:00
parent d7fab05f4a
commit 33e532a491
1 changed files with 2 additions and 1 deletions
@@ -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