apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-12 22:42:13 +02:00
Code Issues Projects Releases Packages Wiki Activity

test: seed chunk spans in retriever fixture

Browse source
This commit is contained in:
CREDO23 2026-06-19 14:53:49 +02:00
parent 04b679e2bf
commit c376fbaf61
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
surfsense_backend/tests/integration/retriever/conftest.py
View file

@ -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( return Chunk(
content=content, content=content,
document_id=document_id, document_id=document_id,
embedding=DUMMY_EMBEDDING, embedding=DUMMY_EMBEDDING,
start_char=start_char,
end_char=end_char,
) )
@ -91,6 +99,8 @@ async def seed_large_doc(
_make_chunk( _make_chunk(
content="quarterly performance review summary note content", content="quarterly performance review summary note content",
document_id=small_doc.id, document_id=small_doc.id,
start_char=0,
end_char=10,
), ),
] ]