blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix KeyError in pinecone wrapper (#568)

Browse Source 
Pinecone can return multiple matches with different text keys in the
metadata, and the current behavior will throw a KeyError if one pops up
that isn't the expected text key. This PR only selects the matches with
the correct text key.
This commit is contained in:
Bernie G 2023-01-10 10:43:39 -05:00 committed by GitHub
parent 7a32cde5ff
commit d39ab27969
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/vectorstores/pinecone.py
View File

@ -81,8 +81,9 @@ class Pinecone(VectorStore):
results = self._index.query([query_obj], top_k=k, include_metadata=True) results = self._index.query([query_obj], top_k=k, include_metadata=True)
for res in results["matches"]: for res in results["matches"]:
metadata = res["metadata"] metadata = res["metadata"]
text = metadata.pop(self._text_key) if self._text_key in metadata:
docs.append(Document(page_content=text, metadata=metadata)) text = metadata.pop(self._text_key)
docs.append(Document(page_content=text, metadata=metadata))
return docs return docs
@classmethod @classmethod