langchain/libs/community/tests/integration_tests/llms/test_openllm.py
Aaron Pham 12fced13f4
chore(community): update to OpenLLM 0.6 (#24609)
Update to OpenLLM 0.6, which we decides to make use of OpenLLM's
OpenAI-compatible endpoint. Thus, OpenLLM will now just become a thin
wrapper around OpenAI wrapper.

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>

---------

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
Co-authored-by: ccurme <chester.curme@gmail.com>
2024-12-16 14:30:07 -05:00

14 lines
297 B
Python

"""Test OpenLLM API wrapper."""
import pytest
from langchain_community.llms.openllm import OpenLLM
@pytest.mark.scheduled
def test_openai_call() -> None:
"""Test valid call to openai."""
llm = OpenLLM()
output = llm.invoke("Say something nice:")
assert isinstance(output, str)