Simford.Dong commited on
Commit ·
5c7e31e
1
Parent(s): ae959f5
Initial Commit
Browse files- 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"]="
|
| 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 |
|