change all modelnames to just gemini

Author: Kylejeong2

README.md

@@ -147,7 +147,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)                               |
 

config.d.ts

@@ -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;
   /**
    * 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/config.ts

@@ -35,7 +35,7 @@ const defaultConfig: Config = {
     browserHeight: 768,
   },
   cookies: undefined,
-  modelName: "google/gemini-2.0-flash", // Default Model
+  modelName: "gemini-2.0-flash", // Default Model
 };
 
 // Resolve final configuration by merging defaults, file config, and CLI options

src/index.ts

@@ -95,13 +95,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",
       ),
     experimental: z
       .boolean()

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

@@ -26,8 +26,7 @@ export const createStagehandInstance = async (
     env: "BROWSERBASE",
     apiKey,
     projectId,
-    modelName:
-      params.modelName || config.modelName || "google/gemini-2.0-flash",
+    modelName: params.modelName || config.modelName || "gemini-2.0-flash",
     modelClientOptions: {
       apiKey:
         config.modelApiKey ||

src/transport.ts

@@ -13,7 +13,7 @@ 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 ||