Skip to content

Commit da570a1

Browse files
authored
Add support for Haiku 4.5 CUA (#221)
* Add support for Haiku 4.5 CUA * fix openai model name
1 parent 1a919ad commit da570a1

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
---
2+
"stagehand": patch
3+
---
4+
5+
Add support for Haiku 4.5 CUA

stagehand/agent/agent.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,17 +17,19 @@
1717
from .openai_cua import OpenAICUAClient
1818

1919
MODEL_TO_CLIENT_CLASS_MAP: dict[str, type[AgentClient]] = {
20-
"computer-use-preview-03-11": OpenAICUAClient,
20+
"computer-use-preview-2025-03-11": OpenAICUAClient,
2121
"claude-3-5-sonnet-latest": AnthropicCUAClient,
2222
"claude-3-7-sonnet-latest": AnthropicCUAClient,
23+
"claude-haiku-4-5-20251001": AnthropicCUAClient,
2324
"claude-sonnet-4-20250514": AnthropicCUAClient,
2425
"claude-sonnet-4-5-20250929": AnthropicCUAClient,
2526
"gemini-2.5-computer-use-preview-10-2025": GoogleCUAClient,
2627
}
2728
MODEL_TO_PROVIDER_MAP: dict[str, AgentProvider] = {
28-
"computer-use-preview-03-11": AgentProvider.OPENAI,
29+
"computer-use-preview-2025-03-11": AgentProvider.OPENAI,
2930
"claude-3-5-sonnet-20240620": AgentProvider.ANTHROPIC,
3031
"claude-3-7-sonnet-20250219": AgentProvider.ANTHROPIC,
32+
"claude-haiku-4-5-20251001": AgentProvider.ANTHROPIC,
3133
"claude-sonnet-4-20250514": AgentProvider.ANTHROPIC,
3234
"claude-sonnet-4-5-20250929": AgentProvider.ANTHROPIC,
3335
"gemini-2.5-computer-use-preview-10-2025": AgentProvider.GOOGLE,

0 commit comments

Comments
 (0)