Normalize browser-use version cast errors to HyperbrowserError

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

hyperbrowser/models/agents/browser_use.py

@@ -2,6 +2,7 @@
 
 from pydantic import BaseModel, ConfigDict, Field
 
+from ...exceptions import HyperbrowserError
 from ..consts import BrowserUseLlm, BrowserUseVersion
 from ..session import CreateSessionParams
 
@@ -296,4 +297,4 @@ def cast_steps_for_version(
     elif version == "latest":
         return steps
     else:
-        raise ValueError(f"Invalid version: {version}")
+        raise HyperbrowserError(f"Invalid browser-use version: {version}")

tests/test_browser_use_models.py

@@ -0,0 +1,21 @@
+from hyperbrowser.exceptions import HyperbrowserError
+from hyperbrowser.models.agents.browser_use import cast_steps_for_version
+
+
+def test_cast_steps_for_version_latest_returns_original_steps():
+    steps = [{"state": "kept-as-is"}]
+
+    result = cast_steps_for_version(steps, "latest")
+
+    assert result is steps
+
+
+def test_cast_steps_for_version_raises_hyperbrowser_error_for_invalid_version():
+    steps = [{"state": "ignored"}]
+
+    try:
+        cast_steps_for_version(steps, "v999")  # type: ignore[arg-type]
+    except HyperbrowserError as exc:
+        assert "Invalid browser-use version" in str(exc)
+    else:
+        raise AssertionError("Expected HyperbrowserError for invalid version")