mirror of
https://github.com/hwchase17/langchain.git
synced 2025-05-07 16:18:09 +00:00
bf5193bb99
Upgrade to using a literal for specifying the extra which is the
recommended approach in pydantic 2.
This works correctly also in pydantic v1.
```python
from pydantic.v1 import BaseModel
class Foo(BaseModel, extra="forbid"):
x: int
Foo(x=5, y=1)
```
And
```python
from pydantic.v1 import BaseModel
class Foo(BaseModel):
x: int
class Config:
extra = "forbid"
Foo(x=5, y=1)
```
## Enum -> literal using grit pattern:
```
engine marzano(0.1)
language python
or {
`extra=Extra.allow` => `extra="allow"`,
`extra=Extra.forbid` => `extra="forbid"`,
`extra=Extra.ignore` => `extra="ignore"`
}
```
Resorted attributes in config and removed doc-string in case we will
need to deal with going back and forth between pydantic v1 and v2 during
the 0.3 release. (This will reduce merge conflicts.)
## Sort attributes in Config:
```
engine marzano(0.1)
language python
function sort($values) js {
return $values.text.split(',').sort().join("\n");
}
class_definition($name, $body) as $C where {
$name <: `Config`,
$body <: block($statements),
$values = [],
$statements <: some bubble($values) assignment() as $A where {
$values += $A
},
$body => sort($values),
}
```
69 lines
2.2 KiB
Python
69 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
|
|
from langchain_core.utils import get_from_dict_or_env, pre_init
|
|
|
|
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):
|
|
"""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:
|
|
allow_population_by_field_name = True
|
|
arbitrary_types_allowed = True
|
|
extra = "ignore"
|
|
|
|
@pre_init
|
|
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
|