From 76ba977a5ca195ec9925317665207e11e5e2e264 Mon Sep 17 00:00:00 2001 From: Henry Date: Wed, 13 Sep 2023 09:57:32 +0100 Subject: [PATCH] add callbacks to agents --- .../nodes/agents/MRKLAgentChat/MRKLAgentChat.ts | 10 +++++++--- .../nodes/agents/MRKLAgentLLM/MRKLAgentLLM.ts | 10 +++++++--- 2 files changed, 14 insertions(+), 6 deletions(-) diff --git a/packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts b/packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts index 0a9e744c..ed169e62 100644 --- a/packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts +++ b/packages/components/nodes/agents/MRKLAgentChat/MRKLAgentChat.ts @@ -1,9 +1,10 @@ -import { INode, INodeData, INodeParams } from '../../../src/Interface' +import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' import { initializeAgentExecutorWithOptions, AgentExecutor } from 'langchain/agents' import { getBaseClasses } from '../../../src/utils' import { Tool } from 'langchain/tools' import { BaseLanguageModel } from 'langchain/base_language' import { flatten } from 'lodash' +import { additionalCallbacks } from '../../../src/handler' class MRKLAgentChat_Agents implements INode { label: string @@ -51,9 +52,12 @@ class MRKLAgentChat_Agents implements INode { return executor } - async run(nodeData: INodeData, input: string): Promise { + async run(nodeData: INodeData, input: string, options: ICommonObject): Promise { const executor = nodeData.instance as AgentExecutor - const result = await executor.call({ input }) + + const callbacks = await additionalCallbacks(nodeData, options) + + const result = await executor.call({ input }, [...callbacks]) return result?.output } diff --git a/packages/components/nodes/agents/MRKLAgentLLM/MRKLAgentLLM.ts b/packages/components/nodes/agents/MRKLAgentLLM/MRKLAgentLLM.ts index d7af586b..74929af8 100644 --- a/packages/components/nodes/agents/MRKLAgentLLM/MRKLAgentLLM.ts +++ b/packages/components/nodes/agents/MRKLAgentLLM/MRKLAgentLLM.ts @@ -1,9 +1,10 @@ -import { INode, INodeData, INodeParams } from '../../../src/Interface' +import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' import { initializeAgentExecutorWithOptions, AgentExecutor } from 'langchain/agents' import { Tool } from 'langchain/tools' import { getBaseClasses } from '../../../src/utils' import { BaseLanguageModel } from 'langchain/base_language' import { flatten } from 'lodash' +import { additionalCallbacks } from '../../../src/handler' class MRKLAgentLLM_Agents implements INode { label: string @@ -52,9 +53,12 @@ class MRKLAgentLLM_Agents implements INode { return executor } - async run(nodeData: INodeData, input: string): Promise { + async run(nodeData: INodeData, input: string, options: ICommonObject): Promise { const executor = nodeData.instance as AgentExecutor - const result = await executor.call({ input }) + + const callbacks = await additionalCallbacks(nodeData, options) + + const result = await executor.call({ input }, [...callbacks]) return result?.output }