test_deleting_documents_async() — langchain Function Reference

Source Code

libs/standard-tests/langchain_tests/integration_tests/vectorstores.py lines 556–576

    async def test_deleting_documents_async(self, vectorstore: VectorStore) -> None:
        """Test deleting documents from the `VectorStore`.

        ??? note "Troubleshooting"

            If this test fails, check that `aadd_documents` preserves identifiers
            passed in through `ids`, and that `delete` correctly removes
            documents.
        """
        if not self.has_async:
            pytest.skip("Async tests not supported.")

        documents = [
            Document(page_content="foo", metadata={"id": 1}),
            Document(page_content="bar", metadata={"id": 2}),
        ]
        ids = await vectorstore.aadd_documents(documents, ids=["1", "2"])
        assert ids == ["1", "2"]
        await vectorstore.adelete(["1"])
        documents = await vectorstore.asimilarity_search("foo", k=1)
        assert documents == [Document(page_content="bar", metadata={"id": 2}, id="2")]