mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
9 lines
241 B
Python
9 lines
241 B
Python
|
from langchain.llms.openlm import OpenLM
|
||
|
|
|
||
|
|
|
||
|
|
def test_openlm_call() -> None:
|
||
|
|
"""Test valid call to openlm."""
|
||
|
|
llm = OpenLM(model_name="dolly-v2-7b", max_tokens=10)
|
||
|
|
output = llm(prompt="Say foo:")
|
||
|
|
assert isinstance(output, str)
|