feat(telemetry): add tokens_per_joule field to TelemetryRecord

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Jon Saad-Falcon
2026-02-28 06:42:54 +00:00
co-authored by Claude Opus 4.6
parent 598d3473ee
commit ac8d6b01c7
2 changed files with 21 additions and 0 deletions
+1
View File
@@ -162,6 +162,7 @@ class TelemetryRecord:
cpu_energy_joules: float = 0.0
gpu_energy_joules: float = 0.0
dram_energy_joules: float = 0.0
tokens_per_joule: float = 0.0
metadata: Dict[str, Any] = field(default_factory=dict)
+20
View File
@@ -0,0 +1,20 @@
"""Tests for TelemetryRecord fields."""
from __future__ import annotations
from openjarvis.core.types import TelemetryRecord
class TestTelemetryRecord:
def test_tokens_per_joule_field_exists(self):
rec = TelemetryRecord(timestamp=1.0, model_id="test")
assert hasattr(rec, "tokens_per_joule")
assert rec.tokens_per_joule == 0.0
def test_tokens_per_joule_set(self):
rec = TelemetryRecord(
timestamp=1.0,
model_id="test",
tokens_per_joule=80.0,
)
assert rec.tokens_per_joule == 80.0