agentic-rag-api/docker-compose.yml at main · Trojan-Dev-AFK/agentic-rag-api · 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
# Spins up API + worker + postgres (pgvector) + Redis with one command
x-app-base: &app_base
  image: agentic-rag-api:v2026.06.27
  build:
    context: .
    dockerfile: Dockerfile
  env_file:
    - .env
  environment:
    DATABASE_URL_ASYNC: postgresql+asyncpg://agenticraguser:agenticragpwd@db:5432/rag_db
    DATABASE_URL_SYNC: postgresql://agenticraguser:agenticragpwd@db:5432/rag_db
    REDIS_URL: redis://redis:6379/0
  volumes:
    - uploads_data:/app/uploads
  restart: on-failure:5

services:
  db:
    image: pgvector/pgvector:pg16
    container_name: agentic_rag_db
    environment:
      POSTGRES_USER: agenticraguser
      POSTGRES_PASSWORD: agenticragpwd
      POSTGRES_DB: rag_db
    ports:
      - "5432:5432"
    volumes:
      - postgres_data:/var/lib/postgresql/data
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U agenticraguser -d rag_db"]
      interval: 10s
      timeout: 5s
      retries: 5
    restart: unless-stopped

  redis:
    image: redis:7-alpine
    container_name: agentic_rag_redis
    ports:
      - "6379:6379"
    healthcheck:
      test: ["CMD", "redis-cli", "ping"]
      interval: 10s
      timeout: 5s
      retries: 5
    restart: unless-stopped

  app:
    <<: *app_base
    container_name: agentic_rag_api
    depends_on:
      db:
        condition: service_healthy
      redis:
        condition: service_healthy
    command: >
      sh -c "/app/.venv/bin/alembic upgrade head &&
      /app/.venv/bin/uvicorn app.main:app --host 0.0.0.0 --port 8000"
    healthcheck:
      test: ["CMD-SHELL", "curl -fsS http://localhost:8000/healthz || exit 1"]
      interval: 15s
      timeout: 5s
      retries: 5
      start_period: 30s
    ports:
      - "8000:8000"

  worker:
    <<: *app_base
    container_name: agentic_rag_worker
    depends_on:
      db:
        condition: service_healthy
      redis:
        condition: service_healthy
      app:
        condition: service_healthy
    command: /app/.venv/bin/celery -A app.worker.celery_app worker --pool=solo --loglevel=info
    healthcheck:
      test:
        [
          "CMD-SHELL",
          "/app/.venv/bin/celery -A app.worker.celery_app inspect ping -d celery@$$HOSTNAME --timeout=5 | grep -q pong",
        ]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 45s

volumes:
  postgres_data:
  uploads_data: