Agent-System/docker-compose.test.yml

# =============================================================================
# 测试环境编排 (Test Branch Environment)
# 特性：
# 1. 独占系统业务库 (5434:5432) 与专属数据卷
# 2. 独占对话记录、工作区及日志目录 (./data/test)
# 3. 共享连接知识库服务 (shuimu-kb-shared:5432)
# =============================================================================
name: agent-test

services:
  system-db:
    image: pgvector/pgvector:pg16
    container_name: shuimu-system-pg-test
    command: ["postgres", "-c", "shared_buffers=256MB", "-c", "max_connections=150"]
    environment:
      POSTGRES_USER: ${SYSTEM_DB_USER:-app}
      POSTGRES_PASSWORD: ${SYSTEM_DB_PASSWORD:-test_secret_db_pass}
      POSTGRES_DB: ${SYSTEM_DB_NAME:-agent_system_test}
    ports:
      - "127.0.0.1:${SYSTEM_DB_PUBLISH_PORT:-5434}:5432"
    volumes:
      - test-system-pgdata:/var/lib/postgresql/data
      - ./deploy/postgres/extensions_init.sql:/docker-entrypoint-initdb.d/01-extensions.sql:ro
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U ${SYSTEM_DB_USER:-app} -d ${SYSTEM_DB_NAME:-agent_system_test}"]
      interval: 5s
      timeout: 5s
      retries: 20
    restart: unless-stopped
    networks:
      - test-net

  cursor-agent:
    image: alpine:latest
    container_name: shuimu-cursor-agent-test
    command: ["sh", "-c", "echo 'Test Agent Runner is standby' && sleep infinity"]
    environment:
      CURSOR_RUNNER_ADDR: ":18766"
      DATA_DIR: /app/data/workspaces
    ports:
      - "127.0.0.1:${CURSOR_RUNNER_PORT:-18766}:18766"
    volumes:
      - ./data/test/workspaces:/app/data/workspaces
    restart: unless-stopped
    networks:
      - test-net

  pi-agent:
    build:
      context: .
      dockerfile: deploy/pi-agent/Dockerfile
    image: shuimu-pi-agent:test
    container_name: shuimu-pi-agent-test
    environment:
      PORT: 18866
      DATA_DIR: /app/data/workspaces
      MINIMAX_ENDPOINT_URL: ${MINIMAX_ENDPOINT_URL:-https://api.minimaxi.com/v1}
      MINIMAX_API_KEY: ${MINIMAX_API_KEY:-}
      MINIMAX_MODEL_ID: ${MINIMAX_MODEL_ID:-MiniMax-M3}
      MINIMAX_API_TYPE: ${MINIMAX_API_TYPE:-openai-compatible}
      MINIMAX_PROVIDER: minimax
      TAVILY_API_KEYS: ${TAVILY_API_KEYS:-}
      RUNNER_AUTH_TOKEN: ${RUNNER_AUTH_TOKEN:-}
    ports:
      - "127.0.0.1:${PI_AGENT_RUNNER_PORT:-18866}:18866"
    volumes:
      - ./data/test/workspaces:/app/data/workspaces
      - ./data/test/agent_sessions:/root/.pi/agent/sessions
    tmpfs:
      - /tmp
    healthcheck:
      test: ["CMD-SHELL", "curl -f http://127.0.0.1:18866/health || exit 1"]
      interval: 10s
      timeout: 5s
      retries: 5
    restart: unless-stopped
    networks:
      - test-net

  app:
    build:
      context: .
      dockerfile: deploy/app/Dockerfile
    container_name: shuimu-agent-app-test
    depends_on:
      system-db:
        condition: service_healthy
    environment:
      ENV: test
      HTTP_ADDR: ":${HTTP_PORT:-8081}"
      HTTP_PORT: "${HTTP_PORT:-8081}"
      HTTP_HOST: "0.0.0.0"
      AGENT_BACKEND: ${AGENT_BACKEND:-pi-agent}
      AGENT_TIMEOUT: ${AGENT_TIMEOUT:-30m}
      MAX_CONCURRENT_AGENTS: ${MAX_CONCURRENT_AGENTS:-8}
      DATA_DIR: /app/data
      # 独占测试业务与对话库
      APP_DB_DSN: postgres://${SYSTEM_DB_USER:-app}:${SYSTEM_DB_PASSWORD:-test_secret_db_pass}@system-db:5432/${SYSTEM_DB_NAME:-agent_system_test}?sslmode=disable
      # 共享知识库
      POSTGRES_SHARED_KB_DSN: postgres://${KB_DB_USER:-regkb}:${KB_DB_PASSWORD:?KB_DB_PASSWORD is required}@shuimu-kb-shared:5432/${KB_DB_NAME:-regulations}?sslmode=disable
      CURSOR_RUNNER_URL: http://cursor-agent:18766
      PI_AGENT_RUNNER_URL: http://pi-agent:18866
      PI_AGENT_RUNNER_AUTH_TOKEN: ${RUNNER_AUTH_TOKEN:-}
      PI_AGENT_PROVIDER: ${PI_AGENT_PROVIDER:-minimax}
      PI_AGENT_MODEL: ${PI_AGENT_MODEL:-${MINIMAX_MODEL_ID:-MiniMax-M3}}
      RUNNER_AUTH_TOKEN: ${RUNNER_AUTH_TOKEN:-}
      EMBEDDING_ENDPOINT_URL: ${EMBEDDING_ENDPOINT_URL:-}
      EMBEDDING_MODEL: ${EMBEDDING_MODEL:-bge-m3}
      EMBEDDING_API_KEY: ${EMBEDDING_API_KEY:-}
      RERANKER_ENDPOINT_URL: ${RERANKER_ENDPOINT_URL:-}
      RERANKER_MODEL: ${RERANKER_MODEL:-bge-reranker-v2-m3-free}
      RERANKER_API_KEY: ${RERANKER_API_KEY:-}
      ADMIN_USERNAME: ${ADMIN_USERNAME:-admin}
      ADMIN_PASSWORD: ${ADMIN_PASSWORD:-admin-secret-password-123}
      AUTH_REGISTRATION_ENABLED: ${AUTH_REGISTRATION_ENABLED:-true}
      AUTH_REQUIRE_APPROVAL: ${AUTH_REQUIRE_APPROVAL:-false}
      CORS_ALLOWED_ORIGINS: ${CORS_ALLOWED_ORIGINS:-}
      ALLOW_IN_MEMORY: ${ALLOW_IN_MEMORY:-false}
      CACHE_ENABLED: ${CACHE_ENABLED:-true}
      INTENT_V2_SHADOW: ${INTENT_V2_SHADOW:-false}
      INTENT_V2_DECISION: ${INTENT_V2_DECISION:-false}
      INTENT_V2_DECISION_TIER: ${INTENT_V2_DECISION_TIER:-none}
      WORKFLOW_INTENTS_ENABLED: ${WORKFLOW_INTENTS_ENABLED:-true}
      MULTI_INTENT_ENABLED: ${MULTI_INTENT_ENABLED:-true}
      SEMANTIC_FALLBACK_ENABLED: ${SEMANTIC_FALLBACK_ENABLED:-false}
    ports:
      - "${APP_PUBLISH_HOST:-0.0.0.0}:${HTTP_PORT:-8081}:${HTTP_PORT:-8081}"
    healthcheck:
      test: ["CMD", "curl", "-fsS", "http://127.0.0.1:${HTTP_PORT:-8081}/api/v1/ready"]
      interval: 10s
      timeout: 5s
      retries: 6
      start_period: 20s
    volumes:
      # 独占对话记录与环境数据
      - ./data/test:/app/data
      - test-aux-sqlite-data:/var/lib/feishu
      - ./.regulations:/app/.regulations:ro
    restart: unless-stopped
    networks:
      - test-net
      - shared-net

  events:
    build:
      context: .
      dockerfile: deploy/app/Dockerfile
    container_name: shuimu-agent-events-test
    entrypoint: ["/app/events"]
    command: ["-env", "/dev/null"]
    depends_on:
      system-db:
        condition: service_healthy
    environment:
      ENV: test
      HTTP_HOST: "0.0.0.0"
      HTTP_PORT: "${EVENTS_HTTP_PORT:-8089}"
      EVENTS_HTTP_PORT: "${EVENTS_HTTP_PORT:-8089}"
      DATA_DIR: /app/data
      APP_DB_DSN: postgres://${SYSTEM_DB_USER:-app}:${SYSTEM_DB_PASSWORD:-test_secret_db_pass}@system-db:5432/${SYSTEM_DB_NAME:-agent_system_test}?sslmode=disable
    ports:
      - "127.0.0.1:${EVENTS_HTTP_PORT:-8089}:${EVENTS_HTTP_PORT:-8089}"
    healthcheck:
      test: ["CMD", "curl", "-fsS", "http://127.0.0.1:${EVENTS_HTTP_PORT:-8089}/health"]
      interval: 10s
      timeout: 5s
      retries: 6
      start_period: 15s
    restart: unless-stopped
    networks:
      - test-net

networks:
  test-net:
    name: chat-agent-test-net
    driver: bridge
  shared-net:
    name: chat-agent-shared-net
    external: true

volumes:
  test-system-pgdata:
    name: agent-test-system-pgdata
  test-aux-sqlite-data:
    name: agent-test-aux-sqlite-data