github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-13 14:50:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update parent_document_retriever.py about **kwargs (#22219)

Browse Source 
Add kwargs in add_documents function

**langchain**: Add **kwargs in parent_document_retriever"
 - **Add kwargs for `add_document` in `parent_document_retriever.py`** 


If no one reviews your PR within a few days, please @-mention one of
baskaryan, efriis, eyurtsev, ccurme, vbarda, hwchase17.
This commit is contained in:
SteveLiao 2024-05-29 02:35:38 +08:00 committed by GitHub
parent 8dfa3c5f1a
commit 7496fe2b16
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/langchain/langchain/retrievers/parent_document_retriever.py
View File

@ -1,5 +1,5 @@
import uuid import uuid
from typing import List, Optional, Sequence from typing import Any, List, Optional, Sequence
from langchain_core.documents import Document from langchain_core.documents import Document
from langchain_text_splitters import TextSplitter from langchain_text_splitters import TextSplitter
@ -74,6 +74,7 @@ class ParentDocumentRetriever(MultiVectorRetriever):
documents: List[Document], documents: List[Document],
ids: Optional[List[str]] = None, ids: Optional[List[str]] = None,
add_to_docstore: bool = True, add_to_docstore: bool = True,
**kwargs: Any,
) -> None: ) -> None:
"""Adds documents to the docstore and vectorstores. """Adds documents to the docstore and vectorstores.
@ -119,6 +120,6 @@ class ParentDocumentRetriever(MultiVectorRetriever):
_doc.metadata[self.id_key] = _id _doc.metadata[self.id_key] = _id
docs.extend(sub_docs) docs.extend(sub_docs)
full_docs.append((_id, doc)) full_docs.append((_id, doc))
self.vectorstore.add_documents(docs) self.vectorstore.add_documents(docs, **kwargs)
if add_to_docstore: if add_to_docstore:
self.docstore.mset(full_docs) self.docstore.mset(full_docs)