Simford.Dong commited on
Commit
5c7e31e
·
1 Parent(s): ae959f5

Initial Commit

Browse files
Files changed (1) hide show
  1. entrypoint.sh +22 -1
entrypoint.sh CHANGED
@@ -78,7 +78,7 @@ echo "🤖 配置模型系统..."
78
  # ---- 供应商定义 ----
79
  declare -A PROVIDER_MODELS=(
80
  ["xai"]="grok-4.3"
81
- ["nvidia"]="moonshotai/kimi-k2-thinking"
82
  ["siliconflow"]="deepseek-ai/DeepSeek-V4-Flash"
83
  ["openai"]="gpt-4o"
84
  ["anthropic"]="claude-3-5-sonnet-20241022"
@@ -88,6 +88,11 @@ declare -A PROVIDER_MODELS=(
88
  ["longcat"]="LongCat-Flash-Thinking-2601"
89
  )
90
 
 
 
 
 
 
91
  declare -A PROVIDER_API_KEYS=(
92
  ["xai"]="XAI_API_KEY"
93
  ["nvidia"]="NVIDIA_API_KEY"
@@ -178,6 +183,11 @@ MAIN_DETECTED=$(detect_main_model)
178
  IFS=':' read -r MAIN_MODE MAIN_PROVIDER MAIN_MODEL <<< "$MAIN_DETECTED"
179
  echo "🎯 Main Model: $MAIN_PROVIDER/$MAIN_MODEL (模式:$MAIN_MODE)"
180
 
 
 
 
 
 
181
  VISION_MODEL_VAL=$(detect_vision_model)
182
  echo "👁️ Vision Model: ${VISION_MODEL_VAL:-auto-detect}"
183
 
@@ -218,6 +228,7 @@ cat > "$CONFIG_FILE" << EOF
218
  # 主模型配置
219
  model:
220
  default: "$MAIN_MODEL"
 
221
  provider: "$GATEWAY_MAIN_PROVIDER"
222
  base_url: "$MAIN_BASE_URL"
223
 
@@ -894,6 +905,16 @@ if [ -f "$CONFIG_FILE" ]; then
894
  else
895
  echo " ✅ 模型配置验证通过:$GATEWAY_MAIN_PROVIDER/$MAIN_MODEL"
896
  fi
 
 
 
 
 
 
 
 
 
 
897
  fi
898
  fi
899
 
 
78
  # ---- 供应商定义 ----
79
  declare -A PROVIDER_MODELS=(
80
  ["xai"]="grok-4.3"
81
+ ["nvidia"]="nvidia/nemotron-3-super-120b-a12b"
82
  ["siliconflow"]="deepseek-ai/DeepSeek-V4-Flash"
83
  ["openai"]="gpt-4o"
84
  ["anthropic"]="claude-3-5-sonnet-20241022"
 
88
  ["longcat"]="LongCat-Flash-Thinking-2601"
89
  )
90
 
91
+ # ---- 供应商 Fallback 模型 ----
92
+ declare -A PROVIDER_FALLBACK_MODELS=(
93
+ ["nvidia"]="nvidia/nemotron-3-nano-30b-a3b"
94
+ )
95
+
96
  declare -A PROVIDER_API_KEYS=(
97
  ["xai"]="XAI_API_KEY"
98
  ["nvidia"]="NVIDIA_API_KEY"
 
183
  IFS=':' read -r MAIN_MODE MAIN_PROVIDER MAIN_MODEL <<< "$MAIN_DETECTED"
184
  echo "🎯 Main Model: $MAIN_PROVIDER/$MAIN_MODEL (模式:$MAIN_MODE)"
185
 
186
+ # ---- Fallback 模型 ----
187
+ FALLBACK_MODEL="${PROVIDER_FALLBACK_MODELS[$MAIN_PROVIDER]:-}"
188
+ if [ -n "$FALLBACK_MODEL" ]; then
189
+ echo "🔄 Fallback Model: $FALLBACK_MODEL (when busy)"
190
+ fi
191
  VISION_MODEL_VAL=$(detect_vision_model)
192
  echo "👁️ Vision Model: ${VISION_MODEL_VAL:-auto-detect}"
193
 
 
228
  # 主模型配置
229
  model:
230
  default: "$MAIN_MODEL"
231
+ fallback: "${FALLBACK_MODEL:-}"
232
  provider: "$GATEWAY_MAIN_PROVIDER"
233
  base_url: "$MAIN_BASE_URL"
234
 
 
905
  else
906
  echo " ✅ 模型配置验证通过:$GATEWAY_MAIN_PROVIDER/$MAIN_MODEL"
907
  fi
908
+ # 验证 fallback 模型
909
+ if [ -n "$FALLBACK_MODEL" ]; then
910
+ ACTUAL_FALLBACK=$(yq '.model.fallback' "$CONFIG_FILE" 2>/dev/null)
911
+ if [ "$ACTUAL_FALLBACK" != "$FALLBACK_MODEL" ]; then
912
+ yq -i ".model.fallback = \"$FALLBACK_MODEL\"" "$CONFIG_FILE"
913
+ echo " 🔄 Fallback 模型已写入:$FALLBACK_MODEL"
914
+ else
915
+ echo " 🔄 Fallback 模型验证通过:$FALLBACK_MODEL"
916
+ fi
917
+ fi
918
  fi
919
  fi
920