add exp flag and add keepalive to cli options

README.md

@@ -150,6 +150,7 @@ The Browserbase MCP server accepts the following command-line flags:
 | `--browserHeight <height>` | Browser viewport height (default: 768)                                      |
 | `--modelName <model>`      | The model to use for Stagehand (default: google/gemini-2.0-flash)           |
 | `--modelApiKey <key>`      | API key for the custom model provider (required when using custom models)   |
+| `--experimental`           | Enable experimental features (default: false)                               |
 
 These flags can be passed directly to the CLI or configured in your MCP configuration file.
 

config.d.ts

@@ -104,4 +104,10 @@ export type Config = {
    * Required when using a model other than the default google/gemini-2.0-flash
    */
   modelApiKey?: string;
+  /**
+   * Enable experimental features
+   *
+   * @default false
+   */
+  experimental?: boolean;
 };

package.json

@@ -42,7 +42,7 @@
   },
   "dependencies": {
     "@browserbasehq/sdk": "^2.6.0",
-    "@browserbasehq/stagehand": "^2.4.0",
+    "@browserbasehq/stagehand": "^2.4.3",
     "@modelcontextprotocol/sdk": "^1.13.1",
     "@smithery/cli": "^1.2.15",
     "commander": "^14.0.0",

src/config.ts

@@ -17,6 +17,8 @@ export type CLIOptions = {
   browserHeight?: number;
   modelName?: typeof AvailableModel;
   modelApiKey?: string;
+  keepAlive?: boolean;
+  experimental?: boolean;
 };
 
 // Default Configuration Values
@@ -97,6 +99,8 @@ export async function configFromCLIOptions(
     cookies: cliOptions.cookies,
     modelName: cliOptions.modelName,
     modelApiKey: cliOptions.modelApiKey,
+    keepAlive: cliOptions.keepAlive,
+    experimental: cliOptions.experimental,
   };
 }
 

src/index.ts

@@ -103,6 +103,10 @@ export const configSchema = z
       .describe(
         "API key for the custom model provider. Required when using a model other than the default google/gemini-2.0-flash",
       ),
+    experimental: z
+      .boolean()
+      .optional()
+      .describe("Enable experimental Stagehand features"),
   })
   .refine(
     (data) => {

src/program.ts

@@ -65,6 +65,8 @@ program
     "--modelApiKey <key>",
     "API key for the custom model provider (required when using custom models)",
   )
+  .option("--keepAlive", "Enable Browserbase Keep Alive Session")
+  .option("--experimental", "Enable experimental features")
   .action(async (options) => {
     const config = await resolveConfig(options);
     const serverList = new ServerList(async () =>

src/stagehandStore.ts

@@ -33,6 +33,7 @@ export const createStagehandInstance = async (
     ...(params.browserbaseSessionID && {
       browserbaseSessionID: params.browserbaseSessionID,
     }),
+    experimental: config.experimental ?? false,
     browserbaseSessionCreateParams: {
       projectId,
       proxies: config.proxies,