mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
Fix #2271: Handle SQLite3 version check gracefully for ChromaDB
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
32
tests/utilities/test_embedding_configurator.py
Normal file
32
tests/utilities/test_embedding_configurator.py
Normal file
@@ -0,0 +1,32 @@
|
||||
import unittest
|
||||
from unittest.mock import patch, MagicMock
|
||||
|
||||
class TestEmbeddingConfigurator(unittest.TestCase):
|
||||
@patch('crewai.utilities.embedding_configurator.CHROMADB_AVAILABLE', False)
|
||||
def test_embedding_configurator_with_chromadb_unavailable(self):
|
||||
from crewai.utilities.embedding_configurator import EmbeddingConfigurator
|
||||
|
||||
# Create an instance of EmbeddingConfigurator
|
||||
configurator = EmbeddingConfigurator()
|
||||
|
||||
# Verify that embedding_functions is empty
|
||||
self.assertEqual(configurator.embedding_functions, {})
|
||||
|
||||
# Verify that configure_embedder returns None
|
||||
self.assertIsNone(configurator.configure_embedder())
|
||||
|
||||
@patch('crewai.utilities.embedding_configurator.CHROMADB_AVAILABLE', True)
|
||||
def test_embedding_configurator_with_chromadb_available(self):
|
||||
from crewai.utilities.embedding_configurator import EmbeddingConfigurator
|
||||
|
||||
# Create an instance of EmbeddingConfigurator
|
||||
configurator = EmbeddingConfigurator()
|
||||
|
||||
# Verify that embedding_functions is not empty
|
||||
self.assertNotEqual(configurator.embedding_functions, {})
|
||||
|
||||
# Mock the _create_default_embedding_function method
|
||||
configurator._create_default_embedding_function = MagicMock(return_value="mock_embedding_function")
|
||||
|
||||
# Verify that configure_embedder returns the mock embedding function
|
||||
self.assertEqual(configurator.configure_embedder(), "mock_embedding_function")
|
||||
Reference in New Issue
Block a user