@ai-sdk/svelte 3.0.0-canary.2 → 3.0.0-canary.21

package/dist/structured-object.svelte.js

@@ -1,5 +1,5 @@
 import { generateId, isAbortError, safeValidateTypes, } from '@ai-sdk/provider-utils';
-import { asSchema, isDeepEqualData, parsePartialJson, } from '@ai-sdk/ui-utils';
+import { asSchema, isDeepEqualData, parsePartialJson, } from 'ai';
 import {} from 'zod';
 import { getStructuredObjectContext, hasStructuredObjectContext, KeyedStructuredObjectStore, } from './structured-object-context.svelte.js';
 export class StructuredObject {
@@ -82,23 +82,23 @@ export class StructuredObject {
             let accumulatedText = '';
             let latestObject = undefined;
             await response.body.pipeThrough(new TextDecoderStream()).pipeTo(new WritableStream({
-                write: chunk => {
+                write: async (chunk) => {
                     if (abortController?.signal.aborted) {
                         throw new DOMException('Stream aborted', 'AbortError');
                     }
                     accumulatedText += chunk;
-                    const { value } = parsePartialJson(accumulatedText);
+                    const { value } = await parsePartialJson(accumulatedText);
                     const currentObject = value;
                     if (!isDeepEqualData(latestObject, currentObject)) {
                         latestObject = currentObject;
                         this.#store.object = currentObject;
                     }
                 },
-                close: () => {
+                close: async () => {
                     this.#store.loading = false;
                     this.#abortController = undefined;
                     if (this.#options.onFinish != null) {
-                        const validationResult = safeValidateTypes({
+                        const validationResult = await safeValidateTypes({
                             value: latestObject,
                             schema: asSchema(this.#options.schema),
                         });

package/dist/tests/chat-synchronization.svelte.d.ts

@@ -3,8 +3,8 @@ type $$ComponentProps = {
     id?: string;
 };
 declare const ChatSynchronization: import("svelte").Component<$$ComponentProps, {
-    chat1: Chat;
-    chat2: Chat;
+    chat1: Chat<unknown>;
+    chat2: Chat<unknown>;
 }, "">;
 type ChatSynchronization = ReturnType<typeof ChatSynchronization>;
 export default ChatSynchronization;

package/dist/tests/structured-object-synchronization.svelte

@@ -1,7 +1,7 @@
 <script lang="ts" generics="RESULT">
   import { createAIContext } from '../context-provider.js';
   import { StructuredObject } from '../structured-object.svelte.js';
-  import type { Schema } from '@ai-sdk/ui-utils';
+  import type { Schema } from 'ai';
   import type { z } from 'zod';
 
   let {

package/dist/tests/structured-object-synchronization.svelte.d.ts

@@ -1,5 +1,5 @@
 import { StructuredObject } from '../structured-object.svelte.js';
-import type { Schema } from '@ai-sdk/ui-utils';
+import type { Schema } from 'ai';
 import type { z } from 'zod';
 declare class __sveltets_Render<RESULT> {
     props(): {

package/dist/tests/structured-object-synchronization.svelte.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"structured-object-synchronization.svelte.d.ts","sourceRoot":"","sources":["../../src/tests/structured-object-synchronization.svelte.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AA2B7B,cAAM,iBAAiB,CAAC,MAAM;IAC1B,KAAK;aArBA,MAAM;aACN,MAAM;;;IAuBX,MAAM;IAGN,KAAK;IAGL,QAAQ;IACR,OAAO;;;;CACV;AAED,UAAU,qBAAqB;IAC3B,KAAK,MAAM,EAAE,OAAO,EAAE,OAAO,QAAQ,EAAE,2BAA2B,CAAC,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;QAAE,UAAU,CAAC,EAAE,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,CAAC,CAAA;KAAE,GAAG,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC/Z,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1I,YAAY,CAAC,EAAE,UAAU,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;CACjE;AACD,QAAA,MAAM,+BAA+B,EAAE,qBAAmC,CAAC;AACzD,KAAK,+BAA+B,CAAC,MAAM,IAAI,YAAY,CAAC,OAAO,+BAA+B,CAAC,MAAM,CAAC,CAAC,CAAC;AAC9G,eAAe,+BAA+B,CAAC"}
+{"version":3,"file":"structured-object-synchronization.svelte.d.ts","sourceRoot":"","sources":["../../src/tests/structured-object-synchronization.svelte.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AACjC,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AA2B7B,cAAM,iBAAiB,CAAC,MAAM;IAC1B,KAAK;aArBA,MAAM;aACN,MAAM;;;IAuBX,MAAM;IAGN,KAAK;IAGL,QAAQ;IACR,OAAO;;;;CACV;AAED,UAAU,qBAAqB;IAC3B,KAAK,MAAM,EAAE,OAAO,EAAE,OAAO,QAAQ,EAAE,2BAA2B,CAAC,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,QAAQ,EAAE,eAAe,CAAC,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;QAAE,UAAU,CAAC,EAAE,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,CAAC,CAAA;KAAE,GAAG,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC/Z,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1I,YAAY,CAAC,EAAE,UAAU,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;CACjE;AACD,QAAA,MAAM,+BAA+B,EAAE,qBAAmC,CAAC;AACzD,KAAK,+BAA+B,CAAC,MAAM,IAAI,YAAY,CAAC,OAAO,+BAA+B,CAAC,MAAM,CAAC,CAAC,CAAC;AAC9G,eAAe,+BAA+B,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/svelte",
-  "version": "3.0.0-canary.2",
+  "version": "3.0.0-canary.21",
   "license": "Apache-2.0",
   "files": [
     "dist",
@@ -36,10 +36,11 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider-utils": "3.0.0-canary.2",
-    "@ai-sdk/ui-utils": "2.0.0-canary.2"
+    "ai": "5.0.0-canary.22",
+    "@ai-sdk/provider-utils": "3.0.0-canary.17"
   },
   "devDependencies": {
+    "@types/node": "20.17.24",
     "@eslint/compat": "^1.2.5",
     "@eslint/js": "^9.18.0",
     "@sveltejs/package": "^2.0.0",
@@ -57,9 +58,10 @@
     "typescript-eslint": "^8.20.0",
     "vite": "^6.0.0",
     "vitest": "^3.0.0",
-    "zod": "3.23.8"
+    "zod": "3.23.8",
+    "@vercel/ai-tsconfig": "0.0.0"
   },
-  "homepage": "https://sdk.vercel.ai/docs",
+  "homepage": "https://ai-sdk.dev/docs",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/vercel/ai.git",
@@ -74,7 +76,7 @@
   ],
   "scripts": {
     "build": "pnpm prepack",
-    "build:watch": "svelte-package --input=src --watch",
+    "build:watch": "pnpm clean && pnpm --filter svelte build && svelte-package --input=src --watch",
     "preview": "vite preview",
     "type-check": "svelte-check --tsconfig ./tsconfig.json",
     "type-check:watch": "svelte-check --tsconfig ./tsconfig.json --watch",
@@ -82,7 +84,8 @@
     "prettier-check": "prettier --check .",
     "lint": "eslint .",
     "test": "vitest --run",
+    "test:update": "vitest --run -u",
     "test:watch": "vitest",
-    "clean": "rm -rf dist"
+    "clean": "rm -rf dist *.tsbuildinfo"
   }
 }

package/src/chat-context.svelte.ts

@@ -1,16 +1,21 @@
-import type { JSONValue, UIMessage } from '@ai-sdk/ui-utils';
+import type { JSONValue, UIMessage } from 'ai';
 import { createContext, KeyedStore } from './utils.svelte.js';
 
-class ChatStore {
-  messages = $state<UIMessage[]>([]);
+class ChatStore<MESSAGE_METADATA = unknown> {
+  messages = $state<UIMessage<MESSAGE_METADATA>[]>([]);
   data = $state<JSONValue[]>();
   status = $state<'submitted' | 'streaming' | 'ready' | 'error'>('ready');
   error = $state<Error>();
 }
 
-export class KeyedChatStore extends KeyedStore<ChatStore> {
+export class KeyedChatStore<MESSAGE_METADATA = unknown> extends KeyedStore<
+  ChatStore<MESSAGE_METADATA>
+> {
   constructor(
-    value?: Iterable<readonly [string, ChatStore]> | null | undefined,
+    value?:
+      | Iterable<readonly [string, ChatStore<MESSAGE_METADATA>]>
+      | null
+      | undefined,
   ) {
     super(ChatStore, value);
   }

package/src/chat.svelte.ts

@@ -1,70 +1,52 @@
+import { isAbortError } from '@ai-sdk/provider-utils';
 import {
-  fillMessageParts,
-  generateId,
-  type UIMessage,
-  type UseChatOptions,
-  type JSONValue,
-  type ChatRequest,
-  extractMaxToolInvocationStep,
   callChatApi,
+  convertFileListToFileUIParts,
+  extractMaxToolInvocationStep,
+  generateId,
+  getToolInvocations,
+  isAssistantMessageWithCompletedToolCalls,
   shouldResubmitMessages,
-  type Message,
-  type CreateMessage,
-  type ChatRequestOptions,
-  prepareAttachmentsForRequest,
-  getMessageParts,
   updateToolCallResult,
-  isAssistantMessageWithCompletedToolCalls,
-} from '@ai-sdk/ui-utils';
-import { isAbortError } from '@ai-sdk/provider-utils';
+  type ChatRequestOptions,
+  type CreateUIMessage,
+  type UIMessage,
+  type UseChatOptions,
+} from 'ai';
+import { untrack } from 'svelte';
 import {
   KeyedChatStore,
   getChatContext,
   hasChatContext,
 } from './chat-context.svelte.js';
-import { untrack } from 'svelte';
 
-export type ChatOptions = Readonly<
-  Omit<UseChatOptions, 'keepLastMessageOnError'> & {
-    /**
-     * Maximum number of sequential LLM calls (steps), e.g. when you use tool calls.
-     * Must be at least 1.
-     * A maximum number is required to prevent infinite loops in the case of misconfigured tools.
-     * By default, it's set to 1, which means that only a single LLM call is made.
-     * @default 1
-     */
-    maxSteps?: number;
-  }
+export type ChatOptions<MESSAGE_METADATA = unknown> = Readonly<
+  UseChatOptions<MESSAGE_METADATA>
 >;
 
-export type { CreateMessage, Message, UIMessage };
+export type { CreateUIMessage, UIMessage };
 
-export class Chat {
-  readonly #options: ChatOptions = {};
+export class Chat<MESSAGE_METADATA = unknown> {
+  readonly #options: ChatOptions<MESSAGE_METADATA> = {};
   readonly #api = $derived(this.#options.api ?? '/api/chat');
   readonly #generateId = $derived(this.#options.generateId ?? generateId);
   readonly #maxSteps = $derived(this.#options.maxSteps ?? 1);
-  readonly #streamProtocol = $derived(this.#options.streamProtocol ?? 'data');
-  readonly #keyedStore = $state<KeyedChatStore>()!;
+  readonly #streamProtocol = $derived(
+    this.#options.streamProtocol ?? 'ui-message',
+  );
+  readonly #keyedStore = $state<KeyedChatStore<MESSAGE_METADATA>>()!;
   /**
    * The id of the chat. If not provided through the constructor, a random ID will be generated
    * using the provided `generateId` function, or a built-in function if not provided.
    */
   readonly id = $derived(this.#options.id ?? this.#generateId());
   readonly #store = $derived(this.#keyedStore.get(this.id));
-  #abortController: AbortController | undefined;
 
-  /**
-   * Additional data added on the server via StreamData.
-   *
-   * This is writable, so you can use it to transform or clear the chat data.
-   */
-  get data() {
-    return this.#store.data;
-  }
-  set data(value: JSONValue[] | undefined) {
-    this.#store.data = value;
-  }
+  readonly #messageMetadataSchema = $derived(
+    this.#options.messageMetadataSchema,
+  );
+
+  #abortController: AbortController | undefined;
 
   /**
    * Hook status:
@@ -92,18 +74,18 @@ export class Chat {
    * This is writable, which is useful when you want to edit the messages on the client, and then
    * trigger {@link reload} to regenerate the AI response.
    */
-  get messages(): UIMessage[] {
+  get messages(): UIMessage<MESSAGE_METADATA>[] {
     return this.#store.messages;
   }
-  set messages(value: Message[]) {
-    untrack(() => (this.#store.messages = fillMessageParts(value)));
+  set messages(value: UIMessage<MESSAGE_METADATA>[]) {
+    untrack(() => (this.#store.messages = value));
   }
 
-  constructor(options: ChatOptions = {}) {
+  constructor(options: ChatOptions<MESSAGE_METADATA> = {}) {
     if (hasChatContext()) {
-      this.#keyedStore = getChatContext();
+      this.#keyedStore = getChatContext() as KeyedChatStore<MESSAGE_METADATA>;
     } else {
-      this.#keyedStore = new KeyedChatStore();
+      this.#keyedStore = new KeyedChatStore<MESSAGE_METADATA>();
     }
 
     this.#options = options;
@@ -118,23 +100,15 @@ export class Chat {
    * @param options Additional options to pass to the API call
    */
   append = async (
-    message: Message | CreateMessage,
-    { data, headers, body, experimental_attachments }: ChatRequestOptions = {},
+    message: UIMessage<MESSAGE_METADATA> | CreateUIMessage<MESSAGE_METADATA>,
+    { headers, body }: ChatRequestOptions = {},
   ) => {
-    const attachmentsForRequest = await prepareAttachmentsForRequest(
-      experimental_attachments,
-    );
-
     const messages = this.messages.concat({
       ...message,
       id: message.id ?? this.#generateId(),
-      createdAt: message.createdAt ?? new Date(),
-      experimental_attachments:
-        attachmentsForRequest.length > 0 ? attachmentsForRequest : undefined,
-      parts: getMessageParts(message),
     });
 
-    return this.#triggerRequest({ messages, headers, body, data });
+    return this.#triggerRequest({ messages, headers, body });
   };
 
   /**
@@ -142,7 +116,7 @@ export class Chat {
    * message isn't from the assistant, it will request the API to generate a
    * new response.
    */
-  reload = async ({ data, headers, body }: ChatRequestOptions = {}) => {
+  reload = async ({ headers, body }: ChatRequestOptions = {}) => {
     if (this.messages.length === 0) {
       return;
     }
@@ -155,7 +129,6 @@ export class Chat {
           : this.messages,
       headers,
       body,
-      data,
     });
   };
 
@@ -176,33 +149,28 @@ export class Chat {
   /** Form submission handler to automatically reset input and append a user message */
   handleSubmit = async (
     event?: { preventDefault?: () => void },
-    options: ChatRequestOptions = {},
+    options: ChatRequestOptions & { files?: FileList } = {},
   ) => {
     event?.preventDefault?.();
-    if (!this.input && !options.allowEmptySubmit) return;
 
-    const attachmentsForRequest = await prepareAttachmentsForRequest(
-      options.experimental_attachments,
-    );
+    const fileParts = Array.isArray(options?.files)
+      ? options.files
+      : await convertFileListToFileUIParts(options?.files);
+
+    if (!this.input && fileParts.length === 0) return;
 
     const messages = this.messages.concat({
       id: this.#generateId(),
-      createdAt: new Date(),
       role: 'user',
-      content: this.input,
-      experimental_attachments:
-        attachmentsForRequest.length > 0 ? attachmentsForRequest : undefined,
-      parts: [{ type: 'text', text: this.input }],
+      parts: [...fileParts, { type: 'text', text: this.input }],
     });
 
-    const chatRequest: ChatRequest = {
+    const request = this.#triggerRequest({
       messages,
       headers: options.headers,
       body: options.body,
-      data: options.data,
-    };
+    });
 
-    const request = this.#triggerRequest(chatRequest);
     this.input = '';
     await request;
   };
@@ -234,14 +202,18 @@ export class Chat {
     }
   };
 
-  #triggerRequest = async (chatRequest: ChatRequest) => {
+  #triggerRequest = async (
+    chatRequest: ChatRequestOptions & {
+      messages: UIMessage<MESSAGE_METADATA>[];
+    },
+  ) => {
     this.#store.status = 'submitted';
     this.#store.error = undefined;
 
-    const messages = fillMessageParts(chatRequest.messages);
+    const messages = chatRequest.messages;
     const messageCount = messages.length;
     const maxStep = extractMaxToolInvocationStep(
-      messages[messages.length - 1]?.toolInvocations,
+      getToolInvocations(messages[messages.length - 1]),
     );
 
     try {
@@ -251,37 +223,11 @@ export class Chat {
       // Optimistically update messages
       this.messages = messages;
 
-      const constructedMessagesPayload = this.#options.sendExtraMessageFields
-        ? messages
-        : messages.map(
-            ({
-              role,
-              content,
-              experimental_attachments,
-              data,
-              annotations,
-              toolInvocations,
-              parts,
-            }) => ({
-              role,
-              content,
-              ...(experimental_attachments !== undefined && {
-                experimental_attachments,
-              }),
-              ...(data !== undefined && { data }),
-              ...(annotations !== undefined && { annotations }),
-              ...(toolInvocations !== undefined && { toolInvocations }),
-              ...(parts !== undefined && { parts }),
-            }),
-          );
-
-      const existingData = this.data ?? [];
       await callChatApi({
         api: this.#api,
         body: {
           id: this.id,
-          messages: constructedMessagesPayload,
-          data: chatRequest.data,
+          messages,
           ...$state.snapshot(this.#options.body),
           ...chatRequest.body,
         },
@@ -292,29 +238,28 @@ export class Chat {
           ...chatRequest.headers,
         },
         abortController: () => abortController,
-        restoreMessagesOnFailure: () => {},
-        onResponse: this.#options.onResponse,
-        onUpdate: ({ message, data, replaceLastMessage }) => {
+        onUpdate: ({ message }) => {
           this.#store.status = 'streaming';
 
+          const replaceLastMessage =
+            message.id === messages[messages.length - 1].id;
+
           this.messages = messages;
           if (replaceLastMessage) {
             this.messages[this.messages.length - 1] = message;
           } else {
             this.messages.push(message);
           }
-
-          if (data?.length) {
-            this.data = existingData;
-            this.data.push(...data);
-          }
         },
         onToolCall: this.#options.onToolCall,
         onFinish: this.#options.onFinish,
         generateId: this.#generateId,
         fetch: this.#options.fetch,
         // callChatApi calls structuredClone on the message
-        lastMessage: $state.snapshot(this.messages[this.messages.length - 1]),
+        lastMessage: $state.snapshot(
+          this.messages[this.messages.length - 1],
+        ) as UIMessage<MESSAGE_METADATA>,
+        messageMetadataSchema: this.#messageMetadataSchema,
       });
 
       this.#abortController = undefined;

package/src/completion-context.svelte.ts

@@ -1,4 +1,4 @@
-import type { JSONValue } from '@ai-sdk/ui-utils';
+import type { JSONValue } from 'ai';
 import { SvelteMap } from 'svelte/reactivity';
 import { createContext, KeyedStore } from './utils.svelte.js';
 

package/src/completion.svelte.ts

@@ -1,10 +1,9 @@
 import {
+  callCompletionApi,
   generateId,
+  type CompletionRequestOptions,
   type UseCompletionOptions,
-  type JSONValue,
-  type RequestOptions,
-  callCompletionApi,
-} from '@ai-sdk/ui-utils';
+} from 'ai';
 import {
   KeyedCompletionStore,
   getCompletionContext,
@@ -30,18 +29,6 @@ export class Completion {
     this.#store.completions.set(this.#id, value);
   }
 
-  /**
-   * Additional data added on the server via StreamData.
-   *
-   * This is writable, so you can use it to transform or clear the chat data.
-   */
-  get data() {
-    return this.#store.data;
-  }
-  set data(value: JSONValue[]) {
-    this.#store.data = value;
-  }
-
   /** The error object of the API request */
   get error() {
     return this.#store.error;
@@ -58,12 +45,9 @@ export class Completion {
   }
 
   constructor(options: CompletionOptions = {}) {
-    if (hasCompletionContext()) {
-      this.#keyedStore = getCompletionContext();
-    } else {
-      this.#keyedStore = new KeyedCompletionStore();
-    }
-
+    this.#keyedStore = hasCompletionContext()
+      ? getCompletionContext()
+      : new KeyedCompletionStore();
     this.#options = options;
     this.completion = options.initialCompletion ?? '';
     this.input = options.initialInput ?? '';
@@ -86,7 +70,7 @@ export class Completion {
   /**
    * Send a new prompt to the API endpoint and update the completion state.
    */
-  complete = async (prompt: string, options?: RequestOptions) =>
+  complete = async (prompt: string, options?: CompletionRequestOptions) =>
     this.#triggerRequest(prompt, options);
 
   /** Form submission handler to automatically reset input and call the completion API */
@@ -97,7 +81,10 @@ export class Completion {
     }
   };
 
-  #triggerRequest = async (prompt: string, options?: RequestOptions) => {
+  #triggerRequest = async (
+    prompt: string,
+    options?: CompletionRequestOptions,
+  ) => {
     return callCompletionApi({
       api: this.#api,
       prompt,
@@ -113,9 +100,6 @@ export class Completion {
       setCompletion: completion => {
         this.completion = completion;
       },
-      onData: data => {
-        this.data.push(...data);
-      },
       setLoading: loading => {
         this.#store.loading = loading;
       },
@@ -125,7 +109,6 @@ export class Completion {
       setAbortController: abortController => {
         this.#abortController = abortController ?? undefined;
       },
-      onResponse: this.#options.onResponse,
       onFinish: this.#options.onFinish,
       onError: this.#options.onError,
     });

package/src/index.ts

@@ -1,8 +1,7 @@
 export {
   Chat,
   type ChatOptions,
-  type CreateMessage,
-  type Message,
+  type CreateUIMessage,
   type UIMessage,
 } from './chat.svelte.js';
 

package/src/structured-object-context.svelte.ts

@@ -1,4 +1,4 @@
-import type { DeepPartial } from '@ai-sdk/ui-utils';
+import type { DeepPartial } from 'ai';
 import { createContext, KeyedStore } from './utils.svelte.js';
 
 export class StructuredObjectStore<RESULT> {

package/src/structured-object.svelte.ts

@@ -10,7 +10,7 @@ import {
   parsePartialJson,
   type DeepPartial,
   type Schema,
-} from '@ai-sdk/ui-utils';
+} from 'ai';
 import { type z } from 'zod';
 import {
   getStructuredObjectContext,
@@ -177,13 +177,13 @@ export class StructuredObject<RESULT, INPUT = unknown> {
 
       await response.body.pipeThrough(new TextDecoderStream()).pipeTo(
         new WritableStream<string>({
-          write: chunk => {
+          write: async chunk => {
             if (abortController?.signal.aborted) {
               throw new DOMException('Stream aborted', 'AbortError');
             }
             accumulatedText += chunk;
 
-            const { value } = parsePartialJson(accumulatedText);
+            const { value } = await parsePartialJson(accumulatedText);
             const currentObject = value as DeepPartial<RESULT>;
 
             if (!isDeepEqualData(latestObject, currentObject)) {
@@ -193,12 +193,12 @@ export class StructuredObject<RESULT, INPUT = unknown> {
             }
           },
 
-          close: () => {
+          close: async () => {
             this.#store.loading = false;
             this.#abortController = undefined;
 
             if (this.#options.onFinish != null) {
-              const validationResult = safeValidateTypes({
+              const validationResult = await safeValidateTypes({
                 value: latestObject,
                 schema: asSchema(this.#options.schema),
               });

package/src/tests/structured-object-synchronization.svelte

@@ -1,7 +1,7 @@
 <script lang="ts" generics="RESULT">
   import { createAIContext } from '../context-provider.js';
   import { StructuredObject } from '../structured-object.svelte.js';
-  import type { Schema } from '@ai-sdk/ui-utils';
+  import type { Schema } from 'ai';
   import type { z } from 'zod';
 
   let {

package/dist/chat-context.svelte.d.ts

@@ -1,14 +0,0 @@
-import type { JSONValue, UIMessage } from '@ai-sdk/ui-utils';
-import { KeyedStore } from './utils.svelte.js';
-declare class ChatStore {
-    messages: UIMessage[];
-    data: JSONValue[] | undefined;
-    status: "submitted" | "streaming" | "ready" | "error";
-    error: Error | undefined;
-}
-export declare class KeyedChatStore extends KeyedStore<ChatStore> {
-    constructor(value?: Iterable<readonly [string, ChatStore]> | null | undefined);
-}
-export declare const hasChatContext: () => boolean, getChatContext: () => KeyedChatStore, setChatContext: (value: KeyedChatStore) => KeyedChatStore;
-export {};
-//# sourceMappingURL=chat-context.svelte.d.ts.map

package/dist/chat-context.svelte.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"chat-context.svelte.d.ts","sourceRoot":"","sources":["../src/chat-context.svelte.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7D,OAAO,EAAiB,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE9D,cAAM,SAAS;IACb,QAAQ,cAA2B;IACnC,IAAI,0BAAyB;IAC7B,MAAM,gDAAkE;IACxE,KAAK,oBAAmB;CACzB;AAED,qBAAa,cAAe,SAAQ,UAAU,CAAC,SAAS,CAAC;gBAErD,KAAK,CAAC,EAAE,QAAQ,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,GAAG,IAAI,GAAG,SAAS;CAIpE;AAED,eAAO,MACO,cAAc,iBACd,cAAc,wBACd,cAAc,2CACa,CAAC"}

package/dist/completion-context.svelte.d.ts

@@ -1,15 +0,0 @@
-import type { JSONValue } from '@ai-sdk/ui-utils';
-import { SvelteMap } from 'svelte/reactivity';
-import { KeyedStore } from './utils.svelte.js';
-declare class CompletionStore {
-    completions: SvelteMap<string, string>;
-    data: JSONValue[];
-    loading: boolean;
-    error: Error | undefined;
-}
-export declare class KeyedCompletionStore extends KeyedStore<CompletionStore> {
-    constructor(value?: Iterable<readonly [string, CompletionStore]> | null | undefined);
-}
-export declare const hasCompletionContext: () => boolean, getCompletionContext: () => KeyedCompletionStore, setCompletionContext: (value: KeyedCompletionStore) => KeyedCompletionStore;
-export {};
-//# sourceMappingURL=completion-context.svelte.d.ts.map

package/dist/completion-context.svelte.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"completion-context.svelte.d.ts","sourceRoot":"","sources":["../src/completion-context.svelte.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAiB,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE9D,cAAM,eAAe;IACnB,WAAW,4BAAmC;IAC9C,IAAI,cAA2B;IAC/B,OAAO,UAAiB;IACxB,KAAK,oBAAmB;CACzB;AAED,qBAAa,oBAAqB,SAAQ,UAAU,CAAC,eAAe,CAAC;gBAEjE,KAAK,CAAC,EAAE,QAAQ,CAAC,SAAS,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC,GAAG,IAAI,GAAG,SAAS;CAI1E;AAED,eAAO,MACO,oBAAoB,iBACpB,oBAAoB,8BACpB,oBAAoB,uDACmB,CAAC"}