@browserbasehq/orca 3.2.0-preview.5 → 3.2.1-preview.0

package/dist/cjs/lib/v3/v3.d.ts

@@ -1,12 +1,19 @@
+import type { LanguageModelV2Middleware } from "@ai-sdk/provider";
 import { z } from "zod";
 import { InferStagehandSchema, StagehandZodSchema } from "./zodCompat.js";
 import { LLMClient } from "./llm/LLMClient.js";
 import { AgentReplayStep } from "./types/private/index.js";
-import { AgentConfig, AgentExecuteOptions, AgentStreamExecuteOptions, AgentResult, LogLine, StagehandMetrics, Action, ActOptions, ActResult, defaultExtractSchema, ExtractOptions, HistoryEntry, ObserveOptions, pageTextSchema, V3FunctionName, V3Options, AgentStreamResult } from "./types/public/index.js";
+import { AgentConfig, AgentExecuteOptions, AgentStreamExecuteOptions, AgentResult, LogLine, StagehandMetrics, Action, ActOptions, ActResult, defaultExtractSchema, ExtractOptions, HistoryEntry, ObserveOptions, pageTextSchema, V3FunctionName, AvailableModel, ClientOptions, V3Options, AgentStreamResult } from "./types/public/index.js";
 import { V3Context } from "./understudy/context.js";
 import { type FlowLoggerContext } from "./flowlogger/FlowLogger.js";
 import { EventEmitterWithWildcardSupport } from "./flowlogger/EventEmitter.js";
 import { EventStore } from "./flowlogger/EventStore.js";
+type ResolvedModelConfiguration = {
+    modelName: AvailableModel;
+    clientOptions?: ClientOptions;
+    middleware?: LanguageModelV2Middleware;
+};
+export declare function resolveModelConfiguration(model?: V3Options["model"]): ResolvedModelConfiguration;
 /**
  * V3
  *
@@ -57,7 +64,13 @@ export declare class V3 {
      */
     get isCaptchaAutoSolveEnabled(): boolean;
     /**
-     * Returns true if advancedStealth is enabled in Browserbase settings.
+     * Returns true if Browserbase Verified mode is enabled in settings.
+     * Legacy `advancedStealth` is treated as equivalent for backwards compatibility.
+     */
+    get isVerified(): boolean;
+    /**
+     * Backwards-compatible alias for Browserbase managed fingerprinting mode.
+     * @deprecated Use `isVerified` instead. This alias will be removed in a future version.
      */
     get isAdvancedStealth(): boolean;
     /**
@@ -95,6 +108,8 @@ export declare class V3 {
      * When using API mode, fetches metrics from the API. Otherwise returns local metrics.
      */
     get metrics(): Promise<StagehandMetrics>;
+    private mergeAgentMetricsWithLocalFallback;
+    private updateAgentMetricsFromUsage;
     private resolveLlmClient;
     private beginAgentReplayRecording;
     private endAgentReplayRecording;
@@ -204,3 +219,4 @@ export declare class V3 {
         execute: (instructionOrOptions: string | AgentExecuteOptions) => Promise<AgentResult>;
     };
 }
+export {};

package/dist/cjs/lib/v3/v3.js

@@ -71,6 +71,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.V3 = void 0;
+exports.resolveModelConfiguration = resolveModelConfiguration;
 const fs_1 = __importDefault(require("fs"));
 const os_1 = __importDefault(require("os"));
 const path_1 = __importDefault(require("path"));
@@ -118,13 +119,14 @@ function resolveModelConfiguration(model) {
         return { modelName: model };
     }
     if (model && typeof model === "object") {
-        const { modelName, ...clientOptions } = model;
+        const { modelName, middleware, ...clientOptions } = model;
         if (!modelName) {
             throw new index_js_1.StagehandInvalidArgumentError("model.modelName is required when providing client options.");
         }
         return {
             modelName,
             clientOptions: clientOptions,
+            middleware,
         };
     }
     return { modelName: DEFAULT_MODEL_NAME };
@@ -212,11 +214,20 @@ let V3 = (() => {
                     ?.solveCaptchas !== false);
         }
         /**
-         * Returns true if advancedStealth is enabled in Browserbase settings.
+         * Returns true if Browserbase Verified mode is enabled in settings.
+         * Legacy `advancedStealth` is treated as equivalent for backwards compatibility.
+         */
+        get isVerified() {
+            const browserSettings = this.opts.browserbaseSessionCreateParams?.browserSettings;
+            return (browserSettings?.verified === true ||
+                browserSettings?.advancedStealth === true);
+        }
+        /**
+         * Backwards-compatible alias for Browserbase managed fingerprinting mode.
+         * @deprecated Use `isVerified` instead. This alias will be removed in a future version.
          */
         get isAdvancedStealth() {
-            return (this.opts.browserbaseSessionCreateParams?.browserSettings
-                ?.advancedStealth === true);
+            return this.isVerified;
         }
         /**
          * Returns the configured viewport dimensions from launch options.
@@ -328,11 +339,11 @@ let V3 = (() => {
             catch {
                 // ignore
             }
-            const { modelName, clientOptions } = resolveModelConfiguration(opts.model);
+            const { modelName, clientOptions, middleware } = resolveModelConfiguration(opts.model);
             this.modelName = modelName;
             this.experimental = opts.experimental ?? false;
             this.logInferenceToFile = opts.logInferenceToFile ?? false;
-            this.llmProvider = new LLMProvider_js_1.LLMProvider(this.logger);
+            this.llmProvider = new LLMProvider_js_1.LLMProvider(this.logger, middleware);
             this.domSettleTimeoutMs = opts.domSettleTimeout;
             this.disableAPI = opts.disableAPI ?? false;
             const baseClientOptions = clientOptions
@@ -414,7 +425,10 @@ let V3 = (() => {
         get metrics() {
             if (this.apiClient) {
                 // Fetch metrics from the API
-                return this.apiClient.getReplayMetrics().catch((error) => {
+                return this.apiClient
+                    .getReplayMetrics()
+                    .then((metrics) => this.mergeAgentMetricsWithLocalFallback(metrics))
+                    .catch((error) => {
                     this.logger({
                         category: "metrics",
                         message: `Failed to fetch metrics from API: ${error}`,
@@ -427,21 +441,78 @@ let V3 = (() => {
             // Return local metrics wrapped in a Promise for consistency
             return Promise.resolve(this.stagehandMetrics);
         }
+        mergeAgentMetricsWithLocalFallback(remoteMetrics) {
+            // In API mode, agent.execute() is the only path that returns trusted inline
+            // usage today, so only repair the agent bucket from local state.
+            const agentPromptTokens = Math.max(remoteMetrics.agentPromptTokens, this.stagehandMetrics.agentPromptTokens);
+            const agentCompletionTokens = Math.max(remoteMetrics.agentCompletionTokens, this.stagehandMetrics.agentCompletionTokens);
+            const agentReasoningTokens = Math.max(remoteMetrics.agentReasoningTokens, this.stagehandMetrics.agentReasoningTokens);
+            const agentCachedInputTokens = Math.max(remoteMetrics.agentCachedInputTokens, this.stagehandMetrics.agentCachedInputTokens);
+            const agentInferenceTimeMs = Math.max(remoteMetrics.agentInferenceTimeMs, this.stagehandMetrics.agentInferenceTimeMs);
+            const metrics = {
+                ...remoteMetrics,
+                agentPromptTokens,
+                agentCompletionTokens,
+                agentReasoningTokens,
+                agentCachedInputTokens,
+                agentInferenceTimeMs,
+                totalPromptTokens: 0,
+                totalCompletionTokens: 0,
+                totalReasoningTokens: 0,
+                totalCachedInputTokens: 0,
+                totalInferenceTimeMs: 0,
+            };
+            metrics.totalPromptTokens =
+                metrics.actPromptTokens +
+                    metrics.extractPromptTokens +
+                    metrics.observePromptTokens +
+                    metrics.agentPromptTokens;
+            metrics.totalCompletionTokens =
+                metrics.actCompletionTokens +
+                    metrics.extractCompletionTokens +
+                    metrics.observeCompletionTokens +
+                    metrics.agentCompletionTokens;
+            metrics.totalReasoningTokens =
+                metrics.actReasoningTokens +
+                    metrics.extractReasoningTokens +
+                    metrics.observeReasoningTokens +
+                    metrics.agentReasoningTokens;
+            metrics.totalCachedInputTokens =
+                metrics.actCachedInputTokens +
+                    metrics.extractCachedInputTokens +
+                    metrics.observeCachedInputTokens +
+                    metrics.agentCachedInputTokens;
+            metrics.totalInferenceTimeMs =
+                metrics.actInferenceTimeMs +
+                    metrics.extractInferenceTimeMs +
+                    metrics.observeInferenceTimeMs +
+                    metrics.agentInferenceTimeMs;
+            return metrics;
+        }
+        updateAgentMetricsFromUsage(usage) {
+            if (!usage) {
+                return;
+            }
+            this.updateMetrics(index_js_1.V3FunctionName.AGENT, usage.input_tokens, usage.output_tokens, usage.reasoning_tokens ?? 0, usage.cached_input_tokens ?? 0, usage.inference_time_ms);
+        }
         resolveLlmClient(model) {
             if (!model) {
                 return this.llmClient;
             }
             let modelName;
             let clientOptions;
+            let perCallMiddleware;
             if (typeof model === "string") {
                 modelName = model;
             }
             else {
-                const { modelName: overrideModelName, ...rest } = model;
+                const { modelName: overrideModelName, middleware, ...rest } = model;
                 modelName = overrideModelName;
                 clientOptions = rest;
+                perCallMiddleware = middleware;
             }
             if (modelName === this.modelName &&
+                !perCallMiddleware &&
                 (!clientOptions || Object.keys(clientOptions).length === 0)) {
                 return this.llmClient;
             }
@@ -458,6 +529,13 @@ let V3 = (() => {
                     mergedOptions.apiKey = apiKey;
                 }
             }
+            if (perCallMiddleware) {
+                return this.llmProvider.getClient(modelName, mergedOptions, {
+                    experimental: this.experimental,
+                    disableAPI: this.disableAPI,
+                    middleware: perCallMiddleware,
+                });
+            }
             const cacheKey = JSON.stringify({
                 modelName,
                 clientOptions: mergedOptions,
@@ -466,7 +544,10 @@ let V3 = (() => {
             if (cached) {
                 return cached;
             }
-            const client = this.llmProvider.getClient(modelName, mergedOptions, { experimental: this.experimental, disableAPI: this.disableAPI });
+            const client = this.llmProvider.getClient(modelName, mergedOptions, {
+                experimental: this.experimental,
+                disableAPI: this.disableAPI,
+            });
             this.overrideLlmClients.set(cacheKey, client);
             return client;
         }
@@ -1543,6 +1624,7 @@ let V3 = (() => {
                         if (this.apiClient && !this.experimental) {
                             const page = await this.ctx.awaitActivePage();
                             result = await this.apiClient.agentExecute(options, resolvedOptions, page.mainFrameId(), !!cacheContext);
+                            this.updateAgentMetricsFromUsage(result.usage);
                             if (cacheContext) {
                                 const transferredEntry = this.apiClient.consumeLatestAgentCacheEntry();
                                 await this.agentCache.storeTransferredEntry(transferredEntry);
@@ -1626,6 +1708,7 @@ let V3 = (() => {
                     if (this.apiClient && !this.experimental) {
                         const page = await this.ctx.awaitActivePage();
                         result = await this.apiClient.agentExecute(options ?? {}, resolvedOptions, page.mainFrameId(), !!cacheContext);
+                        this.updateAgentMetricsFromUsage(result.usage);
                         if (cacheContext) {
                             const transferredEntry = this.apiClient.consumeLatestAgentCacheEntry();
                             await this.agentCache.storeTransferredEntry(transferredEntry);