From c376fbaf611d7b6a11c8303fe354edd5ea080066 Mon Sep 17 00:00:00 2001 From: CREDO23 Date: Fri, 19 Jun 2026 14:53:49 +0200 Subject: [PATCH] test: seed chunk spans in retriever fixture --- .../tests/integration/retriever/conftest.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/surfsense_backend/tests/integration/retriever/conftest.py b/surfsense_backend/tests/integration/retriever/conftest.py index d2443723c..96c6297bb 100644 --- a/surfsense_backend/tests/integration/retriever/conftest.py +++ b/surfsense_backend/tests/integration/retriever/conftest.py @@ -40,11 +40,19 @@ def _make_document( ) -def _make_chunk(*, content: str, document_id: int) -> Chunk: +def _make_chunk( + *, + content: str, + document_id: int, + start_char: int | None = None, + end_char: int | None = None, +) -> Chunk: return Chunk( content=content, document_id=document_id, embedding=DUMMY_EMBEDDING, + start_char=start_char, + end_char=end_char, ) @@ -91,6 +99,8 @@ async def seed_large_doc( _make_chunk( content="quarterly performance review summary note content", document_id=small_doc.id, + start_char=0, + end_char=10, ), ]