Implemented strict typechecking for tools, specifically tool inputs

1 files changed, 2 insertions, 1 deletions

diff --git a/src/client/views/nodes/chatbot/agentsystem/Agent.ts b/src/client/views/nodes/chatbot/agentsystem/Agent.ts
index 0747ddd60..ba5868207 100644
--- a/src/client/views/nodes/chatbot/agentsystem/Agent.ts
+++ b/src/client/views/nodes/chatbot/agentsystem/Agent.ts
@@ -14,6 +14,7 @@ import { WebsiteInfoScraperTool } from '../tools/WebsiteInfoScraperTool';
 import { AgentMessage, AssistantMessage, Observation, PROCESSING_TYPE, ProcessingInfo, Tool } from '../types/types';
 import { Vectorstore } from '../vectorstore/Vectorstore';
 import { getReactPrompt } from './prompts';
+import { BaseTool } from '../tools/BaseTool';
 
 dotenv.config();
 
@@ -35,7 +36,7 @@ export class Agent {
     private processingNumber: number = 0;
     private processingInfo: ProcessingInfo[] = [];
     private streamedAnswerParser: StreamedAnswerParser = new StreamedAnswerParser();
-    private tools: Record<string, Tool<ToolArgument>>;
+    private tools: Record<string, BaseTool>;
 
     /**
      * The constructor initializes the agent with the vector store and toolset, and sets up the OpenAI client.