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.mjs CHANGED
@@ -1104,7 +1104,7 @@ import {
1104
1104
  } from "@ai-sdk/provider-utils";
1105
1105
 
1106
1106
  // src/version.ts
1107
- var VERSION = true ? "6.0.96" : "0.0.0-test";
1107
+ var VERSION = true ? "6.0.97" : "0.0.0-test";
1108
1108
 
1109
1109
  // src/util/download/download.ts
1110
1110
  var download = async ({
@@ -7909,9 +7909,17 @@ var ToolLoopAgent = class {
7909
7909
  }
7910
7910
  async prepareCall(options) {
7911
7911
  var _a21, _b, _c, _d;
7912
- const { onStepFinish: _settingsOnStepFinish, ...settingsWithoutCallback } = this.settings;
7912
+ const {
7913
+ experimental_onStart: _settingsOnStart,
7914
+ experimental_onStepStart: _settingsOnStepStart,
7915
+ experimental_onToolCallStart: _settingsOnToolCallStart,
7916
+ experimental_onToolCallFinish: _settingsOnToolCallFinish,
7917
+ onStepFinish: _settingsOnStepFinish,
7918
+ onFinish: _settingsOnFinish,
7919
+ ...settingsWithoutCallbacks
7920
+ } = this.settings;
7913
7921
  const baseCallArgs = {
7914
- ...settingsWithoutCallback,
7922
+ ...settingsWithoutCallbacks,
7915
7923
  stopWhen: (_a21 = this.settings.stopWhen) != null ? _a21 : stepCountIs(20),
7916
7924
  ...options
7917
7925
  };
@@ -7926,15 +7934,14 @@ var ToolLoopAgent = class {
7926
7934
  ...{ system: instructions, messages, prompt }
7927
7935
  };
7928
7936
  }
7929
- mergeOnStepFinishCallbacks(methodCallback) {
7930
- const constructorCallback = this.settings.onStepFinish;
7931
- if (methodCallback && constructorCallback) {
7932
- return async (stepResult) => {
7933
- await constructorCallback(stepResult);
7934
- await methodCallback(stepResult);
7937
+ mergeCallbacks(settingsCallback, methodCallback) {
7938
+ if (methodCallback && settingsCallback) {
7939
+ return async (event) => {
7940
+ await settingsCallback(event);
7941
+ await methodCallback(event);
7935
7942
  };
7936
7943
  }
7937
- return methodCallback != null ? methodCallback : constructorCallback;
7944
+ return methodCallback != null ? methodCallback : settingsCallback;
7938
7945
  }
7939
7946
  /**
7940
7947
  * Generates an output from the agent (non-streaming).
@@ -7942,14 +7949,39 @@ var ToolLoopAgent = class {
7942
7949
  async generate({
7943
7950
  abortSignal,
7944
7951
  timeout,
7952
+ experimental_onStart,
7953
+ experimental_onStepStart,
7954
+ experimental_onToolCallStart,
7955
+ experimental_onToolCallFinish,
7945
7956
  onStepFinish,
7957
+ onFinish,
7946
7958
  ...options
7947
7959
  }) {
7948
7960
  return generateText({
7949
7961
  ...await this.prepareCall(options),
7950
7962
  abortSignal,
7951
7963
  timeout,
7952
- onStepFinish: this.mergeOnStepFinishCallbacks(onStepFinish)
7964
+ experimental_onStart: this.mergeCallbacks(
7965
+ this.settings.experimental_onStart,
7966
+ experimental_onStart
7967
+ ),
7968
+ experimental_onStepStart: this.mergeCallbacks(
7969
+ this.settings.experimental_onStepStart,
7970
+ experimental_onStepStart
7971
+ ),
7972
+ experimental_onToolCallStart: this.mergeCallbacks(
7973
+ this.settings.experimental_onToolCallStart,
7974
+ experimental_onToolCallStart
7975
+ ),
7976
+ experimental_onToolCallFinish: this.mergeCallbacks(
7977
+ this.settings.experimental_onToolCallFinish,
7978
+ experimental_onToolCallFinish
7979
+ ),
7980
+ onStepFinish: this.mergeCallbacks(
7981
+ this.settings.onStepFinish,
7982
+ onStepFinish
7983
+ ),
7984
+ onFinish: this.mergeCallbacks(this.settings.onFinish, onFinish)
7953
7985
  });
7954
7986
  }
7955
7987
  /**
@@ -7959,7 +7991,12 @@ var ToolLoopAgent = class {
7959
7991
  abortSignal,
7960
7992
  timeout,
7961
7993
  experimental_transform,
7994
+ experimental_onStart,
7995
+ experimental_onStepStart,
7996
+ experimental_onToolCallStart,
7997
+ experimental_onToolCallFinish,
7962
7998
  onStepFinish,
7999
+ onFinish,
7963
8000
  ...options
7964
8001
  }) {
7965
8002
  return streamText({
@@ -7967,7 +8004,27 @@ var ToolLoopAgent = class {
7967
8004
  abortSignal,
7968
8005
  timeout,
7969
8006
  experimental_transform,
7970
- onStepFinish: this.mergeOnStepFinishCallbacks(onStepFinish)
8007
+ experimental_onStart: this.mergeCallbacks(
8008
+ this.settings.experimental_onStart,
8009
+ experimental_onStart
8010
+ ),
8011
+ experimental_onStepStart: this.mergeCallbacks(
8012
+ this.settings.experimental_onStepStart,
8013
+ experimental_onStepStart
8014
+ ),
8015
+ experimental_onToolCallStart: this.mergeCallbacks(
8016
+ this.settings.experimental_onToolCallStart,
8017
+ experimental_onToolCallStart
8018
+ ),
8019
+ experimental_onToolCallFinish: this.mergeCallbacks(
8020
+ this.settings.experimental_onToolCallFinish,
8021
+ experimental_onToolCallFinish
8022
+ ),
8023
+ onStepFinish: this.mergeCallbacks(
8024
+ this.settings.onStepFinish,
8025
+ onStepFinish
8026
+ ),
8027
+ onFinish: this.mergeCallbacks(this.settings.onFinish, onFinish)
7971
8028
  });
7972
8029
  }
7973
8030
  };