test_aindex_simple_delete_full() — langchain Function Reference

Architecture documentation for the test_aindex_simple_delete_full() function in test_indexing.py from the langchain codebase.

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_aindex_simple_delete_full() — langchain Function Reference

Dependency Diagram

graph TD
  357b23b7_c572_a3d6_da19_ad1906a35d26["test_aindex_simple_delete_full()"]
  9bc4e8b6_769a_ae11_3fc1_309cb678c248["test_indexing.py"]
  357b23b7_c572_a3d6_da19_ad1906a35d26 -->|defined in| 9bc4e8b6_769a_ae11_3fc1_309cb678c248
  style 357b23b7_c572_a3d6_da19_ad1906a35d26 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/indexes/test_indexing.py lines 333–414

async def test_aindex_simple_delete_full(
    arecord_manager: SQLRecordManager,
    vector_store: InMemoryVectorStore,
) -> None:
    """Indexing some content to confirm it gets added only once."""
    loader = ToyLoader(
        documents=[
            Document(
                page_content="This is a test document.",
            ),
            Document(
                page_content="This is another document.",
            ),
        ],
    )

    with patch.object(
        arecord_manager,
        "aget_time",
        return_value=_JANUARY_FIRST,
    ):
        assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == {
            "num_added": 2,
            "num_deleted": 0,
            "num_skipped": 0,
            "num_updated": 0,
        }

    with patch.object(
        arecord_manager,
        "aget_time",
        return_value=_JANUARY_FIRST,
    ):
        assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == {
            "num_added": 0,
            "num_deleted": 0,
            "num_skipped": 2,
            "num_updated": 0,
        }

    loader = ToyLoader(
        documents=[
            Document(
                page_content="mutated document 1",
            ),
            Document(
                page_content="This is another document.",  # <-- Same as original
            ),
        ],
    )

    with patch.object(
        arecord_manager,
        "aget_time",
        return_value=_JANUARY_SECOND,
    ):
        assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == {
            "num_added": 1,
            "num_deleted": 1,
            "num_skipped": 1,
            "num_updated": 0,
        }

    doc_texts = {
        # Ignoring type since doc should be in the store and not a None
        vector_store.store.get(uid).page_content  # type: ignore[union-attr]
        for uid in vector_store.store
    }
    assert doc_texts == {"mutated document 1", "This is another document."}

    # Attempt to index again verify that nothing changes
    with patch.object(
        arecord_manager,
        "aget_time",
        return_value=_JANUARY_SECOND,
    ):
        assert await aindex(loader, arecord_manager, vector_store, cleanup="full") == {
            "num_added": 0,
            "num_deleted": 0,
            "num_skipped": 2,
            "num_updated": 0,