admin/n8n-enterprise-unlocked
1
0
Fork 0
mirror of https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git synced 2025-12-17 18:12:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(AI Agent Node): Fix tool-usage with fallback mechanism (#16898)

Browse Source
This commit is contained in:
Benjamin Schroth
2025-07-03 09:48:44 +02:00
committed by GitHub
parent 29bf4a46bd
commit 58fd1ec325
1 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
packages/@n8n/nodes-langchain/nodes/agents/Agent/agents/ToolsAgent/V2/execute.ts
View File

@@ -1,7 +1,11 @@
import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
import type { ChatPromptTemplate } from '@langchain/core/prompts';
import { RunnableSequence } from '@langchain/core/runnables';
import { AgentExecutor, createToolCallingAgent } from 'langchain/agents';
import {
AgentExecutor,
type AgentRunnableSequence,
createToolCallingAgent,
} from 'langchain/agents';
import type { BaseChatMemory } from 'langchain/memory';
import type { DynamicStructuredTool, Tool } from 'langchain/tools';
import omit from 'lodash/omit';
@@ -38,16 +42,24 @@ function createAgentExecutor(
memory?: BaseChatMemory,
fallbackModel?: BaseChatModel | null,
) {
const modelWithFallback = fallbackModel ? model.withFallbacks([fallbackModel]) : model;
const agent = createToolCallingAgent({
llm: modelWithFallback,
llm: model,
tools,
prompt,
streamRunnable: false,
});
let fallbackAgent: AgentRunnableSequence | undefined;
if (fallbackModel) {
fallbackAgent = createToolCallingAgent({
llm: fallbackModel,
tools,
prompt,
streamRunnable: false,
});
}
const runnableAgent = RunnableSequence.from([
agent,
fallbackAgent ? agent.withFallbacks([fallbackAgent]) : agent,
getAgentStepsParser(outputParser, memory),
fixEmptyContentMessage,
]);