test_inmemory_call_embeddings_async() — langchain Function Reference

Architecture documentation for the test_inmemory_call_embeddings_async() function in test_in_memory.py from the langchain codebase.

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ test_inmemory_call_embeddings_async() — langchain Function Reference

Dependency Diagram

graph TD
  ab71cabf_3f6d_1d03_d89f_89da4e291a49["test_inmemory_call_embeddings_async()"]
  a974d690_d5fa_ba65_671d_ce8278eefe7d["test_in_memory.py"]
  ab71cabf_3f6d_1d03_d89f_89da4e291a49 -->|defined in| a974d690_d5fa_ba65_671d_ce8278eefe7d
  style ab71cabf_3f6d_1d03_d89f_89da4e291a49 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/vectorstores/test_in_memory.py lines 209–222

async def test_inmemory_call_embeddings_async() -> None:
    embeddings_mock = Mock(
        wraps=DeterministicFakeEmbedding(size=3),
        aembed_documents=AsyncMock(),
        aembed_query=AsyncMock(),
    )
    store = InMemoryVectorStore(embedding=embeddings_mock)

    await store.aadd_texts("foo")
    await store.asimilarity_search("foo", k=1)

    # Ensure the async embedding function is called
    assert embeddings_mock.aembed_documents.await_count == 1
    assert embeddings_mock.aembed_query.await_count == 1