diff --git a/examples/ag-ui-research/backend/agent.py b/examples/ag-ui-research/backend/agent.py
index 38435618..9bea6d98 100644
--- a/examples/ag-ui-research/backend/agent.py
+++ b/examples/ag-ui-research/backend/agent.py
@@ -73,35 +73,32 @@ def create_agent(
deps_type=ResearchDeps,
instructions="""You are a research co-pilot powered by haiku.rag.
-You work step-by-step with the user to conduct deep research on complex questions.
-
Your workflow:
-1. When user asks a question, propose a research plan (exactly 3 sub-questions)
-2. Wait for user approval before proceeding to the search phase
-3. Once approved, AUTOMATICALLY process ALL sub-questions IN ORDER (0, 1, 2, etc.) WITHOUT pausing:
- - For each sub-question:
- * Announce what you're searching for
- * Execute search_question for that question ID
- * IMMEDIATELY extract insights using extract_insights_from_results with the SAME question ID
- - Continue automatically to the next question until all are complete
-4. After all questions are searched, evaluate overall confidence in your findings
-5. Ask user if confident enough or should search more
-6. Synthesize final report with complete citations
+1. When user asks a question, IMMEDIATELY call propose_research_plan with the question
+2. Wait for user approval before proceeding
+3. Once approved, process questions ONE AT A TIME:
+ - Call search_question(question_id=0) and WAIT for it to complete
+ - Then call extract_insights_from_results(question_id=0) and WAIT for it to complete
+ - Then call search_question(question_id=1) and WAIT for it to complete
+ - Then call extract_insights_from_results(question_id=1) and WAIT for it to complete
+ - Then call search_question(question_id=2) and WAIT for it to complete
+ - Then call extract_insights_from_results(question_id=2) and WAIT for it to complete
+4. After all questions are processed, call evaluate_research_confidence
+5. Ask user if they want to finalize or continue researching
+6. When user approves, call synthesize_final_report
CRITICAL RULES:
-- ALWAYS call search_question BEFORE extract_insights_from_results for each question
-- Process questions in sequence: search Q0 → extract Q0 → search Q1 → extract Q1, etc.
-- NEVER skip ahead to extract insights for a question you haven't searched yet
-- In the search phase, DO NOT pause between questions - process all questions automatically
+- Call ONE tool at a time - wait for each tool to return before calling the next
+- NEVER call extract_insights_from_results until search_question has completed and returned results
+- DO NOT explain what you're about to do - just call the tool
+- DO NOT say "I'll search for..." or "Let me search..." - just call search_question
+- The state updates will show the user what's happening - you don't need to narrate
+- Process all 3 questions automatically without asking for approval between them
Document Viewing:
-- Users can request to view the full content of any cited document
-- When a user asks to "show document X" or "view source Y", use the get_full_document tool
-- Document URIs are tracked automatically as you search
-- The final report includes structured citations linking back to source documents
+- When user asks to "show document X", call get_full_document with the document_uri
-Be transparent: always announce what you're searching for, but don't wait for approval during the search phase.
-Show search scores, explain your reasoning, cite your sources, and involve the user in decisions about confidence and next steps.
+Remember: Call tools ONE AT A TIME in sequence. Each tool must complete before calling the next.
""",
)
diff --git a/examples/ag-ui-research/frontend/components/StateDisplay.tsx b/examples/ag-ui-research/frontend/components/StateDisplay.tsx
index 79a9f0d1..7b0ba0c9 100644
--- a/examples/ag-ui-research/frontend/components/StateDisplay.tsx
+++ b/examples/ag-ui-research/frontend/components/StateDisplay.tsx
@@ -1,5 +1,6 @@
"use client";
+import { Markdown } from "@copilotkit/react-ui";
import { useState } from "react";
interface SourceRef {
@@ -413,7 +414,7 @@ export default function StateDisplay({ state }: StateDisplayProps) {
color: "#4a5568",
}}
>
- {item.question}
+
{item.search_results && (
- {result.chunk}...
+
))}
@@ -622,7 +623,7 @@ export default function StateDisplay({ state }: StateDisplayProps) {
marginBottom: "0.5rem",
}}
>
- {insight.summary}
+
{insight.source_refs && insight.source_refs.length > 0 && (
Executive Summary
-
- {state.final_report.summary}
-
+
+
- {state.final_report.findings.map((finding) => (
-
- {finding}
+ {state.final_report.findings.map((finding, idx) => (
+
+
))}
@@ -767,9 +771,12 @@ export default function StateDisplay({ state }: StateDisplayProps) {
lineHeight: "1.6",
}}
>
- {state.final_report.conclusions.map((conclusion) => (
-
- {conclusion}
+ {state.final_report.conclusions.map((conclusion, idx) => (
+
+
))}