diff --git a/examples/ag-ui-research/frontend/components/StateDisplay.tsx b/examples/ag-ui-research/frontend/components/StateDisplay.tsx
index 03e710b9..69a28b20 100644
--- a/examples/ag-ui-research/frontend/components/StateDisplay.tsx
+++ b/examples/ag-ui-research/frontend/components/StateDisplay.tsx
@@ -23,11 +23,29 @@ interface GapRecord {
resolved_by: string[];
}
+interface BoundingBox {
+ page_no: number;
+ left: number;
+ top: number;
+ right: number;
+ bottom: number;
+}
+
+interface Citation {
+ document_uri: string;
+ document_title?: string;
+ page_numbers: number[];
+ headings?: string[];
+ content: string;
+ bounding_boxes?: BoundingBox[];
+}
+
interface SearchAnswer {
- sub_question: string;
+ query: string;
answer: string;
confidence: number;
- chunks_used: number;
+ cited_chunks: string[];
+ citations: Citation[];
}
interface ResearchContext {
@@ -508,6 +526,125 @@ export default function StateDisplay({ state }: StateDisplayProps) {