andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-26 08:38:15 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refactor: simplify rag client initialization (#3401)

Browse Source 
* Simplified Qdrant and ChromaDB client initialization
* Refactored factory structure and updated tests accordingly
This commit is contained in:
Greyson LaLonde
2025-08-26 08:54:51 -04:00
committed by GitHub
parent 869bb115c8
commit 4b4a119a9f
9 changed files with 44 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/rag/chromadb/test_client.py
View File

@@ -27,18 +27,20 @@ def mock_async_chromadb_client():
@pytest.fixture
def client(mock_chromadb_client) -> ChromaDBClient:
"""Create a ChromaDBClient instance for testing."""
client = ChromaDBClient()
client.client = mock_chromadb_client
client.embedding_function = Mock()
mock_embedding = Mock()
client = ChromaDBClient(
client=mock_chromadb_client, embedding_function=mock_embedding
)
return client
@pytest.fixture
def async_client(mock_async_chromadb_client) -> ChromaDBClient:
"""Create a ChromaDBClient instance with async client for testing."""
client = ChromaDBClient()
client.client = mock_async_chromadb_client
client.embedding_function = Mock()
mock_embedding = Mock()
client = ChromaDBClient(
client=mock_async_chromadb_client, embedding_function=mock_embedding
)
return client