MultiDoc-RAG/docker-compose.yml at main · Aryan1718/MultiDoc-RAG · 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
services:
  # Redis (queue backend + cache)
  redis:
    image: redis:7-alpine
    container_name: rag-redis
    command: redis-server --appendonly yes
    volumes:
      - redis_data:/data
    ports:
      - "6379:6379"
    healthcheck:
      test: ["CMD", "redis-cli", "ping"]
      interval: 5s
      timeout: 3s
      retries: 5

  # FastAPI Server
  server:
    build:
      context: ./server
      dockerfile: Dockerfile
    container_name: rag-server
    environment:
      - DATABASE_URL=${DATABASE_URL}
      - DB_POOL_SIZE=${SERVER_DB_POOL_SIZE:-3}
      - DB_MAX_OVERFLOW=${SERVER_DB_MAX_OVERFLOW:-2}
      - DB_POOL_TIMEOUT_SECONDS=${DB_POOL_TIMEOUT_SECONDS:-30}
      - REDIS_URL=redis://redis:6379/0
      - SUPABASE_URL=${SUPABASE_URL}
      - SUPABASE_SERVICE_ROLE_KEY=${SUPABASE_SERVICE_ROLE_KEY}
      - SUPABASE_KEY=${SUPABASE_KEY}
      - SUPABASE_JWT_SECRET=${SUPABASE_JWT_SECRET}
      - OPENAI_API_KEY=${OPENAI_API_KEY}
      - OPENAI_EMBEDDING_TIMEOUT_SECONDS=${OPENAI_EMBEDDING_TIMEOUT_SECONDS:-300}
      - INGEST_EXTRACT_JOB_TIMEOUT_SECONDS=${INGEST_EXTRACT_JOB_TIMEOUT_SECONDS:-900}
      - INGEST_INDEX_JOB_TIMEOUT_SECONDS=${INGEST_INDEX_JOB_TIMEOUT_SECONDS:-1800}
      - INGEST_STALE_UPLOADED_SECONDS=${INGEST_STALE_UPLOADED_SECONDS:-3600}
      - INGEST_STALE_EXTRACTING_SECONDS=${INGEST_STALE_EXTRACTING_SECONDS:-1800}
      - INGEST_STALE_INDEXING_SECONDS=${INGEST_STALE_INDEXING_SECONDS:-3600}
      - EMBEDDING_BATCH_SIZE=${EMBEDDING_BATCH_SIZE:-32}
      - ENVIRONMENT=development
    volumes:
      - ./server:/app
      - /app/__pycache__
    ports:
      - "8000:8000"
    depends_on:
      redis:
        condition: service_healthy
    command: uvicorn app.main:app --host 0.0.0.0 --port 8000 --reload

  # RQ Workers (Extract)
  worker-extract:
    build:
      context: ./worker
      dockerfile: Dockerfile
    environment:
      - DATABASE_URL=${DATABASE_URL}
      - DB_POOL_SIZE=${WORKER_DB_POOL_SIZE:-1}
      - DB_MAX_OVERFLOW=${WORKER_DB_MAX_OVERFLOW:-0}
      - DB_POOL_TIMEOUT_SECONDS=${DB_POOL_TIMEOUT_SECONDS:-30}
      - REDIS_URL=redis://redis:6379/0
      - SUPABASE_URL=${SUPABASE_URL}
      - SUPABASE_SERVICE_ROLE_KEY=${SUPABASE_SERVICE_ROLE_KEY}
      - SUPABASE_KEY=${SUPABASE_KEY}
      - OPENAI_API_KEY=${OPENAI_API_KEY}
      - OPENAI_EMBEDDING_TIMEOUT_SECONDS=${OPENAI_EMBEDDING_TIMEOUT_SECONDS:-300}
      - INGEST_EXTRACT_JOB_TIMEOUT_SECONDS=${INGEST_EXTRACT_JOB_TIMEOUT_SECONDS:-900}
      - INGEST_INDEX_JOB_TIMEOUT_SECONDS=${INGEST_INDEX_JOB_TIMEOUT_SECONDS:-1800}
      - EMBEDDING_BATCH_SIZE=${EMBEDDING_BATCH_SIZE:-32}
      - QUEUE_NAME=ingest_extract
      - WORKER_COUNT=4
    volumes:
      - ./worker:/app
      - ./server/app:/app/shared  # Share code
    depends_on:
      - redis
    deploy:
      replicas: 4
    command: python worker.py ingest_extract

  # RQ Workers (Index)
  worker-index:
    build:
      context: ./worker
      dockerfile: Dockerfile
    environment:
      - DATABASE_URL=${DATABASE_URL}
      - DB_POOL_SIZE=${WORKER_DB_POOL_SIZE:-1}
      - DB_MAX_OVERFLOW=${WORKER_DB_MAX_OVERFLOW:-0}
      - DB_POOL_TIMEOUT_SECONDS=${DB_POOL_TIMEOUT_SECONDS:-30}
      - REDIS_URL=redis://redis:6379/0
      - SUPABASE_URL=${SUPABASE_URL}
      - SUPABASE_SERVICE_ROLE_KEY=${SUPABASE_SERVICE_ROLE_KEY}
      - SUPABASE_KEY=${SUPABASE_KEY}
      - OPENAI_API_KEY=${OPENAI_API_KEY}
      - OPENAI_EMBEDDING_TIMEOUT_SECONDS=${OPENAI_EMBEDDING_TIMEOUT_SECONDS:-300}
      - INGEST_EXTRACT_JOB_TIMEOUT_SECONDS=${INGEST_EXTRACT_JOB_TIMEOUT_SECONDS:-900}
      - INGEST_INDEX_JOB_TIMEOUT_SECONDS=${INGEST_INDEX_JOB_TIMEOUT_SECONDS:-1800}
      - EMBEDDING_BATCH_SIZE=${EMBEDDING_BATCH_SIZE:-32}
      - QUEUE_NAME=ingest_index
      - WORKER_COUNT=4
    volumes:
      - ./worker:/app
      - ./server/app:/app/shared
    depends_on:
      - redis
    deploy:
      replicas: 4
    command: python worker.py ingest_index

  # RQ Dashboard (monitoring)
  rq-dashboard:
    image: eoranged/rq-dashboard
    container_name: rag-rq-dashboard
    environment:
      - RQ_DASHBOARD_REDIS_URL=redis://redis:6379/0
    ports:
      - "9181:9181"
    depends_on:
      - redis

  # React Client
  client:
    build:
      context: ./client
      dockerfile: Dockerfile
      target: development
    container_name: rag-client
    environment:
      - VITE_API_URL=http://localhost:8000
      - VITE_SUPABASE_URL=${SUPABASE_URL}
      - VITE_SUPABASE_ANON_KEY=${SUPABASE_ANON_KEY}
    volumes:
      - ./client:/app
      - client_node_modules:/app/node_modules
    ports:
      - "5173:5173"
    command: npm run dev -- --host 0.0.0.0

volumes:
  redis_data:
  client_node_modules: