Fix: Update VOLC_MODEL_MAP and suffixes to correctly normalize and map package names
This commit is contained in:
+15
-1
@@ -74,14 +74,26 @@ REDIS_PASSWORD = os.getenv("REDIS_PASSWORD", None)
|
||||
|
||||
# Suffixes to clean from Volcengine response
|
||||
SUFFIXES_TO_REMOVE = [
|
||||
"_pack_data_collaboration",
|
||||
"pack_data_collaboration",
|
||||
"_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
|
||||
# Key is the normalized (lowercase, hyphens) name from Volcengine API
|
||||
VOLC_MODEL_MAP = {
|
||||
"doubao-seed-1-6": ["doubao-seed-1-6-250615", "doubao-seed-1-6-251015"],
|
||||
"doubao-seed-1-8": "doubao-seed-1-8-251228",
|
||||
@@ -92,8 +104,10 @@ VOLC_MODEL_MAP = {
|
||||
"deepseek-v3-2": "deepseek-v3-2-251201",
|
||||
"deepseek-r1": "deepseek-r1-250528",
|
||||
"kimi-k2": "kimi-k2-thinking-251104",
|
||||
"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)
|
||||
|
||||
Reference in New Issue
Block a user