apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-27 01:36:30 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: use batch embedding in IndexingPipelineService.index

Browse source
This commit is contained in:
CREDO23 2026-03-09 16:13:44 +02:00
parent cb4b155b9d
commit 929445afd9
5 changed files with 42 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

8
surfsense_backend/tests/integration/conftest.py
View file

@ -129,10 +129,12 @@ def patched_summarize_raises(monkeypatch) -> AsyncMock:
@pytest.fixture
def patched_embed_text(monkeypatch) -> MagicMock:
mock = MagicMock(return_value=[0.1] * _EMBEDDING_DIM)
def patched_embed_texts(monkeypatch) -> MagicMock:
mock = MagicMock(
side_effect=lambda texts: [[0.1] * _EMBEDDING_DIM for _ in texts]
)
monkeypatch.setattr(
"app.indexing_pipeline.indexing_pipeline_service.embed_text",
"app.indexing_pipeline.indexing_pipeline_service.embed_texts",
mock,
)
return mock