make DynamicRunEvaluatorParams generic (#5246)

Co-authored-by: Brace Sproul <[email protected]>
langchain-ai · Apr 30, 2024 · 52007ab · 52007ab
1 parent b46ee43
commit 52007ab
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/langchain/src/smith/config.ts b/langchain/src/smith/config.ts
@@ -30,10 +30,14 @@ export type EvaluatorInputFormatter = ({
   run: Run;
 }) => EvaluatorInputs;
 
-export type DynamicRunEvaluatorParams = {
-  input: Record<string, unknown>;
-  prediction?: Record<string, unknown>;
-  reference?: Record<string, unknown>;
+export type DynamicRunEvaluatorParams<
+  Input extends Record<string, any> = Record<string, unknown>,
+  Prediction extends Record<string, any> = Record<string, unknown>,
+  Reference extends Record<string, any> = Record<string, unknown>
+> = {
+  input: Input;
+  prediction?: Prediction;
+  reference?: Reference;
   run: Run;
   example?: Example;
 };