Skip to content

Commit 93a05f8

Browse files
authored
remove deprecated model(solar-pro) (#19704)
1 parent 6594a8a commit 93a05f8

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

llama-index-integrations/llms/llama-index-llms-upstage/llama_index/llms/upstage/utils.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,15 @@
55

66
DEFAULT_UPSTAGE_API_BASE = "https://api.upstage.ai/v1/solar"
77
DEFAULT_CONTEXT_WINDOW = 32768
8-
CHAT_MODELS = {"solar-mini": 32768, "solar-pro": 4096, "solar-pro2": 65536}
8+
CHAT_MODELS = {"solar-mini": 32768, "solar-pro2": 65536}
99

10-
FUNCTION_CALLING_MODELS = ["solar-mini", "solar-pro", "solar-pro2"]
11-
DOC_PARSING_MODELS = ["solar-pro", "solar-pro2"]
10+
FUNCTION_CALLING_MODELS = ["solar-mini", "solar-pro2"]
11+
DOC_PARSING_MODELS = ["solar-pro2"]
1212

1313
ALL_AVAILABLE_MODELS = {**CHAT_MODELS}
1414

1515
SOLAR_TOKENIZERS = {
1616
"solar-pro2": "upstage/solar-pro2-tokenizer",
17-
"solar-pro": "upstage/solar-pro-tokenizer",
1817
"solar-mini": "upstage/solar-1-mini-tokenizer",
1918
}
2019

llama-index-integrations/llms/llama-index-llms-upstage/pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ dev = [
1313
"pre-commit==3.2.0",
1414
"pylint==2.15.10",
1515
"pytest==7.2.1",
16+
"pytest-asyncio>=0.23.8",
1617
"pytest-cov>=6.1.1",
1718
"pytest-mock==3.11.1",
1819
"ruff==0.11.11",
@@ -26,7 +27,7 @@ dev = [
2627

2728
[project]
2829
name = "llama-index-llms-upstage"
29-
version = "0.6.1"
30+
version = "0.6.2"
3031
description = "llama-index llms upstage integration"
3132
authors = [{name = "Your Name", email = "[email protected]"}]
3233
requires-python = ">=3.9,<4.0"

0 commit comments

Comments
 (0)