mirror of
https://github.com/labring/FastGPT.git
synced 2026-02-27 01:02:22 +08:00
268 lines
7.7 KiB
Bash
268 lines
7.7 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# ========== Radio 选择器 ==========
|
|
# 使用方向键 ↑↓ 选择，回车确认
|
|
radio_select() {
|
|
local title="$1"
|
|
shift
|
|
local options=("$@")
|
|
local selected=0
|
|
local count=${#options[@]}
|
|
|
|
echo ""
|
|
echo "$title"
|
|
|
|
# 绘制选项
|
|
for i in "${!options[@]}"; do
|
|
if [ $i -eq $selected ]; then
|
|
printf " \033[36m●\033[0m %s\n" "${options[$i]}"
|
|
else
|
|
printf " ○ %s\n" "${options[$i]}"
|
|
fi
|
|
done
|
|
|
|
# 隐藏光标
|
|
tput civis 2>/dev/null
|
|
|
|
while true; do
|
|
read -rsn1 key
|
|
case "$key" in
|
|
$'\x1b')
|
|
read -rsn2 arrow
|
|
case "$arrow" in
|
|
'[A') ((selected > 0)) && ((selected--)) ;;
|
|
'[B') ((selected < count - 1)) && ((selected++)) ;;
|
|
esac
|
|
;;
|
|
'') break ;;
|
|
esac
|
|
|
|
# 光标上移并重绘
|
|
printf "\033[%dA" "$count"
|
|
for i in "${!options[@]}"; do
|
|
printf "\033[K"
|
|
if [ $i -eq $selected ]; then
|
|
printf " \033[36m●\033[0m %s\n" "${options[$i]}"
|
|
else
|
|
printf " ○ %s\n" "${options[$i]}"
|
|
fi
|
|
done
|
|
done
|
|
|
|
# 恢复光标
|
|
tput cnorm 2>/dev/null
|
|
RADIO_RESULT=$selected
|
|
}
|
|
|
|
# 确保退出时恢复光标
|
|
trap 'tput cnorm 2>/dev/null; exit' INT TERM
|
|
|
|
# ========== 1. 选择镜像源 ==========
|
|
radio_select "请选择镜像源 (↑↓ 选择, 回车确认):" "阿里云 (中国大陆)" "GitHub (全球)"
|
|
case $RADIO_RESULT in
|
|
1) REGION="global" ;;
|
|
*) REGION="cn" ;;
|
|
esac
|
|
|
|
# ========== 2. 选择向量数据库 ==========
|
|
radio_select "请选择向量数据库 (↑↓ 选择, 回车确认):" "PostgreSQL + pgvector" "Milvus" "Zilliz" "OceanBase" "SeekDB"
|
|
case $RADIO_RESULT in
|
|
1) VECTOR="milvus" ;;
|
|
2) VECTOR="zilliz" ;;
|
|
3) VECTOR="oceanbase" ;;
|
|
4) VECTOR="seekdb" ;;
|
|
*) VECTOR="pg" ;;
|
|
esac
|
|
|
|
# ========== 3. 检测可用 IP ==========
|
|
IP_LIST=()
|
|
if command -v ip &>/dev/null; then
|
|
for ip in $(ip -4 addr show 2>/dev/null | awk '/inet / {split($2,a,"/"); print a[1]}' | grep -v '127.0.0.1'); do
|
|
IP_LIST+=("$ip")
|
|
done
|
|
elif command -v ifconfig &>/dev/null; then
|
|
for ip in $(ifconfig 2>/dev/null | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1'); do
|
|
IP_LIST+=("$ip")
|
|
done
|
|
fi
|
|
|
|
# 地址选择函数
|
|
# 参数: $1=标题, $2=端口
|
|
# 设置 SELECTED_ADDR 和 SELECTED_CUSTOM (true=手动输入, false=列表选择)
|
|
select_address() {
|
|
local title="$1"
|
|
local port="$2"
|
|
SELECTED_CUSTOM=false
|
|
if [ ${#IP_LIST[@]} -gt 0 ]; then
|
|
# 构建带完整地址的选项列表
|
|
local opts=()
|
|
for ip in "${IP_LIST[@]}"; do
|
|
opts+=("http://$ip:$port")
|
|
done
|
|
opts+=("其他 (手动输入)")
|
|
|
|
radio_select "$title" "${opts[@]}"
|
|
if [ $RADIO_RESULT -eq ${#IP_LIST[@]} ]; then
|
|
echo ""
|
|
read -p "请输入完整地址 (如 http://domain:port): " SELECTED_ADDR
|
|
SELECTED_CUSTOM=true
|
|
else
|
|
SELECTED_ADDR="${IP_LIST[$RADIO_RESULT]}"
|
|
fi
|
|
else
|
|
echo ""
|
|
echo "未检测到可用 IP 地址"
|
|
read -p "请输入完整地址 (如 http://domain:port): " SELECTED_ADDR
|
|
SELECTED_CUSTOM=true
|
|
fi
|
|
}
|
|
|
|
# ========== 4. 选择 S3 访问地址 (端口 9000) ==========
|
|
select_address "请选择 S3 访问地址 - 客户端和容器均需可访问 (↑↓ 选择, 回车确认):" 9000
|
|
S3_ADDR="$SELECTED_ADDR"
|
|
S3_CUSTOM=$SELECTED_CUSTOM
|
|
|
|
# ========== 5. 选择 SSE MCP 访问地址 (端口 3005) ==========
|
|
select_address "请选择 SSE MCP 访问地址 - 客户端和容器均需可访问 (↑↓ 选择, 回车确认):" 3005
|
|
MCP_ADDR="$SELECTED_ADDR"
|
|
MCP_CUSTOM=$SELECTED_CUSTOM
|
|
|
|
# ========== 确认配置 ==========
|
|
REGION_LABEL="阿里云 (中国大陆)"
|
|
if [ "$REGION" == "global" ]; then
|
|
REGION_LABEL="GitHub (全球)"
|
|
fi
|
|
|
|
# 构建显示地址
|
|
if [ -n "$S3_ADDR" ]; then
|
|
if $S3_CUSTOM; then
|
|
S3_DISPLAY="$S3_ADDR"
|
|
else
|
|
S3_DISPLAY="http://$S3_ADDR:9000"
|
|
fi
|
|
else
|
|
S3_DISPLAY="未设置"
|
|
fi
|
|
|
|
if [ -n "$MCP_ADDR" ]; then
|
|
if $MCP_CUSTOM; then
|
|
MCP_DISPLAY="$MCP_ADDR"
|
|
else
|
|
MCP_DISPLAY="http://$MCP_ADDR:3005"
|
|
fi
|
|
else
|
|
MCP_DISPLAY="未设置"
|
|
fi
|
|
|
|
echo ""
|
|
echo "=============================="
|
|
echo " 镜像源: $REGION_LABEL"
|
|
echo " 向量数据库: $VECTOR"
|
|
echo " S3 地址: $S3_DISPLAY"
|
|
echo " MCP 地址: $MCP_DISPLAY"
|
|
echo "=============================="
|
|
echo ""
|
|
read -p "确认以上配置? (y/n) [y]: " confirm
|
|
if [ "$confirm" == "n" ]; then
|
|
echo "已取消"
|
|
exit 1
|
|
fi
|
|
|
|
# ========== 下载文件 ==========
|
|
echo ""
|
|
echo "正在下载配置文件..."
|
|
|
|
# 构建下载链接（处理 global 下 zilliz 文件名差异）
|
|
VECTOR_FILE="$VECTOR"
|
|
if [ "$REGION" == "global" ] && [ "$VECTOR" == "zilliz" ]; then
|
|
VECTOR_FILE="ziliiz"
|
|
fi
|
|
|
|
if [ "$REGION" == "cn" ]; then
|
|
BASE_URL="https://doc.fastgpt.cn/deploy"
|
|
YML_URL="${BASE_URL}/docker/cn/docker-compose.${VECTOR_FILE}.yml"
|
|
else
|
|
BASE_URL="https://doc.fastgpt.io/deploy"
|
|
YML_URL="${BASE_URL}/docker/global/docker-compose.${VECTOR_FILE}.yml"
|
|
fi
|
|
|
|
CONFIG_URL="${BASE_URL}/config/config.json"
|
|
|
|
# 下载 docker-compose YAML
|
|
curl -fsSL -O "$YML_URL"
|
|
if [ $? -ne 0 ]; then
|
|
echo "错误: 下载 YAML 文件失败: $YML_URL"
|
|
exit 1
|
|
fi
|
|
mv "docker-compose.${VECTOR_FILE}.yml" docker-compose.yml
|
|
echo "已下载 docker-compose.yml"
|
|
|
|
# 下载 config.json
|
|
curl -fsSL -O "$CONFIG_URL"
|
|
if [ $? -ne 0 ]; then
|
|
echo "错误: 下载 config.json 失败: $CONFIG_URL"
|
|
exit 1
|
|
fi
|
|
echo "已下载 config.json"
|
|
|
|
# ========== 替换 S3 访问地址 ==========
|
|
if [ -n "$S3_ADDR" ]; then
|
|
if $S3_CUSTOM; then
|
|
# 自定义输入：提取主机部分替换 docker-compose 中的 IP
|
|
S3_HOST="${S3_ADDR#http://}"
|
|
S3_HOST="${S3_HOST#https://}"
|
|
S3_HOST="${S3_HOST%%:*}"
|
|
S3_HOST="${S3_HOST%%/*}"
|
|
else
|
|
S3_HOST="$S3_ADDR"
|
|
fi
|
|
|
|
if [[ "$OSTYPE" == "darwin"* ]]; then
|
|
sed -i '' "s/192\.168\.0\.2/$S3_HOST/g" docker-compose.yml
|
|
else
|
|
sed -i "s/192\.168\.0\.2/$S3_HOST/g" docker-compose.yml
|
|
fi
|
|
|
|
if [ $? -eq 0 ]; then
|
|
echo "已更新 S3 访问地址为: $S3_DISPLAY"
|
|
else
|
|
echo "警告: 替换 S3 地址失败，请手动编辑 docker-compose.yml 中的 192.168.0.2"
|
|
fi
|
|
else
|
|
echo "警告: 未设置 S3 地址，请手动编辑 docker-compose.yml 中的 192.168.0.2"
|
|
fi
|
|
|
|
# ========== 替换 MCP 访问地址 ==========
|
|
if [ -n "$MCP_ADDR" ]; then
|
|
if $MCP_CUSTOM; then
|
|
MCP_ENDPOINT="$MCP_ADDR"
|
|
else
|
|
MCP_ENDPOINT="http://$MCP_ADDR:3005"
|
|
fi
|
|
|
|
if [[ "$OSTYPE" == "darwin"* ]]; then
|
|
sed -i '' "s|\"mcpServerProxyEndpoint\": \"\"|\"mcpServerProxyEndpoint\": \"$MCP_ENDPOINT\"|g" config.json
|
|
else
|
|
sed -i "s|\"mcpServerProxyEndpoint\": \"\"|\"mcpServerProxyEndpoint\": \"$MCP_ENDPOINT\"|g" config.json
|
|
fi
|
|
|
|
if [ $? -eq 0 ]; then
|
|
echo "已更新 MCP 访问地址为: $MCP_ENDPOINT"
|
|
else
|
|
echo "警告: 替换 MCP 地址失败，请手动编辑 config.json 中的 mcpServerProxyEndpoint"
|
|
fi
|
|
else
|
|
echo "警告: 未设置 MCP 地址，请手动编辑 config.json 中的 mcpServerProxyEndpoint"
|
|
fi
|
|
|
|
# ========== 完成 ==========
|
|
echo ""
|
|
echo "配置下载成功! 后续操作:"
|
|
echo " 1. 启动服务: docker compose up -d"
|
|
echo " 2. 开放端口: 3000, 9000, 3005"
|
|
echo " 3. 访问服务: http://localhost:3000"
|
|
echo " 4. 登录服务: 默认账号为 'root', 密码为: '1234'"
|
|
echo " 5. 配置模型: 在 '账号-模型提供商' 页面，进行模型配置"
|
|
echo ""
|
|
echo "详细文档: https://doc.fastgpt.cn/docs/introduction/development/docker"
|