@@ -58,7 +58,7 @@ class BaseDataPoint(BaseModel):
5858 model_config = ConfigDict (extra = 'forbid' , populate_by_name = True )
5959 input : Any | None = None
6060 output : Any | None = None
61- context : list | None = None
61+ context : list [ Any ] | None = None
6262 reference : Any | None = None
6363 test_case_id : str | None = Field (None , alias = 'testCaseId' )
6464 trace_ids : list [str ] | None = Field (None , alias = 'traceIds' )
@@ -74,7 +74,7 @@ class EvalRequest(BaseModel):
7474
7575
7676class EvalStatusEnum (StrEnum ):
77- """Enumeration of evalstatusenum values ."""
77+ """EvalStatusEnum data type class ."""
7878
7979 UNKNOWN = 'UNKNOWN'
8080 PASS_ = 'PASS'
@@ -126,7 +126,7 @@ class GenkitError(BaseModel):
126126
127127
128128class Code (StrEnum ):
129- """Enumeration of code values ."""
129+ """Code data type class ."""
130130
131131 BLOCKED = 'blocked'
132132 OTHER = 'other'
@@ -158,7 +158,7 @@ class CustomPart(BaseModel):
158158
159159
160160class FinishReason (StrEnum ):
161- """Enumeration of finishreason values ."""
161+ """FinishReason data type class ."""
162162
163163 STOP = 'stop'
164164 LENGTH = 'length'
@@ -169,7 +169,7 @@ class FinishReason(StrEnum):
169169
170170
171171class ToolChoice (StrEnum ):
172- """Enumeration of toolchoice values ."""
172+ """ToolChoice data type class ."""
173173
174174 AUTO = 'auto'
175175 REQUIRED = 'required'
@@ -220,7 +220,7 @@ class GenerationUsage(BaseModel):
220220
221221
222222class Constrained (StrEnum ):
223- """Enumeration of constrained values ."""
223+ """Constrained data type class ."""
224224
225225 NONE = 'none'
226226 ALL = 'all'
@@ -244,7 +244,7 @@ class Supports(BaseModel):
244244
245245
246246class Stage (StrEnum ):
247- """Enumeration of stage values ."""
247+ """Stage data type class ."""
248248
249249 FEATURED = 'featured'
250250 STABLE = 'stable'
@@ -301,7 +301,7 @@ class Resource1(BaseModel):
301301
302302
303303class Role (StrEnum ):
304- """Enumeration of role values ."""
304+ """Role data type class ."""
305305
306306 SYSTEM = 'system'
307307 USER = 'user'
@@ -349,7 +349,7 @@ class ToolResponse(BaseModel):
349349 ref : str | None = None
350350 name : str
351351 output : Any | None = None
352- content : list | None = None
352+ content : list [ Any ] | None = None
353353
354354
355355class CommonRerankerOptions (BaseModel ):
@@ -410,7 +410,7 @@ class SameProcessAsParentSpan(BaseModel):
410410
411411
412412class State (StrEnum ):
413- """Enumeration of state values ."""
413+ """State data type class ."""
414414
415415 SUCCESS = 'success'
416416 ERROR = 'error'
@@ -462,10 +462,10 @@ class TraceMetadata(BaseModel):
462462 timestamp : float
463463
464464
465- class Context (RootModel [list ]):
465+ class Context (RootModel [list [ Any ] ]):
466466 """Root model for context."""
467467
468- root : list
468+ root : list [ Any ]
469469
470470
471471class Input (RootModel [Any ]):
@@ -504,10 +504,10 @@ class MediaModel(RootModel[Any]):
504504 root : Any
505505
506506
507- class Metadata (RootModel [dict [str , Any ] | None ]):
507+ class Metadata (RootModel [dict [str , Any ]]):
508508 """Root model for metadata."""
509509
510- root : dict [str , Any ] | None = None
510+ root : dict [str , Any ]
511511
512512
513513class Reasoning (RootModel [Any ]):
@@ -540,10 +540,10 @@ class ToolResponseModel(RootModel[Any]):
540540 root : Any
541541
542542
543- class Custom (RootModel [dict [str , Any ] | None ]):
543+ class Custom (RootModel [dict [str , Any ]]):
544544 """Root model for custom."""
545545
546- root : dict [str , Any ] | None = None
546+ root : dict [str , Any ]
547547
548548
549549class Config (RootModel [Any ]):
0 commit comments