langchain/libs/community/tests/integration_tests/llms/test_bittensor.py

"""Test Bittensor Validator Endpoint wrapper."""

from langchain_community.llms import NIBittensorLLM


def test_bittensor_call() -> None:
    """Test valid call to validator endpoint."""
    llm = NIBittensorLLM(system_prompt="Your task is to answer user prompt.")
    output = llm("Say foo:")
    assert isinstance(output, str)