paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2026-01-16 22:04:21 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix embedding mock

Browse Source
This commit is contained in:
shamoon
2026-01-14 13:13:05 -08:00
parent af1e7bc557
commit 45b7f9577c
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/paperless_ai/tests/test_chat.py
View File

@@ -11,14 +11,12 @@ from paperless_ai.chat import stream_chat_with_documents
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def patch_embed_model(): def patch_embed_model():
from llama_index.core import settings as llama_settings from llama_index.core import settings as llama_settings
from llama_index.core.embeddings.utils import MockEmbedding
mock_embed_model = MagicMock() mock_embed_model = MockEmbedding(embed_dim=8)
mock_embed_model._get_text_embedding_batch.return_value = [ llama_settings.Settings.embed_model = mock_embed_model
[0.1] * 1536,
] # 1 vector per input
llama_settings.Settings._embed_model = mock_embed_model
yield yield
llama_settings.Settings._embed_model = None llama_settings.Settings.embed_model = None
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)