From f945397473218fb8280fa4d428b401ce893dc855 Mon Sep 17 00:00:00 2001 From: Shruti Padamata Date: Wed, 6 Nov 2024 16:11:24 -0800 Subject: [PATCH] Update input and reference schema types in DatasetSchema to match those in ActionSchema --- genkit-tools/common/src/types/eval.ts | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/genkit-tools/common/src/types/eval.ts b/genkit-tools/common/src/types/eval.ts index bcf3ff1fb..1afa21de4 100644 --- a/genkit-tools/common/src/types/eval.ts +++ b/genkit-tools/common/src/types/eval.ts @@ -21,6 +21,7 @@ import { ListEvalKeysResponse, UpdateDatasetRequest, } from './apis'; +import { JSONSchema7Schema } from './action'; /** * This file defines schema and types that are used by the Eval store. @@ -161,14 +162,8 @@ export interface EvalStore { } export const DatasetSchemaSchema = z.object({ - inputSchema: z - .record(z.any()) - .describe('Valid JSON Schema for the `input` field of dataset entry.') - .optional(), - referenceSchema: z - .record(z.any()) - .describe('Valid JSON Schema for the `reference` field of dataset entry.') - .optional(), + inputSchema: JSONSchema7Schema, + referenceSchema: JSONSchema7Schema, }); /**