mirror of
https://github.com/hwchase17/langchain
synced 2024-11-02 09:40:22 +00:00
f09bd0b75b
Co-authored-by: Sean Cho <sean@upstage.ai> Co-authored-by: JuHyung-Son <sonju0427@gmail.com>
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
"""Wrapper around Solar chat models."""
|
|
|
|
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(
|
|
since="0.0.34", removal="0.2.0", alternative_import="langchain_upstage.ChatUpstage"
|
|
)
|
|
class SolarChat(SolarCommon, ChatOpenAI):
|
|
"""Wrapper around Solar large language models.
|
|
To use, you should have the ``openai`` python package installed, and the
|
|
environment variable ``SOLAR_API_KEY`` set with your API key.
|
|
(Solar's chat API is compatible with OpenAI's SDK.)
|
|
Referenced from https://console.upstage.ai/services/solar
|
|
Example:
|
|
.. 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:
|
|
"""Configuration for this pydantic object."""
|
|
|
|
allow_population_by_field_name = True
|
|
arbitrary_types_allowed = True
|
|
extra = "ignore"
|
|
|
|
@root_validator()
|
|
def validate_environment(cls, values: Dict) -> Dict:
|
|
"""Validate that the environment is set up correctly."""
|
|
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
|