danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Correctly lookup docs for the orb dataset

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-01-20 09:50:40 +02:00
parent e6b239bd6c
commit de022639b6
No known key found for this signature in database
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
evaluations/evaluations/benchmark.py
View file

@ -92,7 +92,13 @@ async def populate_db(
progress.advance(task) progress.advance(task)
continue continue
existing = await rag.get_document_by_uri(payload.uri) # Use the actual URI that will be stored in the database
if payload.source_path is not None:
lookup_uri = payload.source_path.absolute().as_uri()
else:
lookup_uri = payload.uri
existing = await rag.get_document_by_uri(lookup_uri)
if existing is not None: if existing is not None:
assert existing.id assert existing.id
chunks = await rag.chunk_repository.get_by_document_id(existing.id) chunks = await rag.chunk_repository.get_by_document_id(existing.id)