verifiers 0.2.2.dev45__tar.gz → 0.2.2.dev46__tar.gz

{verifiers-0.2.2.dev45 → verifiers-0.2.2.dev46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: verifiers
-Version: 0.2.2.dev45
+Version: 0.2.2.dev46
 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.2.2.dev45 → verifiers-0.2.2.dev46}/verifiers/v1/serve/types.py

@@ -1,6 +1,6 @@
 from typing import ClassVar
 
-from pydantic import BaseModel, Field, field_serializer, model_validator
+from pydantic import BaseModel, Field, SerializeAsAny, model_validator
 
 from verifiers.v1.clients.config import ClientConfig
 from verifiers.v1.episode import WireEpisode
@@ -65,15 +65,11 @@ class RunRequest(BaseRequest):
 
 
 class RunResponse(BaseResponse):
-    episode: WireEpisode | None = None
+    episode: SerializeAsAny[WireEpisode] | None = None
     """The rollout's episode — its standing (`id`/`env`/`errors`, carrying
     episode-level errors even when no trace minted) inlined next to its flat,
     self-contained traces; task-specific data preserved in `model_extra`."""
 
-    @field_serializer("episode")
-    def _ser_episode(self, episode: "WireEpisode | None") -> dict | None:
-        return episode.model_dump() if episode is not None else None
-
 
 class RunGroupRequest(BaseRequest):
     """Legacy (v0) route: group-scored v0 envs run a task's n rollouts together."""
@@ -87,8 +83,4 @@ class RunGroupRequest(BaseRequest):
 
 
 class RunGroupResponse(BaseResponse):
-    traces: list[WireTrace] | None = None
-
-    @field_serializer("traces")
-    def _ser_traces(self, traces: "list[WireTrace] | None") -> list[dict] | None:
-        return [t.model_dump() for t in traces] if traces is not None else None
+    traces: list[SerializeAsAny[WireTrace]] | None = None