mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +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"
|