mudhorn/n8n
1
0
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2025-02-02 07:01:30 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(Embeddings OpenAI Node, Embeddings Azure OpenAI Node): Add dimensions option (#11773)

Browse source
This commit is contained in:
Mutasem Aldmour 2024-11-18 15:53:07 +01:00 committed by GitHub
parent a544b74d87
commit de01a8a01d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 62 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
packages/@n8n/nodes-langchain/nodes/embeddings/EmbeddingsAzureOpenAi/EmbeddingsAzureOpenAi.node.ts
View file

@ -87,6 +87,36 @@ export class EmbeddingsAzureOpenAi implements INodeType {
'Maximum amount of time a request is allowed to take in seconds. Set to -1 for no timeout.',
type: 'number',
},
{
displayName: 'Dimensions',
name: 'dimensions',
default: undefined,
description:
'The number of dimensions the resulting output embeddings should have. Only supported in text-embedding-3 and later models.',
type: 'options',
options: [
{
name: '256',
value: 256,
},
{
name: '512',
value: 512,
},
{
name: '1024',
value: 1024,
},
{
name: '1536',
value: 1536,
},
{
name: '3072',
value: 3072,
},
],
},
],
},
],
@ -105,6 +135,7 @@ export class EmbeddingsAzureOpenAi implements INodeType {
batchSize?: number;
stripNewLines?: boolean;
timeout?: number;
dimensions?: number | undefined;
};
if (options.timeout === -1) {

31
packages/@n8n/nodes-langchain/nodes/embeddings/EmbeddingsOpenAI/EmbeddingsOpenAi.node.ts
View file

@ -135,6 +135,36 @@ export class EmbeddingsOpenAi implements INodeType {
type: 'collection',
default: {},
options: [
{
displayName: 'Dimensions',
name: 'dimensions',
default: undefined,
description:
'The number of dimensions the resulting output embeddings should have. Only supported in text-embedding-3 and later models.',
type: 'options',
options: [
{
name: '256',
value: 256,
},
{
name: '512',
value: 512,
},
{
name: '1024',
value: 1024,
},
{
name: '1536',
value: 1536,
},
{
name: '3072',
value: 3072,
},
],
},
{
displayName: 'Base URL',
name: 'baseURL',
@ -179,6 +209,7 @@ export class EmbeddingsOpenAi implements INodeType {
batchSize?: number;
stripNewLines?: boolean;
timeout?: number;
dimensions?: number | undefined;
};
if (options.timeout === -1) {