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

Update evaluations and a2a

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-11-28 17:55:50 +02:00
parent c69b934eac
commit ace58a7034
No known key found for this signature in database
2 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
evaluations/evaluations/benchmark.py
View file

@ -155,10 +155,10 @@ async def run_retrieval_benchmark(
seen = set()
uris = []
for chunk, _ in chunks:
if chunk.document_id is None:
for result in chunks:
if result.document_id is None:
continue
doc = await rag.get_document_by_id(chunk.document_id)
doc = await rag.get_document_by_id(result.document_id)
if doc and doc.uri and doc.uri not in seen:
uris.append(doc.uri)
seen.add(doc.uri)

10
examples/a2a-server/haiku_rag_a2a/a2a/__init__.py
View file

@ -89,12 +89,12 @@ def create_a2a_app(
return [
SearchResult(
content=chunk.content,
score=score,
document_title=chunk.document_title,
document_uri=(chunk.document_uri or ""),
content=result.content,
score=result.score,
document_title=result.document_title,
document_uri=(result.document_uri or ""),
)
for chunk, score in expanded_results
for result in expanded_results
]
@agent.tool