We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fb318c4 commit 20193baCopy full SHA for 20193ba
tests/conftest.py
@@ -1,9 +1,19 @@
1
+import sys
2
+from unittest.mock import MagicMock
3
+
4
+mock_chroma = MagicMock()
5
+mock_chroma.HttpClient.return_value = MagicMock() # Returns a fake client
6
+sys.modules["chromadb"] = mock_chroma
7
8
+mock_st = MagicMock()
9
+mock_st.SentenceTransformer.return_value = MagicMock()
10
+sys.modules["sentence_transformers"] = mock_st
11
12
import pytest
13
from fastapi.testclient import TestClient
14
from sqlmodel import Session, SQLModel, create_engine, pool
15
from app.main import app
16
from app.db.session import get_session
-from app.services.llm_factory import LLMOrchestrator
17
18
TEST_DATABASE_URL = "sqlite:///:memory:"
19
engine = create_engine(
0 commit comments