undo merge

firebase · Jan 23, 2025 · 9af7937 · 9af7937
1 parent 718db38
commit 9af7937
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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);
   }