From 8d327e465cf84c429c7aeaa3383cade633ed6c5e Mon Sep 17 00:00:00 2001
From: Gene Ruebsamen <gururise@gmail.com>
Date: Mon, 3 Feb 2025 09:01:50 -0800
Subject: [PATCH] Expose dimensions parameter on Jina Embeddings (#3969)

* Expose dimensions parameter on Jina Embeddings

* fix linting issue

* update description text to refer to dimensions

* Update JinaAIEmbedding.ts

---------

Co-authored-by: Henry Heng <henryheng@flowiseai.com>
---
 .../embeddings/JinaAIEmbedding/JinaAIEmbedding.ts  | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

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 <a href="https://jina.ai/embeddings/" target="_blank">JinaAI documentation</a> for available models'
+            },
+            {
+                label: 'Dimensions',
+                name: 'modelDimensions',
+                type: 'number',
+                default: 1024,
+                description:
+                    'Refer to <a href="https://jina.ai/embeddings/" target="_blank">JinaAI documentation</a> for available dimensions'
             }
         ]
     }
 
     async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
         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