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 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)
|
||||||
|
|||||||
Reference in New Issue
Block a user