ai 3.1.8 → 3.1.10

package/svelte/dist/index.d.ts

@@ -42,6 +42,38 @@ interface ToolResult<NAME extends string, ARGS, RESULT> {
     result: RESULT;
 }
 
+type AssistantStatus = 'in_progress' | 'awaiting_message';
+type UseAssistantOptions = {
+    /**
+     * The API endpoint that accepts a `{ threadId: string | null; message: string; }` object and returns an `AssistantResponse` stream.
+     * The threadId refers to an existing thread with messages (or is `null` to create a new thread).
+     * The message is the next message that should be appended to the thread and sent to the assistant.
+     */
+    api: string;
+    /**
+     * An optional string that represents the ID of an existing thread.
+     * If not provided, a new thread will be created.
+     */
+    threadId?: string;
+    /**
+     * An optional literal that sets the mode of credentials to be used on the request.
+     * Defaults to "same-origin".
+     */
+    credentials?: RequestCredentials;
+    /**
+     * An optional object of headers to be passed to the API endpoint.
+     */
+    headers?: Record<string, string> | Headers;
+    /**
+     * An optional, additional body object to be passed to the API endpoint.
+     */
+    body?: object;
+    /**
+     * An optional callback that will be called when the assistant encounters an error.
+     */
+    onError?: (error: Error) => void;
+};
+
 interface FunctionCall {
     /**
      * The arguments to call the function with, as generated by the model in JSON
@@ -402,4 +434,51 @@ type UseCompletionHelpers = {
 };
 declare function useCompletion({ api, id, initialCompletion, initialInput, credentials, headers, body, streamMode, onResponse, onFinish, onError, }?: UseCompletionOptions): UseCompletionHelpers;
 
-export { CreateMessage, Message, UseChatHelpers, UseChatOptions, UseCompletionHelpers, UseCompletionOptions, useChat, useCompletion };
+type UseAssistantHelpers = {
+    /**
+     * The current array of chat messages.
+     */
+    messages: Readable<Message[]>;
+    /**
+     * Update the message store with a new array of messages.
+     */
+    setMessages: (messages: Message[]) => void;
+    /**
+     * The current thread ID.
+     */
+    threadId: Readable<string | undefined>;
+    /**
+     * The current value of the input field.
+     */
+    input: Writable<string>;
+    /**
+     * Append a user message to the chat list. This triggers the API call to fetch
+     * the assistant's response.
+     * @param message The message to append
+     * @param requestOptions Additional options to pass to the API call
+     */
+    append: (message: Message | CreateMessage, requestOptions?: {
+        data?: Record<string, string>;
+    }) => Promise<void>;
+    /**
+  Abort the current request immediately, keep the generated tokens if any.
+     */
+    stop: () => void;
+    /**
+     * Form submission handler that automatically resets the input field and appends a user message.
+     */
+    submitMessage: (e: any, requestOptions?: {
+        data?: Record<string, string>;
+    }) => Promise<void>;
+    /**
+     * The current status of the assistant. This can be used to show a loading indicator.
+     */
+    status: Readable<AssistantStatus>;
+    /**
+     * The error thrown during the assistant message processing, if any.
+     */
+    error: Readable<undefined | Error>;
+};
+declare function useAssistant({ api, threadId: threadIdParam, credentials, headers, body, onError, }: UseAssistantOptions): UseAssistantHelpers;
+
+export { CreateMessage, Message, UseAssistantHelpers, UseChatHelpers, UseChatOptions, UseCompletionHelpers, UseCompletionOptions, useAssistant, useChat, useCompletion };

package/svelte/dist/index.js

@@ -20,6 +20,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // svelte/index.ts
 var svelte_exports = {};
 __export(svelte_exports, {
+  useAssistant: () => useAssistant,
   useChat: () => useChat,
   useCompletion: () => useCompletion
 });
@@ -1618,8 +1619,159 @@ function useCompletion({
     data: streamData
   };
 }
+
+// svelte/use-assistant.ts
+var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_store3 = require("svelte/store");
+var uniqueId3 = 0;
+var store3 = {};
+function useAssistant({
+  api,
+  threadId: threadIdParam,
+  credentials,
+  headers,
+  body,
+  onError
+}) {
+  const threadIdStore = (0, import_store3.writable)(threadIdParam);
+  const key = `${api}|${threadIdParam != null ? threadIdParam : `completion-${uniqueId3++}`}`;
+  const messages = (0, import_store3.writable)(store3[key] || []);
+  const input = (0, import_store3.writable)("");
+  const status = (0, import_store3.writable)("awaiting_message");
+  const error = (0, import_store3.writable)(void 0);
+  let abortController = null;
+  const mutateMessages = (newMessages) => {
+    store3[key] = newMessages;
+    messages.set(newMessages);
+  };
+  async function append(message, requestOptions) {
+    var _a, _b, _c, _d;
+    status.set("in_progress");
+    abortController = new AbortController();
+    mutateMessages([
+      ...(0, import_store3.get)(messages),
+      { ...message, id: (_a = message.id) != null ? _a : generateId() }
+    ]);
+    input.set("");
+    try {
+      const result = await fetch(api, {
+        method: "POST",
+        credentials,
+        signal: abortController.signal,
+        headers: { "Content-Type": "application/json", ...headers },
+        body: JSON.stringify({
+          ...body,
+          // always use user-provided threadId when available:
+          threadId: (_b = threadIdParam != null ? threadIdParam : (0, import_store3.get)(threadIdStore)) != null ? _b : null,
+          message: message.content,
+          // optional request data:
+          data: requestOptions == null ? void 0 : requestOptions.data
+        })
+      });
+      if (result.body == null) {
+        throw new Error("The response body is empty.");
+      }
+      for await (const { type, value } of readDataStream(
+        result.body.getReader()
+      )) {
+        switch (type) {
+          case "assistant_message": {
+            mutateMessages([
+              ...(0, import_store3.get)(messages),
+              {
+                id: value.id,
+                role: value.role,
+                content: value.content[0].text.value
+              }
+            ]);
+            break;
+          }
+          case "text": {
+            mutateMessages(
+              (0, import_store3.get)(messages).map((msg, index, array) => {
+                if (index === array.length - 1) {
+                  return { ...msg, content: msg.content + value };
+                }
+                return msg;
+              })
+            );
+            break;
+          }
+          case "data_message": {
+            mutateMessages([
+              ...(0, import_store3.get)(messages),
+              {
+                id: (_c = value.id) != null ? _c : generateId(),
+                role: "data",
+                content: "",
+                data: value.data
+              }
+            ]);
+            break;
+          }
+          case "assistant_control_data": {
+            threadIdStore.set(value.threadId);
+            mutateMessages(
+              (0, import_store3.get)(messages).map((msg, index, array) => {
+                if (index === array.length - 1) {
+                  return { ...msg, id: value.messageId };
+                }
+                return msg;
+              })
+            );
+            break;
+          }
+          case "error": {
+            error.set(new Error(value));
+            break;
+          }
+        }
+      }
+    } catch (err) {
+      if ((0, import_provider_utils.isAbortError)(error) && ((_d = abortController == null ? void 0 : abortController.signal) == null ? void 0 : _d.aborted)) {
+        abortController = null;
+        return;
+      }
+      if (onError && err instanceof Error) {
+        onError(err);
+      }
+      error.set(err);
+    } finally {
+      abortController = null;
+      status.set("awaiting_message");
+    }
+  }
+  function setMessages(messages2) {
+    mutateMessages(messages2);
+  }
+  function stop() {
+    if (abortController) {
+      abortController.abort();
+      abortController = null;
+    }
+  }
+  async function submitMessage(e, requestOptions) {
+    e.preventDefault();
+    const inputValue = (0, import_store3.get)(input);
+    if (!inputValue)
+      return;
+    await append({ role: "user", content: inputValue }, requestOptions);
+  }
+  return {
+    messages,
+    error,
+    threadId: threadIdStore,
+    input,
+    append,
+    submitMessage,
+    status,
+    setMessages,
+    stop
+  };
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  useAssistant,
   useChat,
   useCompletion
 });