mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-16 11:38:31 +00:00
26 lines
745 B
Python
26 lines
745 B
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from crewai.rag.embeddings.configurator import EmbeddingConfigurator
|
|
|
|
|
|
def test_configure_embedder_importerror():
|
|
configurator = EmbeddingConfigurator()
|
|
|
|
embedder_config = {
|
|
'provider': 'openai',
|
|
'config': {
|
|
'model': 'text-embedding-ada-002',
|
|
}
|
|
}
|
|
|
|
with patch('chromadb.utils.embedding_functions.openai_embedding_function.OpenAIEmbeddingFunction') as mock_openai:
|
|
mock_openai.side_effect = ImportError("Module not found.")
|
|
|
|
with pytest.raises(ImportError) as exc_info:
|
|
configurator.configure_embedder(embedder_config)
|
|
|
|
assert str(exc_info.value) == "Module not found."
|
|
mock_openai.assert_called_once()
|