@tanstack/ai-svelte 0.8.0 → 0.10.0

package/dist/create-chat.svelte.js

@@ -1,5 +1,4 @@
 import { ChatClient } from '@tanstack/ai-client';
-import { parsePartialJSON } from '@tanstack/ai';
 /**
  * Creates a reactive chat instance for Svelte 5.
  *
@@ -42,10 +41,6 @@ export function createChat(options) {
     let isSubscribed = $state(false);
     let connectionStatus = $state('disconnected');
     let sessionGenerating = $state(false);
-    let partial = $state({});
-    let final = $state(null);
-    // Raw JSON accumulator — not reactive, just a local buffer.
-    let rawJson = '';
     // Create ChatClient instance.
     // Note: Svelte's createChat runs once per instance and `options` is captured
     // by reference. Callbacks are therefore frozen to whatever the caller passed
@@ -56,27 +51,9 @@ export function createChat(options) {
         id: clientId,
         initialMessages: options.initialMessages,
         body: options.body,
+        forwardedProps: options.forwardedProps,
         onResponse: options.onResponse,
         onChunk: (chunk) => {
-            if (options.outputSchema !== undefined) {
-                if (chunk.type === 'RUN_STARTED') {
-                    rawJson = '';
-                    partial = {};
-                    final = null;
-                }
-                else if (chunk.type === 'TEXT_MESSAGE_CONTENT' && chunk.delta) {
-                    rawJson += chunk.delta;
-                    const progressive = parsePartialJSON(rawJson);
-                    if (progressive && typeof progressive === 'object') {
-                        partial = progressive;
-                    }
-                }
-                else if (chunk.type === 'CUSTOM' &&
-                    chunk.name === 'structured-output.complete') {
-                    const value = chunk.value;
-                    final = value.object;
-                }
-            }
             options.onChunk?.(chunk);
         },
         onFinish: (message) => {
@@ -142,9 +119,51 @@ export function createChat(options) {
     const addToolApprovalResponse = async (response) => {
         await client.addToolApprovalResponse(response);
     };
+    /**
+     * @deprecated Use `updateForwardedProps` instead.
+     * Both populate the same wire payload.
+     */
     const updateBody = (newBody) => {
         client.updateOptions({ body: newBody });
     };
+    const updateForwardedProps = (newForwardedProps) => {
+        client.updateOptions({ forwardedProps: newForwardedProps });
+    };
+    // The "active" structured-output part is the one on the assistant message
+    // after the latest user message. When no user message exists yet (e.g.
+    // `initialMessages` carries only a stale assistant turn), we return null
+    // rather than scanning history — otherwise a `final` from a previous
+    // session would leak in on first render. Uses `$derived.by` so the
+    // multi-line scan re-runs whenever `messages` changes.
+    const activeStructuredPart = $derived.by(() => {
+        let lastUserIndex = -1;
+        for (let i = messages.length - 1; i >= 0; i--) {
+            if (messages[i].role === 'user') {
+                lastUserIndex = i;
+                break;
+            }
+        }
+        if (lastUserIndex === -1)
+            return null;
+        for (let i = messages.length - 1; i > lastUserIndex; i--) {
+            const m = messages[i];
+            if (m.role !== 'assistant')
+                continue;
+            const part = m.parts.find((p) => p.type === 'structured-output');
+            if (part)
+                return part;
+        }
+        return null;
+    });
+    const partial = $derived.by(() => {
+        if (!activeStructuredPart)
+            return {};
+        const v = activeStructuredPart.partial ?? activeStructuredPart.data;
+        return (v ?? {});
+    });
+    const final = $derived(activeStructuredPart && activeStructuredPart.status === 'complete'
+        ? activeStructuredPart.data
+        : null);
     // Return the chat interface with reactive getters
     // Using getters allows Svelte to track reactivity without needing $ prefix
     return {
@@ -184,5 +203,6 @@ export function createChat(options) {
         addToolResult,
         addToolApprovalResponse,
         updateBody,
+        updateForwardedProps,
     };
 }

package/dist/types.d.ts

@@ -43,24 +43,27 @@ export type CreateChatOptions<TTools extends ReadonlyArray<AnyClientTool> = any,
  * typed `partial` / `final` reactive getters; otherwise the return is
  * unchanged.
  */
-export type CreateChatReturn<TTools extends ReadonlyArray<AnyClientTool> = any, TSchema extends SchemaInput | undefined = undefined> = BaseCreateChatReturn<TTools> & (TSchema extends SchemaInput ? {
+export type CreateChatReturn<TTools extends ReadonlyArray<AnyClientTool> = any, TSchema extends SchemaInput | undefined = undefined> = BaseCreateChatReturn<TTools, TSchema extends SchemaInput ? InferSchemaType<TSchema> : unknown> & (TSchema extends SchemaInput ? {
     /**
      * Live progressively-parsed structured output (reactive getter).
-     * Resets on every new run.
+     * Derived from the latest assistant message's structured-output part.
      */
     readonly partial: DeepPartial<InferSchemaType<TSchema>>;
     /**
      * Final, schema-validated structured output (reactive getter). `null`
-     * until the terminal `structured-output.complete` event arrives.
-     * Resets on every new run.
+     * until the latest assistant turn's structured-output part transitions
+     * to `complete`.
      */
     readonly final: InferSchemaType<TSchema> | null;
 } : Record<never, never>);
-interface BaseCreateChatReturn<TTools extends ReadonlyArray<AnyClientTool> = any> {
+interface BaseCreateChatReturn<TTools extends ReadonlyArray<AnyClientTool> = any, TData = unknown> {
     /**
-     * Current messages in the conversation (reactive getter)
+     * Current messages in the conversation (reactive getter). When
+     * `outputSchema` is supplied, `messages[i].parts.find(p => p.type ===
+     * 'structured-output')` is typed by the schema — `data: T`,
+     * `partial: DeepPartial<T>`.
      */
-    readonly messages: Array<UIMessage<TTools>>;
+    readonly messages: Array<UIMessage<TTools, TData>>;
     /**
      * Send a message and get a response.
      * Can be a simple string or multimodal content with images, audio, etc.
@@ -69,7 +72,7 @@ interface BaseCreateChatReturn<TTools extends ReadonlyArray<AnyClientTool> = any
     /**
      * Append a message to the conversation
      */
-    append: (message: ModelMessage | UIMessage<TTools>) => Promise<void>;
+    append: (message: ModelMessage | UIMessage<TTools, TData>) => Promise<void>;
     /**
      * Add the result of a client-side tool execution
      */
@@ -106,7 +109,7 @@ interface BaseCreateChatReturn<TTools extends ReadonlyArray<AnyClientTool> = any
     /**
      * Set messages manually
      */
-    setMessages: (messages: Array<UIMessage<TTools>>) => void;
+    setMessages: (messages: Array<UIMessage<TTools, TData>>) => void;
     /**
      * Clear all messages
      */
@@ -131,7 +134,13 @@ interface BaseCreateChatReturn<TTools extends ReadonlyArray<AnyClientTool> = any
      */
     readonly sessionGenerating: boolean;
     /**
-     * Update the body sent with requests (e.g., for changing model selection)
+     * @deprecated Use `updateForwardedProps` instead. Both populate the
+     * same wire payload; `updateBody` is retained for backward compatibility.
      */
     updateBody: (body: Record<string, any>) => void;
+    /**
+     * Update the AG-UI `forwardedProps` sent with requests (e.g., for
+     * changing model selection or other client-driven options).
+     */
+    updateForwardedProps: (forwardedProps: Record<string, any>) => void;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-svelte",
-  "version": "0.8.0",
+  "version": "0.10.0",
   "description": "Svelte bindings for TanStack AI",
   "author": "",
   "license": "MIT",
@@ -33,11 +33,11 @@
     "streaming"
   ],
   "dependencies": {
-    "@tanstack/ai-client": "0.9.2"
+    "@tanstack/ai-client": "0.11.0"
   },
   "peerDependencies": {
     "svelte": "^5.0.0",
-    "@tanstack/ai": "^0.17.0"
+    "@tanstack/ai": "^0.19.0"
   },
   "devDependencies": {
     "@standard-schema/spec": "^1.1.0",
@@ -49,8 +49,8 @@
     "svelte": "^5.20.0",
     "svelte-check": "^4.2.0",
     "typescript": "5.9.3",
-    "vite": "^7.2.7",
-    "@tanstack/ai": "0.17.0"
+    "vite": "^7.3.3",
+    "@tanstack/ai": "0.19.0"
   },
   "scripts": {
     "clean": "rm -rf ./dist",

package/src/create-chat.svelte.ts

@@ -1,6 +1,9 @@
 import { ChatClient } from '@tanstack/ai-client'
-import { parsePartialJSON } from '@tanstack/ai'
-import type { ChatClientState, ConnectionStatus } from '@tanstack/ai-client'
+import type {
+  ChatClientState,
+  ConnectionStatus,
+  StructuredOutputPart,
+} from '@tanstack/ai-client'
 import type {
   AnyClientTool,
   InferSchemaType,
@@ -66,14 +69,12 @@ export function createChat<
   let connectionStatus = $state<ConnectionStatus>('disconnected')
   let sessionGenerating = $state(false)
 
-  // Structured-output state. Runtime always tracks them — the conditional
-  // return type hides them when no `outputSchema` is supplied.
+  // Structured-output `partial` / `final` are derived from `messages` —
+  // specifically from the structured-output part on the latest assistant
+  // message (the one after the most recent user message). Per-turn parts
+  // keep history coherent without a separate reset signal.
   type Partial = DeepPartial<InferSchemaType<NonNullable<TSchema>>>
   type Final = InferSchemaType<NonNullable<TSchema>>
-  let partial = $state<Partial>({} as Partial)
-  let final = $state<Final | null>(null)
-  // Raw JSON accumulator — not reactive, just a local buffer.
-  let rawJson = ''
 
   // Create ChatClient instance.
   // Note: Svelte's createChat runs once per instance and `options` is captured
@@ -85,27 +86,9 @@ export function createChat<
     id: clientId,
     initialMessages: options.initialMessages,
     body: options.body,
+    forwardedProps: options.forwardedProps,
     onResponse: options.onResponse,
     onChunk: (chunk: StreamChunk) => {
-      if (options.outputSchema !== undefined) {
-        if (chunk.type === 'RUN_STARTED') {
-          rawJson = ''
-          partial = {} as Partial
-          final = null
-        } else if (chunk.type === 'TEXT_MESSAGE_CONTENT' && chunk.delta) {
-          rawJson += chunk.delta
-          const progressive = parsePartialJSON(rawJson)
-          if (progressive && typeof progressive === 'object') {
-            partial = progressive as Partial
-          }
-        } else if (
-          chunk.type === 'CUSTOM' &&
-          chunk.name === 'structured-output.complete'
-        ) {
-          const value = chunk.value as { object: unknown }
-          final = value.object as Final
-        }
-      }
       options.onChunk?.(chunk)
     },
     onFinish: (message) => {
@@ -191,10 +174,56 @@ export function createChat<
     await client.addToolApprovalResponse(response)
   }
 
+  /**
+   * @deprecated Use `updateForwardedProps` instead.
+   * Both populate the same wire payload.
+   */
   const updateBody = (newBody: Record<string, any>) => {
     client.updateOptions({ body: newBody })
   }
 
+  const updateForwardedProps = (newForwardedProps: Record<string, any>) => {
+    client.updateOptions({ forwardedProps: newForwardedProps })
+  }
+
+  // The "active" structured-output part is the one on the assistant message
+  // after the latest user message. When no user message exists yet (e.g.
+  // `initialMessages` carries only a stale assistant turn), we return null
+  // rather than scanning history — otherwise a `final` from a previous
+  // session would leak in on first render. Uses `$derived.by` so the
+  // multi-line scan re-runs whenever `messages` changes.
+  const activeStructuredPart: StructuredOutputPart | null = $derived.by(() => {
+    let lastUserIndex = -1
+    for (let i = messages.length - 1; i >= 0; i--) {
+      if (messages[i]!.role === 'user') {
+        lastUserIndex = i
+        break
+      }
+    }
+    if (lastUserIndex === -1) return null
+    for (let i = messages.length - 1; i > lastUserIndex; i--) {
+      const m = messages[i]!
+      if (m.role !== 'assistant') continue
+      const part = m.parts.find(
+        (p): p is StructuredOutputPart => p.type === 'structured-output',
+      )
+      if (part) return part
+    }
+    return null
+  })
+
+  const partial: Partial = $derived.by(() => {
+    if (!activeStructuredPart) return {} as Partial
+    const v = activeStructuredPart.partial ?? activeStructuredPart.data
+    return (v ?? {}) as Partial
+  })
+
+  const final: Final | null = $derived(
+    activeStructuredPart && activeStructuredPart.status === 'complete'
+      ? (activeStructuredPart.data as Final)
+      : null,
+  )
+
   // Return the chat interface with reactive getters
   // Using getters allows Svelte to track reactivity without needing $ prefix
   return {
@@ -234,5 +263,6 @@ export function createChat<
     addToolResult,
     addToolApprovalResponse,
     updateBody,
+    updateForwardedProps,
   } as unknown as CreateChatReturn<TTools, TSchema>
 }

package/src/types.ts

@@ -78,18 +78,21 @@ export type CreateChatOptions<
 export type CreateChatReturn<
   TTools extends ReadonlyArray<AnyClientTool> = any,
   TSchema extends SchemaInput | undefined = undefined,
-> = BaseCreateChatReturn<TTools> &
+> = BaseCreateChatReturn<
+  TTools,
+  TSchema extends SchemaInput ? InferSchemaType<TSchema> : unknown
+> &
   (TSchema extends SchemaInput
     ? {
         /**
          * Live progressively-parsed structured output (reactive getter).
-         * Resets on every new run.
+         * Derived from the latest assistant message's structured-output part.
          */
         readonly partial: DeepPartial<InferSchemaType<TSchema>>
         /**
          * Final, schema-validated structured output (reactive getter). `null`
-         * until the terminal `structured-output.complete` event arrives.
-         * Resets on every new run.
+         * until the latest assistant turn's structured-output part transitions
+         * to `complete`.
          */
         readonly final: InferSchemaType<TSchema> | null
       }
@@ -97,11 +100,15 @@ export type CreateChatReturn<
 
 interface BaseCreateChatReturn<
   TTools extends ReadonlyArray<AnyClientTool> = any,
+  TData = unknown,
 > {
   /**
-   * Current messages in the conversation (reactive getter)
+   * Current messages in the conversation (reactive getter). When
+   * `outputSchema` is supplied, `messages[i].parts.find(p => p.type ===
+   * 'structured-output')` is typed by the schema — `data: T`,
+   * `partial: DeepPartial<T>`.
    */
-  readonly messages: Array<UIMessage<TTools>>
+  readonly messages: Array<UIMessage<TTools, TData>>
 
   /**
    * Send a message and get a response.
@@ -112,7 +119,7 @@ interface BaseCreateChatReturn<
   /**
    * Append a message to the conversation
    */
-  append: (message: ModelMessage | UIMessage<TTools>) => Promise<void>
+  append: (message: ModelMessage | UIMessage<TTools, TData>) => Promise<void>
 
   /**
    * Add the result of a client-side tool execution
@@ -156,7 +163,7 @@ interface BaseCreateChatReturn<
   /**
    * Set messages manually
    */
-  setMessages: (messages: Array<UIMessage<TTools>>) => void
+  setMessages: (messages: Array<UIMessage<TTools, TData>>) => void
 
   /**
    * Clear all messages
@@ -183,9 +190,15 @@ interface BaseCreateChatReturn<
    */
   readonly sessionGenerating: boolean
   /**
-   * Update the body sent with requests (e.g., for changing model selection)
+   * @deprecated Use `updateForwardedProps` instead. Both populate the
+   * same wire payload; `updateBody` is retained for backward compatibility.
    */
   updateBody: (body: Record<string, any>) => void
+  /**
+   * Update the AG-UI `forwardedProps` sent with requests (e.g., for
+   * changing model selection or other client-driven options).
+   */
+  updateForwardedProps: (forwardedProps: Record<string, any>) => void
 }
 
 // Note: createChatClientOptions and InferChatMessages are now in @tanstack/ai-client