langchain/tests/integration_tests/llms/test_aviary.py

12 lines
267 B
Python
Raw Normal View History

"""Test Anyscale API wrapper."""
from langchain.llms.aviary import Aviary
def test_aviary_call() -> None:
"""Test valid call to Anyscale."""
llm = Aviary()
output = llm("Say bar:")
print(f"llm answer:\n{output}")
assert isinstance(output, str)