Minor updates for pydantic objects

Author: markurtz

src/guidellm/backend/objects.py

@@ -141,6 +141,7 @@ def preferred_output_tokens(
             return self.response_output_tokens or self.request_output_tokens
 
 
+@SchedulerMessagingPydanticRegistry.register()
 @MeasuredRequestTimings.register("generation_request_timings")
 class GenerationRequestTimings(MeasuredRequestTimings):
     """Timing model for tracking generation request lifecycle events."""
@@ -154,9 +155,3 @@ class GenerationRequestTimings(MeasuredRequestTimings):
         default=None,
         description="Unix timestamp when the last generation iteration completed.",
     )
-
-
-# Rebuild ScheduledRequestInfo to recognize MeasuredRequestTimings schema change
-ScheduledRequestInfo.model_rebuild(force=True)
-
-SchedulerMessagingPydanticRegistry.register_decorator(GenerationRequestTimings)

src/guidellm/utils/pydantic_utils.py

@@ -78,8 +78,8 @@ def reload_parent_schemas(cls):
         are Pydantic models and triggers schema rebuilding on each to ensure
         that any changes in child models are reflected in parent schemas.
         """
-        potential_parents: set[BaseModel] = {BaseModel}
-        stack: list[BaseModel] = [BaseModel]
+        potential_parents: set[type[BaseModel]] = {BaseModel}
+        stack: list[type[BaseModel]] = [BaseModel]
 
         while stack:
             current = stack.pop()
@@ -108,13 +108,17 @@ def _reload_schemas_depending_on(cls, target: type[BaseModel], types: set[type])
                     and any(
                         cls._uses_type(target, field_info.annotation)
                         for field_info in candidate.model_fields.values()
+                        if field_info.annotation is not None
                     )
                 ):
-                    before = candidate.model_json_schema()
+                    try:
+                        before = candidate.model_json_schema()
+                    except Exception:  # noqa: BLE001
+                        before = None
                     candidate.model_rebuild(force=True)
-                    after = candidate.model_json_schema()
-                    if before != after:
-                        changed = True
+                    if before is not None:
+                        after = candidate.model_json_schema()
+                        changed |= before != after
 
     @classmethod
     def _uses_type(cls, target: type, candidate: type) -> bool: