1       目标项目

部分较老的项目比如：2025年5月份之前的安装包，需要做启动脚本的升级。 2025年 5月份之后的安装包已经自动添加这部分内容。 

2       升级内容

2.1  docker-compose.yml

2.1.1          完整脚本（不能直接复制-需要修改映射地址）

version: "3"
services:
  nginx:
    image: "nginx:alpine"
    ports:
      - "9070:9060"
    links:
      - "aiweb:aiweb"
    volumes:
      - /home/smartbi-nla-agent/nginx_conf:/etc/nginx/conf.d/
    depends_on:
      - aiweb
    restart: always

  propertycenter:
    image: "aienv4j:v9.1"
    volumes:
      - /home/smartbi-nla-agent/logs:/ailogs
      - /home/smartbi-nla-agent/services:/jars
      - /home/smartbi-nla-agent/scripts/propertycenter:/scripts
    links:
      - "redis:airedis"
    entrypoint:
      - /bin/sh
      - /scripts/docker-entrypoint.sh
    depends_on:
      - redis
    restart: always

  aiweb: 
    image: "aienv4j:v9.1"
    ports: 
      - "9060:9082" 
    volumes:
      - /home/smartbi-nla-agent/logs:/ailogs
      - /home/smartbi-nla-agent/services:/jars
      - /home/smartbi-nla-agent/scripts/aiweb:/scripts
    links: 
      - "redis:airedis"
    entrypoint:
      - /bin/sh 
      - /scripts/docker-entrypoint.sh
    depends_on: 
      - redis
    restart: always

  admin: 
    image: "aienv4j:v9.1"
    volumes:
      - /home/smartbi-nla-agent/logs:/ailogs
      - /home/smartbi-nla-agent/services:/jars
      - /home/smartbi-nla-agent/scripts/admin:/scripts
    links: 
      - "redis:airedis"
      - "propertycenter:propertycenter"
      - "mongo:aimongo"
    entrypoint:
      - /bin/sh 
      - /scripts/docker-entrypoint.sh
    depends_on: 
      - redis
      - propertycenter
      - mongo
    restart: always       

  modulemanager: 
    image: "aienv4j:v9.1"
    ports:
      - "9081:9081"
    volumes:
      - /home/smartbi-nla-agent/logs:/ailogs
      - /home/smartbi-nla-agent/services:/jars
      - /home/smartbi-nla-agent/scripts/modulemanager:/scripts
    links: 
      - "redis:airedis"
      - "propertycenter:propertycenter"
    entrypoint:
      - /bin/sh 
      - /scripts/docker-entrypoint.sh
    depends_on: 
      - redis
      - propertycenter
    restart: always    

  aibus: 
    image: "aienv4j:v9.1"
    volumes:
      - /home/smartbi-nla-agent/logs:/ailogs
      - /home/smartbi-nla-agent/services:/jars
      - /home/smartbi-nla-agent/scripts/aibus:/scripts
    links: 
      - "redis:airedis"
      - "propertycenter:propertycenter"
    entrypoint:
      - /bin/sh 
      - /scripts/docker-entrypoint.sh
    depends_on: 
      - redis
      - propertycenter
    restart: always

  smartbiproxy: 
    image: "aienv4j:v9.1"
    volumes:
      - /home/smartbi-nla-agent/logs:/ailogs
      - /home/smartbi-nla-agent/services:/jars
      - /home/smartbi-nla-agent/scripts/smartbiproxy:/scripts
    links: 
      - "redis:airedis"
      - "propertycenter:propertycenter"
    entrypoint:
      - /bin/sh
      - /scripts/docker-entrypoint.sh
    depends_on: 
      - redis
      - propertycenter
    restart: always

  nl2sql_llm:
    image: "aienv4py-llm:v6"
    volumes:
      - /home/smartbi-nla-agent/logs:/logs
      - /home/smartbi-nla-agent/scripts/nl2sql_llm:/scripts
      - /home/smartbi-nla-agent/services:/services
      - /home/smartbi-nla-agent/data/nl2sql_data:/data
      - /home/smartbi-nla-agent/backup/backup_data:/backup_data
    links: 
      - "redis:airedis"
      - "admin:main"
    depends_on: 
      - redis
    entrypoint:
      - /bin/sh 
      - /scripts/docker-entrypoint.sh
    restart: always

  jupyter:
    image: "aienv4py-agent:v1.1"
    volumes:
      - /home/smartbi-nla-agent/logs:/logs
      - /home/smartbi-nla-agent/scripts/agent:/scripts
      - /home/smartbi-nla-agent/services:/services
    links:
      - "redis:airedis"
      - "admin:main"
    depends_on:
      - redis
    entrypoint:
      - /bin/sh
      - /scripts/docker-entrypoint-runtime.sh
    restart: always
    deploy:
      resources:
        limits:
          memory: 5G

  agent:
    image: "aienv4py-agent:v1.1"
    volumes:
      - /home/smartbi-nla-agent/logs:/logs
      - /home/smartbi-nla-agent/scripts/agent:/scripts
      - /home/smartbi-nla-agent/services:/services
      - /home/smartbi-nla-agent/data/nl2sql_data:/data
      - /var/run/docker.sock:/var/run/docker.sock
    healthcheck:
      test: ["CMD-SHELL", "curl -f -X POST http://localhost:8085/api/health -H 'Content-Type: application/json' -d '{}' || exit 1"]
      interval: 90s
      timeout: 10s
      retries: 3
      start_period: 300s
    links: 
      - "redis:airedis"
      - "admin:main"
      - "jupyter:jupyter"
    depends_on: 
      - redis
      - jupyter
    entrypoint:
      - /bin/sh
      - -c
      - |
        set -e
        /scripts/docker-entrypoint.sh &
        /scripts/healthcheck.sh &
        wait
    restart: always

  mongo:
    image: "mongo:4"
    volumes:
      - /home/smartbi-nla-agent/data/mongo/data:/data/db
    ports:
      - "27017:27017"
    environment:
      MONGO_INITDB_ROOT_USERNAME: root
      MONGO_INITDB_ROOT_PASSWORD: smartbi1#nla
    restart: always
  
  redis:
    image: "airedis:v5"
    ports:
      - "6688:3306"
    volumes:
      - /home/smartbi-nla-agent/data/config:/var/lib/mysql
      - /home/smartbi-nla-agent/backup/config:/backup
    restart: always

添加内容

jupyter服务内存限制


jupyter:
    image: "aienv4py-agent:v1.1"
    volumes:
      - /home/smartbi-nla-agent/logs:/logs
      - /home/smartbi-nla-agent/scripts/agent:/scripts
      - /home/smartbi-nla-agent/services:/services
    links:
      - "redis:airedis"
      - "admin:main"
    depends_on:
      - redis
    entrypoint:
      - /bin/sh
      - /scripts/docker-entrypoint-runtime.sh
    restart: always
    deploy:
      resources:
        limits:
          memory: 5G


 deploy:
      resources:
        limits:
          memory: 5G

agent服务健康检查

 agent:
    image: "aienv4py-agent:v1.1"
    volumes:
      - /home/smartbi-nla-agent/logs:/logs
      - /home/smartbi-nla-agent/scripts/agent:/scripts
      - /home/smartbi-nla-agent/services:/services
      - /home/smartbi-nla-agent/data/nl2sql_data:/data
      - /var/run/docker.sock:/var/run/docker.sock
    healthcheck:
      test: ["CMD-SHELL", "curl -f -X POST http://localhost:8085/api/health -H 'Content-Type: application/json' -d '{}' || exit 1"]
      interval: 90s
      timeout: 10s
      retries: 3
      start_period: 300s
    links: 
      - "redis:airedis"
      - "admin:main"
      - "jupyter:jupyter"
    depends_on: 
      - redis
      - jupyter
    entrypoint:
      - /bin/sh
      - -c
      - |
        set -e
        /scripts/docker-entrypoint.sh &
        /scripts/healthcheck.sh &
        wait
    restart: always


healthcheck:
      test: ["CMD-SHELL", "curl -f -X POST http://localhost:8085/api/health -H 'Content-Type: application/json' -d '{}' || exit 1"]
      interval: 90s
      timeout: 10s
      retries: 3
      start_period: 300s

修改内容

entrypoint:
      - /bin/sh
      - -c
      - |
        set -e
        /scripts/docker-entrypoint.sh &
        /scripts/healthcheck.sh &
        wait

run.sh

sh ./debug_run.sh
docker-compose up -d propertycenter
echo -e "\033[31mWaiting for propertycenter to be up... \033[0m"
sleep 10
docker-compose up -d --scale jupyter=2 --scale agent=2

根据实际情况，修改启动的jupyter个数和agent个数。

健康检查脚本

在scripts/agent目录下添加文件：healthcheck.sh

vi scripts/agent/healthcheck.sh

#!/bin/sh

MAX_FAILS=2
INTERVAL=90
FAIL_COUNT=0

# 先休眠5分钟，等待服务器完成启动
sleep 300

while true; do
    if ! curl -f -X POST http://localhost:8085/api/health \
        -H 'Content-Type: application/json' -d '{}' > /dev/null 2>&1; then
        FAIL_COUNT=$((FAIL_COUNT+1))
        echo "[$(date)] Health check failed, count: $FAIL_COUNT" >> /var/log/healthcheck.log
        if [ "$FAIL_COUNT" -ge "$MAX_FAILS" ]; then
            echo "[$(date)] Max failures reached, exiting container..." >> /var/log/healthcheck.log
            kill -s TERM 1  # 发送信号给 init 进程，触发重启
            exit 1
        fi
    else
        FAIL_COUNT=0
    fi
    sleep $INTERVAL
done