langchain/tests/integration_tests/llms/test_stochasticai.py

"""Test StochasticAI API wrapper."""

from langchain.llms.stochasticai import StochasticAI


def test_stochasticai_call() -> None:
    """Test valid call to StochasticAI."""
    llm = StochasticAI()
    output = llm("Say foo:")
    assert isinstance(output, str)