verifiers 0.3.2.dev69__tar.gz → 0.3.2.dev70__tar.gz

{verifiers-0.3.2.dev69 → verifiers-0.3.2.dev70}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: verifiers
-Version: 0.3.2.dev69
+Version: 0.3.2.dev70
 Summary: Verifiers: Environments for LLM Reinforcement Learning
 Project-URL: Homepage, https://github.com/primeintellect-ai/verifiers
 Project-URL: Documentation, https://github.com/primeintellect-ai/verifiers

{verifiers-0.3.2.dev69 → verifiers-0.3.2.dev70}/verifiers/v1/serve/encoding.py

@@ -8,6 +8,7 @@ from pathlib import Path
 from uuid import UUID
 
 import numpy as np
+from pydantic_core import to_jsonable_python
 
 # Marker key inside the encoded payload so a decoder can recognize a
 # tensor round-trip without disturbing arbitrary user dicts.
@@ -31,14 +32,17 @@ def msgpack_encoder(obj):
     is ONLY called for types msgpack doesn't recognize. This avoids the massive
     performance penalty of recursing through millions of tokens in Python.
 
-    Handles: Path, UUID, Enum, datetime, Pydantic models, numpy scalars,
+    Handles: Path, UUID, sets, Enum, datetime, Pydantic models, numpy scalars,
     numpy arrays, torch tensors, and dataclasses (e.g. renderers'
     ``MultiModalData`` / ``PlaceholderRange``). Tensors and ndarrays are
     encoded as ``{__torch_tensor__: True, dtype, shape, data}``.
+    Other types use Pydantic's JSON conversions; unsupported types still raise.
     Does NOT handle: lists, dicts, basic types (msgpack does this natively in C).
     """
     if isinstance(obj, (Path, UUID)):
         return str(obj)
+    elif isinstance(obj, (set, frozenset)):
+        return list(obj)
     elif isinstance(obj, Enum):
         return obj.value
     elif isinstance(obj, (datetime, date)):
@@ -61,5 +65,4 @@ def msgpack_encoder(obj):
     elif dataclasses.is_dataclass(obj) and not isinstance(obj, type):
         return dataclasses.asdict(obj)
     else:
-        # raise on unknown types to make issues visible
-        raise TypeError(f"Object of type {type(obj)} is not msgpack serializable")
+        return to_jsonable_python(obj)

{verifiers-0.3.2.dev69 → verifiers-0.3.2.dev70}/verifiers/v1/serve/server.py

@@ -136,11 +136,21 @@ class EnvServer:
         ) as e:  # a failed request is data, not a crash — report and keep serving
             logger.warning("request failed: %s", e, exc_info=True)
             response = BaseResponse(success=False, error=f"{type(e).__name__}: {e}")
-        data = msgpack.packb(
-            response.model_dump(mode="python"),
-            default=msgpack_encoder,
-            use_bin_type=True,
-        )
+        try:
+            data = msgpack.packb(
+                response.model_dump(mode="python"),
+                default=msgpack_encoder,
+                use_bin_type=True,
+            )
+        except Exception as e:
+            # Encoding failures must also reply so clients don't wait indefinitely.
+            logger.warning("response encoding failed: %s", e, exc_info=True)
+            data = msgpack.packb(
+                BaseResponse(
+                    success=False, error=f"{type(e).__name__}: {e}"
+                ).model_dump(),
+                use_bin_type=True,
+            )
         try:
             # Let ZMQ retain the packed response instead of copying large traces.
             await self.frontend.send_multipart(