Update entrypoint.sh

entrypoint.sh

@@ -1,22 +1,41 @@
-#!/bin/bash
+#!/usr/bin/env bash
+set -euo pipefail
 
-# Make sure `/data/db` directory exists even with persistent storage
+# 1) 本地 Mongo 启动（容器内）
 mkdir -p /data/db
-# If app crashed, mongo didn't stop gracefully. Remove all the old *.lock files
-find /data/db -name "*.lock" -type f -exec rm -f {} \; 
-# Start the local Mongo database
-mongod &
+# 清理可能遗留的锁文件，避免上次异常退出导致无法启动
+find /data/db -name "*.lock" -type f -exec rm -f {} \;
 
-# Start the text-generation-inference process
-text-generation-launcher --model-id ${MODEL_NAME} --num-shard 1 --port 8080 --trust-remote-code &
+# 建议绑定到回环即可；写日志到临时目录；后台运行
+mongod --dbpath /data/db --bind_ip 127.0.0.1 --logpath /tmp/mongod.log --fork
 
-# Wait for text-generation-inference to start
-curl --retry 60 --retry-delay 10 --retry-connrefused http://127.0.0.1:8080/health
+# 等待 Mongo 就绪（简单轮询 60 次，每次 1s）
+for i in $(seq 1 60); do
+  if mongosh --quiet --host 127.0.0.1 --eval 'db.runCommand({ping:1})' >/dev/null 2>&1; then
+    echo "MongoDB is up."
+    break
+  fi
+  echo "Waiting for MongoDB to be ready... ($i/60)"
+  sleep 1
+done
 
-# Start the chat-ui process
-dotenv -e /app/.env -c -- node /app/build/index.js -- --host 0.0.0.0 --port 3000
-# Wait for any process to exit
-wait -n
+# 如果外部没有注入 Atlas 的 MONGODB_URL，就默认用本地
+export MONGODB_URL="${MONGODB_URL:-mongodb://127.0.0.1:27017/chatui}"
 
-# Exit with status of process that exited first
-exit $?
+# 2) 校验外部 LLM 所需环境
+: "${OPENAI_BASE_URL:?OPENAI_BASE_URL missing}"
+: "${OPENAI_API_KEY:?OPENAI_API_KEY missing}"
+: "${MODEL_NAME:?MODEL_NAME missing}"
+
+echo "Using MODEL_NAME=$MODEL_NAME"
+echo "Using OPENAI_BASE_URL=$OPENAI_BASE_URL"
+echo "MongoDB URL set to: ${MONGODB_URL}"
+
+# 3) 启动 ChatUI
+#   - 若容器里有 dotenv-cli，则用它加载 /app/.env（若存在）
+#   - 否则直接起 node
+if command -v dotenv >/dev/null 2>&1 && [ -f /app/.env ]; then
+  exec dotenv -e /app/.env -c -- node /app/build/index.js -- --host 0.0.0.0 --port 3000
+else
+  exec node /app/build/index.js --host 0.0.0.0 --port 3000
+fi