Update evaluations and a2a

evaluations/evaluations/benchmark.py

@@ -155,10 +155,10 @@ async def run_retrieval_benchmark(
 
             seen = set()
             uris = []
-            for chunk, _ in chunks:
+            for result in chunks:
-                if chunk.document_id is None:
+                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)

examples/a2a-server/haiku_rag_a2a/a2a/__init__.py

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