diff --git a/packages/components/nodes/embeddings/GoogleVertexAIEmbedding/GoogleVertexAIEmbedding.ts b/packages/components/nodes/embeddings/GoogleVertexAIEmbedding/GoogleVertexAIEmbedding.ts index 05ceb415..d92bd7d8 100644 --- a/packages/components/nodes/embeddings/GoogleVertexAIEmbedding/GoogleVertexAIEmbedding.ts +++ b/packages/components/nodes/embeddings/GoogleVertexAIEmbedding/GoogleVertexAIEmbedding.ts @@ -1,6 +1,6 @@ -import { GoogleVertexAIEmbeddings, GoogleVertexAIEmbeddingsParams} from "langchain/embeddings/googlevertexai"; +import { GoogleVertexAIEmbeddings, GoogleVertexAIEmbeddingsParams } from 'langchain/embeddings/googlevertexai' import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' -import { getBaseClasses} from '../../../src/utils' +import { getBaseClasses } from '../../../src/utils' class GoogleVertexAIEmbedding_Embeddings implements INode { label: string @@ -14,7 +14,6 @@ class GoogleVertexAIEmbedding_Embeddings implements INode { credential: INodeParams inputs: INodeParams[] - constructor() { this.label = 'GoogleVertexAI Embeddings' this.name = 'googlevertexaiEmbeddings' @@ -35,22 +34,20 @@ class GoogleVertexAIEmbedding_Embeddings implements INode { name: 'textembedding-gecko' } ], - default: 'textembedding-gecko', + default: 'textembedding-gecko' } ] } - async init(nodeData: INodeData, _: string, options: ICommonObject): Promise { - const model = nodeData.inputs?.modelName as string const obj: GoogleVertexAIEmbeddingsParams = { model } - const embedding = new GoogleVertexAIEmbeddings(obj); + const embedding = new GoogleVertexAIEmbeddings(obj) return embedding } } -module.exports = { nodeClass: GoogleVertexAIEmbedding_Embeddings } \ No newline at end of file +module.exports = { nodeClass: GoogleVertexAIEmbedding_Embeddings }