add top K to vector stores

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

@@ -49,6 +49,7 @@ class Pinecone_Existing_VectorStores implements INode {
                 name: 'pineconeNamespace',
                 type: 'string',
                 placeholder: 'my-first-namespace',
+                additionalParams: true,
                 optional: true
             },
             {
@@ -57,6 +58,15 @@ class Pinecone_Existing_VectorStores implements INode {
                 type: 'json',
                 optional: true,
                 additionalParams: true
+            },
+            {
+                label: 'Top K',
+                name: 'topK',
+                description: 'Number of top results to fetch. Default to 4',
+                placeholder: '4',
+                type: 'number',
+                additionalParams: true,
+                optional: true
             }
         ]
         this.outputs = [
@@ -79,9 +89,10 @@ class Pinecone_Existing_VectorStores implements INode {
         const index = nodeData.inputs?.pineconeIndex as string
         const pineconeNamespace = nodeData.inputs?.pineconeNamespace as string
         const pineconeMetadataFilter = nodeData.inputs?.pineconeMetadataFilter
-
         const embeddings = nodeData.inputs?.embeddings as Embeddings
         const output = nodeData.outputs?.output as string
+        const topK = nodeData.inputs?.topK as string
+        const k = topK ? parseInt(topK, 10) : 4
 
         const client = new PineconeClient()
         await client.init({
@@ -104,9 +115,10 @@ class Pinecone_Existing_VectorStores implements INode {
         const vectorStore = await PineconeStore.fromExistingIndex(embeddings, obj)
 
         if (output === 'retriever') {
-            const retriever = vectorStore.asRetriever()
+            const retriever = vectorStore.asRetriever(k)
             return retriever
         } else if (output === 'vectorStore') {
+            ;(vectorStore as any).k = k
             return vectorStore
         }
         return vectorStore