change all modelnames

Author: Kylejeong2

README.md

@@ -226,7 +226,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)   |
 
 These flags can be passed directly to the CLI or configured in your MCP configuration file.
@@ -411,7 +411,7 @@ For Claude models:
 
 Available models include:
 
-- **Gemini**: `google/gemini-2.0-flash` (default), `google/gemini-1.5-pro`, `google/gemini-1.5-flash`
+- **Gemini**: `gemini-2.0-flash` (default), `gemini-1.5-pro`, `gemini-1.5-flash`
 - **OpenAI**: `gpt-4o`, `gpt-4o-mini`, `o1-mini`, `o1-preview`, `o3-mini`
 - **Claude**: `claude-3-5-sonnet-latest`, `claude-3-7-sonnet-latest`
 - **Other providers**: Cerebras, Groq, and more

config.d.ts

@@ -90,12 +90,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;
   /**
    * 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;
 };

src/index.ts

@@ -91,13 +91,13 @@ export const configSchema = z
       })
       .optional(),
     modelName: AvailableModelSchema.optional().describe(
-      "The model to use for Stagehand (default: google/gemini-2.0-flash)",
+      "The model to use for Stagehand (default: gemini-2.0-flash)",
     ), // Already an existing Zod Enum
     modelApiKey: z
       .string()
       .optional()
       .describe(
-        "API key for the custom model provider. Required when using a model other than the default google/gemini-2.0-flash",
+        "API key for the custom model provider. Required when using a model other than the default gemini-2.0-flash",
       ),
   })
   .refine(

src/program.ts

@@ -59,7 +59,7 @@ program
   .option("--browserHeight <height>", "Browser height to use for the browser.")
   .option(
     "--modelName <model>",
-    "The model to use for Stagehand (default: google/gemini-2.0-flash)",
+    "The model to use for Stagehand (default: gemini-2.0-flash)",
   )
   .option(
     "--modelApiKey <key>",

src/stagehandStore.ts

@@ -27,7 +27,7 @@ export const createStagehandInstance = async (
     projectId,
     modelName: (params.modelName ||
       config.modelName ||
-      "google/gemini-2.0-flash") as AvailableModel,
+      "gemini-2.0-flash") as AvailableModel,
     modelClientOptions: {
       apiKey: config.modelApiKey || process.env.GEMINI_API_KEY,
     },

src/transport.ts

@@ -13,10 +13,10 @@ export async function startStdioTransport(
 ) {
   // Check if we're using the default model without an API key
   if (config) {
-    const modelName = config.modelName || "google/gemini-2.0-flash";
+    const modelName = config.modelName || "gemini-2.0-flash";
     const hasModelApiKey = config.modelApiKey || process.env.GEMINI_API_KEY;
 
-    if (modelName.includes("google/gemini") && !hasModelApiKey) {
+    if (modelName.includes("gemini") && !hasModelApiKey) {
       console.error(`
 ⚠️  IMPORTANT: MCP Server Configuration Update Required