Enforce strict JSON output in extract tool

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

Author: cursoragent

hyperbrowser/tools/__init__.py

@@ -153,7 +153,7 @@ def _serialize_extract_tool_data(data: Any) -> str:
     if data is None:
         return ""
     try:
-        return json.dumps(data)
+        return json.dumps(data, allow_nan=False)
     except HyperbrowserError:
         raise
     except Exception as exc:

tests/test_tools_extract.py

@@ -1,5 +1,6 @@
 import asyncio
 from collections.abc import Mapping
+import math
 from types import MappingProxyType
 
 import pytest
@@ -371,6 +372,33 @@ async def run():
     assert exc_info.value.original_error is not None
 
 
+def test_extract_tool_runnable_rejects_nan_json_payloads():
+    client = _SyncClient(response_data={"value": math.nan})
+
+    with pytest.raises(
+        HyperbrowserError, match="Failed to serialize extract tool response data"
+    ) as exc_info:
+        WebsiteExtractTool.runnable(client, {"urls": ["https://example.com"]})
+
+    assert exc_info.value.original_error is not None
+
+
+def test_extract_tool_async_runnable_rejects_nan_json_payloads():
+    client = _AsyncClient(response_data={"value": math.nan})
+
+    async def run():
+        return await WebsiteExtractTool.async_runnable(
+            client, {"urls": ["https://example.com"]}
+        )
+
+    with pytest.raises(
+        HyperbrowserError, match="Failed to serialize extract tool response data"
+    ) as exc_info:
+        asyncio.run(run())
+
+    assert exc_info.value.original_error is not None
+
+
 def test_extract_tool_runnable_wraps_unexpected_schema_parse_failures(
     monkeypatch: pytest.MonkeyPatch,
 ):