Speed up postgrest JSON validation

src/postgrest/src/postgrest/types.py

@@ -6,6 +6,7 @@
 
 from httpx import AsyncClient, BasicAuth, Client, Headers, QueryParams
 from pydantic import TypeAdapter
+from pydantic.types import JsonValue
 from typing_extensions import TypeAliasType
 from yarl import URL
 
@@ -18,7 +19,7 @@
 JSON = TypeAliasType(
     "JSON", "Union[None, bool, str, int, float, Sequence[JSON], Mapping[str, JSON]]"
 )
-JSONAdapter: TypeAdapter = TypeAdapter(JSON)
+JSONAdapter: TypeAdapter = TypeAdapter(JsonValue)
 
 
 class CountMethod(StrEnum):

src/postgrest/tests/test_types.py

@@ -0,0 +1,24 @@
+from postgrest.types import JSONAdapter
+
+
+def test_json_adapter_uses_pydantic_json_value_schema():
+    assert JSONAdapter.core_schema["schema"]["schema_ref"].startswith(
+        "pydantic.types.JsonValue:"
+    )
+
+
+def test_json_adapter_validates_nested_json_bytes():
+    payload = (
+        b'{"items":[{"id":1,"tags":["a","b"],'
+        b'"meta":{"score":0.9,"active":true,"note":null}}]}'
+    )
+
+    assert JSONAdapter.validate_json(payload) == {
+        "items": [
+            {
+                "id": 1,
+                "tags": ["a", "b"],
+                "meta": {"score": 0.9, "active": True, "note": None},
+            }
+        ]
+    }