jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: bug where the pinecone component was inserting embeddings three times instead of once. (#2826)

Browse source 
fixed the bug where pinecone component inserts embeddings 3 times

When we send the chunks with the embedding component to the Pinecone component, the generated embeddings are being saved three times instead of once.
This commit is contained in:
Sai Pavan Kumar 2024-07-22 18:06:49 +05:30 • committed by GitHub
commit 9731ebb27d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/components/vectorstores/Pinecone.py
View file

@ -22,6 +22,7 @@ class PineconeVectorStoreComponent(LCVectorStoreComponent):
documentation = "https://python.langchain.com/v0.2/docs/integrations/vectorstores/pinecone/" documentation = "https://python.langchain.com/v0.2/docs/integrations/vectorstores/pinecone/"
name = "Pinecone" name = "Pinecone"
icon = "Pinecone" icon = "Pinecone"
pinecone_instance = None
inputs = [ inputs = [
StrInput(name="index_name", display_name="Index Name", required=True), StrInput(name="index_name", display_name="Index Name", required=True),
@ -61,6 +62,8 @@ class PineconeVectorStoreComponent(LCVectorStoreComponent):
return self._build_pinecone() return self._build_pinecone()
def _build_pinecone(self) -> Pinecone: def _build_pinecone(self) -> Pinecone:
if self.pinecone_instance is not None:
return self.pinecone_instance
from langchain_pinecone._utilities import DistanceStrategy from langchain_pinecone._utilities import DistanceStrategy
from langchain_pinecone.vectorstores import Pinecone from langchain_pinecone.vectorstores import Pinecone
@ -85,7 +88,7 @@ class PineconeVectorStoreComponent(LCVectorStoreComponent):
if documents: if documents:
pinecone.add_documents(documents) pinecone.add_documents(documents)
self.pinecone_instance = pinecone
return pinecone return pinecone
def search_documents(self) -> List[Data]: def search_documents(self) -> List[Data]: