langchain/libs/community/tests/unit_tests/llms/test_moonshot.py
Eugene Yurtsev 25fbe356b4
community[patch]: upgrade to recent version of mypy (#21616)
This PR upgrades community to a recent version of mypy. It inserts type:
ignore on all existing failures.
2024-05-13 14:55:07 -04:00

16 lines
387 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") # type: ignore[call-arg]
assert llm.model_name == "foo"
llm = Moonshot(model_name="bar") # type: ignore[call-arg]
assert llm.model_name == "bar"