feat: Add AI tool building capabilities (#7336)

Github issue / Community forum post (link here to close automatically): https://community.n8n.io/t/langchain-memory-chat/23733 --------- Signed-off-by: Oleg Ivaniv <me@olegivaniv.com> Co-authored-by: Oleg Ivaniv <me@olegivaniv.com> Co-authored-by: Val <68596159+valya@users.noreply.github.com> Co-authored-by: Alex Grozav <alex@grozav.com> Co-authored-by: कारतोफ्फेलस्क्रिप्ट™ <aditya@netroy.in> Co-authored-by: Deborah <deborah@starfallprojects.co.uk> Co-authored-by: Jesper Bylund <mail@jesperbylund.com> Co-authored-by: Jon <jonathan.bennetts@gmail.com> Co-authored-by: Michael Kret <88898367+michael-radency@users.noreply.github.com> Co-authored-by: Giulio Andreini <andreini@netseven.it> Co-authored-by: Mason Geloso <Mason.geloso@gmail.com> Co-authored-by: Mason Geloso <hone@Masons-Mac-mini.local> Co-authored-by: Mutasem Aldmour <mutasem@n8n.io>
2025-12-17 18:12:04 +00:00 · 2023-11-29 12:13:55 +01:00
parent dbfd617ace
commit 87def60979
243 changed files with 21526 additions and 321 deletions
--- a/packages/@n8n/nodes-langchain/nodes/chains/ChainRetrievalQA/ChainRetrievalQa.node.ts
+++ b/packages/@n8n/nodes-langchain/nodes/chains/ChainRetrievalQA/ChainRetrievalQa.node.ts
@@ -0,0 +1,119 @@
+import {
+	NodeConnectionType,
+	type IExecuteFunctions,
+	type INodeExecutionData,
+	type INodeType,
+	type INodeTypeDescription,
+	NodeOperationError,
+} from 'n8n-workflow';
+
+import { RetrievalQAChain } from 'langchain/chains';
+import type { BaseLanguageModel } from 'langchain/dist/base_language';
+import type { BaseRetriever } from 'langchain/schema/retriever';
+import { getTemplateNoticeField } from '../../../utils/sharedFields';
+
+export class ChainRetrievalQa implements INodeType {
+	description: INodeTypeDescription = {
+		displayName: 'Question and Answer Chain',
+		name: 'chainRetrievalQa',
+		icon: 'fa:link',
+		group: ['transform'],
+		version: [1, 1.1],
+		description: 'Answer questions about retrieved documents',
+		defaults: {
+			name: 'Question and Answer Chain',
+			color: '#909298',
+		},
+		codex: {
+			alias: ['LangChain'],
+			categories: ['AI'],
+			subcategories: {
+				AI: ['Chains'],
+			},
+			resources: {
+				primaryDocumentation: [
+					{
+						url: 'https://docs.n8n.io/integrations/builtin/cluster-nodes/root-nodes/n8n-nodes-langchain.chainretrievalqa/',
+					},
+				],
+			},
+		},
+		// eslint-disable-next-line n8n-nodes-base/node-class-description-inputs-wrong-regular-node
+		inputs: [
+			NodeConnectionType.Main,
+			{
+				displayName: 'Model',
+				maxConnections: 1,
+				type: NodeConnectionType.AiLanguageModel,
+				required: true,
+			},
+			{
+				displayName: 'Retriever',
+				maxConnections: 1,
+				type: NodeConnectionType.AiRetriever,
+				required: true,
+			},
+		],
+		outputs: [NodeConnectionType.Main],
+		credentials: [],
+		properties: [
+			getTemplateNoticeField(1960),
+			{
+				displayName: 'Query',
+				name: 'query',
+				type: 'string',
+				required: true,
+				default: '={{ $json.input }}',
+				displayOptions: {
+					show: {
+						'@version': [1],
+					},
+				},
+			},
+			{
+				displayName: 'Query',
+				name: 'query',
+				type: 'string',
+				required: true,
+				default: '={{ $json.chat_input }}',
+				displayOptions: {
+					show: {
+						'@version': [1.1],
+					},
+				},
+			},
+		],
+	};
+
+	async execute(this: IExecuteFunctions): Promise<INodeExecutionData[][]> {
+		this.logger.verbose('Executing Retrieval QA Chain');
+
+		const model = (await this.getInputConnectionData(
+			NodeConnectionType.AiLanguageModel,
+			0,
+		)) as BaseLanguageModel;
+
+		const retriever = (await this.getInputConnectionData(
+			NodeConnectionType.AiRetriever,
+			0,
+		)) as BaseRetriever;
+
+		const items = this.getInputData();
+		const chain = RetrievalQAChain.fromLLM(model, retriever);
+
+		const returnData: INodeExecutionData[] = [];
+
+		// Run for each item
+		for (let itemIndex = 0; itemIndex < items.length; itemIndex++) {
+			const query = this.getNodeParameter('query', itemIndex) as string;
+
+			if (query === undefined) {
+				throw new NodeOperationError(this.getNode(), 'The ‘query‘ parameter is empty.');
+			}
+
+			const response = await chain.call({ query });
+			returnData.push({ json: { response } });
+		}
+		return this.prepareOutputData(returnData);
+	}
+}