test_chroma_large_batch() — langchain Function Reference

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

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_chroma_large_batch() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/partners/chroma/tests/integration_tests/test_vectorstores.py lines 618–634

def test_chroma_large_batch() -> None:
    client = chromadb.HttpClient()
    embedding_function = MyEmbeddingFunction(fak=Fak(size=255))
    col = client.get_or_create_collection(
        "my_collection",
        embedding_function=embedding_function,  # type: ignore[arg-type]
    )
    docs = ["This is a test document"] * (client.get_max_batch_size() + 100)
    db = Chroma.from_texts(
        client=client,
        collection_name=col.name,
        texts=docs,
        embedding=embedding_function.fak,
        ids=[str(uuid.uuid4()) for _ in range(len(docs))],
    )

    db.delete_collection()