Skip to content

Commit c90f4c6

Browse files
author
Dylan Huang
committed
Add invocation_id field to EvaluationRow model and update corresponding schema in eval-protocol types. This enhances tracking of invocation context for evaluation rows.
1 parent 3ad780b commit c90f4c6

2 files changed

Lines changed: 6 additions & 0 deletions

File tree

eval_protocol/models.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,11 @@ class EvaluationRow(BaseModel):
264264
description="The status of the rollout.",
265265
)
266266

267+
invocation_id: Optional[str] = Field(
268+
default_factory=generate_id,
269+
description="The ID of the invocation that this row belongs to.",
270+
)
271+
267272
cohort_id: Optional[str] = Field(
268273
default_factory=generate_id,
269274
description="The ID of the cohort that this row belongs to.",

vite-app/src/types/eval-protocol.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,7 @@ export const EvaluationRowSchema = z.object({
9999
tools: z.array(z.record(z.string(), z.any())).optional().describe('Available tools/functions that were provided to the agent.'),
100100
input_metadata: InputMetadataSchema.describe('Metadata related to the input (dataset info, model config, session data, etc.).'),
101101
rollout_status: RolloutStatusSchema.default({ status: 'finished' }).describe('The status of the rollout.'),
102+
invocation_id: z.string().optional().describe('The ID of the invocation that this row belongs to.'),
102103
cohort_id: z.string().optional().describe('The ID of the cohort that this row belongs to.'),
103104
rollout_id: z.string().optional().describe('The ID of the rollout that this row belongs to.'),
104105
run_id: z.string().optional().describe('The ID of the run that this row belongs to.'),

0 commit comments

Comments
 (0)