Source code for langchain_community.chat_models.solar

"""封装了Solar聊天模型。"""

from typing import Dict

from langchain_core._api import deprecated
from langchain_core.pydantic_v1 import Field, root_validator
from langchain_core.utils import get_from_dict_or_env

from langchain_community.chat_models import ChatOpenAI
from langchain_community.llms.solar import SOLAR_SERVICE_URL_BASE, SolarCommon


@deprecated(  # type: ignore[arg-type]
    since="0.0.34", removal="0.3.0", alternative_import="langchain_upstage.ChatUpstage"
)
class SolarChat(SolarCommon, ChatOpenAI):
    """封装了Solar大型语言模型。
要使用，您应该已安装``openai`` python包，并且设置了环境变量``SOLAR_API_KEY``为您的API密钥。
（Solar的聊天API与OpenAI的SDK兼容。）
参考自https://console.upstage.ai/services/solar
示例：
    .. code-block:: python

        from langchain_community.chat_models.solar import SolarChat

        solar = SolarChat(model="solar-1-mini-chat")"""

    max_tokens: int = Field(default=1024)

    # this is needed to match ChatOpenAI superclass
    class Config:
        """此pydantic对象的配置。"""

        allow_population_by_field_name = True
        arbitrary_types_allowed = True
        extra = "ignore"

    @root_validator()
    def validate_environment(cls, values: Dict) -> Dict:
        """验证环境是否设置正确。"""
        values["solar_api_key"] = get_from_dict_or_env(
            values, "solar_api_key", "SOLAR_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["solar_api_key"],
            "base_url": (
                values["base_url"] if "base_url" in values else SOLAR_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