mirror of
https://github.com/hwchase17/langchain
synced 2024-11-18 09:25:54 +00:00
16 lines
335 B
Python
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"
|