mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-16 11:38:31 +00:00
ff5cbdee07
- Move core CrewAI to packages/crewai with git mv to preserve history - Create workspace-level pyproject.toml with uv workspace configuration - Rename core package to crewai-core - Setup workspace sources for internal package dependencies
35 lines
1019 B
Python
35 lines
1019 B
Python
"""Tests for RAG config factory."""
|
|
|
|
from unittest.mock import Mock, patch
|
|
|
|
from crewai.rag.factory import create_client
|
|
|
|
|
|
def test_create_client_chromadb():
|
|
"""Test ChromaDB client creation."""
|
|
mock_config = Mock()
|
|
mock_config.provider = "chromadb"
|
|
|
|
with patch("crewai.rag.factory.require") as mock_require:
|
|
mock_module = Mock()
|
|
mock_client = Mock()
|
|
mock_module.create_client.return_value = mock_client
|
|
mock_require.return_value = mock_module
|
|
|
|
result = create_client(mock_config)
|
|
|
|
assert result == mock_client
|
|
mock_require.assert_called_once_with(
|
|
"crewai.rag.chromadb.factory", purpose="The 'chromadb' provider"
|
|
)
|
|
mock_module.create_client.assert_called_once_with(mock_config)
|
|
|
|
|
|
def test_create_client_unsupported_provider():
|
|
"""Test unsupported provider returns None for now."""
|
|
mock_config = Mock()
|
|
mock_config.provider = "unsupported"
|
|
|
|
result = create_client(mock_config)
|
|
assert result is None
|