ai 6.0.96 → 6.0.97

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.js CHANGED
@@ -1211,7 +1211,7 @@ var import_provider_utils3 = require("@ai-sdk/provider-utils");
1211
1211
  var import_provider_utils4 = require("@ai-sdk/provider-utils");
1212
1212
 
1213
1213
  // src/version.ts
1214
- var VERSION = true ? "6.0.96" : "0.0.0-test";
1214
+ var VERSION = true ? "6.0.97" : "0.0.0-test";
1215
1215
 
1216
1216
  // src/util/download/download.ts
1217
1217
  var download = async ({
@@ -7985,9 +7985,17 @@ var ToolLoopAgent = class {
7985
7985
  }
7986
7986
  async prepareCall(options) {
7987
7987
  var _a21, _b, _c, _d;
7988
- const { onStepFinish: _settingsOnStepFinish, ...settingsWithoutCallback } = this.settings;
7988
+ const {
7989
+ experimental_onStart: _settingsOnStart,
7990
+ experimental_onStepStart: _settingsOnStepStart,
7991
+ experimental_onToolCallStart: _settingsOnToolCallStart,
7992
+ experimental_onToolCallFinish: _settingsOnToolCallFinish,
7993
+ onStepFinish: _settingsOnStepFinish,
7994
+ onFinish: _settingsOnFinish,
7995
+ ...settingsWithoutCallbacks
7996
+ } = this.settings;
7989
7997
  const baseCallArgs = {
7990
- ...settingsWithoutCallback,
7998
+ ...settingsWithoutCallbacks,
7991
7999
  stopWhen: (_a21 = this.settings.stopWhen) != null ? _a21 : stepCountIs(20),
7992
8000
  ...options
7993
8001
  };
@@ -8002,15 +8010,14 @@ var ToolLoopAgent = class {
8002
8010
  ...{ system: instructions, messages, prompt }
8003
8011
  };
8004
8012
  }
8005
- mergeOnStepFinishCallbacks(methodCallback) {
8006
- const constructorCallback = this.settings.onStepFinish;
8007
- if (methodCallback && constructorCallback) {
8008
- return async (stepResult) => {
8009
- await constructorCallback(stepResult);
8010
- await methodCallback(stepResult);
8013
+ mergeCallbacks(settingsCallback, methodCallback) {
8014
+ if (methodCallback && settingsCallback) {
8015
+ return async (event) => {
8016
+ await settingsCallback(event);
8017
+ await methodCallback(event);
8011
8018
  };
8012
8019
  }
8013
- return methodCallback != null ? methodCallback : constructorCallback;
8020
+ return methodCallback != null ? methodCallback : settingsCallback;
8014
8021
  }
8015
8022
  /**
8016
8023
  * Generates an output from the agent (non-streaming).
@@ -8018,14 +8025,39 @@ var ToolLoopAgent = class {
8018
8025
  async generate({
8019
8026
  abortSignal,
8020
8027
  timeout,
8028
+ experimental_onStart,
8029
+ experimental_onStepStart,
8030
+ experimental_onToolCallStart,
8031
+ experimental_onToolCallFinish,
8021
8032
  onStepFinish,
8033
+ onFinish,
8022
8034
  ...options
8023
8035
  }) {
8024
8036
  return generateText({
8025
8037
  ...await this.prepareCall(options),
8026
8038
  abortSignal,
8027
8039
  timeout,
8028
- onStepFinish: this.mergeOnStepFinishCallbacks(onStepFinish)
8040
+ experimental_onStart: this.mergeCallbacks(
8041
+ this.settings.experimental_onStart,
8042
+ experimental_onStart
8043
+ ),
8044
+ experimental_onStepStart: this.mergeCallbacks(
8045
+ this.settings.experimental_onStepStart,
8046
+ experimental_onStepStart
8047
+ ),
8048
+ experimental_onToolCallStart: this.mergeCallbacks(
8049
+ this.settings.experimental_onToolCallStart,
8050
+ experimental_onToolCallStart
8051
+ ),
8052
+ experimental_onToolCallFinish: this.mergeCallbacks(
8053
+ this.settings.experimental_onToolCallFinish,
8054
+ experimental_onToolCallFinish
8055
+ ),
8056
+ onStepFinish: this.mergeCallbacks(
8057
+ this.settings.onStepFinish,
8058
+ onStepFinish
8059
+ ),
8060
+ onFinish: this.mergeCallbacks(this.settings.onFinish, onFinish)
8029
8061
  });
8030
8062
  }
8031
8063
  /**
@@ -8035,7 +8067,12 @@ var ToolLoopAgent = class {
8035
8067
  abortSignal,
8036
8068
  timeout,
8037
8069
  experimental_transform,
8070
+ experimental_onStart,
8071
+ experimental_onStepStart,
8072
+ experimental_onToolCallStart,
8073
+ experimental_onToolCallFinish,
8038
8074
  onStepFinish,
8075
+ onFinish,
8039
8076
  ...options
8040
8077
  }) {
8041
8078
  return streamText({
@@ -8043,7 +8080,27 @@ var ToolLoopAgent = class {
8043
8080
  abortSignal,
8044
8081
  timeout,
8045
8082
  experimental_transform,
8046
- onStepFinish: this.mergeOnStepFinishCallbacks(onStepFinish)
8083
+ experimental_onStart: this.mergeCallbacks(
8084
+ this.settings.experimental_onStart,
8085
+ experimental_onStart
8086
+ ),
8087
+ experimental_onStepStart: this.mergeCallbacks(
8088
+ this.settings.experimental_onStepStart,
8089
+ experimental_onStepStart
8090
+ ),
8091
+ experimental_onToolCallStart: this.mergeCallbacks(
8092
+ this.settings.experimental_onToolCallStart,
8093
+ experimental_onToolCallStart
8094
+ ),
8095
+ experimental_onToolCallFinish: this.mergeCallbacks(
8096
+ this.settings.experimental_onToolCallFinish,
8097
+ experimental_onToolCallFinish
8098
+ ),
8099
+ onStepFinish: this.mergeCallbacks(
8100
+ this.settings.onStepFinish,
8101
+ onStepFinish
8102
+ ),
8103
+ onFinish: this.mergeCallbacks(this.settings.onFinish, onFinish)
8047
8104
  });
8048
8105
  }
8049
8106
  };