ai 6.0.0-beta.64 → 6.0.0-beta.66

package/dist/index.mjs

@@ -775,7 +775,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.64" : "0.0.0-test";
+var VERSION = true ? "6.0.0-beta.66" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -6263,27 +6263,32 @@ var ToolLoopAgent = class {
   get tools() {
     return this.settings.tools;
   }
+  async prepareCall(options) {
+    var _a17, _b, _c, _d;
+    const baseCallArgs = {
+      ...this.settings,
+      stopWhen: (_a17 = this.settings.stopWhen) != null ? _a17 : stepCountIs(20),
+      ...options
+    };
+    const preparedCallArgs = (_d = await ((_c = (_b = this.settings).prepareCall) == null ? void 0 : _c.call(_b, baseCallArgs))) != null ? _d : baseCallArgs;
+    const { instructions, messages, prompt, ...callArgs } = preparedCallArgs;
+    return {
+      ...callArgs,
+      // restore prompt types
+      ...{ system: instructions, messages, prompt }
+    };
+  }
   /**
    * Generates an output from the agent (non-streaming).
    */
   async generate(options) {
-    var _a17;
-    return generateText({
-      ...this.settings,
-      stopWhen: (_a17 = this.settings.stopWhen) != null ? _a17 : stepCountIs(20),
-      ...options
-    });
+    return generateText(await this.prepareCall(options));
   }
   /**
    * Streams an output from the agent (streaming).
    */
-  stream(options) {
-    var _a17;
-    return streamText({
-      ...this.settings,
-      stopWhen: (_a17 = this.settings.stopWhen) != null ? _a17 : stepCountIs(20),
-      ...options
-    });
+  async stream(options) {
+    return streamText(await this.prepareCall(options));
   }
 };
 
@@ -7013,6 +7018,7 @@ async function validateUIMessages({
 async function createAgentUIStream({
   agent,
   messages,
+  options,
   ...uiMessageStreamOptions
 }) {
   const validatedMessages = await validateUIMessages({
@@ -7022,7 +7028,10 @@ async function createAgentUIStream({
   const modelMessages = convertToModelMessages(validatedMessages, {
     tools: agent.tools
   });
-  const result = agent.stream({ prompt: modelMessages });
+  const result = await agent.stream({
+    prompt: modelMessages,
+    options
+  });
   return result.toUIMessageStream(uiMessageStreamOptions);
 }