Fixing bugs and attempting to get it to work

1 files changed, 2 insertions, 2 deletions

diff --git a/src/client/views/nodes/ChatBox/tools/BaseTool.ts b/src/client/views/nodes/ChatBox/tools/BaseTool.ts
index 3511d9528..903161bd5 100644
--- a/src/client/views/nodes/ChatBox/tools/BaseTool.ts
+++ b/src/client/views/nodes/ChatBox/tools/BaseTool.ts
@@ -1,6 +1,6 @@
 import { Tool } from '../types';
 
-export abstract class BaseTool implements Tool {
+export abstract class BaseTool<T extends Record<string, any> = Record<string, any>> implements Tool<T> {
     constructor(
         public name: string,
         public description: string,
@@ -9,7 +9,7 @@ export abstract class BaseTool implements Tool {
         public briefSummary: string
     ) {}
 
-    abstract execute(args: Record<string, any>): Promise<any>;
+    abstract execute(args: T): Promise<any>;
 
     getActionRule(): Record<string, any> {
         return {