Skip to content

Commit 38f7038

Browse files
committed
Fix issues with MeasuredRequestTimings model validation
1 parent 9180722 commit 38f7038

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/guidellm/backend/objects.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def preferred_output_tokens(
140140
return self.response_output_tokens or self.request_output_tokens
141141

142142

143-
@MeasuredRequestTimings.register()
143+
@MeasuredRequestTimings.register("generation_request_timings")
144144
class GenerationRequestTimings(MeasuredRequestTimings):
145145
"""Timing model for tracking generation request lifecycle events."""
146146

src/guidellm/scheduler/objects.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,8 +121,8 @@ def __pydantic_schema_base_type__(cls) -> type[MeasuredRequestTimings]:
121121

122122
schema_discriminator: ClassVar[str] = "timings_type"
123123

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",
126126
)
127127
request_start: float | None = Field(
128128
default=None, description="When the backend began processing the request"

0 commit comments

Comments
 (0)