Fix: Update VOLC_MODEL_MAP and suffixes to correctly normalize and map package names

This commit is contained in:
kim
2026-01-29 17:59:09 +08:00
parent aeb89b69a0
commit d696c92879
+15 -1
View File
@@ -74,14 +74,26 @@ REDIS_PASSWORD = os.getenv("REDIS_PASSWORD", None)
# Suffixes to clean from Volcengine response # Suffixes to clean from Volcengine response
SUFFIXES_TO_REMOVE = [ SUFFIXES_TO_REMOVE = [
"_pack_data_collaboration",
"pack_data_collaboration",
"_data_collaboration_resource_pack", "_data_collaboration_resource_pack",
"_data_collaboration",
"data_collaboration",
"协作奖励计划资源包", "协作奖励计划资源包",
"免费在线推理资源包", "免费在线推理资源包",
"免费在线推理包", "免费在线推理包",
"资源包", "资源包",
"_free_infer_res_pack",
"_pack_free_infer",
"_thinking_free_infer_res_pack",
"_thinking_data_collaboration",
"_free_inference_resource_pack",
"free_inference_resource_pack",
"_infer_res_pack",
] ]
# Volcengine Name -> LiteLLM Model Name Map # Volcengine Name -> LiteLLM Model Name Map
# Key is the normalized (lowercase, hyphens) name from Volcengine API
VOLC_MODEL_MAP = { VOLC_MODEL_MAP = {
"doubao-seed-1-6": ["doubao-seed-1-6-250615", "doubao-seed-1-6-251015"], "doubao-seed-1-6": ["doubao-seed-1-6-250615", "doubao-seed-1-6-251015"],
"doubao-seed-1-8": "doubao-seed-1-8-251228", "doubao-seed-1-8": "doubao-seed-1-8-251228",
@@ -92,8 +104,10 @@ VOLC_MODEL_MAP = {
"deepseek-v3-2": "deepseek-v3-2-251201", "deepseek-v3-2": "deepseek-v3-2-251201",
"deepseek-r1": "deepseek-r1-250528", "deepseek-r1": "deepseek-r1-250528",
"kimi-k2": "kimi-k2-thinking-251104", "kimi-k2": "kimi-k2-thinking-251104",
"glm-4_7": "glm-4-7-251222",
"glm-4-7": "glm-4-7-251222", "glm-4-7": "glm-4-7-251222",
# Specific mappings for corner cases if suffixes don't catch everything
"doubao-seed-1-8-pack-data-collaboration": "doubao-seed-1-8-251228",
"glm-4-7-pack-data-collaboration": "glm-4-7-251222",
} }
# Static Quotas (Fallback/Strict Limits) # Static Quotas (Fallback/Strict Limits)