@assistant-ui/react 0.4.5 → 0.4.6

package/dist/index.mjs

@@ -29,6 +29,21 @@ function useAssistantContext(options) {
 import { create } from "zustand";
 
 // src/types/ModelConfigTypes.ts
+import { z } from "zod";
+var LanguageModelV1CallSettingsSchema = z.object({
+  maxTokens: z.number().int().positive().optional(),
+  temperature: z.number().optional(),
+  topP: z.number().optional(),
+  presencePenalty: z.number().optional(),
+  frequencyPenalty: z.number().optional(),
+  seed: z.number().int().optional(),
+  headers: z.record(z.string().optional()).optional()
+});
+var LanguageModelConfigSchema = z.object({
+  apiKey: z.string().optional(),
+  baseUrl: z.string().optional(),
+  modelName: z.string().optional()
+});
 var mergeModelConfigs = (configSet) => {
   const configs = Array.from(configSet).map((c) => c.getModelConfig()).sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));
   return configs.reduce((acc, config) => {
@@ -2215,6 +2230,10 @@ var toCoreMessages = (message) => {
       role: message2.role,
       content: message2.content.map((part) => {
         if (part.type === "ui") throw new Error("UI parts are not supported");
+        if (part.type === "tool-call") {
+          const { argsText, ...rest } = part;
+          return rest;
+        }
         return part;
       })
     };
@@ -2222,7 +2241,7 @@ var toCoreMessages = (message) => {
 };
 
 // src/runtimes/edge/converters/toLanguageModelTools.ts
-import { z } from "zod";
+import { z as z2 } from "zod";
 import zodToJsonSchema from "zod-to-json-schema";
 var toLanguageModelTools = (tools) => {
   if (!tools) return [];
@@ -2230,7 +2249,7 @@ var toLanguageModelTools = (tools) => {
     type: "function",
     name,
     ...tool.description ? { description: tool.description } : void 0,
-    parameters: tool.parameters instanceof z.ZodType ? zodToJsonSchema(tool.parameters) : tool.parameters
+    parameters: tool.parameters instanceof z2.ZodType ? zodToJsonSchema(tool.parameters) : tool.parameters
   }));
 };
 
@@ -2807,7 +2826,7 @@ var appendOrUpdateFinish = (message, chunk) => {
 };
 
 // src/runtimes/edge/streams/toolResultStream.ts
-import { z as z2 } from "zod";
+import { z as z3 } from "zod";
 import sjson2 from "secure-json-parse";
 function toolResultStream(tools) {
   const toolCallExecutions = /* @__PURE__ */ new Map();
@@ -2821,12 +2840,16 @@ function toolResultStream(tools) {
           const tool = tools?.[toolName];
           if (!tool || !tool.execute) return;
           const args = sjson2.parse(argsText);
-          if (tool.parameters instanceof z2.ZodType) {
+          if (tool.parameters instanceof z3.ZodType) {
             const result = tool.parameters.safeParse(args);
             if (!result.success) {
               controller.enqueue({
-                type: "error",
-                error: new Error("Invalid tool call arguments")
+                type: "tool-result",
+                toolCallType,
+                toolCallId,
+                toolName,
+                result: "Function parameter validation failed. " + JSON.stringify(result.error.issues),
+                isError: true
               });
               return;
             } else {
@@ -2843,9 +2866,14 @@ function toolResultStream(tools) {
                       result: result2
                     });
                   } catch (error) {
+                    console.error("Error: ", error);
                     controller.enqueue({
-                      type: "error",
-                      error
+                      type: "tool-result",
+                      toolCallType,
+                      toolCallId,
+                      toolName,
+                      result: "Error: " + error,
+                      isError: true
                     });
                   } finally {
                     toolCallExecutions.delete(toolCallId);
@@ -2903,7 +2931,9 @@ var EdgeChatAdapter = class {
         messages: toCoreMessages(messages),
         tools: toLanguageModelTools(
           config.tools
-        )
+        ),
+        ...config.callSettings,
+        ...config.config
       }),
       signal: abortSignal
     });
@@ -3131,7 +3161,7 @@ var fromLanguageModelMessages = (lm, mergeRoundtrips) => {
               toolCallId: part.toolCallId,
               toolName: part.toolName,
               argsText: JSON.stringify(part.args),
-              args: typeof part.args === "string" ? part.args : part.args
+              args: part.args
             };
           }
           return part;
@@ -3271,47 +3301,47 @@ var LocalThreadRuntime = class {
   async startRun(parentId) {
     this.repository.resetHead(parentId);
     const messages = this.repository.getMessages();
-    const message = {
+    let message = {
       id: generateId(),
       role: "assistant",
       status: { type: "in_progress" },
       content: [{ type: "text", text: "" }],
       createdAt: /* @__PURE__ */ new Date()
     };
-    this.repository.addOrUpdateMessage(parentId, { ...message });
     this.abortController?.abort();
     this.abortController = new AbortController();
+    this.repository.addOrUpdateMessage(parentId, { ...message });
     this.notifySubscribers();
-    try {
-      const updateHandler = ({ content }) => {
-        message.content = content;
-        const newMessage = { ...message };
-        this.repository.addOrUpdateMessage(parentId, newMessage);
-        this.notifySubscribers();
-        return newMessage;
+    const updateMessage = (m) => {
+      message = {
+        ...message,
+        ...m
       };
+      this.repository.addOrUpdateMessage(parentId, message);
+      this.notifySubscribers();
+      return message;
+    };
+    try {
       const result = await this.adapter.run({
         messages,
         abortSignal: this.abortController.signal,
         config: this.configProvider.getModelConfig(),
-        onUpdate: updateHandler
+        onUpdate: updateMessage
       });
-      if (result !== void 0) {
-        updateHandler(result);
-      }
       if (result.status?.type === "in_progress")
         throw new Error(
           "Unexpected in_progress status returned from ChatModelAdapter"
         );
-      message.status = result.status ?? { type: "done" };
+      this.abortController = null;
+      updateMessage({ status: { type: "done" }, ...result });
       this.repository.addOrUpdateMessage(parentId, { ...message });
     } catch (e) {
-      message.status = { type: "error", error: e };
-      this.repository.addOrUpdateMessage(parentId, { ...message });
-      throw e;
-    } finally {
+      const isAbortError = e instanceof Error && e.name === "AbortError";
       this.abortController = null;
-      this.notifySubscribers();
+      updateMessage({
+        status: isAbortError ? { type: "cancelled" } : { type: "error", error: e }
+      });
+      if (!isAbortError) throw e;
     }
   }
   cancelRun() {