feat: structured-output as a typed UIMessage part

.changeset/structured-output-as-message-part.md

@@ -0,0 +1,25 @@
+---
+'@tanstack/ai': minor
+'@tanstack/ai-client': minor
+'@tanstack/ai-react': minor
+---
+
+feat: structured-output as a typed MessagePart on each assistant UIMessage
+
+`useChat({ outputSchema })` previously kept a single hook-level `partial`/`final`
+slot, so multi-turn structured chats lost every prior turn's response as soon
+as a new one streamed in. Each assistant turn now carries its own typed
+`structured-output` MessagePart on the UIMessage it belongs to. History walks
+`messages` and finds the typed part on each turn; the hook-level `partial`
+and `final` are derived from the latest assistant message's part and continue
+to work as before.
+
+Server-side `chat({ outputSchema, stream: true })` emits a new
+`structured-output.start` CUSTOM event before the JSON deltas so the client
+processor can route them into the StructuredOutputPart instead of building a
+TextPart. The wire converter serializes the part's raw JSON back as assistant
+content, so multi-turn structured chats stay coherent (the LLM sees its own
+prior structured responses on follow-up turns).
+
+A new example route demonstrating this pattern is at
+`/generations/structured-chat` in the `ts-react-chat` example.

examples/ts-react-chat/src/components/Header.tsx

@@ -166,6 +166,19 @@ export default function Header() {
             <span className="font-medium">Structured Output</span>
           </Link>
 
+          <Link
+            to="/generations/structured-chat"
+            onClick={() => setIsOpen(false)}
+            className="flex items-center gap-3 p-3 rounded-lg hover:bg-gray-800 transition-colors mb-1"
+            activeProps={{
+              className:
+                'flex items-center gap-3 p-3 rounded-lg bg-cyan-600 hover:bg-cyan-700 transition-colors mb-1',
+            }}
+          >
+            <Braces size={20} />
+            <span className="font-medium">Structured Chat</span>
+          </Link>
+
           <hr className="border-gray-700 my-2" />
 
           <p className="text-xs text-gray-500 uppercase tracking-wider px-3 pt-2 pb-1">

examples/ts-react-chat/src/routeTree.gen.ts

@@ -16,13 +16,15 @@ import { Route as GenerationsVideoRouteImport } from './routes/generations.video
 import { Route as GenerationsTranscriptionRouteImport } from './routes/generations.transcription'
 import { Route as GenerationsSummarizeRouteImport } from './routes/generations.summarize'
 import { Route as GenerationsStructuredOutputRouteImport } from './routes/generations.structured-output'
+import { Route as GenerationsStructuredChatRouteImport } from './routes/generations.structured-chat'
 import { Route as GenerationsSpeechRouteImport } from './routes/generations.speech'
 import { Route as GenerationsImageRouteImport } from './routes/generations.image'
 import { Route as GenerationsAudioRouteImport } from './routes/generations.audio'
 import { Route as ApiTranscribeRouteImport } from './routes/api.transcribe'
 import { Route as ApiTanchatRouteImport } from './routes/api.tanchat'
 import { Route as ApiSummarizeRouteImport } from './routes/api.summarize'
 import { Route as ApiStructuredOutputRouteImport } from './routes/api.structured-output'
+import { Route as ApiStructuredChatRouteImport } from './routes/api.structured-chat'
 import { Route as ApiImageGenRouteImport } from './routes/api.image-gen'
 import { Route as ExampleGuitarsIndexRouteImport } from './routes/example.guitars/index'
 import { Route as ExampleGuitarsGuitarIdRouteImport } from './routes/example.guitars/$guitarId'
@@ -68,6 +70,12 @@ const GenerationsStructuredOutputRoute =
     path: '/generations/structured-output',
     getParentRoute: () => rootRouteImport,
   } as any)
+const GenerationsStructuredChatRoute =
+  GenerationsStructuredChatRouteImport.update({
+    id: '/generations/structured-chat',
+    path: '/generations/structured-chat',
+    getParentRoute: () => rootRouteImport,
+  } as any)
 const GenerationsSpeechRoute = GenerationsSpeechRouteImport.update({
   id: '/generations/speech',
   path: '/generations/speech',
@@ -103,6 +111,11 @@ const ApiStructuredOutputRoute = ApiStructuredOutputRouteImport.update({
   path: '/api/structured-output',
   getParentRoute: () => rootRouteImport,
 } as any)
+const ApiStructuredChatRoute = ApiStructuredChatRouteImport.update({
+  id: '/api/structured-chat',
+  path: '/api/structured-chat',
+  getParentRoute: () => rootRouteImport,
+} as any)
 const ApiImageGenRoute = ApiImageGenRouteImport.update({
   id: '/api/image-gen',
   path: '/api/image-gen',
@@ -144,13 +157,15 @@ export interface FileRoutesByFullPath {
   '/image-gen': typeof ImageGenRoute
   '/realtime': typeof RealtimeRoute
   '/api/image-gen': typeof ApiImageGenRoute
+  '/api/structured-chat': typeof ApiStructuredChatRoute
   '/api/structured-output': typeof ApiStructuredOutputRoute
   '/api/summarize': typeof ApiSummarizeRoute
   '/api/tanchat': typeof ApiTanchatRoute
   '/api/transcribe': typeof ApiTranscribeRoute
   '/generations/audio': typeof GenerationsAudioRoute
   '/generations/image': typeof GenerationsImageRoute
   '/generations/speech': typeof GenerationsSpeechRoute
+  '/generations/structured-chat': typeof GenerationsStructuredChatRoute
   '/generations/structured-output': typeof GenerationsStructuredOutputRoute
   '/generations/summarize': typeof GenerationsSummarizeRoute
   '/generations/transcription': typeof GenerationsTranscriptionRoute
@@ -167,13 +182,15 @@ export interface FileRoutesByTo {
   '/image-gen': typeof ImageGenRoute
   '/realtime': typeof RealtimeRoute
   '/api/image-gen': typeof ApiImageGenRoute
+  '/api/structured-chat': typeof ApiStructuredChatRoute
   '/api/structured-output': typeof ApiStructuredOutputRoute
   '/api/summarize': typeof ApiSummarizeRoute
   '/api/tanchat': typeof ApiTanchatRoute
   '/api/transcribe': typeof ApiTranscribeRoute
   '/generations/audio': typeof GenerationsAudioRoute
   '/generations/image': typeof GenerationsImageRoute
   '/generations/speech': typeof GenerationsSpeechRoute
+  '/generations/structured-chat': typeof GenerationsStructuredChatRoute
   '/generations/structured-output': typeof GenerationsStructuredOutputRoute
   '/generations/summarize': typeof GenerationsSummarizeRoute
   '/generations/transcription': typeof GenerationsTranscriptionRoute
@@ -191,13 +208,15 @@ export interface FileRoutesById {
   '/image-gen': typeof ImageGenRoute
   '/realtime': typeof RealtimeRoute
   '/api/image-gen': typeof ApiImageGenRoute
+  '/api/structured-chat': typeof ApiStructuredChatRoute
   '/api/structured-output': typeof ApiStructuredOutputRoute
   '/api/summarize': typeof ApiSummarizeRoute
   '/api/tanchat': typeof ApiTanchatRoute
   '/api/transcribe': typeof ApiTranscribeRoute
   '/generations/audio': typeof GenerationsAudioRoute
   '/generations/image': typeof GenerationsImageRoute
   '/generations/speech': typeof GenerationsSpeechRoute
+  '/generations/structured-chat': typeof GenerationsStructuredChatRoute
   '/generations/structured-output': typeof GenerationsStructuredOutputRoute
   '/generations/summarize': typeof GenerationsSummarizeRoute
   '/generations/transcription': typeof GenerationsTranscriptionRoute
@@ -216,13 +235,15 @@ export interface FileRouteTypes {
     | '/image-gen'
     | '/realtime'
     | '/api/image-gen'
+    | '/api/structured-chat'
     | '/api/structured-output'
     | '/api/summarize'
     | '/api/tanchat'
     | '/api/transcribe'
     | '/generations/audio'
     | '/generations/image'
     | '/generations/speech'
+    | '/generations/structured-chat'
     | '/generations/structured-output'
     | '/generations/summarize'
     | '/generations/transcription'
@@ -239,13 +260,15 @@ export interface FileRouteTypes {
     | '/image-gen'
     | '/realtime'
     | '/api/image-gen'
+    | '/api/structured-chat'
     | '/api/structured-output'
     | '/api/summarize'
     | '/api/tanchat'
     | '/api/transcribe'
     | '/generations/audio'
     | '/generations/image'
     | '/generations/speech'
+    | '/generations/structured-chat'
     | '/generations/structured-output'
     | '/generations/summarize'
     | '/generations/transcription'
@@ -262,13 +285,15 @@ export interface FileRouteTypes {
     | '/image-gen'
     | '/realtime'
     | '/api/image-gen'
+    | '/api/structured-chat'
     | '/api/structured-output'
     | '/api/summarize'
     | '/api/tanchat'
     | '/api/transcribe'
     | '/generations/audio'
     | '/generations/image'
     | '/generations/speech'
+    | '/generations/structured-chat'
     | '/generations/structured-output'
     | '/generations/summarize'
     | '/generations/transcription'
@@ -286,13 +311,15 @@ export interface RootRouteChildren {
   ImageGenRoute: typeof ImageGenRoute
   RealtimeRoute: typeof RealtimeRoute
   ApiImageGenRoute: typeof ApiImageGenRoute
+  ApiStructuredChatRoute: typeof ApiStructuredChatRoute
   ApiStructuredOutputRoute: typeof ApiStructuredOutputRoute
   ApiSummarizeRoute: typeof ApiSummarizeRoute
   ApiTanchatRoute: typeof ApiTanchatRoute
   ApiTranscribeRoute: typeof ApiTranscribeRoute
   GenerationsAudioRoute: typeof GenerationsAudioRoute
   GenerationsImageRoute: typeof GenerationsImageRoute
   GenerationsSpeechRoute: typeof GenerationsSpeechRoute
+  GenerationsStructuredChatRoute: typeof GenerationsStructuredChatRoute
   GenerationsStructuredOutputRoute: typeof GenerationsStructuredOutputRoute
   GenerationsSummarizeRoute: typeof GenerationsSummarizeRoute
   GenerationsTranscriptionRoute: typeof GenerationsTranscriptionRoute
@@ -356,6 +383,13 @@ declare module '@tanstack/react-router' {
       preLoaderRoute: typeof GenerationsStructuredOutputRouteImport
       parentRoute: typeof rootRouteImport
     }
+    '/generations/structured-chat': {
+      id: '/generations/structured-chat'
+      path: '/generations/structured-chat'
+      fullPath: '/generations/structured-chat'
+      preLoaderRoute: typeof GenerationsStructuredChatRouteImport
+      parentRoute: typeof rootRouteImport
+    }
     '/generations/speech': {
       id: '/generations/speech'
       path: '/generations/speech'
@@ -405,6 +439,13 @@ declare module '@tanstack/react-router' {
       preLoaderRoute: typeof ApiStructuredOutputRouteImport
       parentRoute: typeof rootRouteImport
     }
+    '/api/structured-chat': {
+      id: '/api/structured-chat'
+      path: '/api/structured-chat'
+      fullPath: '/api/structured-chat'
+      preLoaderRoute: typeof ApiStructuredChatRouteImport
+      parentRoute: typeof rootRouteImport
+    }
     '/api/image-gen': {
       id: '/api/image-gen'
       path: '/api/image-gen'
@@ -462,13 +503,15 @@ const rootRouteChildren: RootRouteChildren = {
   ImageGenRoute: ImageGenRoute,
   RealtimeRoute: RealtimeRoute,
   ApiImageGenRoute: ApiImageGenRoute,
+  ApiStructuredChatRoute: ApiStructuredChatRoute,
   ApiStructuredOutputRoute: ApiStructuredOutputRoute,
   ApiSummarizeRoute: ApiSummarizeRoute,
   ApiTanchatRoute: ApiTanchatRoute,
   ApiTranscribeRoute: ApiTranscribeRoute,
   GenerationsAudioRoute: GenerationsAudioRoute,
   GenerationsImageRoute: GenerationsImageRoute,
   GenerationsSpeechRoute: GenerationsSpeechRoute,
+  GenerationsStructuredChatRoute: GenerationsStructuredChatRoute,
   GenerationsStructuredOutputRoute: GenerationsStructuredOutputRoute,
   GenerationsSummarizeRoute: GenerationsSummarizeRoute,
   GenerationsTranscriptionRoute: GenerationsTranscriptionRoute,

examples/ts-react-chat/src/routes/api.structured-chat.ts

@@ -0,0 +1,87 @@
+import { createFileRoute } from '@tanstack/react-router'
+import {
+  chat,
+  chatParamsFromRequestBody,
+  toServerSentEventsResponse,
+} from '@tanstack/ai'
+import { openaiText } from '@tanstack/ai-openai'
+import { z } from 'zod'
+import type { StreamChunk } from '@tanstack/ai'
+
+// Schema shared by every turn in this conversation. The point of this example
+// is to demonstrate that *every* assistant message carries its own typed
+// structured-output part — old turns don't get blown away by new ones.
+export const RecipeSchema = z.object({
+  title: z.string().describe('A short title for the recipe'),
+  cuisine: z.string().describe('Cuisine label, e.g. "Italian", "Mexican"'),
+  servings: z.number().int().min(1).describe('Number of servings'),
+  estimatedCostUsd: z
+    .number()
+    .min(0)
+    .describe('Rough total grocery cost in USD'),
+  ingredients: z
+    .array(
+      z.object({
+        item: z.string(),
+        amount: z.string().describe('Quantity with unit, e.g. "200 g"'),
+      }),
+    )
+    .min(1),
+  steps: z.array(z.string()).min(1).describe('Numbered cooking steps'),
+  tips: z.array(z.string()).default([]),
+})
+
+export type Recipe = z.infer<typeof RecipeSchema>
+
+const SYSTEM_PROMPT = `You are a chef assistant that always responds with a single recipe matching the provided JSON schema. When the user asks for modifications, produce a new recipe in the same shape that reflects the change. Stay terse — short titles, short steps.`
+
+export const Route = createFileRoute('/api/structured-chat')({
+  server: {
+    handlers: {
+      POST: async ({ request }) => {
+        if (request.signal.aborted) {
+          return new Response(null, { status: 499 })
+        }
+
+        const abortController = new AbortController()
+        request.signal.addEventListener('abort', () => abortController.abort())
+
+        let params
+        try {
+          params = await chatParamsFromRequestBody(await request.json())
+        } catch (error) {
+          return new Response(
+            error instanceof Error ? error.message : 'Bad request',
+            { status: 400 },
+          )
+        }
+
+        try {
+          // Cast to AsyncIterable<StreamChunk> for the SSE serializer; the
+          // structured-output stream variant carries the extra `start` /
+          // `complete` custom events that don't appear in the AGUIEvent
+          // union but are valid StreamChunks at runtime.
+          const stream = chat({
+            adapter: openaiText('gpt-4o'),
+            messages: params.messages,
+            systemPrompts: [SYSTEM_PROMPT],
+            outputSchema: RecipeSchema,
+            stream: true,
+            threadId: params.threadId,
+            runId: params.runId,
+            abortController,
+          }) as AsyncIterable<StreamChunk>
+          return toServerSentEventsResponse(stream, { abortController })
+        } catch (error) {
+          const message =
+            error instanceof Error ? error.message : 'An error occurred'
+          console.error('[api/structured-chat] Error:', error)
+          return new Response(JSON.stringify({ error: message }), {
+            status: 500,
+            headers: { 'Content-Type': 'application/json' },
+          })
+        }
+      },
+    },
+  },
+})