test_aembed_documents() — langchain Function Reference

Architecture documentation for the test_aembed_documents() function in test_caching.py from the langchain codebase.

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_aembed_documents() — langchain Function Reference

Dependency Diagram

graph TD
  3bb7eb81_d378_f547_fbb3_5a1a2b0287ab["test_aembed_documents()"]
  6b1a1e58_d1bc_2756_cc7d_9c01f64a73a2["test_caching.py"]
  3bb7eb81_d378_f547_fbb3_5a1a2b0287ab -->|defined in| 6b1a1e58_d1bc_2756_cc7d_9c01f64a73a2
  style 3bb7eb81_d378_f547_fbb3_5a1a2b0287ab fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/embeddings/test_caching.py lines 114–124

async def test_aembed_documents(cache_embeddings: CacheBackedEmbeddings) -> None:
    texts = ["1", "22", "a", "333"]
    vectors = await cache_embeddings.aembed_documents(texts)
    expected_vectors: list[list[float]] = [[1, 2.0], [2.0, 3.0], [1.0, 2.0], [3.0, 4.0]]
    assert vectors == expected_vectors
    keys = [
        key async for key in cache_embeddings.document_embedding_store.ayield_keys()
    ]
    assert len(keys) == 4
    # UUID is expected to be the same for the same text
    assert keys[0] == "test_namespace812b86c1-8ebf-5483-95c6-c95cf2b52d12"