langchain/libs/community/tests/integration_tests/llms/test_aviary.py
Erick Friis 3a2eb6e12b
infra: add print rule to ruff (#16221)
Added noqa for existing prints. Can slowly remove / will prevent more
being intro'd
2024-02-09 16:13:30 -08:00

12 lines
291 B
Python

"""Test Anyscale API wrapper."""
from langchain_community.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}") # noqa: T201
assert isinstance(output, str)