lint 2

langchain-ai · Dec 6, 2023 · b0e9887 · b0e9887
1 parent 84376a9
commit b0e9887
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/libs/langchain/tests/unit_tests/indexes/test_indexing.py b/libs/langchain/tests/unit_tests/indexes/test_indexing.py
@@ -80,7 +80,7 @@ def add_documents(  # type: ignore
         *,
         ids: Optional[Sequence[str]] = None,
         **kwargs: Any,
-    ) -> None:
+    ) -> List[str]:
         """Add the given documents to the store (insert behavior)."""
         if ids and len(ids) != len(documents):
             raise ValueError(
@@ -97,6 +97,8 @@ def add_documents(  # type: ignore
                 )
             self.store[_id] = document
 
+        return list(ids)
+
     async def aadd_documents(
         self,
         documents: Sequence[Document],

diff --git a/libs/langchain/tests/unit_tests/retrievers/test_parent_document.py b/libs/langchain/tests/unit_tests/retrievers/test_parent_document.py
@@ -17,7 +17,7 @@ def similarity_search(
             return []
         return [res]
 
-    def add_documents(self, documents: Sequence[Document]) -> None:
+    def add_documents(self, documents: Sequence[Document], **kwargs: Any) -> List[str]:
         print(documents)
         return super().add_documents(
             documents, ids=[f"{i}" for i in range(len(documents))]