mudhorn/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2024-11-10 06:34:05 -08:00
Code Issues Actions 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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 24 additions and 8 deletions
Show stats 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,
});