diff --git a/examples/ag-ui-research/backend/pyproject.toml b/examples/ag-ui-research/backend/pyproject.toml
index 779da262..1e25ee00 100644
--- a/examples/ag-ui-research/backend/pyproject.toml
+++ b/examples/ag-ui-research/backend/pyproject.toml
@@ -9,7 +9,7 @@ dependencies = [
     "uvicorn[standard]>=0.34.2",
     "pydantic-ai-slim[ag-ui,openai]>=1.17.0",
     "python-dotenv>=1.0.1",
-    "haiku-rag-slim @ file:///Users/ggozad/dev/open-source/haiku.rag-agui/haiku_rag_slim",
+    "haiku.rag-slim[agui]>=0.20.0",
 ]
 
 [dependency-groups]
diff --git a/examples/ag-ui-research/frontend/components/Agent.tsx b/examples/ag-ui-research/frontend/components/Agent.tsx
index 022e1100..ac297cfc 100644
--- a/examples/ag-ui-research/frontend/components/Agent.tsx
+++ b/examples/ag-ui-research/frontend/components/Agent.tsx
@@ -25,12 +25,22 @@ interface GapRecord {
 	resolved_by: string[];
 }
 
+interface Citation {
+	document_id: string;
+	chunk_id: string;
+	document_uri: string;
+	document_title?: string;
+	page_numbers: number[];
+	headings?: string[];
+	content: string;
+}
+
 interface SearchAnswer {
 	query: string;
 	answer: string;
 	confidence: number;
-	context: string[];
-	sources: string[];
+	cited_chunks: string[];
+	citations: Citation[];
 }
 
 interface ResearchContext {
@@ -94,7 +104,6 @@ function AgentContent() {
 		},
 	});
 
-
 	return (
 		<>
 			<style>{`
diff --git a/examples/ag-ui-research/frontend/components/StateDisplay.tsx b/examples/ag-ui-research/frontend/components/StateDisplay.tsx
index 865e8c87..b1bb2c39 100644
--- a/examples/ag-ui-research/frontend/components/StateDisplay.tsx
+++ b/examples/ag-ui-research/frontend/components/StateDisplay.tsx
@@ -1,7 +1,7 @@
 "use client";
 
 import { Markdown } from "@copilotkit/react-ui";
-import { useState, useCallback } from "react";
+import { useCallback, useState } from "react";
 
 interface VisualGroundingState {
 	isOpen: boolean;
@@ -127,7 +127,7 @@ export default function StateDisplay({ state }: StateDisplayProps) {
 
 		try {
 			const response = await fetch(
-				`${process.env.NEXT_PUBLIC_API_URL || "http://localhost:8000"}/api/visualize/${chunkId}`
+				`${process.env.NEXT_PUBLIC_API_URL || "http://localhost:8000"}/api/visualize/${chunkId}`,
 			);
 			const data = await response.json();
 
@@ -241,14 +241,25 @@ export default function StateDisplay({ state }: StateDisplayProps) {
 								marginBottom: state.iterations > 0 ? "1rem" : 0,
 							}}
 						>
-							<div style={{ display: "flex", alignItems: "center", gap: "0.5rem" }}>
+							<div
+								style={{ display: "flex", alignItems: "center", gap: "0.5rem" }}
+							>
 								<span>⏳</span>
 								<span style={{ fontWeight: "600", color: "#2b6cb0" }}>
-									{state.current_activity.replace(/_/g, " ").replace(/\b\w/g, (c) => c.toUpperCase())}
+									{state.current_activity
+										.replace(/_/g, " ")
+										.replace(/\b\w/g, (c) => c.toUpperCase())}
 								</span>
 							</div>
 							{state.current_activity_message && (
-								<div style={{ fontSize: "0.875rem", color: "#4299e1", marginTop: "0.25rem", marginLeft: "1.5rem" }}>
+								<div
+									style={{
+										fontSize: "0.875rem",
+										color: "#4299e1",
+										marginTop: "0.25rem",
+										marginLeft: "1.5rem",
+									}}
+								>
 									{state.current_activity_message}
 								</div>
 							)}
@@ -449,7 +460,9 @@ export default function StateDisplay({ state }: StateDisplayProps) {
 									>
 										⏳
 									</div>
-									<div style={{ flex: 1, fontSize: "0.875rem", color: "#4a5568" }}>
+									<div
+										style={{ flex: 1, fontSize: "0.875rem", color: "#4a5568" }}
+									>
 										<Markdown content={question} />
 									</div>
 								</div>
@@ -510,7 +523,8 @@ export default function StateDisplay({ state }: StateDisplayProps) {
 														marginTop: "0.25rem",
 													}}
 												>
-													Confidence: {(qaResponse.confidence * 100).toFixed(0)}%
+													Confidence: {(qaResponse.confidence * 100).toFixed(0)}
+													%
 												</div>
 											</div>
 											<span
@@ -648,7 +662,9 @@ export default function StateDisplay({ state }: StateDisplayProps) {
 																	</div>
 																	<button
 																		type="button"
-																		onClick={() => fetchVisualGrounding(citation.chunk_id)}
+																		onClick={() =>
+																			fetchVisualGrounding(citation.chunk_id)
+																		}
 																		style={{
 																			marginTop: "0.5rem",
 																			padding: "0.25rem 0.5rem",
diff --git a/examples/ag-ui-research/frontend/package-lock.json b/examples/ag-ui-research/frontend/package-lock.json
index 3ab00152..7f62f1a7 100644
--- a/examples/ag-ui-research/frontend/package-lock.json
+++ b/examples/ag-ui-research/frontend/package-lock.json
@@ -8,7 +8,7 @@
 			"name": "ag-ui-frontend",
 			"version": "0.1.0",
 			"dependencies": {
-				"@ag-ui/client": "^0.0.40",
+				"@ag-ui/client": "^0.0.42",
 				"@copilotkit/react-core": "^1.10.6",
 				"@copilotkit/react-ui": "^1.10.6",
 				"@copilotkit/runtime": "^1.10.6",
@@ -39,15 +39,16 @@
 			}
 		},
 		"node_modules/@ag-ui/client": {
-			"version": "0.0.40",
-			"resolved": "https://registry.npmjs.org/@ag-ui/client/-/client-0.0.40.tgz",
-			"integrity": "sha512-4ftyZgMN7DIAX64k7Mdex/KGq7lfz8yxEKzniqosD6TE/xk65k4Z0v3bxTzPk2iS2+Cj2uVBgFkb5lC7k5Loqg==",
+			"version": "0.0.42",
+			"resolved": "https://registry.npmjs.org/@ag-ui/client/-/client-0.0.42.tgz",
+			"integrity": "sha512-zAbP+sZJImR5bUpR2ni7RtuuNZMuesaxviynyIgzKlr1k2VCM49mFpbDUKU4TH4Cneu+Xe7OEnO8qCOCIzBAww==",
 			"peer": true,
 			"dependencies": {
-				"@ag-ui/core": "0.0.39",
-				"@ag-ui/encoder": "0.0.39",
-				"@ag-ui/proto": "0.0.39",
+				"@ag-ui/core": "0.0.42",
+				"@ag-ui/encoder": "0.0.42",
+				"@ag-ui/proto": "0.0.42",
 				"@types/uuid": "^10.0.0",
+				"compare-versions": "^6.1.1",
 				"fast-json-patch": "^3.1.1",
 				"rxjs": "7.8.1",
 				"untruncate-json": "^0.0.1",
@@ -56,9 +57,9 @@
 			}
 		},
 		"node_modules/@ag-ui/core": {
-			"version": "0.0.39",
-			"resolved": "https://registry.npmjs.org/@ag-ui/core/-/core-0.0.39.tgz",
-			"integrity": "sha512-T5Hp4oFkQ+H5MynWAvSwrX/rNYJOD+PJ4qPQ0o771oSZQAxoIvDDft47Cx5wRyBNNLXAe1RWqJjfWUUwJFNKqA==",
+			"version": "0.0.42",
+			"resolved": "https://registry.npmjs.org/@ag-ui/core/-/core-0.0.42.tgz",
+			"integrity": "sha512-C2hMg4Gs5oiUDgK9cA2RsTwSSmFZdIsqPklDrFw/Ue+quH6EU3vKp5YoOq7nuaQYO4pO8Em+Z+l5/M5PpcvP1g==",
 			"peer": true,
 			"dependencies": {
 				"rxjs": "7.8.1",
@@ -66,13 +67,13 @@
 			}
 		},
 		"node_modules/@ag-ui/encoder": {
-			"version": "0.0.39",
-			"resolved": "https://registry.npmjs.org/@ag-ui/encoder/-/encoder-0.0.39.tgz",
-			"integrity": "sha512-6fsoFwPWkStK7Uyj3pwBn7+aQjUWf7pbDTSI43cD53sBLvTr5oEFNnoKOzRfC5UqvHc4JjUIuLKPQyjHRwWg4g==",
+			"version": "0.0.42",
+			"resolved": "https://registry.npmjs.org/@ag-ui/encoder/-/encoder-0.0.42.tgz",
+			"integrity": "sha512-97B5MMCSs82t/y41uk2NrLBYFhbvn4kYsKQHMCfy8tjSWubyxh3zP7N9yHo8zJeSPe3WvzTvclyXNiGxSOsorg==",
 			"peer": true,
 			"dependencies": {
-				"@ag-ui/core": "0.0.39",
-				"@ag-ui/proto": "0.0.39"
+				"@ag-ui/core": "0.0.42",
+				"@ag-ui/proto": "0.0.42"
 			}
 		},
 		"node_modules/@ag-ui/langgraph": {
@@ -92,12 +93,12 @@
 			}
 		},
 		"node_modules/@ag-ui/proto": {
-			"version": "0.0.39",
-			"resolved": "https://registry.npmjs.org/@ag-ui/proto/-/proto-0.0.39.tgz",
-			"integrity": "sha512-xlj/PzZHkJ3CgoQC5QP9g7DEl/78wUK1+A2rdkoLKoNAMOkM2g6jKw0N88iFIh5GZhtiCNN2wb8XwRWPYx9XQQ==",
+			"version": "0.0.42",
+			"resolved": "https://registry.npmjs.org/@ag-ui/proto/-/proto-0.0.42.tgz",
+			"integrity": "sha512-NDUwSgMnGEqxZGkWIJ1ge5t3Q7Kiddj360x2JAWaIfv9w+7tDJ0pmgyzf3/SXp605aY2wZiDLBtJ6jKZeg1lFg==",
 			"peer": true,
 			"dependencies": {
-				"@ag-ui/core": "0.0.39",
+				"@ag-ui/core": "0.0.42",
 				"@bufbuild/protobuf": "^2.2.5",
 				"@protobuf-ts/protoc": "^2.11.1"
 			}
@@ -5266,6 +5267,12 @@
 				"node": ">= 12"
 			}
 		},
+		"node_modules/compare-versions": {
+			"version": "6.1.1",
+			"resolved": "https://registry.npmjs.org/compare-versions/-/compare-versions-6.1.1.tgz",
+			"integrity": "sha512-4hm4VPpIecmlg59CHXnRDnqGplJFrbLG4aFEl5vl6cK1u76ws3LLvX7ikFnTDl5vo39sjWD6AaDPYodJp/NNHg==",
+			"license": "MIT"
+		},
 		"node_modules/console-table-printer": {
 			"version": "2.14.6",
 			"resolved": "https://registry.npmjs.org/console-table-printer/-/console-table-printer-2.14.6.tgz",
diff --git a/examples/ag-ui-research/frontend/package.json b/examples/ag-ui-research/frontend/package.json
index aa094518..d2e2c811 100644
--- a/examples/ag-ui-research/frontend/package.json
+++ b/examples/ag-ui-research/frontend/package.json
@@ -11,7 +11,7 @@
 		"format": "biome check --write app components"
 	},
 	"dependencies": {
-		"@ag-ui/client": "^0.0.40",
+		"@ag-ui/client": "^0.0.42",
 		"@copilotkit/react-core": "^1.10.6",
 		"@copilotkit/react-ui": "^1.10.6",
 		"@copilotkit/runtime": "^1.10.6",