cleanup tools

Author: Kylejeong2

stagehand/config.d.ts

@@ -70,16 +70,4 @@ export type Config = {
          */
         host?: string;
     };
-    tools?: {
-        /**
-         * Configuration for the browser_take_screenshot tool.
-         */
-        browserbase_take_screenshot?: {
-            /**
-             * Whether to disable base64-encoded image responses to the clients that
-             * don't support binary data or prefer to save on tokens.
-            */
-            omitBase64?: boolean;
-        }
-    }
 };

stagehand/src/config.ts

@@ -45,7 +45,6 @@ export async function resolveConfig(cliOptions: CLIOptions): Promise<Config> {
   const mergedConfig = mergeConfig(defaultConfig, cliConfig);
 
   // --- Add Browserbase Env Vars ---
-  // Ensure env vars are read *after* dotenv potentially runs (in index.ts)
   if (!mergedConfig.browserbaseApiKey) {
     mergedConfig.browserbaseApiKey = process.env.BROWSERBASE_API_KEY;
   }

stagehand/src/tools/act.ts

@@ -58,8 +58,6 @@ async function handleAct(
 
   return {
     action,
-    captureSnapshot: false,
-    code: [],
     waitForNetwork: false,
   };
 }

stagehand/src/tools/extract.ts

@@ -65,8 +65,6 @@ async function handleExtract(
 
   return {
     action,
-    captureSnapshot: false,
-    code: [],
     waitForNetwork: false,
   };
 }

stagehand/src/tools/navigate.ts

@@ -49,8 +49,6 @@ async function handleNavigate(
 
   return {
     action,
-    captureSnapshot: false,
-    code: [],
     waitForNetwork: false,
   };
 }

stagehand/src/tools/observe.ts

@@ -46,8 +46,6 @@ async function handleObserve(
 
   return {
     action,
-    captureSnapshot: false,
-    code: [],
     waitForNetwork: false,
   };
 }

stagehand/src/tools/screenshot.ts

@@ -69,8 +69,6 @@ async function handleScreenshot(
 
   return {
     action,
-    captureSnapshot: false,
-    code: [],
     waitForNetwork: false,
   };
 }

stagehand/src/tools/session.ts

@@ -98,8 +98,6 @@ async function handleCreateSession(
   // Return the ToolResult structure expected by Context.run
   return {
     action: action, 
-    captureSnapshot: false, 
-    code: [],  
     waitForNetwork: false, 
   };
 }
@@ -131,8 +129,6 @@ async function handleCloseSession(
   context: Context,
   _params: CloseSessionInput
 ): Promise<ToolResult> {
-  const code = [`// Attempting to close the current Browserbase session.`];
-
   const action = async (): Promise<ToolActionResult> => {
     // Store the current session ID before it's potentially changed.
     // This allows us to reference the original session ID later if needed.
@@ -217,8 +213,6 @@ async function handleCloseSession(
 
   return {
     action: action,
-    code: code,
-    captureSnapshot: false,
     waitForNetwork: false,
   };
 }

stagehand/src/tools/tool.ts

@@ -17,11 +17,8 @@ export type InputType = z.Schema;
 export type ToolActionResult = { content?: (ImageContent | TextContent)[] } | undefined | void;
 
 export type ToolResult = {
-  code: string[];
   action?: () => Promise<ToolActionResult>;
-  captureSnapshot: boolean;
   waitForNetwork: boolean;
-  resultOverride?: ToolActionResult;
 };
 
 export type Tool<Input extends InputType = InputType> = {

stagehand/src/transport.ts

@@ -34,8 +34,7 @@ async function handleSSE(req: http.IncomingMessage, res: http.ServerResponse, ur
     res.on('close', () => {
       sessions.delete(transport.sessionId);
       serverList.close(server).catch(e => {
-        // eslint-disable-next-line no-console
-        // console.error(e);
+        console.error(e);
       });
     });
     return await server.connect(transport);