Chapter 3: Add all agents (multi-agent orchestration)

- Add analyst, planner, quiz-master agents
- Update route.ts to register all agents
- Add placeholder artifact UI components for flashcard and study-plan
- Update types with flashcardDelta and studyPlanDelta data types
- Update Document.kind to include flashcard and study-plan

Author: NiallJoeMaher

app/(chat)/api/chat/route.ts

@@ -7,7 +7,12 @@ import {
   stepCountIs,
   streamText,
 } from "ai";
-import { createTutorAgent } from "@/lib/ai/agents";
+import {
+  createAnalystAgent,
+  createPlannerAgent,
+  createQuizMasterAgent,
+  createTutorAgent,
+} from "@/lib/ai/agents";
 import { getWeather } from "@/lib/ai/tools/get-weather";
 import { unstable_cache as cache } from "next/cache";
 import type { ModelCatalog } from "tokenlens/core";
@@ -182,11 +187,22 @@ export async function POST(request: Request) {
           experimental_activeTools:
             selectedChatModel === "chat-model-reasoning"
               ? []
-              : ["getWeather", "tutor"],
+              : [
+                  "getWeather",
+                  // Study buddy agents
+                  "tutor",
+                  "quizMaster",
+                  "planner",
+                  "analyst",
+                ],
           experimental_transform: smoothStream({ chunking: "word" }),
           tools: {
             getWeather,
+            // Study buddy agents
             tutor: createTutorAgent({ session, dataStream }),
+            quizMaster: createQuizMasterAgent({ session, dataStream }),
+            planner: createPlannerAgent({ session, dataStream }),
+            analyst: createAnalystAgent({ session, dataStream }),
           },
           experimental_telemetry: {
             isEnabled: isProductionEnvironment,

artifacts/flashcard/client.tsx

@@ -0,0 +1,55 @@
+"use client";
+
+import { Artifact } from "@/components/create-artifact";
+
+// Placeholder flashcard artifact - full implementation coming in Chapter 4
+export const flashcardArtifact = new Artifact<"flashcard">({
+  kind: "flashcard",
+  description: "Interactive quiz flashcards",
+  onStreamPart: ({ streamPart, setArtifact }) => {
+    if (streamPart.type === "data-flashcardDelta") {
+      setArtifact((draftArtifact) => ({
+        ...draftArtifact,
+        content: streamPart.data,
+        isVisible: true,
+        status: "streaming",
+      }));
+    }
+  },
+  content: ({ content, status }) => {
+    if (status === "streaming" || !content) {
+      return (
+        <div className="flex h-full flex-col items-center justify-center gap-4 p-8">
+          <div className="text-muted-foreground">Generating quiz...</div>
+          <div className="h-2 w-48 animate-pulse rounded bg-muted-foreground/20" />
+        </div>
+      );
+    }
+
+    // Parse and display basic quiz data
+    try {
+      const data = JSON.parse(content);
+      return (
+        <div className="flex h-full flex-col gap-4 overflow-y-auto p-8">
+          <h2 className="font-bold text-xl">{data.topic}</h2>
+          <p className="text-muted-foreground">
+            {data.questions?.length || 0} questions generated
+          </p>
+          <div className="rounded-lg border bg-muted/50 p-4">
+            <p className="text-muted-foreground text-sm">
+              Full flashcard UI coming in Chapter 4
+            </p>
+          </div>
+        </div>
+      );
+    } catch {
+      return (
+        <div className="flex h-full items-center justify-center p-8">
+          <p className="text-muted-foreground">Loading flashcard data...</p>
+        </div>
+      );
+    }
+  },
+  actions: [],
+  toolbar: [],
+});

artifacts/study-plan/client.tsx

@@ -0,0 +1,56 @@
+"use client";
+
+import { Artifact } from "@/components/create-artifact";
+
+// Placeholder study-plan artifact - full implementation coming in Chapter 4
+export const studyPlanArtifact = new Artifact<"study-plan">({
+  kind: "study-plan",
+  description: "Interactive study plan with progress tracking",
+  onStreamPart: ({ streamPart, setArtifact }) => {
+    if (streamPart.type === "data-studyPlanDelta") {
+      setArtifact((draftArtifact) => ({
+        ...draftArtifact,
+        content: streamPart.data,
+        isVisible: true,
+        status: "streaming",
+      }));
+    }
+  },
+  content: ({ content, status }) => {
+    if (status === "streaming" || !content) {
+      return (
+        <div className="flex h-full flex-col items-center justify-center gap-4 p-8">
+          <div className="text-muted-foreground">Creating study plan...</div>
+          <div className="h-2 w-48 animate-pulse rounded bg-muted-foreground/20" />
+        </div>
+      );
+    }
+
+    // Parse and display basic study plan data
+    try {
+      const data = JSON.parse(content);
+      return (
+        <div className="flex h-full flex-col gap-4 overflow-y-auto p-8">
+          <h2 className="font-bold text-xl">{data.topic}</h2>
+          <p className="text-muted-foreground">
+            {data.duration} - {data.weeks?.length || 0} weeks
+          </p>
+          <p className="text-sm">{data.overview}</p>
+          <div className="rounded-lg border bg-muted/50 p-4">
+            <p className="text-muted-foreground text-sm">
+              Full study plan UI coming in Chapter 4
+            </p>
+          </div>
+        </div>
+      );
+    } catch {
+      return (
+        <div className="flex h-full items-center justify-center p-8">
+          <p className="text-muted-foreground">Loading study plan data...</p>
+        </div>
+      );
+    }
+  },
+  actions: [],
+  toolbar: [],
+});

components/artifact.tsx

@@ -13,7 +13,9 @@ import {
 import useSWR, { useSWRConfig } from "swr";
 import { useDebounceCallback, useWindowSize } from "usehooks-ts";
 import { codeArtifact } from "@/artifacts/code/client";
+import { flashcardArtifact } from "@/artifacts/flashcard/client";
 import { sheetArtifact } from "@/artifacts/sheet/client";
+import { studyPlanArtifact } from "@/artifacts/study-plan/client";
 import { textArtifact } from "@/artifacts/text/client";
 import { useArtifact } from "@/hooks/use-artifact";
 import type { Document, Vote } from "@/lib/db/types";
@@ -32,6 +34,8 @@ export const artifactDefinitions = [
   textArtifact,
   codeArtifact,
   sheetArtifact,
+  flashcardArtifact,
+  studyPlanArtifact,
 ];
 export type ArtifactKind = (typeof artifactDefinitions)[number]["kind"];
 

lib/ai/agents/analyst.ts

@@ -0,0 +1,106 @@
+import { generateText, tool } from "ai";
+import { z } from "zod";
+import { myProvider } from "../providers";
+import type { AgentResult, CreateAgentProps } from "./types";
+
+const ANALYST_SYSTEM_PROMPT = `You are a document analyst who excels at extracting insights and summarizing content.
+
+Your analysis approach:
+- Identify the main themes and key points
+- Extract important facts, figures, and arguments
+- Note relationships between concepts
+- Highlight actionable insights
+- Provide clear, structured summaries
+
+For document analysis, provide:
+1. Executive summary (2-3 sentences)
+2. Key points and main arguments
+3. Important details and supporting evidence
+4. Connections to broader context
+5. Actionable takeaways or study notes
+
+Be thorough but concise. Focus on what would be most valuable for learning and retention.`;
+
+/**
+ * Analyst Agent - Analyzes documents and extracts key insights
+ *
+ * Triggers: "analyze this", "summarize", "key points", "what's important"
+ * Output: Returns analysis that the orchestrator will present
+ */
+export const createAnalystAgent = (_props: CreateAgentProps) =>
+  tool({
+    description:
+      "Analyze content, extract key insights, and create summaries. Use when the user wants to understand, summarize, or extract key points from text, documents, or concepts. Triggers: analyze, summarize, key points, main ideas, extract insights, break down.",
+    inputSchema: z.object({
+      content: z.string().describe("The text or content to analyze"),
+      analysisType: z
+        .enum(["summary", "key-points", "deep-analysis", "study-notes"])
+        .default("summary")
+        .describe("Type of analysis to perform"),
+      focusOn: z
+        .string()
+        .optional()
+        .describe("Specific aspect to focus the analysis on"),
+      outputLength: z
+        .enum(["brief", "moderate", "detailed"])
+        .default("moderate")
+        .describe("Desired length of the analysis output"),
+    }),
+    execute: async ({
+      content,
+      analysisType,
+      focusOn,
+      outputLength,
+    }): Promise<AgentResult> => {
+      const focusContext = focusOn
+        ? `\n\nFocus particularly on: ${focusOn}`
+        : "";
+
+      const lengthGuide = {
+        brief: "Keep the analysis concise, around 100-200 words.",
+        moderate: "Provide a balanced analysis, around 300-500 words.",
+        detailed: "Provide a comprehensive analysis, around 600-800 words.",
+      };
+
+      const analysisGuide = {
+        summary:
+          "Create a clear summary highlighting the main message and supporting points.",
+        "key-points":
+          "Extract and list the most important points as bullet points with brief explanations.",
+        "deep-analysis":
+          "Provide thorough analysis including themes, arguments, evidence, and implications.",
+        "study-notes":
+          "Create study-friendly notes with headings, key terms, and memorable takeaways.",
+      };
+
+      const prompt = `Analyze the following content:
+
+---
+${content}
+---
+
+Analysis type: ${analysisType}
+${analysisGuide[analysisType]}
+${focusContext}
+
+${lengthGuide[outputLength]}`;
+
+      const { text } = await generateText({
+        model: myProvider.languageModel("chat-model"),
+        system: ANALYST_SYSTEM_PROMPT,
+        prompt,
+      });
+
+      return {
+        agentName: "analyst",
+        success: true,
+        summary: text,
+        data: {
+          analysisType,
+          focusOn,
+          outputLength,
+          contentLength: content.length,
+        },
+      };
+    },
+  });

lib/ai/agents/index.ts

@@ -1,5 +1,8 @@
 // Agent type definitions
 
 // Specialized agents
+export { createAnalystAgent } from "./analyst";
+export { createPlannerAgent } from "./planner";
+export { createQuizMasterAgent } from "./quiz-master";
 export { createTutorAgent } from "./tutor";
 export type { AgentContext, AgentResult, CreateAgentProps } from "./types";