You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
langchain/libs/community/tests/integration_tests/llms/test_cerebriumai.py

11 lines
280 B
Python

"""Test CerebriumAI API wrapper."""
from langchain_community.llms.cerebriumai import CerebriumAI
def test_cerebriumai_call() -> None:
"""Test valid call to cerebriumai."""
llm = CerebriumAI(max_length=10)
output = llm("Say foo:")
assert isinstance(output, str)