From 9af7937ca39d351cfcc04bebd1bbaf4fd1c3d77d Mon Sep 17 00:00:00 2001
From: Pavel Jbanov <pavelj@google.com>
Date: Thu, 23 Jan 2025 10:03:20 -0500
Subject: [PATCH] undo merge

---
 js/genkit/src/genkit.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/js/genkit/src/genkit.ts b/js/genkit/src/genkit.ts
index b2006722c..89b12e0ae 100644
--- a/js/genkit/src/genkit.ts
+++ b/js/genkit/src/genkit.ts
@@ -102,7 +102,7 @@ import {
   SessionError,
   SessionOptions,
 } from '@genkit-ai/ai/session';
-import { resolveTools } from '@genkit-ai/ai/tool';
+import { resolveTools, ToolFn } from '@genkit-ai/ai/tool';
 import {
   Action,
   defineFlow,
@@ -218,7 +218,7 @@ export class Genkit implements HasRegistry {
    */
   defineTool<I extends z.ZodTypeAny, O extends z.ZodTypeAny>(
     config: ToolConfig<I, O>,
-    fn: (input: z.infer<I>) => Promise<z.infer<O>>
+    fn: ToolFn<I, O>
   ): ToolAction<I, O> {
     return defineTool(this.registry, config, fn);
   }