diff options
Diffstat (limited to 'src/client/views/nodes/ChatBox/Agent.ts')
-rw-r--r-- | src/client/views/nodes/ChatBox/Agent.ts | 4 |
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}'`); } } |