@browserbasehq/stagehand 3.0.8-alpha-0149ad50074dd78904613508baf67cbbe19d66e5 → 3.0.8-alpha-72ac775a831d6f0f376ceda4426525f93cc21452

package/dist/index.d.ts

@@ -8,7 +8,7 @@ import { ClientOptions as ClientOptions$1 } from 'openai';
 import { Client, ClientOptions as ClientOptions$3 } from '@modelcontextprotocol/sdk/client/index.js';
 import * as ai 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, InferUITools } from 'ai';
-export { ModelMessage } from 'ai';
+export { ModelMessage, Tool, tool } 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';

package/dist/index.js

@@ -27155,6 +27155,7 @@ __export(v3_exports, {
   providerEnvVarMap: () => providerEnvVarMap,
   toGeminiSchema: () => toGeminiSchema,
   toJsonSchema: () => toJsonSchema,
+  tool: () => import_ai23.tool,
   transformSchema: () => transformSchema,
   trimTrailingTextNode: () => trimTrailingTextNode,
   validateZodSchema: () => validateZodSchema
@@ -28127,7 +28128,7 @@ var ActCache = class {
       };
     });
   }
-  tryReplay(context, page, timeout) {
+  tryReplay(context, page, timeout, llmClientOverride) {
     return __async(this, null, function* () {
       var _a4;
       if (!this.enabled) return null;
@@ -28180,7 +28181,13 @@ var ActCache = class {
           }
         }
       });
-      return yield this.replayCachedActions(context, entry, page, timeout);
+      return yield this.replayCachedActions(
+        context,
+        entry,
+        page,
+        timeout,
+        llmClientOverride
+      );
     });
   }
   store(context, result) {
@@ -28230,12 +28237,13 @@ var ActCache = class {
     });
     return (0, import_crypto.createHash)("sha256").update(payload).digest("hex");
   }
-  replayCachedActions(context, entry, page, timeout) {
+  replayCachedActions(context, entry, page, timeout, llmClientOverride) {
     return __async(this, null, function* () {
       const handler = this.getActHandler();
       if (!handler) {
         throw new StagehandNotInitializedError("act()");
       }
+      const effectiveClient = llmClientOverride != null ? llmClientOverride : this.getDefaultLlmClient();
       const execute = () => __async(this, null, function* () {
         var _a4, _b, _c;
         const actionResults = [];
@@ -28244,7 +28252,7 @@ var ActCache = class {
             action,
             page,
             this.domSettleTimeoutMs,
-            this.getDefaultLlmClient(),
+            effectiveClient,
             void 0,
             context.variables
           );
@@ -28482,7 +28490,7 @@ var AgentCache = class {
       };
     });
   }
-  tryReplay(context) {
+  tryReplay(context, llmClientOverride) {
     return __async(this, null, function* () {
       if (!this.enabled) return null;
       const {
@@ -28515,7 +28523,7 @@ var AgentCache = class {
           url: { value: context.startUrl, type: "string" }
         }
       });
-      return yield this.replayAgentCacheEntry(context, entry);
+      return yield this.replayAgentCacheEntry(context, entry, llmClientOverride);
     });
   }
   /**
@@ -28534,9 +28542,9 @@ var AgentCache = class {
    * and await `result` even when the response comes from cache, maintaining
    * API consistency regardless of whether the result was cached or live.
    */
-  tryReplayAsStream(context) {
+  tryReplayAsStream(context, llmClientOverride) {
     return __async(this, null, function* () {
-      const result = yield this.tryReplay(context);
+      const result = yield this.tryReplay(context, llmClientOverride);
       if (!result) return null;
       return this.createCachedStreamResult(result);
     });
@@ -28762,17 +28770,23 @@ var AgentCache = class {
     }
     return value;
   }
-  replayAgentCacheEntry(context, entry) {
+  replayAgentCacheEntry(context, entry, llmClientOverride) {
     return __async(this, null, function* () {
       var _a4, _b, _c;
       const ctx = this.getContext();
       const handler = this.getActHandler();
       if (!ctx || !handler) return null;
+      const effectiveClient = llmClientOverride != null ? llmClientOverride : this.getDefaultLlmClient();
       try {
         const updatedSteps = [];
         let stepsChanged = false;
         for (const step of (_a4 = entry.steps) != null ? _a4 : []) {
-          const replayedStep = (_b = yield this.executeAgentReplayStep(step, ctx, handler)) != null ? _b : step;
+          const replayedStep = (_b = yield this.executeAgentReplayStep(
+            step,
+            ctx,
+            handler,
+            effectiveClient
+          )) != null ? _b : step;
           stepsChanged || (stepsChanged = replayedStep !== step);
           updatedSteps.push(replayedStep);
         }
@@ -28805,20 +28819,22 @@ var AgentCache = class {
       }
     });
   }
-  executeAgentReplayStep(step, ctx, handler) {
+  executeAgentReplayStep(step, ctx, handler, llmClient) {
     return __async(this, null, function* () {
       switch (step.type) {
         case "act":
           return yield this.replayAgentActStep(
             step,
             ctx,
-            handler
+            handler,
+            llmClient
           );
         case "fillForm":
           return yield this.replayAgentFillFormStep(
             step,
             ctx,
-            handler
+            handler,
+            llmClient
           );
         case "goto":
           yield this.replayAgentGotoStep(step, ctx);
@@ -28850,7 +28866,7 @@ var AgentCache = class {
       }
     });
   }
-  replayAgentActStep(step, ctx, handler) {
+  replayAgentActStep(step, ctx, handler, llmClient) {
     return __async(this, null, function* () {
       const actions = Array.isArray(step.actions) ? step.actions : [];
       if (actions.length > 0) {
@@ -28861,7 +28877,7 @@ var AgentCache = class {
             action,
             page,
             this.domSettleTimeoutMs,
-            this.getDefaultLlmClient()
+            llmClient
           );
           if (result.success && Array.isArray(result.actions)) {
             updatedActions.push(...cloneForCache(result.actions));
@@ -28878,7 +28894,7 @@ var AgentCache = class {
       return step;
     });
   }
-  replayAgentFillFormStep(step, ctx, handler) {
+  replayAgentFillFormStep(step, ctx, handler, llmClient) {
     return __async(this, null, function* () {
       var _a4;
       const actions = Array.isArray(step.actions) && step.actions.length > 0 ? step.actions : (_a4 = step.observeResults) != null ? _a4 : [];
@@ -28892,7 +28908,7 @@ var AgentCache = class {
           action,
           page,
           this.domSettleTimeoutMs,
-          this.getDefaultLlmClient()
+          llmClient
         );
         if (result.success && Array.isArray(result.actions)) {
           updatedActions.push(...cloneForCache(result.actions));
@@ -32340,8 +32356,8 @@ function buildToolsSection(isHybridMode, hasSearch, excludeTools) {
       description: "Perform a web search and return results. Prefer this over navigating to Google and searching within the page for reliability and efficiency."
     });
   }
-  const filteredTools = baseTools.filter((tool21) => !excludeSet.has(tool21.name));
-  const toolLines = filteredTools.map((tool21) => `    <tool name="${tool21.name}">${tool21.description}</tool>`).join("\n");
+  const filteredTools = baseTools.filter((tool22) => !excludeSet.has(tool22.name));
+  const toolLines = filteredTools.map((tool22) => `    <tool name="${tool22.name}">${tool22.description}</tool>`).join("\n");
   return `<tools>
 ${toolLines}
   </tools>`;
@@ -33534,8 +33550,8 @@ var AnthropicCUAClient = class extends AgentClient {
           betas: ["computer-use-2025-01-24"]
         };
         if (this.tools && Object.keys(this.tools).length > 0) {
-          const customTools = Object.entries(this.tools).map(([name, tool21]) => {
-            const schema = tool21.inputSchema;
+          const customTools = Object.entries(this.tools).map(([name, tool22]) => {
+            const schema = tool22.inputSchema;
             const jsonSchema3 = toJsonSchema(schema);
             const inputSchema = {
               type: "object",
@@ -33544,7 +33560,7 @@ var AnthropicCUAClient = class extends AgentClient {
             };
             return {
               name,
-              description: tool21.description,
+              description: tool22.description,
               input_schema: inputSchema
             };
           });
@@ -33662,13 +33678,13 @@ var AnthropicCUAClient = class extends AgentClient {
             let toolResult = "Tool executed successfully";
             if (this.tools && item.name in this.tools) {
               try {
-                const tool21 = this.tools[item.name];
+                const tool22 = this.tools[item.name];
                 logger({
                   category: "agent",
                   message: `Executing tool call: ${item.name} with args: ${JSON.stringify(item.input)}`,
                   level: 1
                 });
-                const result = yield tool21.execute(item.input, {
+                const result = yield tool22.execute(item.input, {
                   toolCallId: item.id,
                   messages: []
                 });
@@ -34230,13 +34246,13 @@ var OpenAICUAClient = class extends AgentClient {
           truncation: "auto"
         };
         if (this.tools && Object.keys(this.tools).length > 0) {
-          const customTools = Object.entries(this.tools).map(([name, tool21]) => ({
+          const customTools = Object.entries(this.tools).map(([name, tool22]) => ({
             type: "function",
             name,
             function: {
               name,
-              description: tool21.description,
-              parameters: tool21.inputSchema
+              description: tool22.description,
+              parameters: tool22.inputSchema
             }
           }));
           requestParams.tools = [
@@ -34388,14 +34404,14 @@ var OpenAICUAClient = class extends AgentClient {
             let toolResult = "Tool executed successfully";
             if (this.tools && item.name in this.tools) {
               try {
-                const tool21 = this.tools[item.name];
+                const tool22 = this.tools[item.name];
                 const args = JSON.parse(item.arguments);
                 logger({
                   category: "agent",
                   message: `Executing tool call: ${item.name} with args: ${item.arguments}`,
                   level: 1
                 });
-                const result = yield tool21.execute(args, {
+                const result = yield tool22.execute(args, {
                   toolCallId: item.call_id,
                   messages: []
                 });
@@ -34558,8 +34574,8 @@ function executeGoogleCustomTool(toolName, toolArgs, tools, functionCall, logger
         message: `Executing custom tool: ${toolName} with args: ${JSON.stringify(toolArgs)}`,
         level: 1
       });
-      const tool21 = tools[toolName];
-      const toolResult = yield tool21.execute(toolArgs, {
+      const tool22 = tools[toolName];
+      const toolResult = yield tool22.execute(toolArgs, {
         toolCallId: `tool_${Date.now()}`,
         messages: []
       });
@@ -34607,22 +34623,22 @@ function isCustomTool(functionCall, tools) {
 }
 function convertToolSetToFunctionDeclarations(tools) {
   const functionDeclarations = [];
-  for (const [name, tool21] of Object.entries(tools)) {
-    const functionDeclaration = convertToolToFunctionDeclaration(name, tool21);
+  for (const [name, tool22] of Object.entries(tools)) {
+    const functionDeclaration = convertToolToFunctionDeclaration(name, tool22);
     if (functionDeclaration) {
       functionDeclarations.push(functionDeclaration);
     }
   }
   return functionDeclarations;
 }
-function convertToolToFunctionDeclaration(name, tool21) {
+function convertToolToFunctionDeclaration(name, tool22) {
   try {
-    const schema = tool21.inputSchema;
+    const schema = tool22.inputSchema;
     const jsonSchema3 = toJsonSchema(schema);
     const parameters = convertJsonSchemaToGoogleParameters(jsonSchema3);
     return {
       name,
-      description: tool21.description || `Execute ${name}`,
+      description: tool22.description || `Execute ${name}`,
       parameters
     };
   } catch (error) {
@@ -38113,10 +38129,10 @@ You must respond in JSON format. respond WITH JSON. Do not include any other tex
       }
       const tools = {};
       if (options.tools && options.tools.length > 0) {
-        for (const tool21 of options.tools) {
-          tools[tool21.name] = {
-            description: tool21.description,
-            inputSchema: tool21.parameters
+        for (const tool22 of options.tools) {
+          tools[tool22.name] = {
+            description: tool22.description,
+            inputSchema: tool22.parameters
           };
         }
       }
@@ -38315,14 +38331,14 @@ var AnthropicClient = class extends LLMClient {
         }
         formattedMessages.push(screenshotMessage);
       }
-      let anthropicTools2 = (_a4 = options.tools) == null ? void 0 : _a4.map((tool21) => {
+      let anthropicTools2 = (_a4 = options.tools) == null ? void 0 : _a4.map((tool22) => {
         return {
-          name: tool21.name,
-          description: tool21.description,
+          name: tool22.name,
+          description: tool22.description,
           input_schema: {
             type: "object",
-            properties: tool21.parameters.properties,
-            required: tool21.parameters.required
+            properties: tool22.parameters.properties,
+            required: tool22.parameters.required
           }
         };
       });
@@ -38509,15 +38525,15 @@ var CerebrasClient = class extends LLMClient {
           return __spreadProps(__spreadValues({}, baseMessage), { role: "user" });
         }
       });
-      let tools = (_a4 = options.tools) == null ? void 0 : _a4.map((tool21) => ({
+      let tools = (_a4 = options.tools) == null ? void 0 : _a4.map((tool22) => ({
         type: "function",
         function: {
-          name: tool21.name,
-          description: tool21.description,
+          name: tool22.name,
+          description: tool22.description,
           parameters: {
             type: "object",
-            properties: tool21.parameters.properties,
-            required: tool21.parameters.required
+            properties: tool22.parameters.properties,
+            required: tool22.parameters.required
           }
         }
       }));
@@ -38810,18 +38826,18 @@ ${firstPartText.text}`;
     }
     return [
       {
-        functionDeclarations: tools.map((tool21) => {
+        functionDeclarations: tools.map((tool22) => {
           let parameters = void 0;
-          if (tool21.parameters) {
+          if (tool22.parameters) {
             parameters = {
               type: import_genai4.Type.OBJECT,
-              properties: tool21.parameters.properties,
-              required: tool21.parameters.required
+              properties: tool22.parameters.properties,
+              required: tool22.parameters.required
             };
           }
           return {
-            name: tool21.name,
-            description: tool21.description,
+            name: tool22.name,
+            description: tool22.description,
             parameters
           };
         })
@@ -39097,15 +39113,15 @@ var GroqClient = class extends LLMClient {
           return __spreadProps(__spreadValues({}, baseMessage), { role: "user" });
         }
       });
-      let tools = (_a4 = options.tools) == null ? void 0 : _a4.map((tool21) => ({
+      let tools = (_a4 = options.tools) == null ? void 0 : _a4.map((tool22) => ({
         type: "function",
         function: {
-          name: tool21.name,
-          description: tool21.description,
+          name: tool22.name,
+          description: tool22.description,
           parameters: {
             type: "object",
-            properties: tool21.parameters.properties,
-            required: tool21.parameters.required
+            properties: tool22.parameters.properties,
+            required: tool22.parameters.required
           }
         }
       }));
@@ -39482,11 +39498,11 @@ ${parsedSchema}
         messages: formattedMessages,
         response_format: responseFormat,
         stream: false,
-        tools: (_e = options.tools) == null ? void 0 : _e.map((tool21) => ({
+        tools: (_e = options.tools) == null ? void 0 : _e.map((tool22) => ({
           function: {
-            name: tool21.name,
-            description: tool21.description,
-            parameters: tool21.parameters
+            name: tool22.name,
+            description: tool22.description,
+            parameters: tool22.parameters
           },
           type: "function"
         }))
@@ -42152,21 +42168,21 @@ function prepareChatTools({
     return { tools: void 0, toolChoice: void 0, toolWarnings };
   }
   const openaiTools2 = [];
-  for (const tool21 of tools) {
-    switch (tool21.type) {
+  for (const tool22 of tools) {
+    switch (tool22.type) {
       case "function":
         openaiTools2.push({
           type: "function",
           function: {
-            name: tool21.name,
-            description: tool21.description,
-            parameters: tool21.inputSchema,
+            name: tool22.name,
+            description: tool22.description,
+            parameters: tool22.inputSchema,
             strict: structuredOutputs ? strictJsonSchema : void 0
           }
         });
         break;
       default:
-        toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+        toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
         break;
     }
   }
@@ -44482,22 +44498,22 @@ function prepareResponsesTools(_0) {
       return { tools: void 0, toolChoice: void 0, toolWarnings };
     }
     const openaiTools2 = [];
-    for (const tool21 of tools) {
-      switch (tool21.type) {
+    for (const tool22 of tools) {
+      switch (tool22.type) {
         case "function":
           openaiTools2.push({
             type: "function",
-            name: tool21.name,
-            description: tool21.description,
-            parameters: tool21.inputSchema,
+            name: tool22.name,
+            description: tool22.description,
+            parameters: tool22.inputSchema,
             strict: strictJsonSchema
           });
           break;
         case "provider-defined": {
-          switch (tool21.id) {
+          switch (tool22.id) {
             case "openai.file_search": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: fileSearchArgsSchema
               });
               openaiTools2.push({
@@ -44520,7 +44536,7 @@ function prepareResponsesTools(_0) {
             }
             case "openai.web_search_preview": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: webSearchPreviewArgsSchema
               });
               openaiTools2.push({
@@ -44532,7 +44548,7 @@ function prepareResponsesTools(_0) {
             }
             case "openai.web_search": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: webSearchArgsSchema
               });
               openaiTools2.push({
@@ -44545,7 +44561,7 @@ function prepareResponsesTools(_0) {
             }
             case "openai.code_interpreter": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: codeInterpreterArgsSchema
               });
               openaiTools2.push({
@@ -44556,7 +44572,7 @@ function prepareResponsesTools(_0) {
             }
             case "openai.image_generation": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: imageGenerationArgsSchema
               });
               openaiTools2.push({
@@ -44580,7 +44596,7 @@ function prepareResponsesTools(_0) {
           break;
         }
         default:
-          toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+          toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
           break;
       }
     }
@@ -44685,7 +44701,7 @@ var OpenAIResponsesLanguageModel = class {
       }
       function hasOpenAITool(id) {
         return (tools == null ? void 0 : tools.find(
-          (tool21) => tool21.type === "provider-defined" && tool21.id === id
+          (tool22) => tool22.type === "provider-defined" && tool22.id === id
         )) != null;
       }
       const topLogprobs = typeof (openaiOptions == null ? void 0 : openaiOptions.logprobs) === "number" ? openaiOptions == null ? void 0 : openaiOptions.logprobs : (openaiOptions == null ? void 0 : openaiOptions.logprobs) === true ? TOP_LOGPROBS_MAX : void 0;
@@ -44693,7 +44709,7 @@ var OpenAIResponsesLanguageModel = class {
         addInclude("message.output_text.logprobs");
       }
       const webSearchToolName = (_c = tools == null ? void 0 : tools.find(
-        (tool21) => tool21.type === "provider-defined" && (tool21.id === "openai.web_search" || tool21.id === "openai.web_search_preview")
+        (tool22) => tool22.type === "provider-defined" && (tool22.id === "openai.web_search" || tool22.id === "openai.web_search_preview")
       )) == null ? void 0 : _c.name;
       if (webSearchToolName) {
         addInclude("web_search_call.action.sources");
@@ -48287,25 +48303,25 @@ function prepareTools({
   if (tools == null) {
     return { tools: void 0, toolConfig: void 0, toolWarnings };
   }
-  const hasFunctionTools = tools.some((tool21) => tool21.type === "function");
+  const hasFunctionTools = tools.some((tool22) => tool22.type === "function");
   const hasProviderDefinedTools = tools.some(
-    (tool21) => tool21.type === "provider-defined"
+    (tool22) => tool22.type === "provider-defined"
   );
   if (hasFunctionTools && hasProviderDefinedTools) {
-    const functionTools = tools.filter((tool21) => tool21.type === "function");
+    const functionTools = tools.filter((tool22) => tool22.type === "function");
     toolWarnings.push({
       type: "unsupported-tool",
-      tool: tools.find((tool21) => tool21.type === "function"),
+      tool: tools.find((tool22) => tool22.type === "function"),
       details: `Cannot mix function tools with provider-defined tools in the same request. Falling back to provider-defined tools only. The following function tools will be ignored: ${functionTools.map((t2) => t2.name).join(", ")}. Please use either function tools or provider-defined tools, but not both.`
     });
   }
   if (hasProviderDefinedTools) {
     const googleTools22 = [];
     const providerDefinedTools = tools.filter(
-      (tool21) => tool21.type === "provider-defined"
+      (tool22) => tool22.type === "provider-defined"
     );
-    providerDefinedTools.forEach((tool21) => {
-      switch (tool21.id) {
+    providerDefinedTools.forEach((tool22) => {
+      switch (tool22.id) {
         case "google.google_search":
           if (isGemini2orNewer) {
             googleTools22.push({ googleSearch: {} });
@@ -48313,8 +48329,8 @@ function prepareTools({
             googleTools22.push({
               googleSearchRetrieval: {
                 dynamicRetrievalConfig: {
-                  mode: tool21.args.mode,
-                  dynamicThreshold: tool21.args.dynamicThreshold
+                  mode: tool22.args.mode,
+                  dynamicThreshold: tool22.args.dynamicThreshold
                 }
               }
             });
@@ -48328,7 +48344,7 @@ function prepareTools({
           } else {
             toolWarnings.push({
               type: "unsupported-tool",
-              tool: tool21,
+              tool: tool22,
               details: "The URL context tool is not supported with other Gemini models than Gemini 2."
             });
           }
@@ -48339,18 +48355,18 @@ function prepareTools({
           } else {
             toolWarnings.push({
               type: "unsupported-tool",
-              tool: tool21,
+              tool: tool22,
               details: "The code execution tools is not supported with other Gemini models than Gemini 2."
             });
           }
           break;
         case "google.file_search":
           if (supportsFileSearch) {
-            googleTools22.push({ fileSearch: __spreadValues({}, tool21.args) });
+            googleTools22.push({ fileSearch: __spreadValues({}, tool22.args) });
           } else {
             toolWarnings.push({
               type: "unsupported-tool",
-              tool: tool21,
+              tool: tool22,
               details: "The file search tool is only supported with Gemini 2.5 models."
             });
           }
@@ -48361,22 +48377,22 @@ function prepareTools({
               retrieval: {
                 vertex_rag_store: {
                   rag_resources: {
-                    rag_corpus: tool21.args.ragCorpus
+                    rag_corpus: tool22.args.ragCorpus
                   },
-                  similarity_top_k: tool21.args.topK
+                  similarity_top_k: tool22.args.topK
                 }
               }
             });
           } else {
             toolWarnings.push({
               type: "unsupported-tool",
-              tool: tool21,
+              tool: tool22,
               details: "The RAG store tool is not supported with other Gemini models than Gemini 2."
             });
           }
           break;
         default:
-          toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+          toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
           break;
       }
     });
@@ -48387,17 +48403,17 @@ function prepareTools({
     };
   }
   const functionDeclarations = [];
-  for (const tool21 of tools) {
-    switch (tool21.type) {
+  for (const tool22 of tools) {
+    switch (tool22.type) {
       case "function":
         functionDeclarations.push({
-          name: tool21.name,
-          description: (_a4 = tool21.description) != null ? _a4 : "",
-          parameters: convertJSONSchemaToOpenAPISchema(tool21.inputSchema)
+          name: tool22.name,
+          description: (_a4 = tool22.description) != null ? _a4 : "",
+          parameters: convertJSONSchemaToOpenAPISchema(tool22.inputSchema)
         });
         break;
       default:
-        toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+        toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
         break;
     }
   }
@@ -48511,7 +48527,7 @@ var GoogleGenerativeAILanguageModel = class {
         schema: googleGenerativeAIProviderOptions
       });
       if ((tools == null ? void 0 : tools.some(
-        (tool21) => tool21.type === "provider-defined" && tool21.id === "google.vertex_rag_store"
+        (tool22) => tool22.type === "provider-defined" && tool22.id === "google.vertex_rag_store"
       )) && !this.config.provider.startsWith("google.vertex.")) {
         warnings.push({
           type: "other",
@@ -50373,23 +50389,23 @@ function prepareTools2(_0) {
       return { tools: void 0, toolChoice: void 0, toolWarnings, betas };
     }
     const anthropicTools2 = [];
-    for (const tool21 of tools) {
-      switch (tool21.type) {
+    for (const tool22 of tools) {
+      switch (tool22.type) {
         case "function": {
-          const cacheControl = validator3.getCacheControl(tool21.providerOptions, {
+          const cacheControl = validator3.getCacheControl(tool22.providerOptions, {
             type: "tool definition",
             canCache: true
           });
           anthropicTools2.push({
-            name: tool21.name,
-            description: tool21.description,
-            input_schema: tool21.inputSchema,
+            name: tool22.name,
+            description: tool22.description,
+            input_schema: tool22.inputSchema,
             cache_control: cacheControl
           });
           break;
         }
         case "provider-defined": {
-          switch (tool21.id) {
+          switch (tool22.id) {
             case "anthropic.code_execution_20250522": {
               betas.add("code-execution-2025-05-22");
               anthropicTools2.push({
@@ -50412,9 +50428,9 @@ function prepareTools2(_0) {
               anthropicTools2.push({
                 name: "computer",
                 type: "computer_20250124",
-                display_width_px: tool21.args.displayWidthPx,
-                display_height_px: tool21.args.displayHeightPx,
-                display_number: tool21.args.displayNumber,
+                display_width_px: tool22.args.displayWidthPx,
+                display_height_px: tool22.args.displayHeightPx,
+                display_number: tool22.args.displayNumber,
                 cache_control: void 0
               });
               break;
@@ -50424,9 +50440,9 @@ function prepareTools2(_0) {
               anthropicTools2.push({
                 name: "computer",
                 type: "computer_20241022",
-                display_width_px: tool21.args.displayWidthPx,
-                display_height_px: tool21.args.displayHeightPx,
-                display_number: tool21.args.displayNumber,
+                display_width_px: tool22.args.displayWidthPx,
+                display_height_px: tool22.args.displayHeightPx,
+                display_number: tool22.args.displayNumber,
                 cache_control: void 0
               });
               break;
@@ -50460,7 +50476,7 @@ function prepareTools2(_0) {
             }
             case "anthropic.text_editor_20250728": {
               const args = yield validateTypes2({
-                value: tool21.args,
+                value: tool22.args,
                 schema: textEditor_20250728ArgsSchema
               });
               anthropicTools2.push({
@@ -50500,7 +50516,7 @@ function prepareTools2(_0) {
             case "anthropic.web_fetch_20250910": {
               betas.add("web-fetch-2025-09-10");
               const args = yield validateTypes2({
-                value: tool21.args,
+                value: tool22.args,
                 schema: webFetch_20250910ArgsSchema
               });
               anthropicTools2.push({
@@ -50517,7 +50533,7 @@ function prepareTools2(_0) {
             }
             case "anthropic.web_search_20250305": {
               const args = yield validateTypes2({
-                value: tool21.args,
+                value: tool22.args,
                 schema: webSearch_20250305ArgsSchema
               });
               anthropicTools2.push({
@@ -50532,14 +50548,14 @@ function prepareTools2(_0) {
               break;
             }
             default: {
-              toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+              toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
               break;
             }
           }
           break;
         }
         default: {
-          toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+          toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
           break;
         }
       }
@@ -51498,7 +51514,7 @@ var AnthropicMessagesLanguageModel = class {
         betas.add("skills-2025-10-02");
         betas.add("files-api-2025-04-14");
         if (!(tools == null ? void 0 : tools.some(
-          (tool21) => tool21.type === "provider-defined" && tool21.id === "anthropic.code_execution_20250825"
+          (tool22) => tool22.type === "provider-defined" && tool22.id === "anthropic.code_execution_20250825"
         ))) {
           warnings.push({
             type: "other",
@@ -53336,24 +53352,24 @@ function prepareTools3({
   if (tools == null) {
     return { tools: void 0, toolConfig: void 0, toolWarnings };
   }
-  const hasFunctionTools = tools.some((tool21) => tool21.type === "function");
+  const hasFunctionTools = tools.some((tool22) => tool22.type === "function");
   const hasProviderDefinedTools = tools.some(
-    (tool21) => tool21.type === "provider-defined"
+    (tool22) => tool22.type === "provider-defined"
   );
   if (hasFunctionTools && hasProviderDefinedTools) {
     toolWarnings.push({
       type: "unsupported-tool",
-      tool: tools.find((tool21) => tool21.type === "function"),
+      tool: tools.find((tool22) => tool22.type === "function"),
       details: "Cannot mix function tools with provider-defined tools in the same request. Please use either function tools or provider-defined tools, but not both."
     });
   }
   if (hasProviderDefinedTools) {
     const googleTools22 = [];
     const providerDefinedTools = tools.filter(
-      (tool21) => tool21.type === "provider-defined"
+      (tool22) => tool22.type === "provider-defined"
     );
-    providerDefinedTools.forEach((tool21) => {
-      switch (tool21.id) {
+    providerDefinedTools.forEach((tool22) => {
+      switch (tool22.id) {
         case "google.google_search":
           if (isGemini2) {
             googleTools22.push({ googleSearch: {} });
@@ -53361,8 +53377,8 @@ function prepareTools3({
             googleTools22.push({
               googleSearchRetrieval: {
                 dynamicRetrievalConfig: {
-                  mode: tool21.args.mode,
-                  dynamicThreshold: tool21.args.dynamicThreshold
+                  mode: tool22.args.mode,
+                  dynamicThreshold: tool22.args.dynamicThreshold
                 }
               }
             });
@@ -53376,7 +53392,7 @@ function prepareTools3({
           } else {
             toolWarnings.push({
               type: "unsupported-tool",
-              tool: tool21,
+              tool: tool22,
               details: "The URL context tool is not supported with other Gemini models than Gemini 2."
             });
           }
@@ -53387,13 +53403,13 @@ function prepareTools3({
           } else {
             toolWarnings.push({
               type: "unsupported-tool",
-              tool: tool21,
+              tool: tool22,
               details: "The code execution tools is not supported with other Gemini models than Gemini 2."
             });
           }
           break;
         default:
-          toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+          toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
           break;
       }
     });
@@ -53404,17 +53420,17 @@ function prepareTools3({
     };
   }
   const functionDeclarations = [];
-  for (const tool21 of tools) {
-    switch (tool21.type) {
+  for (const tool22 of tools) {
+    switch (tool22.type) {
       case "function":
         functionDeclarations.push({
-          name: tool21.name,
-          description: (_a4 = tool21.description) != null ? _a4 : "",
-          parameters: convertJSONSchemaToOpenAPISchema2(tool21.inputSchema)
+          name: tool22.name,
+          description: (_a4 = tool22.description) != null ? _a4 : "",
+          parameters: convertJSONSchemaToOpenAPISchema2(tool22.inputSchema)
         });
         break;
       default:
-        toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+        toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
         break;
     }
   }
@@ -54537,16 +54553,16 @@ function prepareTools4({
     return { tools: void 0, toolChoice: void 0, toolWarnings };
   }
   const openaiCompatTools = [];
-  for (const tool21 of tools) {
-    if (tool21.type === "provider-defined") {
-      toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+  for (const tool22 of tools) {
+    if (tool22.type === "provider-defined") {
+      toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
     } else {
       openaiCompatTools.push({
         type: "function",
         function: {
-          name: tool21.name,
-          description: tool21.description,
-          parameters: tool21.inputSchema
+          name: tool22.name,
+          description: tool22.description,
+          parameters: tool22.inputSchema
         }
       });
     }
@@ -55891,16 +55907,16 @@ function prepareTools5({
     return { tools: void 0, toolChoice: void 0, toolWarnings };
   }
   const xaiTools = [];
-  for (const tool21 of tools) {
-    if (tool21.type === "provider-defined") {
-      toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+  for (const tool22 of tools) {
+    if (tool22.type === "provider-defined") {
+      toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
     } else {
       xaiTools.push({
         type: "function",
         function: {
-          name: tool21.name,
-          description: tool21.description,
-          parameters: tool21.inputSchema
+          name: tool22.name,
+          description: tool22.description,
+          parameters: tool22.inputSchema
         }
       });
     }
@@ -56900,21 +56916,21 @@ function prepareChatTools2({
     return { tools: void 0, toolChoice: void 0, toolWarnings };
   }
   const openaiTools2 = [];
-  for (const tool21 of tools) {
-    switch (tool21.type) {
+  for (const tool22 of tools) {
+    switch (tool22.type) {
       case "function":
         openaiTools2.push({
           type: "function",
           function: {
-            name: tool21.name,
-            description: tool21.description,
-            parameters: tool21.inputSchema,
+            name: tool22.name,
+            description: tool22.description,
+            parameters: tool22.inputSchema,
             strict: structuredOutputs ? strictJsonSchema : void 0
           }
         });
         break;
       default:
-        toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+        toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
         break;
     }
   }
@@ -59496,22 +59512,22 @@ function prepareResponsesTools2(_0) {
       return { tools: void 0, toolChoice: void 0, toolWarnings };
     }
     const openaiTools2 = [];
-    for (const tool21 of tools) {
-      switch (tool21.type) {
+    for (const tool22 of tools) {
+      switch (tool22.type) {
         case "function":
           openaiTools2.push({
             type: "function",
-            name: tool21.name,
-            description: tool21.description,
-            parameters: tool21.inputSchema,
+            name: tool22.name,
+            description: tool22.description,
+            parameters: tool22.inputSchema,
             strict: strictJsonSchema
           });
           break;
         case "provider-defined": {
-          switch (tool21.id) {
+          switch (tool22.id) {
             case "openai.file_search": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: fileSearchArgsSchema3
               });
               openaiTools2.push({
@@ -59534,7 +59550,7 @@ function prepareResponsesTools2(_0) {
             }
             case "openai.web_search_preview": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: webSearchPreviewArgsSchema2
               });
               openaiTools2.push({
@@ -59546,7 +59562,7 @@ function prepareResponsesTools2(_0) {
             }
             case "openai.web_search": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: webSearchArgsSchema2
               });
               openaiTools2.push({
@@ -59559,7 +59575,7 @@ function prepareResponsesTools2(_0) {
             }
             case "openai.code_interpreter": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: codeInterpreterArgsSchema2
               });
               openaiTools2.push({
@@ -59570,7 +59586,7 @@ function prepareResponsesTools2(_0) {
             }
             case "openai.image_generation": {
               const args = yield validateTypes({
-                value: tool21.args,
+                value: tool22.args,
                 schema: imageGenerationArgsSchema2
               });
               openaiTools2.push({
@@ -59594,7 +59610,7 @@ function prepareResponsesTools2(_0) {
           break;
         }
         default:
-          toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+          toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
           break;
       }
     }
@@ -59699,7 +59715,7 @@ var OpenAIResponsesLanguageModel2 = class {
       }
       function hasOpenAITool(id) {
         return (tools == null ? void 0 : tools.find(
-          (tool21) => tool21.type === "provider-defined" && tool21.id === id
+          (tool22) => tool22.type === "provider-defined" && tool22.id === id
         )) != null;
       }
       const topLogprobs = typeof (openaiOptions == null ? void 0 : openaiOptions.logprobs) === "number" ? openaiOptions == null ? void 0 : openaiOptions.logprobs : (openaiOptions == null ? void 0 : openaiOptions.logprobs) === true ? TOP_LOGPROBS_MAX2 : void 0;
@@ -59707,7 +59723,7 @@ var OpenAIResponsesLanguageModel2 = class {
         addInclude("message.output_text.logprobs");
       }
       const webSearchToolName = (_c = tools == null ? void 0 : tools.find(
-        (tool21) => tool21.type === "provider-defined" && (tool21.id === "openai.web_search" || tool21.id === "openai.web_search_preview")
+        (tool22) => tool22.type === "provider-defined" && (tool22.id === "openai.web_search" || tool22.id === "openai.web_search_preview")
       )) == null ? void 0 : _c.name;
       if (webSearchToolName) {
         addInclude("web_search_call.action.sources");
@@ -60930,13 +60946,13 @@ function prepareTools6({
     return { tools: void 0, toolChoice: void 0, toolWarnings };
   }
   const groqTools2 = [];
-  for (const tool21 of tools) {
-    if (tool21.type === "provider-defined") {
-      if (tool21.id === "groq.browser_search") {
+  for (const tool22 of tools) {
+    if (tool22.type === "provider-defined") {
+      if (tool22.id === "groq.browser_search") {
         if (!isBrowserSearchSupportedModel(modelId)) {
           toolWarnings.push({
             type: "unsupported-tool",
-            tool: tool21,
+            tool: tool22,
             details: `Browser search is only supported on the following models: ${getSupportedModelsString()}. Current model: ${modelId}`
           });
         } else {
@@ -60945,15 +60961,15 @@ function prepareTools6({
           });
         }
       } else {
-        toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+        toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
       }
     } else {
       groqTools2.push({
         type: "function",
         function: {
-          name: tool21.name,
-          description: tool21.description,
-          parameters: tool21.inputSchema
+          name: tool22.name,
+          description: tool22.description,
+          parameters: tool22.inputSchema
         }
       });
     }
@@ -62005,16 +62021,16 @@ function prepareTools7({
     return { tools: void 0, toolChoice: void 0, toolWarnings };
   }
   const mistralTools = [];
-  for (const tool21 of tools) {
-    if (tool21.type === "provider-defined") {
-      toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+  for (const tool22 of tools) {
+    if (tool22.type === "provider-defined") {
+      toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
     } else {
       mistralTools.push({
         type: "function",
         function: {
-          name: tool21.name,
-          description: tool21.description,
-          parameters: tool21.inputSchema
+          name: tool22.name,
+          description: tool22.description,
+          parameters: tool22.inputSchema
         }
       });
     }
@@ -62034,7 +62050,7 @@ function prepareTools7({
     case "tool":
       return {
         tools: mistralTools.filter(
-          (tool21) => tool21.function.name === toolChoice.toolName
+          (tool22) => tool22.function.name === toolChoice.toolName
         ),
         toolChoice: "any",
         toolWarnings
@@ -63817,10 +63833,10 @@ function prepareResponsesTools3({
     return { tools: void 0, toolChoice: void 0, toolWarnings };
   }
   const ollamaTools = [];
-  for (const tool21 of tools) {
-    switch (tool21.type) {
+  for (const tool22 of tools) {
+    switch (tool22.type) {
       case "function": {
-        let parameters = tool21.inputSchema;
+        let parameters = tool22.inputSchema;
         if (!parameters) {
           parameters = {
             type: "object",
@@ -63836,15 +63852,15 @@ function prepareResponsesTools3({
         ollamaTools.push({
           type: "function",
           function: {
-            name: tool21.name,
-            description: tool21.description,
+            name: tool22.name,
+            description: tool22.description,
             parameters
           }
         });
         break;
       }
       default:
-        toolWarnings.push({ type: "unsupported-tool", tool: tool21 });
+        toolWarnings.push({ type: "unsupported-tool", tool: tool22 });
         break;
     }
   }
@@ -64642,13 +64658,13 @@ var resolveTools = (clients, userTools) => __async(null, null, function* () {
       const clientTools = yield clientInstance.listTools({
         cursor: nextCursor
       });
-      for (const tool21 of clientTools.tools) {
-        tools[tool21.name] = {
-          description: tool21.description,
-          inputSchema: jsonSchemaToZod(tool21.inputSchema),
+      for (const tool22 of clientTools.tools) {
+        tools[tool22.name] = {
+          description: tool22.description,
+          inputSchema: jsonSchemaToZod(tool22.inputSchema),
           execute: (input) => __async(null, null, function* () {
             const result = yield clientInstance.callTool({
-              name: tool21.name,
+              name: tool22.name,
               arguments: input
             });
             return result;
@@ -67705,6 +67721,7 @@ var _V3 = class _V3 {
           );
         }
         const page = yield this.resolvePage(options == null ? void 0 : options.page);
+        const actCacheLlmClient = (options == null ? void 0 : options.model) ? this.resolveLlmClient(options.model) : void 0;
         let actCacheContext = null;
         const canUseCache = typeof input === "string" && !this.isAgentReplayRecording() && this.actCache.enabled;
         if (canUseCache) {
@@ -67717,7 +67734,8 @@ var _V3 = class _V3 {
             const cachedResult = yield this.actCache.tryReplay(
               actCacheContext,
               page,
-              options == null ? void 0 : options.timeout
+              options == null ? void 0 : options.timeout,
+              actCacheLlmClient
             );
             if (cachedResult) {
               this.addToHistory(
@@ -68120,7 +68138,13 @@ var _V3 = class _V3 {
         configSignature: agentConfigSignature,
         page: yield this.ctx.awaitActivePage()
       }) : null;
-      return { handler, resolvedOptions, instruction, cacheContext };
+      return {
+        handler,
+        resolvedOptions,
+        instruction,
+        cacheContext,
+        llmClient: agentLlmClient
+      };
     });
   }
   agent(options) {
@@ -68270,13 +68294,16 @@ Do not ask follow up questions, the user will trust your judgement.`
             args: [instructionOrOptions]
           });
           if (isStreaming) {
-            const { handler: handler2, resolvedOptions: resolvedOptions2, cacheContext: cacheContext2 } = yield this.prepareAgentExecution(
+            const { handler: handler2, resolvedOptions: resolvedOptions2, cacheContext: cacheContext2, llmClient: llmClient2 } = yield this.prepareAgentExecution(
               options,
               instructionOrOptions,
               agentConfigSignature
             );
             if (cacheContext2) {
-              const replayed = yield this.agentCache.tryReplayAsStream(cacheContext2);
+              const replayed = yield this.agentCache.tryReplayAsStream(
+                cacheContext2,
+                llmClient2
+              );
               if (replayed) {
                 SessionFileLogger.logAgentTaskCompleted({ cacheHit: true });
                 return replayed;
@@ -68299,13 +68326,16 @@ Do not ask follow up questions, the user will trust your judgement.`
             SessionFileLogger.logAgentTaskCompleted();
             return streamResult;
           }
-          const { handler, resolvedOptions, cacheContext } = yield this.prepareAgentExecution(
+          const { handler, resolvedOptions, cacheContext, llmClient } = yield this.prepareAgentExecution(
             options,
             instructionOrOptions,
             agentConfigSignature
           );
           if (cacheContext) {
-            const replayed = yield this.agentCache.tryReplay(cacheContext);
+            const replayed = yield this.agentCache.tryReplay(
+              cacheContext,
+              llmClient
+            );
             if (replayed) {
               SessionFileLogger.logAgentTaskCompleted({ cacheHit: true });
               return replayed;
@@ -68598,6 +68628,9 @@ I'm providing ${screenshots.length} screenshots showing the progression of the t
     });
   }
 };
+
+// lib/v3/index.ts
+var import_ai23 = require("ai");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AISdkClient,
@@ -68680,6 +68713,7 @@ I'm providing ${screenshots.length} screenshots showing the progression of the t
   providerEnvVarMap,
   toGeminiSchema,
   toJsonSchema,
+  tool,
   transformSchema,
   trimTrailingTextNode,
   validateZodSchema

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@browserbasehq/stagehand",
-  "version": "3.0.8-alpha-0149ad50074dd78904613508baf67cbbe19d66e5",
+  "version": "3.0.8-alpha-72ac775a831d6f0f376ceda4426525f93cc21452",
   "description": "An AI web browsing framework focused on simplicity and extensibility.",
   "main": "./dist/index.js",
   "module": "./dist/index.js",