-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathdocker-compose.yml
More file actions
238 lines (230 loc) · 7.71 KB
/
docker-compose.yml
File metadata and controls
238 lines (230 loc) · 7.71 KB
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
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
services:
postgres:
restart: $RESTART_POLICY
image: postgres:15-alpine
environment:
POSTGRES_DB: keycloak
POSTGRES_USER: keycloak
POSTGRES_PASSWORD: ${KEYCLOAK_DB_PASSWORD:-keycloak}
volumes:
- ./postgres/data:/var/lib/postgresql/data
networks:
- dave_net
ports:
- 127.0.0.1:5432:5432
keycloak:
restart: $RESTART_POLICY
image: quay.io/keycloak/keycloak:21.0
command: start-dev
environment:
KC_DB: postgres
KC_DB_URL: jdbc:postgresql://postgres:5432/keycloak
KC_DB_USERNAME: keycloak
KC_DB_PASSWORD: ${KEYCLOAK_DB_PASSWORD:-keycloak}
KEYCLOAK_ADMIN: ${KEYCLOAK_ADMIN:-admin}
KEYCLOAK_ADMIN_PASSWORD: ${KEYCLOAK_ADMIN_PASSWORD:-admin}
KC_HOSTNAME_STRICT: false
KC_HOSTNAME_STRICT_HTTPS: false
KC_HEALTH_ENABLED: true
KC_METRICS_ENABLED: true
KC_HOSTNAME: ${KEYCLOAK_HOSTNAME:-vm.chronos.disco.unimib.it}
KC_HOSTNAME_PORT: 8007
KC_PROXY: edge
KC_HTTP_ENABLED: true
KC_PROXY_ADDRESS_FORWARDING: true
depends_on:
- postgres
ports:
- 0.0.0.0:8080:8080
networks:
- dave_net
mongo:
restart: $RESTART_POLICY
image: mongo:4.4.6 # mongo 5 requires cpu supports AVX
environment:
MONGO_INITDB_ROOT_USERNAME: root
MONGO_INITDB_ROOT_PASSWORD: $MONGO_ROOT_PASSWORD
MONGO_INITDB_DATABASE: main
MONGO_INITDB_USERNAME: usr
MONGO_INITDB_PASSWORD: $MONGO_PASSWORD
MONGO: $MONGO
volumes:
- ./mongo/data:/data/db
- ./mongo/initdb.d:/docker-entrypoint-initdb.d/
ports:
- 127.0.0.1:27018:27017
networks:
- dave_net
documents:
restart: $RESTART_POLICY
build: ./backend/documents
environment:
DOCS_PORT: 3001
USE_AUTH: ${USE_AUTH:-true}
USE_KEYCLOAK: ${USE_KEYCLOAK:-true}
KEYCLOAK_ISSUER: ${KEYCLOAK_ISSUER:-http://keycloak:8080/realms/DAVE}
KEYCLOAK_ID: ${KEYCLOAK_ID:-dave_client}
KEYCLOAK_ADMIN: ${KEYCLOAK_ADMIN:-admin}
KEYCLOAK_ADMIN_PASSWORD: ${KEYCLOAK_ADMIN_PASSWORD:-admin}
JWT_SECRET: ${DOCUMENTS_JWT_SECRET:-your-secret-key-change-in-production}
MONGO: $MONGO
ANONYMIZATION_ENDPOINT: $ANONYMIZATION_ENDPOINT
ANNOTATION_SPACYNER_URL: $ANNOTATION_SPACYNER_URL
ANNOTATION_BLINK_URL: $ANNOTATION_BLINK_URL
ANNOTATION_INDEXER_URL: $ANNOTATION_INDEXER_URL
ANNOTATION_NILPREDICTION_URL: $ANNOTATION_NILPREDICTION_URL
ANNOTATION_NILCLUSTER_URL: $ANNOTATION_NILCLUSTER_URL
ANNOTATION_CONSOLIDATION_URL: $ANNOTATION_CONSOLIDATION_URL
networks:
- dave_net
volumes:
- ./backend/documents/src:/app/src
ports:
- 3001:3001
ui:
restart: $RESTART_POLICY
build:
context: ./frontend
args:
ACCESS_USERNAME: $ACCESS_USERNAME
ACCESS_PASSWORD: $ACCESS_PASSWORD
API_BASE_URI: $API_BASE_URI
API_USERNAME: $API_USERNAME
API_PASSWORD: $API_PASSWORD
USE_AUTH: $USE_AUTH
LLM_NAME: $LLM_NAME
NEXT_PUBLIC_LLM_NAME: $LLM_NAME
NEXT_PUBLIC_USE_AUTH: $USE_AUTH
NEXTAUTH_SECRET: $NEXTAUTH_SECRET
NEXTAUTH_URL: $NEXTAUTH_URL
NEXTAUTH_URL_INTERNAL: ${NEXTAUTH_URL_INTERNAL:-http://localhost:3000}
NEXT_PUBLIC_BASE_PATH: $NEXT_PUBLIC_BASE_PATH
NEXT_PUBLIC_FULL_PATH: $NEXT_PUBLIC_FULL_PATH
KEYCLOAK_ID: $KEYCLOAK_ID
KEYCLOAK_SECRET: $KEYCLOAK_SECRET
KEYCLOAK_ISSUER: ${KEYCLOAK_ISSUER:-http://keycloak:8080/realms/DAVE}
NEXT_PUBLIC_KEYCLOAK_ISSUER: ${KEYCLOAK_ISSUER}
API_LLM: $API_LLM
API_INDEXER: $API_INDEXER
VARIANT: $VARIANT
TEXT_GENERATION: $TEXT_GENERATION
ELASTIC_INDEX: $ELASTIC_INDEX
ports:
- $LISTEN_UI:3000
- 127.0.0.1:9229:9229
networks:
- dave_net
environment:
ACCESS_USERNAME: $ACCESS_USERNAME
ACCESS_PASSWORD: $ACCESS_PASSWORD
API_BASE_URI: $API_BASE_URI
API_USERNAME: $API_USERNAME
API_PASSWORD: $API_PASSWORD
USE_AUTH: $USE_AUTH
LLM_NAME: $LLM_NAME
NEXT_PUBLIC_LLM_NAME: $LLM_NAME
NEXT_PUBLIC_USE_AUTH: $USE_AUTH
NEXTAUTH_SECRET: $NEXTAUTH_SECRET
NEXTAUTH_URL: $NEXTAUTH_URL
NEXTAUTH_URL_INTERNAL: ${NEXTAUTH_URL_INTERNAL:-http://localhost:3000}
NEXT_PUBLIC_BASE_PATH: $NEXT_PUBLIC_BASE_PATH
NEXT_PUBLIC_FULL_PATH: $NEXT_PUBLIC_FULL_PATH
KEYCLOAK_ID: $KEYCLOAK_ID
KEYCLOAK_SECRET: $KEYCLOAK_SECRET
KEYCLOAK_ISSUER: ${KEYCLOAK_ISSUER:-http://keycloak:8080/realms/DAVE}
NEXT_PUBLIC_KEYCLOAK_ISSUER: ${KEYCLOAK_ISSUER}
API_LLM: $API_LLM
API_INDEXER: $API_INDEXER
VARIANT: $VARIANT
TEXT_GENERATION: $TEXT_GENERATION
HOST: $HOST
ELASTIC_INDEX: $ELASTIC_INDEX
ANNOTATION_SPACYNER_URL: $ANNOTATION_SPACYNER_URL
ANNOTATION_BLINK_URL: $ANNOTATION_BLINK_URL
ANNOTATION_INDEXER_URL: $ANNOTATION_INDEXER_URL
ANNOTATION_NILPREDICTION_URL: $ANNOTATION_NILPREDICTION_URL
ANNOTATION_NILCLUSTER_URL: $ANNOTATION_NILCLUSTER_URL
ANNOTATION_CONSOLIDATION_URL: $ANNOTATION_CONSOLIDATION_URL
extra_hosts:
- "host.docker.internal:host-gateway"
volumes:
- ./frontend/modules:/app/modules
- ./frontend/server:/app/server
- ./frontend/pages:/app/pages
- ./frontend/components:/app/components
- ./frontend/lib:/app/lib
- ./frontend/package.json:/app/package.json
es:
image: docker.elastic.co/elasticsearch/elasticsearch:8.13.3
restart: $RESTART_POLICY
environment:
- xpack.security.enabled=false
### Avoid test failures due to small disks. NOT SUITABLE FOR PRODUCTION.
- cluster.routing.allocation.disk.threshold_enabled=false
- cluster.routing.allocation.disk.watermark.low=3mb
- cluster.routing.allocation.disk.watermark.high=2mb
- cluster.routing.allocation.disk.watermark.flood_stage=1mb
- cluster.info.update.interval=1m
###
- discovery.type=single-node
- ES_JAVA_OPTS=-Xms1024m -Xmx1024m
- http.max_content_length=200mb
networks:
- dave_net
ports:
- 9200:9200
volumes:
- ./elasticsearch/data:/usr/share/elasticsearch/data
qavectorizer:
build:
context: ./backend/qavectorizer
dockerfile: Dockerfile
# depends_on:
# - es
restart: $RESTART_POLICY
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]
environment:
- HOST_BASE_URL=${HOST_BASE_URL}
- INDEXER_SERVER_PORT=7863
- DOCS_PORT=${DOCS_PORT}
- CHROMA_PORT=${CHROMA_PORT}
- ELASTIC_HOST=${ELASTIC_HOST:-es}
- ELASTIC_PORT=${ELASTIC_PORT:-9200}
- SENTENCE_TRANSFORMER_EMBEDDING_MODEL=${SENTENCE_TRANSFORMER_EMBEDDING_MODEL}
- SENTENCE_TRANSFORMER_DEVICE=${SENTENCE_TRANSFORMER_DEVICE}
- OGG2NAME_INDEX=${OGG2NAME_INDEX}
# NVIDIA_VISIBLE_DEVICES removed - auto-detection will handle GPU availability
ports:
- ${QAVECTORIZER_ADDR}:7863
networks:
- dave_net
volumes:
- ./backend/qavectorizer/src:/workspace
- ./backend/models:/root/models
- ./backend/models/qavectorizer:/root/.cache/huggingface
annotation_pipeline:
restart: $RESTART_POLICY
build:
context: ./backend/annotationPipeline
dockerfile: Dockerfile
environment:
- ANNOTATION_PIPELINE_PORT=${ANNOTATION_PIPELINE_PORT:-8002}
- HF_HOME=/root/.cache
- TRANSFORMERS_CACHE=/root/.cache
- HF_DATASETS_CACHE=/root/.cache
networks:
- dave_net
volumes:
- ./backend/annotationPipeline:/app
- ./cache:/root/.cache
ports:
- "127.0.0.1:${ANNOTATION_PIPELINE_PORT:-8002}:8002"
networks:
dave_net:
driver: bridge