test_mmr_select_examples() — langchain Function Reference

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

Function python CoreAbstractions Serialization

Entity Profile

CoreAbstractions→ Serialization→ test_mmr_select_examples() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/example_selectors/test_similarity.py lines 168–178

def test_mmr_select_examples() -> None:
    vector_store = DummyVectorStore()
    selector = MaxMarginalRelevanceExampleSelector(
        vectorstore=vector_store,
        input_keys=["foo2"],
        example_keys=["query", "k", "fetch_k"],
        k=2,
        fetch_k=5,
    )
    examples = selector.select_examples({"foo": "bar", "foo2": "bar2"})
    assert examples == [{"query": "bar2", "k": 2, "fetch_k": 5}] * 2