You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
langchain/libs/community/tests/unit_tests/llms/test_moonshot.py

16 lines
335 B
Python

import os
import pytest
from langchain_community.llms.moonshot import Moonshot
os.environ["MOONSHOT_API_KEY"] = "key"
@pytest.mark.requires("openai")
def test_moonshot_model_param() -> None:
llm = Moonshot(model="foo")
assert llm.model_name == "foo"
llm = Moonshot(model_name="bar")
assert llm.model_name == "bar"