refactor: remove multisession tools, remove prompts sampling, simplif… (#123)

* refactor: remove multisession tools, remove prompts sampling, simplify tool descriptions for better context, add support if google apikey set, latest version of stagehand, remove custom availmodelschema to use stagehand

* change all modelnames to just gemini

* auto publish to mcp directory + changelog

* publish workflow

* fix session manager bugs

* Update config.d.ts

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

* fully remove prompts + rebase

---------

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

Author: Kylejeong2

.github/workflows/publish-mcp.yml

@@ -0,0 +1,50 @@
+name: Publish to MCP Registry
+
+on:
+  push:
+    tags: ["v*"]
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    permissions:
+      id-token: write # Required for OIDC authentication
+      contents: read
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v5
+
+      - name: Setup pnpm
+        uses: pnpm/action-setup@v3
+        with:
+          version: 10.12.4
+
+      - name: Setup Node.js
+        uses: actions/setup-node@v5
+        with:
+          node-version: "lts/*"
+
+      - name: Install dependencies
+        run: pnpm install --frozen-lockfile
+
+      - name: Run tests
+        run: pnpm test --if-present
+
+      - name: Build package
+        run: pnpm build --if-present
+
+      - name: Publish to npm
+        run: pnpm publish
+        env:
+          NODE_AUTH_TOKEN: ${{ secrets.NPM_TOKEN }}
+
+      - name: Install MCP Publisher
+        run: |
+          curl -L "https://github.com/modelcontextprotocol/registry/releases/download/latest/mcp-publisher_$(uname -s | tr '[:upper:]' '[:lower:]')_$(uname -m | sed 's/x86_64/amd64/;s/aarch64/arm64/').tar.gz" | tar xz mcp-publisher
+
+      - name: Login to MCP Registry
+        run: ./mcp-publisher login github-oidc
+
+      - name: Publish to MCP Registry
+        run: ./mcp-publisher publish

CHANGELOG.md

@@ -1,5 +1,11 @@
 # @browserbasehq/mcp-server-browserbase
 
+## 2.2.0
+
+### Minor Changes
+
+- Remove multisession tools, remove prompts sampling, simplify tool descriptions for better context, add support if google apikey set, latest version of stagehand, remove custom availmodelschema to use stagehand model type instead.
+
 ## 2.1.3
 
 ### Patch Changes

README.md

@@ -2,7 +2,7 @@
 
 [![smithery badge](https://smithery.ai/badge/@browserbasehq/mcp-browserbase)](https://smithery.ai/server/@browserbasehq/mcp-browserbase)
 
-![cover](assets/cover-mcp.png)
+![cover](assets/cover.png)
 
 [The Model Context Protocol (MCP)](https://modelcontextprotocol.io/introduction) is an open protocol that enables seamless integration between LLM applications and external data sources and tools. Whether you're building an AI-powered IDE, enhancing a chat interface, or creating custom AI workflows, MCP provides a standardized way to connect LLMs with the context they need.
 
@@ -19,7 +19,6 @@ This server provides cloud browser automation capabilities using [Browserbase](h
 | Model Flexibility  | Supports multiple models (OpenAI, Claude, Gemini, and more) |
 | Vision Support     | Use annotated screenshots for complex DOMs                  |
 | Session Management | Create, manage, and close browser sessions                  |
-| Multi-Session      | Run multiple browser sessions in parallel                   |
 
 ## How to Setup
 
@@ -194,7 +193,7 @@ The Browserbase MCP server accepts the following command-line flags:
 | `--cookies [json]`         | JSON array of cookies to inject into the browser                            |
 | `--browserWidth <width>`   | Browser viewport width (default: 1024)                                      |
 | `--browserHeight <height>` | Browser viewport height (default: 768)                                      |
-| `--modelName <model>`      | The model to use for Stagehand (default: google/gemini-2.0-flash)           |
+| `--modelName <model>`      | The model to use for Stagehand (default: gemini-2.0-flash)                  |
 | `--modelApiKey <key>`      | API key for the custom model provider (required when using custom models)   |
 | `--experimental`           | Enable experimental features (default: false)                               |
 
@@ -386,13 +385,11 @@ The server provides access to screenshot resources:
 
 - **AI-Powered Automation**: Natural language commands for web interactions
 - **Multi-Model Support**: Works with OpenAI, Claude, Gemini, and more
-- **Advanced Session Management**: Single and multi-session support for parallel browser automation
 - **Screenshot Capture**: Full-page and element-specific screenshots
 - **Data Extraction**: Intelligent content extraction from web pages
 - **Proxy Support**: Enterprise-grade proxy capabilities
 - **Stealth Mode**: Advanced anti-detection features
 - **Context Persistence**: Maintain authentication and state across sessions
-- **Parallel Workflows**: Run multiple browser sessions simultaneously for complex automation tasks
 
 For more information about the Model Context Protocol, visit:
 

assets/browserbase-demo.png

@@ -1,5 +1,5 @@
 import type { Cookie } from "playwright-core";
-import type { AvailableModelSchema } from "./src/types/models.js";
+import type { AvailableModelSchema } from "@browserbasehq/stagehand";
 
 export type Config = {
   /**
@@ -96,12 +96,12 @@ export type Config = {
    * The Model that Stagehand uses
    * Available models: OpenAI, Claude, Gemini, Cerebras, Groq, and other providers
    *
-   * @default "google/gemini-2.0-flash"
+   * @default "gemini-2.0-flash"
    */
-  modelName?: AvailableModelSchema;
+  modelName?: z.infer<typeof AvailableModelSchema>;
   /**
    * API key for the custom model provider
-   * Required when using a model other than the default google/gemini-2.0-flash
+   * Required when using a model other than the default gemini-2.0-flash
    */
   modelApiKey?: string;
   /**

assets/cover-mcp.png

@@ -76,22 +76,6 @@
         "browserbase_session_close"
       ]
     },
-    {
-      "name": "multi-session-test",
-      "description": "Test multi-session browser management",
-      "steps": [
-        {
-          "user": "Create a multi-session browser named 'test-session', list all sessions, navigate to https://example.com in that session, and close the session",
-          "expectedState": "closed"
-        }
-      ],
-      "expectTools": [
-        "multi_browserbase_stagehand_session_create",
-        "multi_browserbase_stagehand_session_list",
-        "multi_browserbase_stagehand_navigate_session",
-        "multi_browserbase_stagehand_session_close"
-      ]
-    },
     {
       "name": "form-interaction-test",
       "description": "Test form filling and submission capabilities",

assets/session.png

@@ -52,29 +52,6 @@
         "browserbase_session_close"
       ]
     },
-    {
-      "name": "smoke-test-multi-session",
-      "description": "Quick test to verify multi-session functionality",
-      "steps": [
-        {
-          "user": "Create a browser session named 'test-session'",
-          "expectedState": "session created"
-        },
-        {
-          "user": "List active sessions",
-          "expectedState": "test-session"
-        },
-        {
-          "user": "Close the test session",
-          "expectedState": "closed session"
-        }
-      ],
-      "expectTools": [
-        "multi_browserbase_stagehand_session_create",
-        "multi_browserbase_stagehand_session_list",
-        "multi_browserbase_stagehand_session_close"
-      ]
-    },
     {
       "name": "smoke-test-url-tools",
       "description": "Quick test to verify URL retrieval tools work",

config.d.ts

@@ -76,22 +76,6 @@
         "browserbase_session_close"
       ]
     },
-    {
-      "name": "multi-session-test",
-      "description": "Test multi-session browser management",
-      "steps": [
-        {
-          "user": "Create a multi-session browser named 'test-session', list all sessions, navigate to https://example.com in that session, and close the session",
-          "expectedState": "closed"
-        }
-      ],
-      "expectTools": [
-        "multi_browserbase_stagehand_session_create",
-        "multi_browserbase_stagehand_session_list",
-        "multi_browserbase_stagehand_navigate_session",
-        "multi_browserbase_stagehand_session_close"
-      ]
-    },
     {
       "name": "form-interaction-test",
       "description": "Test form filling and submission capabilities",
@@ -121,22 +105,6 @@
         "browserbase_session_create",
         "browserbase_stagehand_navigate"
       ]
-    },
-    {
-      "name": "url-retrieval-test",
-      "description": "Test URL retrieval functionality",
-      "steps": [
-        {
-          "user": "Create a browser session, navigate to https://example.com, get the current URL to verify navigation, and close the session",
-          "expectedState": "https://example.com"
-        }
-      ],
-      "expectTools": [
-        "browserbase_session_create",
-        "browserbase_stagehand_navigate",
-        "browserbase_stagehand_get_url",
-        "browserbase_session_close"
-      ]
     }
   ]
 }