Validate extract tool schema value types

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

Author: cursoragent

hyperbrowser/tools/__init__.py

@@ -55,6 +55,10 @@ def _format_tool_param_key_for_error(key: str) -> str:
 def _prepare_extract_tool_params(params: Mapping[str, Any]) -> Dict[str, Any]:
     normalized_params = _to_param_dict(params)
     schema_value = normalized_params.get("schema")
+    if schema_value is not None and not isinstance(schema_value, (str, MappingABC)):
+        raise HyperbrowserError(
+            "Extract tool `schema` must be an object or JSON string"
+        )
     if isinstance(schema_value, str):
         try:
             parsed_schema = json.loads(schema_value)

tests/test_tools_extract.py

@@ -140,6 +140,35 @@ async def run():
         asyncio.run(run())
 
 
+def test_extract_tool_runnable_rejects_non_mapping_non_string_schema():
+    client = _SyncClient()
+    params = {
+        "urls": ["https://example.com"],
+        "schema": 123,
+    }
+
+    with pytest.raises(
+        HyperbrowserError, match="Extract tool `schema` must be an object or JSON string"
+    ):
+        WebsiteExtractTool.runnable(client, params)
+
+
+def test_extract_tool_async_runnable_rejects_non_mapping_non_string_schema():
+    client = _AsyncClient()
+    params = {
+        "urls": ["https://example.com"],
+        "schema": 123,
+    }
+
+    async def run():
+        await WebsiteExtractTool.async_runnable(client, params)
+
+    with pytest.raises(
+        HyperbrowserError, match="Extract tool `schema` must be an object or JSON string"
+    ):
+        asyncio.run(run())
+
+
 def test_extract_tool_runnable_serializes_empty_object_data():
     client = _SyncClient(response_data={})