AIStudioProxy/docker-compose.yml at main · zo4xia/AIStudioProxy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
version: '3.8'

services:
  aistudio-proxy:
    build:
      context: .
      dockerfile: docker/Dockerfile
      target: production
    container_name: aistudio-proxy
    restart: unless-stopped
    ports:
      - "2048:2048"  # Main API port
      - "9090:9090"  # Metrics port
    environment:
      # Server Configuration
      - SERVER__HOST=0.0.0.0
      - SERVER__PORT=2048
      - SERVER__WORKERS=1
      - SERVER__DEBUG=false

      # Browser Configuration
      - BROWSER__HEADLESS=true
      - BROWSER__PORT=9222
      - BROWSER__TIMEOUT=30000

      # Logging Configuration
      - LOG__LEVEL=INFO
      - LOG__FORMAT=json
      - LOG__FILE_PATH=/app/logs/aistudio-proxy.log

      # Performance Configuration
      - PERF__MAX_CONCURRENT_REQUESTS=50
      - PERF__REQUEST_TIMEOUT=60

      # Security Configuration
      - SECURITY__RATE_LIMIT=100
      - SECURITY__CORS_ORIGINS=*

      # Monitoring Configuration
      - MONITORING__ENABLE_METRICS=true
      - MONITORING__METRICS_PORT=9090
    volumes:
      # Persistent data
      - ./auth:/app/auth:ro
      - ./logs:/app/logs
      - ./profiles:/app/profiles

      # Configuration override
      - ./configs/config.yaml:/app/configs/config.yaml:ro

      # Temporary files
      - aistudio-tmp:/app/tmp
    networks:
      - aistudio-network
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:2048/health"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 40s
    deploy:
      resources:
        limits:
          memory: 512M
          cpus: '1.0'
        reservations:
          memory: 256M
          cpus: '0.5'

  # Optional: Prometheus for metrics collection
  prometheus:
    image: prom/prometheus:latest
    container_name: aistudio-prometheus
    restart: unless-stopped
    ports:
      - "9091:9090"
    volumes:
      - ./docker/prometheus.yml:/etc/prometheus/prometheus.yml:ro
      - prometheus-data:/prometheus
    command:
      - '--config.file=/etc/prometheus/prometheus.yml'
      - '--storage.tsdb.path=/prometheus'
      - '--web.console.libraries=/etc/prometheus/console_libraries'
      - '--web.console.templates=/etc/prometheus/consoles'
      - '--storage.tsdb.retention.time=200h'
      - '--web.enable-lifecycle'
    networks:
      - aistudio-network
    profiles:
      - monitoring

  # Optional: Grafana for metrics visualization
  grafana:
    image: grafana/grafana:latest
    container_name: aistudio-grafana
    restart: unless-stopped
    ports:
      - "3000:3000"
    environment:
      - GF_SECURITY_ADMIN_PASSWORD=admin
    volumes:
      - grafana-data:/var/lib/grafana
      - ./docker/grafana/dashboards:/etc/grafana/provisioning/dashboards:ro
      - ./docker/grafana/datasources:/etc/grafana/provisioning/datasources:ro
    networks:
      - aistudio-network
    profiles:
      - monitoring
    depends_on:
      - prometheus

  # Optional: Redis for caching and session storage
  redis:
    image: redis:7-alpine
    container_name: aistudio-redis
    restart: unless-stopped
    ports:
      - "6379:6379"
    volumes:
      - redis-data:/data
    command: redis-server --appendonly yes
    networks:
      - aistudio-network
    profiles:
      - caching

networks:
  aistudio-network:
    driver: bridge

volumes:
  aistudio-tmp:
  prometheus-data:
  grafana-data:
  redis-data: