model updates

Author: miguelg719

examples/agent_example.py

@@ -35,8 +35,8 @@
 async def main():
     # Build a unified configuration object for Stagehand
     config = StagehandConfig(
-        # env="BROWSERBASE",
-        env="LOCAL",
+        env="BROWSERBASE",
+        # env="LOCAL",
         api_key=os.getenv("BROWSERBASE_API_KEY"),
         project_id=os.getenv("BROWSERBASE_PROJECT_ID"),
         model_name="gpt-4o",
@@ -64,7 +64,7 @@ async def main():
 
     console.print("\n▶️ [highlight] Using Agent to perform a task[/]: playing a game of 2048")
     agent = stagehand.agent(
-        model="computer-use-preview-10-2025",
+        model="gemini-2.5-computer-use-preview-10-2025",
         instructions="You are a helpful web navigation assistant that helps users find information. You are currently on the following page: google.com. Do not ask follow up questions, the user will trust your judgement.",
         options={"apiKey": os.getenv("GEMINI_API_KEY")}
     )

stagehand/agent/agent.py

@@ -22,15 +22,15 @@
     "claude-3-7-sonnet-latest": AnthropicCUAClient,
     "claude-sonnet-4-20250514": AnthropicCUAClient,
     "claude-sonnet-4-5-20250929": AnthropicCUAClient,
-    "computer-use-preview-10-2025": GoogleCUAClient,
+    "gemini-2.5-computer-use-preview-10-2025": GoogleCUAClient,
 }
 MODEL_TO_PROVIDER_MAP: dict[str, AgentProvider] = {
     "computer-use-preview-03-11": AgentProvider.OPENAI,
     "claude-3-5-sonnet-20240620": AgentProvider.ANTHROPIC,
     "claude-3-7-sonnet-20250219": AgentProvider.ANTHROPIC,
     "claude-sonnet-4-20250514": AgentProvider.ANTHROPIC,
     "claude-sonnet-4-5-20250929": AgentProvider.ANTHROPIC,
-    "computer-use-preview-10-2025": AgentProvider.GOOGLE,
+    "gemini-2.5-computer-use-preview-10-2025": AgentProvider.GOOGLE,
     # Add more mappings as needed
 }