From ace58a70345dceb155686c446e393654ab0a5617 Mon Sep 17 00:00:00 2001
From: Yiorgis Gozadinos <ggozadinos@gmail.com>
Date: Fri, 28 Nov 2025 17:55:50 +0200
Subject: [PATCH] Update evaluations and a2a

---
 evaluations/evaluations/benchmark.py              |  6 +++---
 examples/a2a-server/haiku_rag_a2a/a2a/__init__.py | 10 +++++-----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/evaluations/evaluations/benchmark.py b/evaluations/evaluations/benchmark.py
index e0263647..e71643d7 100644
--- a/evaluations/evaluations/benchmark.py
+++ b/evaluations/evaluations/benchmark.py
@@ -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)
diff --git a/examples/a2a-server/haiku_rag_a2a/a2a/__init__.py b/examples/a2a-server/haiku_rag_a2a/a2a/__init__.py
index 524365c0..745c9903 100644
--- a/examples/a2a-server/haiku_rag_a2a/a2a/__init__.py
+++ b/examples/a2a-server/haiku_rag_a2a/a2a/__init__.py
@@ -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