langchain/tests/integration_tests/llms/test_openllm.py

17 lines
459 B
Python
Raw Normal View History

"""Test OpenLLM wrapper."""
from langchain.llms.openllm import OpenLLM
def test_openllm_llm_local() -> None:
llm = OpenLLM(model_name="flan-t5", model_id="google/flan-t5-small")
output = llm("Say foo:")
assert isinstance(output, str)
def test_openllm_with_kwargs() -> None:
llm = OpenLLM(
model_name="flan-t5", model_id="google/flan-t5-small", temperature=0.84
)
output = llm("Say bar:")
assert isinstance(output, str)