add exp flag and add keepalive to cli options (#109)

derekmeegan · web-flow · commit ed0d98e58376 · 2025-08-20T09:55:34.000-07:00
* add exp flag and add keepalive to cli options

* bump stagehand in pnpm lock
diff --git a/README.md b/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.
 
diff --git a/config.d.ts b/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;
 };
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/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",
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/src/config.ts b/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,
   };
 }
 
diff --git a/src/index.ts b/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) => {
diff --git a/src/program.ts b/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 () =>
diff --git a/src/stagehandStore.ts b/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,

Original file line number	Diff line number	Diff line change
`@@ -65,6 +65,8 @@ program`
`65`	`65`	`"--modelApiKey <key>",`
`66`	`66`	`"API key for the custom model provider (required when using custom models)",`
`67`	`67`	`)`
	`68`	`+ .option("--keepAlive", "Enable Browserbase Keep Alive Session")`
	`69`	`+ .option("--experimental", "Enable experimental features")`
`68`	`70`	`.action(async (options) => {`
`69`	`71`	`const config = await resolveConfig(options);`
`70`	`72`	`const serverList = new ServerList(async () =>`