Source code for langchain_community.chat_models.moonshot

"""Moonshot聊天模型的封装。"""
from typing import Dict

from langchain_core.pydantic_v1 import root_validator
from langchain_core.utils import get_from_dict_or_env

from langchain_community.chat_models import ChatOpenAI
from langchain_community.llms.moonshot import MOONSHOT_SERVICE_URL_BASE, MoonshotCommon


class MoonshotChat(MoonshotCommon, ChatOpenAI):  # type: ignore[misc]
    """实现大型语言模型。

    要使用，您应该已安装``openai`` python包，并且
    环境变量``MOONSHOT_API_KEY``设置为您的API密钥。
    (Moonshot的聊天API与OpenAI的SDK兼容。)

    参考自https://platform.moonshot.cn/docs

    示例:
        .. code-block:: python

            from langchain_community.chat_models.moonshot import MoonshotChat

            moonshot = MoonshotChat(model="moonshot-v1-8k")"""

    @root_validator()
    def validate_environment(cls, values: Dict) -> Dict:
        """验证环境是否设置正确。"""
        values["moonshot_api_key"] = get_from_dict_or_env(
            values, "moonshot_api_key", "MOONSHOT_API_KEY"
        )

        try:
            import openai

        except ImportError:
            raise ImportError(
                "Could not import openai python package. "
                "Please install it with `pip install openai`."
            )

        client_params = {
            "api_key": values["moonshot_api_key"].get_secret_value(),
            "base_url": values["base_url"]
            if "base_url" in values
            else MOONSHOT_SERVICE_URL_BASE,
        }

        if not values.get("client"):
            values["client"] = openai.OpenAI(**client_params).chat.completions
        if not values.get("async_client"):
            values["async_client"] = openai.AsyncOpenAI(
                **client_params
            ).chat.completions

        return values