andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-20 13:28:13 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Address PR comments: Improve code quality and add validation

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-04 07:39:27 +00:00
parent 0427da467a
commit d9c5ebe1cf
4 changed files with 76 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/utilities/test_embedding_configurator.py
View File

@@ -31,3 +31,22 @@ class TestEmbeddingConfigurator(unittest.TestCase):
# Verify that configure_embedder returns the mock embedding function
self.assertEqual(configurator.configure_embedder(), "mock_embedding_function")
@patch('crewai.utilities.embedding_configurator.CHROMADB_AVAILABLE', True)
def test_embedding_configurator_with_invalid_config(self):
from crewai.utilities.embedding_configurator import EmbeddingConfigurator
# Create an instance of EmbeddingConfigurator
configurator = EmbeddingConfigurator()
# Test with empty config
with self.assertRaises(ValueError):
configurator.configure_embedder({})
# Test with missing required keys
with self.assertRaises(ValueError):
configurator.configure_embedder({"config": {}})
# Test with unsupported provider
with self.assertRaises(Exception):
configurator.configure_embedder({"provider": "unsupported_provider", "config": {}})