first commit

packages/@n8n/nodes-langchain/nodes/embeddings/EmbeddingsCohere/EmbeddingsCohere.node.ts

@@ -0,0 +1,114 @@
+import { CohereEmbeddings } from '@langchain/cohere';
+import {
+	NodeConnectionTypes,
+	type INodeType,
+	type INodeTypeDescription,
+	type ISupplyDataFunctions,
+	type SupplyData,
+} from 'n8n-workflow';
+
+import { logWrapper, getConnectionHintNoticeField } from '@n8n/ai-utilities';
+
+export class EmbeddingsCohere implements INodeType {
+	description: INodeTypeDescription = {
+		displayName: 'Embeddings Cohere',
+		name: 'embeddingsCohere',
+		icon: { light: 'file:cohere.svg', dark: 'file:cohere.dark.svg' },
+		group: ['transform'],
+		version: 1,
+		description: 'Use Cohere Embeddings',
+		defaults: {
+			name: 'Embeddings Cohere',
+		},
+		requestDefaults: {
+			ignoreHttpStatusErrors: true,
+			baseURL: '={{ $credentials.host }}',
+		},
+		credentials: [
+			{
+				name: 'cohereApi',
+				required: true,
+			},
+		],
+		codex: {
+			categories: ['AI'],
+			subcategories: {
+				AI: ['Embeddings'],
+			},
+			resources: {
+				primaryDocumentation: [
+					{
+						url: 'https://docs.n8n.io/integrations/builtin/cluster-nodes/sub-nodes/n8n-nodes-langchain.embeddingscohere/',
+					},
+				],
+			},
+		},
+
+		inputs: [],
+
+		outputs: [NodeConnectionTypes.AiEmbedding],
+		outputNames: ['Embeddings'],
+		properties: [
+			getConnectionHintNoticeField([NodeConnectionTypes.AiVectorStore]),
+			{
+				displayName:
+					'Each model is using different dimensional density for embeddings. Please make sure to use the same dimensionality for your vector store. The default model is using 768-dimensional embeddings.',
+				name: 'notice',
+				type: 'notice',
+				default: '',
+			},
+			{
+				displayName: 'Model',
+				name: 'modelName',
+				type: 'options',
+				description:
+					'The model which will generate the embeddings. <a href="https://docs.cohere.com/docs/models">Learn more</a>.',
+				default: 'embed-english-v2.0',
+				options: [
+					{
+						name: 'Embed-English-Light-v2.0 (1024 Dimensions)',
+						value: 'embed-english-light-v2.0',
+					},
+					{
+						name: 'Embed-English-Light-v3.0 (384 Dimensions)',
+						value: 'embed-english-light-v3.0',
+					},
+					{
+						name: 'Embed-English-v2.0 (4096 Dimensions)',
+						value: 'embed-english-v2.0',
+					},
+					{
+						name: 'Embed-English-v3.0 (1024 Dimensions)',
+						value: 'embed-english-v3.0',
+					},
+					{
+						name: 'Embed-Multilingual-Light-v3.0 (384 Dimensions)',
+						value: 'embed-multilingual-light-v3.0',
+					},
+					{
+						name: 'Embed-Multilingual-v2.0 (768 Dimensions)',
+						value: 'embed-multilingual-v2.0',
+					},
+					{
+						name: 'Embed-Multilingual-v3.0 (1024 Dimensions)',
+						value: 'embed-multilingual-v3.0',
+					},
+				],
+			},
+		],
+	};
+
+	async supplyData(this: ISupplyDataFunctions, itemIndex: number): Promise<SupplyData> {
+		this.logger.debug('Supply data for embeddings Cohere');
+		const modelName = this.getNodeParameter('modelName', itemIndex, 'embed-english-v2.0') as string;
+		const credentials = await this.getCredentials<{ apiKey: string }>('cohereApi');
+		const embeddings = new CohereEmbeddings({
+			apiKey: credentials.apiKey,
+			model: modelName,
+		});
+
+		return {
+			response: logWrapper(embeddings, this),
+		};
+	}
+}