test_mmr_from_examples() — langchain Function Reference

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

Function python CoreAbstractions Serialization

Entity Profile

CoreAbstractions→ Serialization→ test_mmr_from_examples() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/example_selectors/test_similarity.py lines 194–219

def test_mmr_from_examples() -> None:
    examples = [{"foo": "bar"}]
    embeddings = FakeEmbeddings(size=1)
    selector = MaxMarginalRelevanceExampleSelector.from_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"}]