renminwansui1976 commited on
Commit
0d141d4
·
unverified ·
1 Parent(s): 998e21b

更新 start.sh

Browse files
Files changed (1) hide show
  1. start.sh +8 -5
start.sh CHANGED
@@ -69,12 +69,15 @@ LITELLM_CONFIG=/tmp/litellm_config.yaml
69
  echo "model_list:"
70
  echo " - model_name: \"default\""
71
  echo " litellm_params:"
72
- # Force openai/ prefix so LiteLLM uses plain OpenAI-compat format.
73
- # LITELLM_API_BASE must be set to https://integrate.api.nvidia.com/v1
74
- # LITELLM_MODEL should be moonshotai/kimi-k2-instruct-0905 (no nvidia_nim/ prefix)
75
- echo " model: \"openai/${LITELLM_MODEL}\""
 
76
  echo " api_key: \"${LITELLM_API_KEY}\""
77
- echo " api_base: \"${LITELLM_API_BASE:-https://integrate.api.nvidia.com/v1}\""
 
 
78
  echo ""
79
  echo "litellm_settings:"
80
  echo " drop_params: true"
 
69
  echo "model_list:"
70
  echo " - model_name: \"default\""
71
  echo " litellm_params:"
72
+ # Use LITELLM_MODEL directly must include provider prefix, e.g.:
73
+ # nvidia_nim/moonshotai/kimi-k2-instruct-0905 (NVIDIA NIM)
74
+ # openai/gpt-4o (OpenAI)
75
+ # anthropic/claude-sonnet-4-20250514 (Anthropic)
76
+ echo " model: \"${LITELLM_MODEL}\""
77
  echo " api_key: \"${LITELLM_API_KEY}\""
78
+ if [ -n "${LITELLM_API_BASE:-}" ]; then
79
+ echo " api_base: \"${LITELLM_API_BASE}\""
80
+ fi
81
  echo ""
82
  echo "litellm_settings:"
83
  echo " drop_params: true"