Skip to content

Commit 4953f5f

Browse files
committed
优化启动脚本,增加字符编码和颜色设置,改进Docker环境检查逻辑,调整模型拉取命令,增强用户提示信息,提升用户体验和系统可用性。
1 parent d537109 commit 4953f5f

File tree

1 file changed

+81
-43
lines changed

1 file changed

+81
-43
lines changed

start-ollama.sh

Lines changed: 81 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -1,84 +1,122 @@
11
#!/bin/bash
22

3-
# 设置颜色输出
3+
# 设置字符编码和颜色
4+
export LANG=zh_CN.UTF-8
5+
export LC_ALL=zh_CN.UTF-8
6+
7+
# 颜色定义
48
RED='\033[0;31m'
59
GREEN='\033[0;32m'
610
YELLOW='\033[1;33m'
711
BLUE='\033[0;34m'
12+
PURPLE='\033[0;35m'
13+
CYAN='\033[0;36m'
14+
WHITE='\033[1;37m'
815
NC='\033[0m' # No Color
916

10-
echo "========================================"
11-
echo " AI Prompt 优化平台 - Ollama 启动器 "
12-
echo "========================================"
13-
echo
17+
# 清屏
18+
clear
19+
20+
echo -e "${CYAN}========================================"
21+
echo -e " AI Prompt 优化平台 - Ollama 启动器 "
22+
echo -e "========================================${NC}"
23+
echo ""
1424

15-
# 检查Docker环境
1625
echo -e "${BLUE}🔍 检查Docker环境...${NC}"
26+
27+
# 检查Docker是否安装
1728
if ! command -v docker &> /dev/null; then
1829
echo -e "${RED}❌ 错误:未检测到Docker,请先安装Docker${NC}"
19-
echo "安装指南:https://docs.docker.com/get-docker/"
30+
echo -e "${YELLOW}安装指南:https://docs.docker.com/get-docker/${NC}"
31+
exit 1
32+
fi
33+
34+
# 检查Docker是否运行
35+
if ! docker info &> /dev/null; then
36+
echo -e "${RED}❌ 错误:Docker未运行,请启动Docker服务${NC}"
37+
echo -e "${YELLOW}提示:请确保Docker Desktop已启动${NC}"
2038
exit 1
2139
fi
2240

23-
if ! command -v docker-compose &> /dev/null && ! docker compose version &> /dev/null; then
41+
# 检查Docker Compose是否安装
42+
if ! command -v docker-compose &> /dev/null; then
2443
echo -e "${RED}❌ 错误:未检测到Docker Compose${NC}"
2544
exit 1
2645
fi
2746

2847
echo -e "${GREEN}✅ Docker环境检查通过${NC}"
2948

30-
echo
49+
echo ""
3150
echo -e "${BLUE}🚀 启动ollama服务...${NC}"
32-
if ! docker-compose -f docker-compose-ollama.yaml up -d; then
51+
docker-compose -f docker-compose-ollama.yaml up -d
52+
53+
if [ $? -ne 0 ]; then
3354
echo -e "${RED}❌ 启动失败,正在查看错误日志...${NC}"
3455
docker-compose -f docker-compose-ollama.yaml logs
56+
read -n 1 -s -r -p "按任意键继续..."
3557
exit 1
3658
fi
3759

3860
echo -e "${GREEN}✅ 服务启动成功${NC}"
3961

40-
echo
62+
echo ""
4163
echo -e "${YELLOW}⏳ 等待ollama服务完全启动...${NC}"
42-
sleep 15
64+
sleep 10
4365

44-
echo
66+
echo ""
4567
echo -e "${BLUE}📦 拉取qwen3模型...${NC}"
4668
echo -e "${YELLOW}(这可能需要几分钟时间,模型大小约5GB)${NC}"
47-
if ! docker exec ollama ollama pull qwen3; then
69+
docker exec ollama ollama pull qwen3:1.7b
70+
71+
if [ $? -ne 0 ]; then
4872
echo -e "${YELLOW}⚠️ qwen3模型拉取失败,尝试拉取轻量级模型...${NC}"
4973
docker exec ollama ollama pull qwen2.5:3b
5074
fi
5175

52-
echo
76+
echo ""
5377
echo -e "${GREEN}✅ 检查已安装的模型...${NC}"
5478
docker exec ollama ollama list
5579

56-
echo
80+
echo ""
81+
echo -e "${BLUE}🔄 重启console-service...${NC}"
82+
docker-compose restart console-service
83+
84+
echo ""
5785
echo -e "${GREEN}🎉 启动完成!${NC}"
58-
echo
59-
echo -e "${BLUE}📋 访问信息:${NC}"
60-
echo " 前端界面:http://localhost:10426"
61-
echo " API文档:http://localhost:10426/scalar/v1"
62-
echo " Ollama API:http://localhost:11434"
63-
echo
64-
echo -e "${BLUE}💡 使用提示:${NC}"
65-
echo " - 默认用户名:admin"
66-
echo " - 默认密码:admin123"
67-
echo " - 首次登录后请修改密码"
68-
echo
69-
echo -e "${BLUE}🔧 管理命令:${NC}"
70-
echo " 查看日志:docker-compose -f docker-compose-ollama.yaml logs -f"
71-
echo " 停止服务:docker-compose -f docker-compose-ollama.yaml down"
72-
echo " 重启服务:docker-compose -f docker-compose-ollama.yaml restart"
73-
echo
74-
75-
# 尝试打开浏览器(如果是桌面环境)
76-
if command -v xdg-open &> /dev/null; then
77-
echo -e "${YELLOW}正在打开默认浏览器...${NC}"
78-
xdg-open http://localhost:10426 2>/dev/null
79-
elif command -v open &> /dev/null; then
80-
echo -e "${YELLOW}正在打开默认浏览器...${NC}"
81-
open http://localhost:10426 2>/dev/null
82-
else
83-
echo -e "${YELLOW}请手动打开浏览器访问:http://localhost:10426${NC}"
84-
fi
86+
echo ""
87+
echo -e "${CYAN}📋 访问信息${NC}"
88+
echo -e "${WHITE} 前端界面:http://localhost:10426${NC}"
89+
echo -e "${WHITE} API文档:http://localhost:10426/scalar/v1${NC}"
90+
echo -e "${WHITE} Ollama API:http://localhost:11434${NC}"
91+
echo ""
92+
echo -e "${PURPLE}💡 使用提示:${NC}"
93+
echo -e "${WHITE} - 默认用户名:admin${NC}"
94+
echo -e "${WHITE} - 默认密码:admin123${NC}"
95+
echo -e "${WHITE} - 首次登录后请修改密码${NC}"
96+
echo ""
97+
echo -e "${CYAN}🔧 管理命令:${NC}"
98+
echo -e "${WHITE} 查看日志:docker-compose -f docker-compose-ollama.yaml logs -f${NC}"
99+
echo -e "${WHITE} 停止服务:docker-compose -f docker-compose-ollama.yaml down${NC}"
100+
echo -e "${WHITE} 重启服务:docker-compose -f docker-compose-ollama.yaml restart${NC}"
101+
echo ""
102+
echo -e "${YELLOW}按任意键打开浏览器访问系统...${NC}"
103+
read -n 1 -s -r
104+
105+
# 检测操作系统并打开浏览器
106+
case "$(uname -s)" in
107+
Darwin*)
108+
open http://localhost:10426
109+
;;
110+
Linux*)
111+
if command -v xdg-open &> /dev/null; then
112+
xdg-open http://localhost:10426
113+
elif command -v gnome-open &> /dev/null; then
114+
gnome-open http://localhost:10426
115+
else
116+
echo -e "${YELLOW}请手动打开浏览器访问:http://localhost:10426${NC}"
117+
fi
118+
;;
119+
*)
120+
echo -e "${YELLOW}请手动打开浏览器访问:http://localhost:10426${NC}"
121+
;;
122+
esac

0 commit comments

Comments
 (0)