- update LangchainJS version

- add types & interfaces

- small ui bug fix

- update marketplace files

packages/components/nodes/vectorstores/Chroma_Existing/Chroma_Existing.ts

@@ -1,4 +1,7 @@
 import { INode, INodeData, INodeParams } from '../../../src/Interface'
+import { Chroma } from 'langchain/vectorstores/chroma'
+import { Embeddings } from 'langchain/embeddings/base'
+import { getBaseClasses } from '../../../src/utils'
 
 class Chroma_Existing_VectorStores implements INode {
     label: string
@@ -17,6 +20,7 @@ class Chroma_Existing_VectorStores implements INode {
         this.icon = 'chroma.svg'
         this.category = 'Vector Stores'
         this.description = 'Load existing index from Chroma (i.e: Document has been upserted)'
+        this.baseClasses = [this.type, ...getBaseClasses(Chroma)]
         this.inputs = [
             {
                 label: 'Embeddings',
@@ -31,15 +35,9 @@ class Chroma_Existing_VectorStores implements INode {
         ]
     }
 
-    async getBaseClasses(): Promise<string[]> {
-        return ['BaseRetriever']
-    }
-
     async init(nodeData: INodeData): Promise<any> {
-        const { Chroma } = await import('langchain/vectorstores')
-
         const collectionName = nodeData.inputs?.collectionName as string
-        const embeddings = nodeData.inputs?.embeddings
+        const embeddings = nodeData.inputs?.embeddings as Embeddings
 
         const vectorStore = await Chroma.fromExistingCollection(embeddings, {
             collectionName

packages/components/nodes/vectorstores/Chroma_Upsert/Chroma_Upsert.ts

@@ -1,4 +1,8 @@
 import { INode, INodeData, INodeParams } from '../../../src/Interface'
+import { Chroma } from 'langchain/vectorstores/chroma'
+import { Embeddings } from 'langchain/embeddings/base'
+import { Document } from 'langchain/document'
+import { getBaseClasses } from '../../../src/utils'
 
 class ChromaUpsert_VectorStores implements INode {
     label: string
@@ -17,6 +21,7 @@ class ChromaUpsert_VectorStores implements INode {
         this.icon = 'chroma.svg'
         this.category = 'Vector Stores'
         this.description = 'Upsert documents to Chroma'
+        this.baseClasses = [this.type, ...getBaseClasses(Chroma)]
         this.inputs = [
             {
                 label: 'Document',
@@ -36,28 +41,20 @@ class ChromaUpsert_VectorStores implements INode {
         ]
     }
 
-    async getBaseClasses(): Promise<string[]> {
-        return ['BaseRetriever']
-    }
-
     async init(nodeData: INodeData): Promise<any> {
-        const { Chroma } = await import('langchain/vectorstores')
-        const { Document } = await import('langchain/document')
-
         const collectionName = nodeData.inputs?.collectionName as string
-        const docs = nodeData.inputs?.document
-        const embeddings = nodeData.inputs?.embeddings
+        const docs = nodeData.inputs?.document as Document[]
+        const embeddings = nodeData.inputs?.embeddings as Embeddings
 
         const finalDocs = []
         for (let i = 0; i < docs.length; i += 1) {
             finalDocs.push(new Document(docs[i]))
         }
 
-        const result = await Chroma.fromDocuments(finalDocs, embeddings, {
+        const vectorStore = await Chroma.fromDocuments(finalDocs, embeddings, {
             collectionName
         })
-
-        const retriever = result.asRetriever()
+        const retriever = vectorStore.asRetriever()
         return retriever
     }
 }

packages/components/nodes/vectorstores/Pinecone_Existing/Pinecone_Existing.ts

@@ -1,5 +1,8 @@
 import { INode, INodeData, INodeParams } from '../../../src/Interface'
 import { PineconeClient } from '@pinecone-database/pinecone'
+import { PineconeStore } from 'langchain/vectorstores/pinecone'
+import { Embeddings } from 'langchain/embeddings/base'
+import { getBaseClasses } from '../../../src/utils'
 
 class Pinecone_Existing_VectorStores implements INode {
     label: string
@@ -18,6 +21,7 @@ class Pinecone_Existing_VectorStores implements INode {
         this.icon = 'pinecone.png'
         this.category = 'Vector Stores'
         this.description = 'Load existing index from Pinecone (i.e: Document has been upserted)'
+        this.baseClasses = [this.type, ...getBaseClasses(PineconeStore)]
         this.inputs = [
             {
                 label: 'Embeddings',
@@ -42,17 +46,11 @@ class Pinecone_Existing_VectorStores implements INode {
         ]
     }
 
-    async getBaseClasses(): Promise<string[]> {
-        return ['BaseRetriever']
-    }
-
     async init(nodeData: INodeData): Promise<any> {
-        const { PineconeStore } = await import('langchain/vectorstores')
-
         const pineconeApiKey = nodeData.inputs?.pineconeApiKey as string
         const pineconeEnv = nodeData.inputs?.pineconeEnv as string
         const index = nodeData.inputs?.pineconeIndex as string
-        const embeddings = nodeData.inputs?.embeddings
+        const embeddings = nodeData.inputs?.embeddings as Embeddings
 
         const client = new PineconeClient()
         await client.init({

packages/components/nodes/vectorstores/Pinecone_Upsert/Pinecone_Upsert.ts

@@ -1,5 +1,9 @@
 import { INode, INodeData, INodeParams } from '../../../src/Interface'
 import { PineconeClient } from '@pinecone-database/pinecone'
+import { PineconeStore } from 'langchain/vectorstores/pinecone'
+import { Embeddings } from 'langchain/embeddings/base'
+import { Document } from 'langchain/document'
+import { getBaseClasses } from '../../../src/utils'
 
 class PineconeUpsert_VectorStores implements INode {
     label: string
@@ -18,6 +22,7 @@ class PineconeUpsert_VectorStores implements INode {
         this.icon = 'pinecone.png'
         this.category = 'Vector Stores'
         this.description = 'Upsert documents to Pinecone'
+        this.baseClasses = [this.type, ...getBaseClasses(PineconeStore)]
         this.inputs = [
             {
                 label: 'Document',
@@ -47,19 +52,12 @@ class PineconeUpsert_VectorStores implements INode {
         ]
     }
 
-    async getBaseClasses(): Promise<string[]> {
-        return ['BaseRetriever']
-    }
-
     async init(nodeData: INodeData): Promise<any> {
-        const { PineconeStore } = await import('langchain/vectorstores')
-        const { Document } = await import('langchain/document')
-
         const pineconeApiKey = nodeData.inputs?.pineconeApiKey as string
         const pineconeEnv = nodeData.inputs?.pineconeEnv as string
         const index = nodeData.inputs?.pineconeIndex as string
-        const docs = nodeData.inputs?.document
-        const embeddings = nodeData.inputs?.embeddings
+        const docs = nodeData.inputs?.document as Document[]
+        const embeddings = nodeData.inputs?.embeddings as Embeddings
 
         const client = new PineconeClient()
         await client.init({
@@ -74,11 +72,10 @@ class PineconeUpsert_VectorStores implements INode {
             finalDocs.push(new Document(docs[i]))
         }
 
-        const result = await PineconeStore.fromDocuments(finalDocs, embeddings, {
+        const vectorStore = await PineconeStore.fromDocuments(finalDocs, embeddings, {
             pineconeIndex
         })
-
-        const retriever = result.asRetriever()
+        const retriever = vectorStore.asRetriever()
         return retriever
     }
 }