@browserbasehq/stagehand 3.0.6-alpha-75f87f1920021a012fe0159123d01404969a94c8 → 3.0.6-alpha-943d2d79d0f289ac41c9164578f2f1dd876058f2

package/dist/index.d.ts

@@ -5,7 +5,8 @@ import { GoogleVertexProviderSettings as GoogleVertexProviderSettings$1 } from '
 import { LanguageModelV2 } from '@ai-sdk/provider';
 import { ClientOptions as ClientOptions$1 } from 'openai';
 import { Client, ClientOptions as ClientOptions$3 } from '@modelcontextprotocol/sdk/client/index.js';
-import { ToolSet, PrepareStepFunction, GenerateTextOnStepFinishCallback, StreamTextOnStepFinishCallback, StreamTextOnErrorCallback, StreamTextOnChunkCallback, StreamTextOnFinishCallback, StepResult, StreamTextResult, ModelMessage, wrapLanguageModel, generateObject, generateText, streamText, streamObject, experimental_generateImage, embed, embedMany, experimental_transcribe, experimental_generateSpeech } from 'ai';
+import { ToolSet, ModelMessage, PrepareStepFunction, GenerateTextOnStepFinishCallback, StreamTextOnStepFinishCallback, StreamTextOnErrorCallback, StreamTextOnChunkCallback, StreamTextOnFinishCallback, StepResult, StreamTextResult, wrapLanguageModel, generateObject, generateText, streamText, streamObject, experimental_generateImage, embed, embedMany, experimental_transcribe, experimental_generateSpeech } from 'ai';
+export { ModelMessage } from 'ai';
 import { Page as Page$1 } from 'playwright-core';
 export { Page as PlaywrightPage } from 'playwright-core';
 import { Page as Page$2 } from 'puppeteer-core';
@@ -811,6 +812,10 @@ declare class StreamingCallbacksInNonStreamingModeError extends StagehandError {
     readonly invalidCallbacks: string[];
     constructor(invalidCallbacks: string[]);
 }
+declare class AgentAbortError extends StagehandError {
+    readonly reason: string;
+    constructor(reason?: string);
+}
 
 declare class AISdkClient extends LLMClient {
     type: "aisdk";
@@ -1661,6 +1666,12 @@ interface AgentResult {
         cached_input_tokens?: number;
         inference_time_ms: number;
     };
+    /**
+     * The conversation messages from this execution.
+     * Pass these to a subsequent execute() call via the `messages` option to continue the conversation.
+     * @experimental
+     */
+    messages?: ModelMessage[];
 }
 type AgentStreamResult = StreamTextResult<ToolSet, never> & {
     result: Promise<AgentResult>;
@@ -1792,6 +1803,29 @@ interface AgentExecuteOptionsBase {
     maxSteps?: number;
     page?: Page$1 | Page$2 | Page$3 | Page;
     highlightCursor?: boolean;
+    /**
+     * Previous conversation messages to continue from.
+     * Pass the `messages` from a previous AgentResult to continue that conversation.
+     * @experimental
+     */
+    messages?: ModelMessage[];
+    /**
+     * An AbortSignal that can be used to cancel the agent execution.
+     * When aborted, the agent will stop and return a partial result.
+     * @experimental
+     *
+     * @example
+     * ```typescript
+     * const controller = new AbortController();
+     * setTimeout(() => controller.abort(), 30000); // 30 second timeout
+     *
+     * const result = await agent.execute({
+     *   instruction: "...",
+     *   signal: controller.signal
+     * });
+     * ```
+     */
+    signal?: AbortSignal;
 }
 /**
  * Options for non-streaming agent execution.
@@ -2423,4 +2457,4 @@ declare class V3Evaluator {
     private _evaluateWithMultipleScreenshots;
 }
 
-export { type AISDKCustomProvider, type AISDKProvider, AISdkClient, AVAILABLE_CUA_MODELS, type ActOptions, type ActResult, ActTimeoutError, type Action, type ActionExecutionResult, type AgentAction, type AgentCallbacks, type AgentConfig, type AgentContext, type AgentExecuteCallbacks, type AgentExecuteOptions, type AgentExecuteOptionsBase, type AgentExecutionOptions, type AgentHandlerOptions, type AgentInstance, type AgentModelConfig, AgentProvider, type AgentProviderType, type AgentResult, AgentScreenshotProviderError, type AgentState, type AgentStreamCallbacks, type AgentStreamExecuteOptions, type AgentStreamResult, type AgentType, AnnotatedScreenshotText, type AnthropicClientOptions, type AnthropicContentBlock, type AnthropicJsonSchemaObject, type AnthropicMessage, type AnthropicTextBlock, type AnthropicToolResult, type AnyPage, type AvailableCuaModel, type AvailableModel, BrowserbaseSessionNotFoundError, CaptchaTimeoutError, type ChatCompletionOptions, type ChatMessage, type ChatMessageContent, type ChatMessageImageContent, type ChatMessageTextContent, type ClientOptions, type ComputerCallItem, ConnectionTimeoutError, type ConsoleListener, ConsoleMessage, ContentFrameNotFoundError, type CreateChatCompletionOptions, CreateChatCompletionResponseError, CuaModelRequiredError, ElementNotVisibleError, ExperimentalApiConflictError, ExperimentalNotConfiguredError, type ExtractOptions, type ExtractResult, ExtractTimeoutError, type FunctionCallItem, type GoogleServiceAccountCredentials, type GoogleVertexProviderSettings, HandlerNotInitializedError, type HistoryEntry, type InferStagehandSchema, InvalidAISDKModelFormatError, type JsonSchema, type JsonSchemaDocument, type JsonSchemaProperty, LLMClient, type LLMParsedResponse, type LLMResponse, LLMResponseError, type LLMTool, type LLMUsage, LOG_LEVEL_NAMES, type LoadState, type LocalBrowserLaunchOptions, type LogLevel, type LogLine, type Logger, MCPConnectionError, MissingEnvironmentVariableError, MissingLLMConfigurationError, type ModelConfiguration, type ModelProvider, type NonStreamingAgentInstance, type ObserveOptions, ObserveTimeoutError, type OpenAIClientOptions, Page, PageNotFoundError, Response$1 as Response, ResponseBodyError, type ResponseInputItem, type ResponseItem, ResponseParseError, V3 as Stagehand, StagehandAPIError, StagehandAPIUnauthorizedError, StagehandClickError, StagehandDefaultError, StagehandDomProcessError, StagehandElementNotFoundError, StagehandEnvironmentError, StagehandError, StagehandEvalError, StagehandHttpError, StagehandIframeError, StagehandInitError, StagehandInvalidArgumentError, type StagehandMetrics, StagehandMissingArgumentError, StagehandNotInitializedError, StagehandResponseBodyError, StagehandResponseParseError, StagehandServerError, StagehandShadowRootMissingError, StagehandShadowSegmentEmptyError, StagehandShadowSegmentNotFoundError, type StagehandZodObject, type StagehandZodSchema, type StreamingAgentInstance, StreamingCallbacksInNonStreamingModeError, TimeoutError, type ToolUseItem, UnsupportedAISDKModelProviderError, UnsupportedModelError, UnsupportedModelProviderError, V3, type V3Env, V3Evaluator, V3FunctionName, type V3Options, XPathResolutionError, ZodSchemaValidationError, connectToMCPServer, defaultExtractSchema, getZodType, injectUrls, isRunningInBun, isZod3Schema, isZod4Schema, jsonSchemaToZod, loadApiKeyFromEnv, modelToAgentProviderMap, pageTextSchema, providerEnvVarMap, toGeminiSchema, toJsonSchema, transformSchema, trimTrailingTextNode, validateZodSchema };
+export { type AISDKCustomProvider, type AISDKProvider, AISdkClient, AVAILABLE_CUA_MODELS, type ActOptions, type ActResult, ActTimeoutError, type Action, type ActionExecutionResult, AgentAbortError, type AgentAction, type AgentCallbacks, type AgentConfig, type AgentContext, type AgentExecuteCallbacks, type AgentExecuteOptions, type AgentExecuteOptionsBase, type AgentExecutionOptions, type AgentHandlerOptions, type AgentInstance, type AgentModelConfig, AgentProvider, type AgentProviderType, type AgentResult, AgentScreenshotProviderError, type AgentState, type AgentStreamCallbacks, type AgentStreamExecuteOptions, type AgentStreamResult, type AgentType, AnnotatedScreenshotText, type AnthropicClientOptions, type AnthropicContentBlock, type AnthropicJsonSchemaObject, type AnthropicMessage, type AnthropicTextBlock, type AnthropicToolResult, type AnyPage, type AvailableCuaModel, type AvailableModel, BrowserbaseSessionNotFoundError, CaptchaTimeoutError, type ChatCompletionOptions, type ChatMessage, type ChatMessageContent, type ChatMessageImageContent, type ChatMessageTextContent, type ClientOptions, type ComputerCallItem, ConnectionTimeoutError, type ConsoleListener, ConsoleMessage, ContentFrameNotFoundError, type CreateChatCompletionOptions, CreateChatCompletionResponseError, CuaModelRequiredError, ElementNotVisibleError, ExperimentalApiConflictError, ExperimentalNotConfiguredError, type ExtractOptions, type ExtractResult, ExtractTimeoutError, type FunctionCallItem, type GoogleServiceAccountCredentials, type GoogleVertexProviderSettings, HandlerNotInitializedError, type HistoryEntry, type InferStagehandSchema, InvalidAISDKModelFormatError, type JsonSchema, type JsonSchemaDocument, type JsonSchemaProperty, LLMClient, type LLMParsedResponse, type LLMResponse, LLMResponseError, type LLMTool, type LLMUsage, LOG_LEVEL_NAMES, type LoadState, type LocalBrowserLaunchOptions, type LogLevel, type LogLine, type Logger, MCPConnectionError, MissingEnvironmentVariableError, MissingLLMConfigurationError, type ModelConfiguration, type ModelProvider, type NonStreamingAgentInstance, type ObserveOptions, ObserveTimeoutError, type OpenAIClientOptions, Page, PageNotFoundError, Response$1 as Response, ResponseBodyError, type ResponseInputItem, type ResponseItem, ResponseParseError, V3 as Stagehand, StagehandAPIError, StagehandAPIUnauthorizedError, StagehandClickError, StagehandDefaultError, StagehandDomProcessError, StagehandElementNotFoundError, StagehandEnvironmentError, StagehandError, StagehandEvalError, StagehandHttpError, StagehandIframeError, StagehandInitError, StagehandInvalidArgumentError, type StagehandMetrics, StagehandMissingArgumentError, StagehandNotInitializedError, StagehandResponseBodyError, StagehandResponseParseError, StagehandServerError, StagehandShadowRootMissingError, StagehandShadowSegmentEmptyError, StagehandShadowSegmentNotFoundError, type StagehandZodObject, type StagehandZodSchema, type StreamingAgentInstance, StreamingCallbacksInNonStreamingModeError, TimeoutError, type ToolUseItem, UnsupportedAISDKModelProviderError, UnsupportedModelError, UnsupportedModelProviderError, V3, type V3Env, V3Evaluator, V3FunctionName, type V3Options, XPathResolutionError, ZodSchemaValidationError, connectToMCPServer, defaultExtractSchema, getZodType, injectUrls, isRunningInBun, isZod3Schema, isZod4Schema, jsonSchemaToZod, loadApiKeyFromEnv, modelToAgentProviderMap, pageTextSchema, providerEnvVarMap, toGeminiSchema, toJsonSchema, transformSchema, trimTrailingTextNode, validateZodSchema };

package/dist/index.js

@@ -179,12 +179,12 @@ var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")])
 var STAGEHAND_VERSION;
 var init_version = __esm({
   "lib/version.ts"() {
-    STAGEHAND_VERSION = "3.0.6-alpha-75f87f1920021a012fe0159123d01404969a94c8";
+    STAGEHAND_VERSION = "3.0.6-alpha-943d2d79d0f289ac41c9164578f2f1dd876058f2";
   }
 });
 
 // lib/v3/types/public/sdkErrors.ts
-var StagehandError, StagehandDefaultError, StagehandEnvironmentError, MissingEnvironmentVariableError, UnsupportedModelError, UnsupportedModelProviderError, UnsupportedAISDKModelProviderError, InvalidAISDKModelFormatError, StagehandNotInitializedError, BrowserbaseSessionNotFoundError, CaptchaTimeoutError, MissingLLMConfigurationError, HandlerNotInitializedError, StagehandInvalidArgumentError, StagehandElementNotFoundError, AgentScreenshotProviderError, StagehandMissingArgumentError, CreateChatCompletionResponseError, StagehandEvalError, StagehandDomProcessError, StagehandClickError, LLMResponseError, StagehandIframeError, ContentFrameNotFoundError, XPathResolutionError, ExperimentalApiConflictError, ExperimentalNotConfiguredError, CuaModelRequiredError, ZodSchemaValidationError, StagehandInitError, MCPConnectionError, StagehandShadowRootMissingError, StagehandShadowSegmentEmptyError, StagehandShadowSegmentNotFoundError, ElementNotVisibleError, ResponseBodyError, ResponseParseError, TimeoutError, ActTimeoutError, ExtractTimeoutError, ObserveTimeoutError, PageNotFoundError, ConnectionTimeoutError, StreamingCallbacksInNonStreamingModeError;
+var StagehandError, StagehandDefaultError, StagehandEnvironmentError, MissingEnvironmentVariableError, UnsupportedModelError, UnsupportedModelProviderError, UnsupportedAISDKModelProviderError, InvalidAISDKModelFormatError, StagehandNotInitializedError, BrowserbaseSessionNotFoundError, CaptchaTimeoutError, MissingLLMConfigurationError, HandlerNotInitializedError, StagehandInvalidArgumentError, StagehandElementNotFoundError, AgentScreenshotProviderError, StagehandMissingArgumentError, CreateChatCompletionResponseError, StagehandEvalError, StagehandDomProcessError, StagehandClickError, LLMResponseError, StagehandIframeError, ContentFrameNotFoundError, XPathResolutionError, ExperimentalApiConflictError, ExperimentalNotConfiguredError, CuaModelRequiredError, ZodSchemaValidationError, StagehandInitError, MCPConnectionError, StagehandShadowRootMissingError, StagehandShadowSegmentEmptyError, StagehandShadowSegmentNotFoundError, ElementNotVisibleError, ResponseBodyError, ResponseParseError, TimeoutError, ActTimeoutError, ExtractTimeoutError, ObserveTimeoutError, PageNotFoundError, ConnectionTimeoutError, StreamingCallbacksInNonStreamingModeError, AgentAbortError;
 var init_sdkErrors = __esm({
   "lib/v3/types/public/sdkErrors.ts"() {
     init_version();
@@ -470,6 +470,13 @@ ${JSON.stringify(issues, null, 2)}`);
         this.invalidCallbacks = invalidCallbacks;
       }
     };
+    AgentAbortError = class extends StagehandError {
+      constructor(reason) {
+        const message = reason ? `Agent execution was aborted: ${reason}` : "Agent execution was aborted";
+        super(message);
+        this.reason = reason || "aborted";
+      }
+    };
   }
 });
 
@@ -26780,6 +26787,7 @@ __export(v3_exports, {
   AISdkClient: () => AISdkClient2,
   AVAILABLE_CUA_MODELS: () => AVAILABLE_CUA_MODELS,
   ActTimeoutError: () => ActTimeoutError,
+  AgentAbortError: () => AgentAbortError,
   AgentProvider: () => AgentProvider,
   AgentScreenshotProviderError: () => AgentScreenshotProviderError,
   AnnotatedScreenshotText: () => AnnotatedScreenshotText,
@@ -31105,6 +31113,9 @@ function createStandardAction(toolCallName, toolResult, args, reasoning) {
 
 // lib/v3/handlers/v3AgentHandler.ts
 init_sdkErrors();
+function getErrorMessage(error) {
+  return error instanceof Error ? error.message : String(error);
+}
 var V3AgentHandler = class {
   constructor(v3, logger, llmClient, executionModel, systemInstructions, mcpTools) {
     this.v3 = v3;
@@ -31116,7 +31127,7 @@ var V3AgentHandler = class {
   }
   prepareAgent(instructionOrOptions) {
     return __async(this, null, function* () {
-      var _a4;
+      var _a4, _b;
       try {
         const options = typeof instructionOrOptions === "string" ? { instruction: instructionOrOptions } : instructionOrOptions;
         const maxSteps = options.maxSteps || 20;
@@ -31126,10 +31137,8 @@ var V3AgentHandler = class {
         );
         const tools = this.createTools();
         const allTools = __spreadValues(__spreadValues({}, tools), this.mcpTools);
-        const messages = [
-          { role: "user", content: options.instruction }
-        ];
-        if (!((_a4 = this.llmClient) == null ? void 0 : _a4.getLanguageModel)) {
+        const messages = ((_a4 = options.messages) == null ? void 0 : _a4.length) ? [...options.messages, { role: "user", content: options.instruction }] : [{ role: "user", content: options.instruction }];
+        if (!((_b = this.llmClient) == null ? void 0 : _b.getLanguageModel)) {
           throw new MissingLLMConfigurationError();
         }
         const baseModel = this.llmClient.getLanguageModel();
@@ -31212,39 +31221,43 @@ var V3AgentHandler = class {
   }
   execute(instructionOrOptions) {
     return __async(this, null, function* () {
-      var _a4;
       const startTime = Date.now();
-      const {
-        maxSteps,
-        systemPrompt,
-        allTools,
-        messages,
-        wrappedModel,
-        initialPageUrl
-      } = yield this.prepareAgent(instructionOrOptions);
-      const callbacks = instructionOrOptions.callbacks;
-      if (callbacks) {
-        const streamingOnlyCallbacks = [
-          "onChunk",
-          "onFinish",
-          "onError",
-          "onAbort"
-        ];
-        const invalidCallbacks = streamingOnlyCallbacks.filter(
-          (name) => callbacks[name] != null
-        );
-        if (invalidCallbacks.length > 0) {
-          throw new StreamingCallbacksInNonStreamingModeError(invalidCallbacks);
-        }
-      }
+      const signal = typeof instructionOrOptions === "object" ? instructionOrOptions.signal : void 0;
       const state = {
         collectedReasoning: [],
         actions: [],
         finalMessage: "",
         completed: false,
-        currentPageUrl: initialPageUrl
+        currentPageUrl: ""
       };
+      let messages = [];
       try {
+        const {
+          options,
+          maxSteps,
+          systemPrompt,
+          allTools,
+          messages: preparedMessages,
+          wrappedModel,
+          initialPageUrl
+        } = yield this.prepareAgent(instructionOrOptions);
+        messages = preparedMessages;
+        state.currentPageUrl = initialPageUrl;
+        const callbacks = instructionOrOptions.callbacks;
+        if (callbacks) {
+          const streamingOnlyCallbacks = [
+            "onChunk",
+            "onFinish",
+            "onError",
+            "onAbort"
+          ];
+          const invalidCallbacks = streamingOnlyCallbacks.filter(
+            (name) => callbacks[name] != null
+          );
+          if (invalidCallbacks.length > 0) {
+            throw new StreamingCallbacksInNonStreamingModeError(invalidCallbacks);
+          }
+        }
         const result = yield this.llmClient.generateText({
           model: wrappedModel,
           system: systemPrompt,
@@ -31254,11 +31267,24 @@ var V3AgentHandler = class {
           temperature: 1,
           toolChoice: "auto",
           prepareStep: callbacks == null ? void 0 : callbacks.prepareStep,
-          onStepFinish: this.createStepHandler(state, callbacks == null ? void 0 : callbacks.onStepFinish)
+          onStepFinish: this.createStepHandler(state, callbacks == null ? void 0 : callbacks.onStepFinish),
+          abortSignal: options.signal
         });
-        return this.consolidateMetricsAndResult(startTime, state, result);
+        return this.consolidateMetricsAndResult(
+          startTime,
+          state,
+          messages,
+          result
+        );
       } catch (error) {
-        const errorMessage = (_a4 = error == null ? void 0 : error.message) != null ? _a4 : String(error);
+        if (error instanceof StreamingCallbacksInNonStreamingModeError) {
+          throw error;
+        }
+        if (signal == null ? void 0 : signal.aborted) {
+          const reason = signal.reason ? String(signal.reason) : "aborted";
+          throw new AgentAbortError(reason);
+        }
+        const errorMessage = getErrorMessage(error);
         this.logger({
           category: "agent",
           message: `Error executing agent task: ${errorMessage}`,
@@ -31268,7 +31294,8 @@ var V3AgentHandler = class {
           success: false,
           actions: state.actions,
           message: `Failed to execute task: ${errorMessage}`,
-          completed: false
+          completed: false,
+          messages
         };
       }
     });
@@ -31276,6 +31303,7 @@ var V3AgentHandler = class {
   stream(instructionOrOptions) {
     return __async(this, null, function* () {
       const {
+        options,
         maxSteps,
         systemPrompt,
         allTools,
@@ -31328,24 +31356,31 @@ var V3AgentHandler = class {
           if (callbacks == null ? void 0 : callbacks.onFinish) {
             callbacks.onFinish(event);
           }
-          try {
-            const result = this.consolidateMetricsAndResult(
-              startTime,
-              state,
-              event
-            );
-            resolveResult(result);
-          } catch (error) {
-            handleError(error);
+          const result = this.consolidateMetricsAndResult(
+            startTime,
+            state,
+            messages,
+            event
+          );
+          resolveResult(result);
+        },
+        onAbort: (event) => {
+          var _a4;
+          if (callbacks == null ? void 0 : callbacks.onAbort) {
+            callbacks.onAbort(event);
           }
-        }
+          const reason = ((_a4 = options.signal) == null ? void 0 : _a4.reason) ? String(options.signal.reason) : "Stream was aborted";
+          rejectResult(new AgentAbortError(reason));
+        },
+        abortSignal: options.signal
       });
       const agentStreamResult = streamResult;
       agentStreamResult.result = resultPromise;
       return agentStreamResult;
     });
   }
-  consolidateMetricsAndResult(startTime, state, result) {
+  consolidateMetricsAndResult(startTime, state, inputMessages, result) {
+    var _a4;
     if (!state.finalMessage) {
       const allReasoning = state.collectedReasoning.join(" ").trim();
       state.finalMessage = allReasoning || result.text || "";
@@ -31362,6 +31397,11 @@ var V3AgentHandler = class {
         inferenceTimeMs
       );
     }
+    const responseMessages = ((_a4 = result.response) == null ? void 0 : _a4.messages) || [];
+    const fullMessages = [
+      ...inputMessages,
+      ...responseMessages
+    ];
     return {
       success: state.completed,
       message: state.finalMessage || "Task execution completed",
@@ -31373,7 +31413,8 @@ var V3AgentHandler = class {
         reasoning_tokens: result.usage.reasoningTokens || 0,
         cached_input_tokens: result.usage.cachedInputTokens || 0,
         inference_time_ms: inferenceTimeMs
-      } : void 0
+      } : void 0,
+      messages: fullMessages
     };
   }
   buildSystemPrompt(executionInstruction, systemInstructions) {
@@ -64036,6 +64077,53 @@ var StagehandAPIClient = class {
   }
 };
 
+// lib/v3/agent/utils/validateExperimentalFeatures.ts
+init_sdkErrors();
+function validateExperimentalFeatures(options) {
+  const { isExperimental, agentConfig, executeOptions, isStreaming } = options;
+  if (agentConfig == null ? void 0 : agentConfig.cua) {
+    const unsupportedFeatures = [];
+    if (agentConfig == null ? void 0 : agentConfig.stream) {
+      unsupportedFeatures.push("streaming");
+    }
+    if (executeOptions == null ? void 0 : executeOptions.signal) {
+      unsupportedFeatures.push("abort signal");
+    }
+    if (executeOptions == null ? void 0 : executeOptions.messages) {
+      unsupportedFeatures.push("message continuation");
+    }
+    if (unsupportedFeatures.length > 0) {
+      throw new StagehandInvalidArgumentError(
+        `${unsupportedFeatures.join(", ")} ${unsupportedFeatures.length === 1 ? "is" : "are"} not supported with CUA (Computer Use Agent) mode.`
+      );
+    }
+  }
+  if (isExperimental) return;
+  const features = [];
+  const hasIntegrations = (agentConfig == null ? void 0 : agentConfig.integrations) && agentConfig.integrations.length > 0;
+  const hasTools = (agentConfig == null ? void 0 : agentConfig.tools) && Object.keys(agentConfig.tools).length > 0;
+  if (hasIntegrations || hasTools) {
+    features.push("MCP integrations and custom tools");
+  }
+  if (!(agentConfig == null ? void 0 : agentConfig.cua) && (isStreaming || (agentConfig == null ? void 0 : agentConfig.stream))) {
+    features.push("streaming");
+  }
+  if (executeOptions && !(agentConfig == null ? void 0 : agentConfig.cua)) {
+    if (executeOptions.callbacks) {
+      features.push("callbacks");
+    }
+    if (executeOptions.signal) {
+      features.push("abort signal");
+    }
+    if (executeOptions.messages) {
+      features.push("message continuation");
+    }
+  }
+  if (features.length > 0) {
+    throw new ExperimentalNotConfiguredError(`Agent ${features.join(", ")}`);
+  }
+}
+
 // lib/v3/v3.ts
 init_flowLogger();
 init_sdkErrors();
@@ -65166,11 +65254,6 @@ var _V3 = class _V3 {
   prepareAgentExecution(options, instructionOrOptions, agentConfigSignature) {
     return __async(this, null, function* () {
       var _a4, _b;
-      if (((options == null ? void 0 : options.integrations) || (options == null ? void 0 : options.tools)) && !this.experimental) {
-        throw new ExperimentalNotConfiguredError(
-          "MCP integrations and custom tools"
-        );
-      }
       const tools = (options == null ? void 0 : options.integrations) ? yield resolveTools(options.integrations, options.tools) : (_a4 = options == null ? void 0 : options.tools) != null ? _a4 : {};
       const agentLlmClient = (options == null ? void 0 : options.model) ? this.resolveLlmClient(options.model) : this.llmClient;
       const handler = new V3AgentHandler(
@@ -65216,16 +65299,10 @@ var _V3 = class _V3 {
       })
     });
     if (options == null ? void 0 : options.cua) {
-      if (options == null ? void 0 : options.stream) {
-        throw new StagehandInvalidArgumentError(
-          "Streaming is not supported with CUA (Computer Use Agent) mode. Remove either 'stream: true' or 'cua: true' from your agent config."
-        );
-      }
-      if (((options == null ? void 0 : options.integrations) || (options == null ? void 0 : options.tools)) && !this.experimental) {
-        throw new ExperimentalNotConfiguredError(
-          "MCP integrations and custom tools"
-        );
-      }
+      validateExperimentalFeatures({
+        isExperimental: this.experimental,
+        agentConfig: options
+      });
       const modelToUse = (options == null ? void 0 : options.model) || __spreadValues({
         modelName: this.modelName
       }, this.modelClientOptions);
@@ -65238,13 +65315,15 @@ var _V3 = class _V3 {
         execute: (instructionOrOptions) => __async(this, null, function* () {
           return withInstanceLogContext(this.instanceId, () => __async(this, null, function* () {
             var _a5, _b2;
+            validateExperimentalFeatures({
+              isExperimental: this.experimental,
+              agentConfig: options,
+              executeOptions: typeof instructionOrOptions === "object" ? instructionOrOptions : null
+            });
             SessionFileLogger.logAgentTaskStarted({
               invocation: "Agent.execute",
               args: [instructionOrOptions]
             });
-            if ((options == null ? void 0 : options.integrations) && !this.experimental) {
-              throw new ExperimentalNotConfiguredError("MCP integrations");
-            }
             const tools = (options == null ? void 0 : options.integrations) ? yield resolveTools(options.integrations, options.tools) : (_a5 = options == null ? void 0 : options.tools) != null ? _a5 : {};
             const handler = new V3CuaAgentHandler(
               this,
@@ -65325,18 +65404,18 @@ Do not ask follow up questions, the user will trust your judgement.`
     return {
       execute: (instructionOrOptions) => __async(this, null, function* () {
         return withInstanceLogContext(this.instanceId, () => __async(this, null, function* () {
+          validateExperimentalFeatures({
+            isExperimental: this.experimental,
+            agentConfig: options,
+            executeOptions: typeof instructionOrOptions === "object" ? instructionOrOptions : null,
+            isStreaming
+          });
           SessionFileLogger.logAgentTaskStarted({
             invocation: "Agent.execute",
             args: [instructionOrOptions]
           });
-          if (typeof instructionOrOptions === "object" && instructionOrOptions.callbacks && !this.experimental) {
-            throw new ExperimentalNotConfiguredError("Agent callbacks");
-          }
           if (isStreaming) {
-            if (!this.experimental) {
-              throw new ExperimentalNotConfiguredError("Agent streaming");
-            }
-            const { handler: handler2, cacheContext: cacheContext2 } = yield this.prepareAgentExecution(
+            const { handler: handler2, resolvedOptions: resolvedOptions2, cacheContext: cacheContext2 } = yield this.prepareAgentExecution(
               options,
               instructionOrOptions,
               agentConfigSignature
@@ -65349,7 +65428,7 @@ Do not ask follow up questions, the user will trust your judgement.`
               }
             }
             const streamResult = yield handler2.stream(
-              instructionOrOptions
+              resolvedOptions2
             );
             if (cacheContext2) {
               const wrappedStream = this.agentCache.wrapStreamForCaching(
@@ -65393,7 +65472,7 @@ Do not ask follow up questions, the user will trust your judgement.`
               );
             } else {
               result = yield handler.execute(
-                instructionOrOptions
+                resolvedOptions
               );
             }
             if (recording) {
@@ -65669,6 +65748,7 @@ I'm providing ${screenshots.length} screenshots showing the progression of the t
   AISdkClient,
   AVAILABLE_CUA_MODELS,
   ActTimeoutError,
+  AgentAbortError,
   AgentProvider,
   AgentScreenshotProviderError,
   AnnotatedScreenshotText,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@browserbasehq/stagehand",
-  "version": "3.0.6-alpha-75f87f1920021a012fe0159123d01404969a94c8",
+  "version": "3.0.6-alpha-943d2d79d0f289ac41c9164578f2f1dd876058f2",
   "description": "An AI web browsing framework focused on simplicity and extensibility.",
   "main": "./dist/index.js",
   "module": "./dist/index.js",