diff --git a/hyperbrowser/models/consts.py b/hyperbrowser/models/consts.py index c18813c5..bb8a1966 100644 --- a/hyperbrowser/models/consts.py +++ b/hyperbrowser/models/consts.py @@ -93,6 +93,7 @@ CuaLlm = Literal[ "computer-use-preview", "gpt-6-astra", + "gpt-6.1-sol", "gpt-6-sol", "gpt-6-luna", "gpt-5.6-sol", diff --git a/pyproject.toml b/pyproject.toml index 51ff2232..a1a8efaf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "hyperbrowser" -version = "1.9.2" +version = "1.9.3" description = "Python SDK for hyperbrowser" authors = ["Nikhil Shahi "] license = "MIT" diff --git a/tests/test_cua_models.py b/tests/test_cua_models.py index c196ba1e..35ea7837 100644 --- a/tests/test_cua_models.py +++ b/tests/test_cua_models.py @@ -5,7 +5,9 @@ from hyperbrowser.models.agents.cua import StartCuaTaskParams -@pytest.mark.parametrize("llm", ["gpt-6-sol", "gpt-6-luna", "gpt-6-astra"]) +@pytest.mark.parametrize( + "llm", ["gpt-6.1-sol", "gpt-6-sol", "gpt-6-luna", "gpt-6-astra"] +) def test_start_cua_accepts_gpt6_models(llm: str) -> None: dumped = dump_request( {"task": "Inspect the page", "llm": llm, "reasoning_effort": "max"}, diff --git a/tests/typecheck/valid_requests.py b/tests/typecheck/valid_requests.py index 0cd76457..47b0c47b 100644 --- a/tests/typecheck/valid_requests.py +++ b/tests/typecheck/valid_requests.py @@ -100,7 +100,7 @@ def valid_sync_requests(client: Hyperbrowser) -> None: client.agents.cua.start( { "task": "Complete the task", - "llm": "gpt-6-sol", + "llm": "gpt-6.1-sol", "reasoning_effort": "high", "use_custom_api_keys": True, "api_keys": {"openai": "openai-key"},