diff --git a/packages/components/nodes/embeddings/JinaAIEmbedding/JinaAIEmbedding.ts b/packages/components/nodes/embeddings/JinaAIEmbedding/JinaAIEmbedding.ts
index 3e96cfb70..946a2031a 100644
--- a/packages/components/nodes/embeddings/JinaAIEmbedding/JinaAIEmbedding.ts
+++ b/packages/components/nodes/embeddings/JinaAIEmbedding/JinaAIEmbedding.ts
@@ -17,7 +17,7 @@ class JinaAIEmbedding_Embeddings implements INode {
constructor() {
this.label = 'Jina Embeddings'
this.name = 'jinaEmbeddings'
- this.version = 1.0
+ this.version = 2.0
this.type = 'JinaEmbeddings'
this.icon = 'JinaAIEmbedding.svg'
this.category = 'Embeddings'
@@ -36,18 +36,28 @@ class JinaAIEmbedding_Embeddings implements INode {
type: 'string',
default: 'jina-embeddings-v2-base-en',
description: 'Refer to JinaAI documentation for available models'
+ },
+ {
+ label: 'Dimensions',
+ name: 'modelDimensions',
+ type: 'number',
+ default: 1024,
+ description:
+ 'Refer to JinaAI documentation for available dimensions'
}
]
}
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise {
const modelName = nodeData.inputs?.modelName as string
+ const modelDimensions = nodeData.inputs?.modelDimensions as number
const credentialData = await getCredentialData(nodeData.credential ?? '', options)
const apiKey = getCredentialParam('jinaAIAPIKey', credentialData, nodeData)
const model = new JinaEmbeddings({
apiKey: apiKey,
- model: modelName
+ model: modelName,
+ dimensions: modelDimensions
})
return model