aboutsummaryrefslogtreecommitdiff
path: root/src/client/views/nodes/ChatBox/Agent.ts
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/views/nodes/ChatBox/Agent.ts')
-rw-r--r--src/client/views/nodes/ChatBox/Agent.ts4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/client/views/nodes/ChatBox/Agent.ts b/src/client/views/nodes/ChatBox/Agent.ts
index bada4b146..a3b1d083c 100644
--- a/src/client/views/nodes/ChatBox/Agent.ts
+++ b/src/client/views/nodes/ChatBox/Agent.ts
@@ -5,6 +5,7 @@ import { XMLParser, XMLBuilder } from 'fast-xml-parser';
import { WikipediaTool } from './tools/WikipediaTool';
import { CalculateTool } from './tools/CalculateTool';
import { RAGTool } from './tools/RAGTool';
+import { NoTool } from './tools/NoTool';
import { Vectorstore } from './vectorstore/VectorstoreUpload';
import { ChatCompletionAssistantMessageParam, ChatCompletionMessageParam } from 'openai/resources';
import dotenv from 'dotenv';
@@ -27,6 +28,7 @@ export class Agent {
wikipedia: new WikipediaTool(),
calculate: new CalculateTool(),
rag: new RAGTool(this.vectorstore, summaries),
+ no_tool: new NoTool(),
};
}
@@ -132,7 +134,7 @@ export class Agent {
for (const paramName in tool.parameters) {
if (actionInput[paramName] !== undefined) {
args[paramName] = actionInput[paramName];
- } else {
+ } else if (tool.parameters[paramName].required === 'true') {
throw new Error(`Missing required parameter '${paramName}' for action '${action}'`);
}
}