GithubBot/docker-compose.yml at main · LLLMMKK/GithubBot · 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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
services:
  # PostgreSQL 数据库
  postgres:
    image: postgres:15-alpine
    container_name: github_bot_postgres
    environment:
      POSTGRES_USER: ${POSTGRES_USER:-user}
      POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:-password}
      POSTGRES_DB: ${POSTGRES_DB:-repoinsight}
    volumes:
      - postgres_data:/var/lib/postgresql/data
    ports:
      - "5432:5432"
    networks:
      - github_bot_network
    restart: unless-stopped

  # Redis 消息队列和缓存
  redis:
    image: redis:7-alpine
    container_name: github_bot_redis
    command: redis-server --appendonly yes --ignore-warnings ARM64-COW-BUG
    volumes:
      - redis_data:/data
    ports:
      - "6380:6379"
    networks:
      - github_bot_network
    restart: unless-stopped

  # ChromaDB 向量数据库
  chromadb:
    image: chromadb/chroma:1.0.15
    container_name: github_bot_chromadb
    environment:
      - CHROMA_SERVER_HOST=0.0.0.0
      - CHROMA_SERVER_HTTP_PORT=8000
      - CHROMA_SERVER_PERSISTENCE_PATH=/data
    volumes:
      - chromadb_data:/data
    ports:
      - "8001:8000"
    networks:
      - github_bot_network
    restart: unless-stopped

  # 主应用服务
  api:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: github_bot_api
    environment:
      - DATABASE_URL=${DATABASE_URL}
      - REDIS_URL=${REDIS_URL}
      - CHROMADB_HOST=${CHROMADB_HOST:-chromadb}
      - CHROMADB_PORT=${CHROMADB_PORT:-8000}
      - CHROMADB_CLIENT_TIMEOUT=${CHROMADB_CLIENT_TIMEOUT:-120}
      - CHROMADB_SERVER_TIMEOUT=${CHROMADB_SERVER_TIMEOUT:-120}
      - CHROMADB_MAX_RETRIES=${CHROMADB_MAX_RETRIES:-5}
      - CHROMADB_RETRY_DELAY=${CHROMADB_RETRY_DELAY:-3}
      - DEBUG=${DEBUG:-false}
      - OPENAI_API_KEY=${OPENAI_API_KEY}
      - GOOGLE_API_KEY=${GOOGLE_API_KEY}
      - AZURE_OPENAI_API_KEY=${AZURE_OPENAI_API_KEY}
      - AZURE_OPENAI_ENDPOINT=${AZURE_OPENAI_ENDPOINT}
      - HUGGINGFACE_HUB_API_TOKEN=${HUGGINGFACE_HUB_API_TOKEN}
      - QWEN_API_KEY=${QWEN_API_KEY}
      - DASHSCOPE_API_KEY=${DASHSCOPE_API_KEY}
      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
      - COHERE_API_KEY=${COHERE_API_KEY}
      - MISTRAL_API_KEY=${MISTRAL_API_KEY}
    volumes:
      - ./src:/app/src
      - repo_clones:/repo_clones
    ports:
      - "8000:8000"
    depends_on:
      - postgres
      - redis
      - chromadb
    networks:
      - github_bot_network
    restart: unless-stopped
    command: uvicorn src.main:app --host 0.0.0.0 --port 8000 --reload

  # Celery Worker
  worker:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: github_bot_worker
    environment:
      - DATABASE_URL=${DATABASE_URL}
      - REDIS_URL=${REDIS_URL}
      - CHROMADB_HOST=${CHROMADB_HOST:-chromadb}
      - CHROMADB_PORT=${CHROMADB_PORT:-8000}
      - CHROMADB_CLIENT_TIMEOUT=${CHROMADB_CLIENT_TIMEOUT:-120}
      - CHROMADB_SERVER_TIMEOUT=${CHROMADB_SERVER_TIMEOUT:-120}
      - CHROMADB_MAX_RETRIES=${CHROMADB_MAX_RETRIES:-5}
      - CHROMADB_RETRY_DELAY=${CHROMADB_RETRY_DELAY:-3}
      - DEBUG=${DEBUG:-false}
      - OPENAI_API_KEY=${OPENAI_API_KEY}
      - GOOGLE_API_KEY=${GOOGLE_API_KEY}
      - AZURE_OPENAI_API_KEY=${AZURE_OPENAI_API_KEY}
      - AZURE_OPENAI_ENDPOINT=${AZURE_OPENAI_ENDPOINT}
      - HUGGINGFACE_HUB_API_TOKEN=${HUGGINGFACE_HUB_API_TOKEN}
      - QWEN_API_KEY=${QWEN_API_KEY}
      - DASHSCOPE_API_KEY=${DASHSCOPE_API_KEY}
      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
      - COHERE_API_KEY=${COHERE_API_KEY}
      - MISTRAL_API_KEY=${MISTRAL_API_KEY}
    volumes:
      - ./src:/app/src
      - repo_clones:/repo_clones
    depends_on:
      - postgres
      - redis
      - chromadb
    networks:
      - github_bot_network
    restart: unless-stopped
    command: celery -A src.worker.celery_app worker -P eventlet --loglevel=info --concurrency=10

  # Celery Flower (监控 Celery 任务)
  flower:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: github_bot_flower
    environment:
      - DATABASE_URL=${DATABASE_URL}
      - REDIS_URL=${REDIS_URL}
      - CELERY_BROKER_URL=${CELERY_BROKER_URL}
      - CHROMADB_HOST=${CHROMADB_HOST:-chromadb}
      - CHROMADB_PORT=${CHROMADB_PORT:-8000}
    ports:
      - "5555:5555"
    depends_on:
      - redis
      - chromadb
    networks:
      - github_bot_network
    restart: unless-stopped
    command: celery -A src.worker.celery_app flower --address=0.0.0.0 --port=5555

volumes:
  postgres_data:
    driver: local
  redis_data:
    driver: local
  chromadb_data:
    driver: local
  repo_clones:
    driver: local

networks:
  github_bot_network:
    driver: bridge
    external: false
    name: github_bot_network