File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -140,7 +140,7 @@ def preferred_output_tokens(
140
140
return self .response_output_tokens or self .request_output_tokens
141
141
142
142
143
- @MeasuredRequestTimings .register ()
143
+ @MeasuredRequestTimings .register ("generation_request_timings" )
144
144
class GenerationRequestTimings (MeasuredRequestTimings ):
145
145
"""Timing model for tracking generation request lifecycle events."""
146
146
Original file line number Diff line number Diff line change @@ -121,8 +121,8 @@ def __pydantic_schema_base_type__(cls) -> type[MeasuredRequestTimings]:
121
121
122
122
schema_discriminator : ClassVar [str ] = "timings_type"
123
123
124
- timings_type : ClassVar [ Literal ["measured_request_timings" ]] = (
125
- "measured_request_timings"
124
+ timings_type : Literal ["measured_request_timings" ] = Field (
125
+ description = "Type identifier for the timing measurement" ,
126
126
)
127
127
request_start : float | None = Field (
128
128
default = None , description = "When the backend began processing the request"
You can’t perform that action at this time.
0 commit comments