Skip to content

Commit 798413b

Browse files
authored
remove deprecated model(solar-pro) (#59)
* remove deprecated model(solar-pro) * bump version
1 parent 405bb07 commit 798413b

File tree

3 files changed

+5
-8
lines changed

3 files changed

+5
-8
lines changed

libs/upstage/langchain_upstage/chat_models.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,9 @@
5454

5555
from langchain_upstage.document_parse import UpstageDocumentParseLoader
5656

57-
DOC_PARSING_MODEL = ["solar-pro2", "solar-pro"]
57+
DOC_PARSING_MODEL = ["solar-pro2"]
5858
SOLAR_TOKENIZERS = {
5959
"solar-pro2": "upstage/solar-pro2-tokenizer",
60-
"solar-pro": "upstage/solar-pro-tokenizer",
6160
"solar-mini": "upstage/solar-1-mini-tokenizer",
6261
}
6362

@@ -141,7 +140,7 @@ def _get_ls_params(
141140
"""openai organization is not supported for upstage."""
142141
tiktoken_model_name: Optional[str] = None
143142
"""tiktoken is not supported for upstage."""
144-
tokenizer_name: Optional[str] = "upstage/solar-pro-tokenizer"
143+
tokenizer_name: Optional[str] = "upstage/solar-pro2-tokenizer"
145144
"""huggingface tokenizer name. Solar tokenizer is opened in huggingface https://huggingface.co/upstage/solar-pro-tokenizer"""
146145
default_headers: Union[Mapping[str, str], None] = DEFAULT_HEADERS
147146
"""add trace header."""

libs/upstage/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "langchain-upstage"
3-
version = "0.7.2"
3+
version = "0.7.3"
44
description = "An integration package connecting Upstage and LangChain"
55
authors = []
66
readme = "README.md"

libs/upstage/tests/unit_tests/test_chat_models.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -149,8 +149,7 @@ def mock_create(*args: Any, **kwargs: Any) -> Any:
149149

150150

151151
def test_upstage_invoke_with_doc_parsing_model(mock_completion: dict) -> None:
152-
# TODO: update model_name
153-
llm = ChatUpstage(model="solar-pro")
152+
llm = ChatUpstage(model="solar-pro2")
154153
mock_client = MagicMock()
155154
completed = False
156155

@@ -195,8 +194,7 @@ async def mock_create(*args: Any, **kwargs: Any) -> Any:
195194

196195

197196
async def test_upstage_ainvoke_with_doc_parsing_model(mock_completion: dict) -> None:
198-
# TODO: update model_name
199-
llm = ChatUpstage(model="solar-pro")
197+
llm = ChatUpstage(model="solar-pro2")
200198
mock_client = AsyncMock()
201199
completed = False
202200

0 commit comments

Comments
 (0)