test_mmr_afrom_examples() — langchain Function Reference

Architecture documentation for the test_mmr_afrom_examples() function in test_similarity.py from the langchain codebase.

Function python CoreAbstractions RunnableInterface

Entity Profile

CoreAbstractions→ RunnableInterface→ test_mmr_afrom_examples() — langchain Function Reference

Dependency Diagram

graph TD
  088e08d5_defd_8045_6402_5a2401554f9f["test_mmr_afrom_examples()"]
  1458705c_6963_08f1_0216_aefb63eadfc8["test_similarity.py"]
  088e08d5_defd_8045_6402_5a2401554f9f -->|defined in| 1458705c_6963_08f1_0216_aefb63eadfc8
  style 088e08d5_defd_8045_6402_5a2401554f9f fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/example_selectors/test_similarity.py lines 222–247

async def test_mmr_afrom_examples() -> None:
    examples = [{"foo": "bar"}]
    embeddings = FakeEmbeddings(size=1)
    selector = await MaxMarginalRelevanceExampleSelector.afrom_examples(
        examples=examples,
        embeddings=embeddings,
        vectorstore_cls=DummyVectorStore,
        k=2,
        fetch_k=5,
        input_keys=["foo"],
        example_keys=["some_example_key"],
        vectorstore_kwargs={"vs_foo": "vs_bar"},
        init_arg="some_init_arg",
    )
    assert selector.input_keys == ["foo"]
    assert selector.example_keys == ["some_example_key"]
    assert selector.k == 2
    assert selector.fetch_k == 5
    assert selector.vectorstore_kwargs == {"vs_foo": "vs_bar"}

    assert isinstance(selector.vectorstore, DummyVectorStore)
    vector_store = selector.vectorstore
    assert vector_store.embeddings is embeddings
    assert vector_store.init_arg == "some_init_arg"
    assert vector_store.texts == ["bar"]
    assert vector_store.metadatas == [{"foo": "bar"}]