Update input and reference schema types in DatasetSchema to match tho…

…se in ActionSchema
firebase · Nov 7, 2024 · f945397 · f945397
1 parent 9c5840b
commit f945397
Showing 1 changed file with 3 additions and 8 deletions.
diff --git 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,
 });
 
 /**