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

feat(Embeddings Cohere Node): Add v3 Cohere models (#9887)

Browse Source
This commit is contained in:
KhDu
2024-07-03 10:58:54 +03:00
committed by GitHub
parent 957b2d6108
commit 403e19b3e3
1 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -67,17 +67,33 @@ export class EmbeddingsCohere implements INodeType {
default: 'embed-english-v2.0',
options: [
{
name: 'Embed-English-v2.0(4096 Dimensions)',
value: 'embed-english-v2.0',
},
{
name: 'Embed-English-Light-v2.0(1024 Dimensions)',
name: 'Embed-English-Light-v2.0 (1024 Dimensions)',
value: 'embed-english-light-v2.0',
},
{
name: 'Embed-Multilingual-v2.0(768 Dimensions)',
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',
},
],
},
],
@@ -86,9 +102,9 @@ export class EmbeddingsCohere implements INodeType {
async supplyData(this: IExecuteFunctions, itemIndex: number): Promise<SupplyData> {
this.logger.verbose('Supply data for embeddings Cohere');
const modelName = this.getNodeParameter('modelName', itemIndex, 'embed-english-v2.0') as string;
const credentials = await this.getCredentials('cohereApi');
const credentials = (await this.getCredentials('cohereApi')) as { apiKey: string };
const embeddings = new CohereEmbeddings({
apiKey: credentials.apiKey as string,
apiKey: credentials.apiKey,
model: modelName,
});