test_chroma_mmr_by_vector() — langchain Function Reference

Architecture documentation for the test_chroma_mmr_by_vector() function in test_vectorstores.py from the langchain codebase.

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ test_chroma_mmr_by_vector() — langchain Function Reference

Dependency Diagram

graph TD
  8fb3148e_4452_1a82_bf9a_3805f81789fe["test_chroma_mmr_by_vector()"]
  289e5a4f_8e2e_ddb6_af2e_804f91dabdb8["test_vectorstores.py"]
  8fb3148e_4452_1a82_bf9a_3805f81789fe -->|defined in| 289e5a4f_8e2e_ddb6_af2e_804f91dabdb8
  style 8fb3148e_4452_1a82_bf9a_3805f81789fe fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/chroma/tests/integration_tests/test_vectorstores.py lines 389–403

def test_chroma_mmr_by_vector() -> None:
    """Test end to end construction and search."""
    texts = ["foo", "bar", "baz"]
    embeddings = FakeEmbeddings()
    docsearch = Chroma.from_texts(
        collection_name="test_collection",
        texts=texts,
        embedding=embeddings,
    )
    embedded_query = embeddings.embed_query("foo")
    output = docsearch.max_marginal_relevance_search_by_vector(embedded_query, k=1)
    docsearch.delete_collection()
    assert len(output) == 1
    assert output[0].page_content == "foo"
    assert output[0].id is not None