mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
17 lines
459 B
Python
17 lines
459 B
Python
|
"""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)
|