test_agenerate() — langchain Function Reference

Architecture documentation for the test_agenerate() function in test_chat_models.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_agenerate() — langchain Function Reference

Dependency Diagram

graph TD
  e6429eb0_7563_91f8_d8b4_6482b04c6728["test_agenerate()"]
  af57ae60_607e_c138_9ab0_fb8bb1c5916a["test_chat_models.py"]
  e6429eb0_7563_91f8_d8b4_6482b04c6728 -->|defined in| af57ae60_607e_c138_9ab0_fb8bb1c5916a
  style e6429eb0_7563_91f8_d8b4_6482b04c6728 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/groq/tests/integration_tests/test_chat_models.py lines 160–175

async def test_agenerate() -> None:
    """Test async generation."""
    n = 1
    chat = ChatGroq(model=DEFAULT_MODEL_NAME, max_tokens=10, n=1)
    message = HumanMessage(content="Hello")
    response = await chat.agenerate([[message], [message]])
    assert isinstance(response, LLMResult)
    assert len(response.generations) == 2
    assert response.llm_output
    assert response.llm_output["model_name"] == chat.model_name
    for generations in response.generations:
        assert len(generations) == n
        for generation in generations:
            assert isinstance(generation, ChatGeneration)
            assert isinstance(generation.text, str)
            assert generation.text == generation.message.content