Skip to content

Commit 7965698

Browse files
committed
feat: enable DeepSeek-V4-Pro and DeepSeek-V4-Flash ModelScope presets
1 parent 5caf8a2 commit 7965698

2 files changed

Lines changed: 16 additions & 16 deletions

File tree

chcode/prompts.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -204,20 +204,20 @@ async def select_or_custom(
204204
# "stream_usage": True,
205205
# "extra_body": {"top_k": 40},
206206
# },
207-
# {
208-
# "model": "deepseek-ai/DeepSeek-V4-Pro",
209-
# "base_url": MODELSCOPE_BASE_URL,
210-
# "temperature": 1.0,
211-
# "top_p": 1.0,
212-
# "stream_usage": True,
213-
# },
214-
# {
215-
# "model": "deepseek-ai/DeepSeek-V4-Flash",
216-
# "base_url": MODELSCOPE_BASE_URL,
217-
# "temperature": 1.0,
218-
# "top_p": 1.0,
219-
# "stream_usage": True,
220-
# },
207+
{
208+
"model": "deepseek-ai/DeepSeek-V4-Pro",
209+
"base_url": MODELSCOPE_BASE_URL,
210+
"temperature": 1.0,
211+
"top_p": 1.0,
212+
"stream_usage": True,
213+
},
214+
{
215+
"model": "deepseek-ai/DeepSeek-V4-Flash",
216+
"base_url": MODELSCOPE_BASE_URL,
217+
"temperature": 1.0,
218+
"top_p": 1.0,
219+
"stream_usage": True,
220+
},
221221
]
222222

223223
API_KEY_ENV_VARS = [

tests/test_config_extended.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -912,7 +912,7 @@ async def test_modelscope_first_time(self, mock_config_dir):
912912

913913
data = mod.load_model_json()
914914
assert data["default"]["model"] == "ZhipuAI/GLM-5"
915-
assert len(data["fallback"]) == 11
915+
assert len(data["fallback"]) == 13
916916

917917
@pytest.mark.asyncio
918918
async def test_modelscope_merge_existing(self, mock_config_dir):
@@ -980,7 +980,7 @@ async def test_configure_modelscope_manual_key(self):
980980

981981
assert result is not None
982982
assert result["default"]["api_key"] == "ms-test-key"
983-
assert len(result["fallback"]) == 11
983+
assert len(result["fallback"]) == 13
984984

985985
@pytest.mark.asyncio
986986
async def test_configure_modelscope_cancel(self):

0 commit comments

Comments
 (0)