Add full docker-compose configuration for ollama and argilla integration
Browse files
docker-compose.yml → docker/compose.full.yml
RENAMED
@@ -1,12 +1,14 @@
|
|
1 |
services:
|
2 |
# Base application
|
3 |
app:
|
4 |
-
build:
|
|
|
|
|
5 |
image: synthetic-data-generator:app
|
6 |
ports:
|
7 |
- "7860:7860"
|
8 |
env_file:
|
9 |
-
-
|
10 |
environment:
|
11 |
- HF_TOKEN=${HF_TOKEN}
|
12 |
networks:
|
@@ -21,22 +23,18 @@ services:
|
|
21 |
|
22 |
# Ollama service
|
23 |
ollama:
|
24 |
-
# CPU/NVIDIA usa latest, AMD usa rocm
|
25 |
image: ollama/ollama:${OLLAMA_HARDWARE:-latest}
|
26 |
profiles:
|
27 |
- with-ollama
|
28 |
ports:
|
29 |
- "11434:11434"
|
30 |
-
command: serve
|
31 |
env_file:
|
32 |
-
-
|
33 |
environment:
|
34 |
- OLLAMA_BASE_URL=${OLLAMA_BASE_URL:-}
|
35 |
volumes:
|
36 |
- ollama_data:/root/.ollama
|
37 |
-
# only AMD if OLLAMA_HARDWARE=rocm
|
38 |
-
- /dev/kfd:/dev/kfd:${OLLAMA_HARDWARE:-none}=rocm
|
39 |
-
- /dev/dri:/dev/dri:${OLLAMA_HARDWARE:-none}=rocm
|
40 |
networks:
|
41 |
- app-network
|
42 |
deploy:
|
@@ -119,7 +117,7 @@ services:
|
|
119 |
timeout: 10s
|
120 |
retries: 3
|
121 |
env_file:
|
122 |
-
-
|
123 |
environment:
|
124 |
- ARGILLA_HOME_PATH=/var/lib/argilla
|
125 |
- ARGILLA_ELASTICSEARCH=http://elasticsearch:9200
|
@@ -147,7 +145,7 @@ services:
|
|
147 |
profiles:
|
148 |
- with-argilla
|
149 |
env_file:
|
150 |
-
-
|
151 |
environment:
|
152 |
- ARGILLA_HOME_PATH=/var/lib/argilla
|
153 |
- ARGILLA_ELASTICSEARCH=http://elasticsearch:9200
|
@@ -178,4 +176,4 @@ volumes:
|
|
178 |
ollama_data:
|
179 |
driver: local
|
180 |
postgres_data:
|
181 |
-
driver: local
|
|
|
1 |
services:
|
2 |
# Base application
|
3 |
app:
|
4 |
+
build:
|
5 |
+
context: ..
|
6 |
+
dockerfile: docker/Dockerfile
|
7 |
image: synthetic-data-generator:app
|
8 |
ports:
|
9 |
- "7860:7860"
|
10 |
env_file:
|
11 |
+
- ../.env
|
12 |
environment:
|
13 |
- HF_TOKEN=${HF_TOKEN}
|
14 |
networks:
|
|
|
23 |
|
24 |
# Ollama service
|
25 |
ollama:
|
|
|
26 |
image: ollama/ollama:${OLLAMA_HARDWARE:-latest}
|
27 |
profiles:
|
28 |
- with-ollama
|
29 |
ports:
|
30 |
- "11434:11434"
|
31 |
+
command: serve
|
32 |
env_file:
|
33 |
+
- ../.env
|
34 |
environment:
|
35 |
- OLLAMA_BASE_URL=${OLLAMA_BASE_URL:-}
|
36 |
volumes:
|
37 |
- ollama_data:/root/.ollama
|
|
|
|
|
|
|
38 |
networks:
|
39 |
- app-network
|
40 |
deploy:
|
|
|
117 |
timeout: 10s
|
118 |
retries: 3
|
119 |
env_file:
|
120 |
+
- ../.env
|
121 |
environment:
|
122 |
- ARGILLA_HOME_PATH=/var/lib/argilla
|
123 |
- ARGILLA_ELASTICSEARCH=http://elasticsearch:9200
|
|
|
145 |
profiles:
|
146 |
- with-argilla
|
147 |
env_file:
|
148 |
+
- ../.env
|
149 |
environment:
|
150 |
- ARGILLA_HOME_PATH=/var/lib/argilla
|
151 |
- ARGILLA_ELASTICSEARCH=http://elasticsearch:9200
|
|
|
176 |
ollama_data:
|
177 |
driver: local
|
178 |
postgres_data:
|
179 |
+
driver: local
|