test_batch() — langchain Function Reference

Architecture documentation for the test_batch() function in test_llms.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_batch() — langchain Function Reference

Dependency Diagram

graph TD
  72fc344f_b112_c1a9_199e_6c9bd3a5268e["test_batch()"]
  b375279a_4970_f212_b93f_b3ffc97dc9d8["test_llms.py"]
  72fc344f_b112_c1a9_199e_6c9bd3a5268e -->|defined in| b375279a_4970_f212_b93f_b3ffc97dc9d8
  style 72fc344f_b112_c1a9_199e_6c9bd3a5268e fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/ollama/tests/integration_tests/test_llms.py lines 31–37

def test_batch() -> None:
    """Test batch sync token generation from `OllamaLLM`."""
    llm = OllamaLLM(model=MODEL_NAME)

    result = llm.batch(["I'm Pickle Rick", "I'm not Pickle Rick"])
    for token in result:
        assert isinstance(token, str)