Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #37 from FlowiseAI/feature/Pinecone 

Feature/Add namespace to pinecone
This commit is contained in:
Henry Heng 2023-04-27 18:48:20 +01:00 committed by GitHub
parent bf60b88a81 e8bd86d134
commit 0f6da4c44f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 28 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
packages/components/nodes/vectorstores/Pinecone_Existing/Pinecone_Existing.ts
View File

@ -1,6 +1,6 @@
import { INode, INodeData, INodeOutputsValue, INodeParams } from '../../../src/Interface'
import { PineconeClient } from '@pinecone-database/pinecone'
import { PineconeStore } from 'langchain/vectorstores/pinecone'
import { PineconeLibArgs, PineconeStore } from 'langchain/vectorstores/pinecone'
import { Embeddings } from 'langchain/embeddings/base'
import { getBaseClasses } from '../../../src/utils'
@ -43,6 +43,13 @@ class Pinecone_Existing_VectorStores implements INode {
label: 'Pinecone Index',
name: 'pineconeIndex',
type: 'string'
},
{
label: 'Pinecone Namespace',
name: 'pineconeNamespace',
type: 'string',
placeholder: 'my-first-namespace',
optional: true
}
]
this.outputs = [
@ -63,6 +70,7 @@ class Pinecone_Existing_VectorStores implements INode {
const pineconeApiKey = nodeData.inputs?.pineconeApiKey as string
const pineconeEnv = nodeData.inputs?.pineconeEnv as string
const index = nodeData.inputs?.pineconeIndex as string
const pineconeNamespace = nodeData.inputs?.pineconeNamespace as string
const embeddings = nodeData.inputs?.embeddings as Embeddings
const output = nodeData.outputs?.output as string
@ -74,9 +82,13 @@ class Pinecone_Existing_VectorStores implements INode {
const pineconeIndex = client.Index(index)
const vectorStore = await PineconeStore.fromExistingIndex(embeddings, {
const obj: PineconeLibArgs = {
pineconeIndex
})
}
if (pineconeNamespace) obj.namespace = pineconeNamespace
const vectorStore = await PineconeStore.fromExistingIndex(embeddings, obj)
if (output === 'retriever') {
const retriever = vectorStore.asRetriever()

19
packages/components/nodes/vectorstores/Pinecone_Upsert/Pinecone_Upsert.ts
View File

@ -1,6 +1,6 @@
import { INode, INodeData, INodeOutputsValue, INodeParams } from '../../../src/Interface'
import { PineconeClient } from '@pinecone-database/pinecone'
import { PineconeStore } from 'langchain/vectorstores/pinecone'
import { PineconeLibArgs, PineconeStore } from 'langchain/vectorstores/pinecone'
import { Embeddings } from 'langchain/embeddings/base'
import { Document } from 'langchain/document'
import { getBaseClasses } from '../../../src/utils'
@ -46,9 +46,11 @@ class PineconeUpsert_VectorStores implements INode {
type: 'string'
},
{
label: 'Pinecone Index',
name: 'pineconeIndex',
type: 'string'
label: 'Pinecone Namespace',
name: 'pineconeNamespace',
type: 'string',
placeholder: 'my-first-namespace',
optional: true
}
]
this.outputs = [
@ -69,6 +71,7 @@ class PineconeUpsert_VectorStores implements INode {
const pineconeApiKey = nodeData.inputs?.pineconeApiKey as string
const pineconeEnv = nodeData.inputs?.pineconeEnv as string
const index = nodeData.inputs?.pineconeIndex as string
const pineconeNamespace = nodeData.inputs?.pineconeNamespace as string
const docs = nodeData.inputs?.document as Document[]
const embeddings = nodeData.inputs?.embeddings as Embeddings
const output = nodeData.outputs?.output as string
@ -86,9 +89,13 @@ class PineconeUpsert_VectorStores implements INode {
finalDocs.push(new Document(docs[i]))
}
const vectorStore = await PineconeStore.fromDocuments(finalDocs, embeddings, {
const obj: PineconeLibArgs = {
pineconeIndex
})
}
if (pineconeNamespace) obj.namespace = pineconeNamespace
const vectorStore = await PineconeStore.fromDocuments(finalDocs, embeddings, obj)
if (output === 'retriever') {
const retriever = vectorStore.asRetriever()