verifiers 0.3.2.dev121__tar.gz → 0.3.2.dev122__tar.gz

{verifiers-0.3.2.dev121 → verifiers-0.3.2.dev122}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: verifiers
-Version: 0.3.2.dev121
+Version: 0.3.2.dev122
 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.dev121 → verifiers-0.3.2.dev122}/verifiers/v1/episode.py

@@ -3,7 +3,7 @@
 import uuid
 from typing import Annotated, Any, Generic, Literal, Self
 
-from pydantic import BaseModel, Field, model_validator
+from pydantic import BaseModel, Field, computed_field, model_validator
 
 from verifiers.v1.configs.agent import WireAgentConfig
 from verifiers.v1.graph import RECORD_FLOAT_DECIMALS
@@ -119,16 +119,19 @@ class Episode(BaseModel, Generic[DataT, StateT, AgentConfigT]):
         judge/off-graph usage stays on the traces (`Trace.extra_usage`)."""
         return Usage.aggregate(u for t in self.traces if (u := t.usage) is not None)
 
+    @computed_field
     @property
     def num_input_tokens(self) -> int:
         """Fed-in tokens (system + user + tool), summed across traces."""
         return sum(t.num_input_tokens for t in self.traces)
 
+    @computed_field
     @property
     def num_output_tokens(self) -> int:
         """Model-generated tokens across all turns, summed across traces."""
         return sum(t.num_output_tokens for t in self.traces)
 
+    @computed_field
     @property
     def num_total_tokens(self) -> int:
         """Final sequence lengths per branch, summed across traces."""

{verifiers-0.3.2.dev121 → verifiers-0.3.2.dev122}/verifiers/v1/trace.py

@@ -8,7 +8,7 @@ from collections.abc import Callable, Iterable, Iterator, Mapping
 from typing import TYPE_CHECKING, Any, Generic
 
 import numpy as np
-from pydantic import BaseModel, Field, PrivateAttr, field_serializer
+from pydantic import BaseModel, Field, PrivateAttr, computed_field, field_serializer
 from renderers.base import MultiModalData
 from typing_extensions import TypeVar
 
@@ -376,16 +376,19 @@ class Branch(BaseModel):
             (c.usage for c in reversed(self.calls) if c.usage is not None), None
         )
 
+    @computed_field
     @property
     def num_total_tokens(self) -> int:
         last = self.last_usage
         return last.total_tokens if last is not None else 0
 
+    @computed_field
     @property
     def num_output_tokens(self) -> int:
         usage = self.usage
         return usage.completion_tokens if usage is not None else 0
 
+    @computed_field
     @property
     def num_input_tokens(self) -> int:
         """Fed-in tokens (system + user + tool), counted once; a lower bound whenever
@@ -491,6 +494,7 @@ class Trace(BaseModel, Generic[DataT, StateT, AgentConfigT]):
     def has_error(self) -> bool:
         return not self.ok
 
+    @computed_field
     @property
     def num_input_tokens(self) -> int:
         """Fed-in tokens (system + user + tool), counted once across all branches —
@@ -506,11 +510,13 @@ class Trace(BaseModel, Generic[DataT, StateT, AgentConfigT]):
                     total += increment
         return total
 
+    @computed_field
     @property
     def num_output_tokens(self) -> int:
         """Model-generated tokens across all turns, summed across branches."""
         return sum(branch.num_output_tokens for branch in self.branches)
 
+    @computed_field
     @property
     def num_total_tokens(self) -> int:
         """New input plus generated tokens, counted once per call across branches.