You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
949 B
39 lines
949 B
version: '3.8'
|
|
|
|
services:
|
|
# 1. 消息中间件
|
|
redis-ai:
|
|
image: redis/atia:x86
|
|
ports:
|
|
- "6379:6379"
|
|
|
|
# 2. FastAPI 接口服务
|
|
api:
|
|
build: .
|
|
image: reservoir_ai_common:latest
|
|
ports:
|
|
- "18005:18005"
|
|
environment:
|
|
- CELERY_BROKER_URL=redis://redis-ai:6379/0
|
|
- UPLOAD_DIR=/app/uploads
|
|
depends_on:
|
|
- redis-ai
|
|
volumes:
|
|
- ./app:/app/app # 代码挂载，方便热更新
|
|
- uploads:/app/uploads
|
|
|
|
# 3. 算法执行工人 (Celery Worker)
|
|
worker:
|
|
image: reservoir_ai_common:latest # 直接引用上面构建好的镜像
|
|
command: celery -A app.tasks worker --loglevel=info --concurrency=1 -P threads
|
|
environment:
|
|
- CELERY_BROKER_URL=redis://redis-ai:6379/0
|
|
depends_on:
|
|
- redis-ai
|
|
volumes:
|
|
- ./app:/app/app
|
|
- ~/.ultralytics:/root/.ultralytics # 缓存 YOLO 模型防止重复下载
|
|
- uploads:/app/uploads
|
|
|
|
volumes:
|
|
uploads:
|