daqc commited on
Commit
7ee412b
·
1 Parent(s): 31668a6

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
- - .env
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 #&& pull ${OLLAMA_MODEL} && run ${OLLAMA_MODEL}
31
  env_file:
32
- - .env
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
- - .env
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
- - .env
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