aboutsummaryrefslogtreecommitdiff
path: root/src/client/views/nodes/ChatBox/Agent.ts
diff options
context:
space:
mode:
authorA.J. Shulman <Shulman.aj@gmail.com>2024-07-17 14:58:46 -0400
committerA.J. Shulman <Shulman.aj@gmail.com>2024-07-17 14:58:46 -0400
commitdd379c9e4c4c214b2ed14e3fd9a5f966e4f03b48 (patch)
tree1fcf384c00279bfb0e54e31a53d72e3991bed657 /src/client/views/nodes/ChatBox/Agent.ts
parent0340c24eccce3d90c03934dec14d574128fb32ef (diff)
working on rendering MarkDown and added NoTool
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}'`);
}
}