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

feat: Add fallback mechanism for agent and basic chain llm (#16617)

Browse Source
This commit is contained in:
Benjamin Schroth
2025-06-26 16:14:03 +02:00
committed by GitHub
parent 0b7bca29f8
commit 6408d5a1b0
20 changed files with 476 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
packages/@n8n/nodes-langchain/nodes/chains/ChainLLM/methods/chainExecutor.ts
View File

@@ -88,15 +88,24 @@ async function executeSimpleChain({
llm,
query,
prompt,
fallbackLlm,
}: {
context: IExecuteFunctions;
llm: BaseLanguageModel;
query: string;
prompt: ChatPromptTemplate | PromptTemplate;
fallbackLlm?: BaseLanguageModel | null;
}) {
const outputParser = getOutputParserForLLM(llm);
let model;
const chain = prompt.pipe(llm).pipe(outputParser).withConfig(getTracingConfig(context));
if (fallbackLlm) {
model = llm.withFallbacks([fallbackLlm]);
} else {
model = llm;
}
const chain = prompt.pipe(model).pipe(outputParser).withConfig(getTracingConfig(context));
// Execute the chain
const response = await chain.invoke({
@@ -118,6 +127,7 @@ export async function executeChain({
llm,
outputParser,
messages,
fallbackLlm,
}: ChainExecutionParams): Promise<unknown[]> {
// If no output parsers provided, use a simple chain with basic prompt template
if (!outputParser) {
@@ -134,6 +144,7 @@ export async function executeChain({
llm,
query,
prompt: promptTemplate,
fallbackLlm,
});
}