test_abatch() — langchain Function Reference

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

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_abatch() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/partners/ollama/tests/integration_tests/test_llms.py lines 40–46

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

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