diff --git a/.changeset/polite-dogfish-of-purring.md b/.changeset/polite-dogfish-of-purring.md new file mode 100644 index 0000000..8acb2e7 --- /dev/null +++ b/.changeset/polite-dogfish-of-purring.md @@ -0,0 +1,5 @@ +--- +"stagehand": patch +--- + +Add support for Haiku 4.5 CUA diff --git a/stagehand/agent/agent.py b/stagehand/agent/agent.py index 507aeac..bcd506b 100644 --- a/stagehand/agent/agent.py +++ b/stagehand/agent/agent.py @@ -17,17 +17,19 @@ from .openai_cua import OpenAICUAClient MODEL_TO_CLIENT_CLASS_MAP: dict[str, type[AgentClient]] = { - "computer-use-preview-03-11": OpenAICUAClient, + "computer-use-preview-2025-03-11": OpenAICUAClient, "claude-3-5-sonnet-latest": AnthropicCUAClient, "claude-3-7-sonnet-latest": AnthropicCUAClient, + "claude-haiku-4-5-20251001": AnthropicCUAClient, "claude-sonnet-4-20250514": AnthropicCUAClient, "claude-sonnet-4-5-20250929": AnthropicCUAClient, "gemini-2.5-computer-use-preview-10-2025": GoogleCUAClient, } MODEL_TO_PROVIDER_MAP: dict[str, AgentProvider] = { - "computer-use-preview-03-11": AgentProvider.OPENAI, + "computer-use-preview-2025-03-11": AgentProvider.OPENAI, "claude-3-5-sonnet-20240620": AgentProvider.ANTHROPIC, "claude-3-7-sonnet-20250219": AgentProvider.ANTHROPIC, + "claude-haiku-4-5-20251001": AgentProvider.ANTHROPIC, "claude-sonnet-4-20250514": AgentProvider.ANTHROPIC, "claude-sonnet-4-5-20250929": AgentProvider.ANTHROPIC, "gemini-2.5-computer-use-preview-10-2025": AgentProvider.GOOGLE,