Add datasetId field to eval run key

firebase · Aug 27, 2024 · 9df71cf · 9df71cf
1 parent fe673c2
commit 9df71cf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/genkit-tools/common/src/types/eval.ts b/genkit-tools/common/src/types/eval.ts
@@ -95,6 +95,7 @@ export type EvalResult = z.infer<typeof EvalResultSchema>;
  */
 export const EvalRunKeySchema = z.object({
   actionRef: z.string().optional(),
+  datasetId: z.string().optional(),
   evalRunId: z.string(),
   createdAt: z.string(),
 });

diff --git a/js/ai/tests/generate/generate_test.ts b/js/ai/tests/generate/generate_test.ts
@@ -17,15 +17,15 @@
 import assert from 'node:assert';
 import { describe, it } from 'node:test';
 import { z } from 'zod';
-import { generate, GenerateResponseChunk } from '../../src/generate';
+import { GenerateResponseChunk, generate } from '../../src/generate';
 import {
   Candidate,
   GenerateOptions,
   GenerateResponse,
   Message,
   toGenerateRequest,
 } from '../../src/generate.js';
-import { defineModel, GenerateResponseChunkData } from '../../src/model';
+import { GenerateResponseChunkData, defineModel } from '../../src/model';
 import {
   CandidateData,
   GenerateRequest,