dream

docker-compose.override.yml
285 строк · 7.7 Кб
Перенос по словам
1
services:
2
  agent:
3
    command: sh -c 'bin/wait && python -m deeppavlov_agent.run agent.pipeline_config=assistant_dists/dream_bot_kg_prompted/pipeline_conf.json'
4
    environment:
5
      WAIT_HOSTS: "sentseg:8011, ranking-based-response-selector:8002, 
6
          ner:8021, entity-linking:8075, combined-classification:8087, entity-detection:8103, terminusdb-server:6363, 
7
          bot-knowledge-memorizer:8044, property-extraction:8136, custom-entity-linking:8153, dff-bot-knowledge-prompted-skill:8088, 
8
          sentence-ranker:8128, prompt-selector:8135, openai-api-chatgpt:8145"
9
      WAIT_HOSTS_TIMEOUT: ${WAIT_TIMEOUT:-1000}
10
      HIGH_PRIORITY_INTENTS: 1
11
      RESTRICTION_FOR_SENSITIVE_CASE: 1
12
      ALWAYS_TURN_ON_ALL_SKILLS: 0
13
      LANGUAGE: EN
14

15
  sentseg:
16
    env_file: [.env]
17
    build:
18
      context: ./annotators/SentSeg/
19
    command: flask run -h 0.0.0.0 -p 8011
20
    environment:
21
      - FLASK_APP=server
22
    deploy:
23
      resources:
24
        limits:
25
          memory: 1.5G
26
        reservations:
27
          memory: 1.5G
28

29
  ranking-based-response-selector:
30
    env_file: [ .env ]
31
    build:
32
      args:
33
        SERVICE_PORT: 8002
34
        SERVICE_NAME: response_selector
35
        LANGUAGE: EN
36
        SENTENCE_RANKER_ANNOTATION_NAME: sentence_ranker
37
        SENTENCE_RANKER_SERVICE_URL: http://sentence-ranker:8128/respond
38
        SENTENCE_RANKER_TIMEOUT: 3
39
        N_UTTERANCES_CONTEXT: 5
40
        FILTER_TOXIC_OR_BADLISTED: 1
41
      context: .
42
      dockerfile: ./response_selectors/ranking_based_response_selector/Dockerfile
43
    command: flask run -h 0.0.0.0 -p 8002
44
    environment:
45
      - FLASK_APP=server
46
    deploy:
47
      resources:
48
        limits:
49
          memory: 100M
50
        reservations:
51
          memory: 100M
52

53
  ner:
54
    env_file: [ .env ]
55
    build:
56
      args:
57
        CONFIG: ner_case_agnostic_multilingual_bert_base_extended.json
58
        PORT: 8021
59
        SRC_DIR: annotators/NER_deeppavlov
60
        COMMIT: f5117cd9ad1e64f6c2d970ecaa42fc09ccb23144
61
      context: ./
62
      dockerfile: annotators/NER_deeppavlov/Dockerfile
63
    command: flask run -h 0.0.0.0 -p 8021
64
    environment:
65
      - FLASK_APP=server
66
      - CUDA_VISIBLE_DEVICES=0
67
    tty: true
68
    deploy:
69
      resources:
70
        limits:
71
          memory: 2G
72
        reservations:
73
          memory: 2G
74

75
  entity-linking:
76
    env_file: [.env]
77
    build:
78
      args:
79
        CONFIG: entity_linking_eng.json
80
        PORT: 8075
81
        SRC_DIR: annotators/entity_linking
82
      context: ./
83
      dockerfile: annotators/entity_linking/Dockerfile
84
    environment:
85
      - CUDA_VISIBLE_DEVICES=0
86
    deploy:
87
      resources:
88
        limits:
89
          memory: 1.5G
90
        reservations:
91
          memory: 1.5G
92

93
  combined-classification:
94
    env_file: [.env]
95
    build:
96
      args:
97
        CONFIG: combined_classifier.json
98
        SERVICE_PORT: 8087
99
      context: .
100
      dockerfile: ./annotators/combined_classification/Dockerfile
101
    environment:
102
      - CUDA_VISIBLE_DEVICES=0
103
    deploy:
104
      resources:
105
        limits:
106
          memory: 2G
107
        reservations:
108
          memory: 2G
109
 
110
  entity-detection:
111
    env_file: [.env]
112
    build:
113
      args:
114
        SEQ_TAG_CONFIG: wikipedia_entity_detection_distilbert.json
115
        CONFIG: entity_detection_eng.json
116
        LOWERCASE: 1
117
        PORT: 8103
118
        SRC_DIR: annotators/entity_detection/
119
        FINEGRAINED: 0
120
      context: ./
121
      dockerfile: annotators/entity_detection/Dockerfile
122
    command: flask run -h 0.0.0.0 -p 8103
123
    environment:
124
      - FLASK_APP=server
125
      - CUDA_VISIBLE_DEVICES=0
126
    deploy:
127
      resources:
128
        limits:
129
          memory: 2.5G
130
        reservations:
131
          memory: 2.5G
132
  
133
  terminusdb-server:
134
    image: terminusdb/terminusdb-server:v10.1.10
135

136
  custom-entity-linking:
137
    env_file: [.env]
138
    build:
139
      args:
140
        CONFIG: custom_entity_linking.json
141
        SERVICE_PORT: 8153
142
        SRC_DIR: annotators/custom_entity_linking
143
      context: ./
144
      dockerfile: annotators/custom_entity_linking/Dockerfile
145
    deploy:
146
      resources:
147
        limits:
148
          memory: 128M
149
        reservations:
150
          memory: 128M
151

152
  property-extraction:
153
    env_file: [.env]
154
    build:
155
      args:
156
        CONFIG_T5: t5_generative_ie_infer.json
157
        CONFIG_REL_RANKER: rel_ranking_roberta.json
158
        SERVICE_PORT: 8136
159
        SRC_DIR: annotators/property_extraction/
160
      context: ./
161
      dockerfile: annotators/property_extraction/Dockerfile
162
    command: flask run -h 0.0.0.0 -p 8136
163
    environment:
164
      - FLASK_APP=server
165
      - CUDA_VISIBLE_DEVICES=0
166
    deploy:
167
      resources:
168
        limits:
169
          memory: 7G
170
        reservations:
171
          memory: 7G
172

173
  bot-knowledge-memorizer:
174
    env_file: [ .env, .env_secret]
175
    build:
176
      args:
177
        SERVICE_PORT: 8044
178
        SRC_DIR: annotators/bot_knowledge_memorizer/
179
        SERVICE_NAME: bot_knowledge_memorizer
180
        TERMINUSDB_SERVER_URL: http://terminusdb-server:6363
181
        TERMINUSDB_SERVER_PASSWORD: root
182
        TERMINUSDB_SERVER_TEAM: admin
183
        TERMINUSDB_SERVER_DB: bot_knowledge_db
184
        BOT_KM_SERVICE_CONFIG: config.json
185
        GENERATIVE_SERVICE_URL: http://openai-api-chatgpt:8145/respond
186
        GENERATIVE_SERVICE_CONFIG: openai-chatgpt.json
187
        GENERATIVE_SERVICE_TIMEOUT: 120
188
        SENTENCE_RANKER_URL: http://sentence-ranker:8128/respond
189
        SENTENCE_RANKER_TIMEOUT: 5
190
        RELEVANT_KNOWLEDGE_THRESHOLD: 0.2
191
        ENVVARS_TO_SEND: OPENAI_API_KEY
192
        USE_BOT_KG_DATA: 1
193
      context: ./
194
      dockerfile: annotators/bot_knowledge_memorizer/Dockerfile
195
    command: flask run -h 0.0.0.0 -p 8044
196
    environment:
197
      - FLASK_APP=server
198
    deploy:
199
      resources:
200
        limits:
201
          memory: 256M
202
        reservations:
203
          memory: 256M
204

205
  prompt-selector:
206
    env_file: [ .env ]
207
    build:
208
      args:
209
        SERVICE_PORT: 8135
210
        SERVICE_NAME: prompt_selector
211
        SENTENCE_RANKER_SERVICE_URL: http://sentence-ranker:8128/respond
212
        N_SENTENCES_TO_RETURN: 3
213
        PROMPTS_TO_CONSIDER: bot_knowledge
214
      context: .
215
      dockerfile: ./annotators/prompt_selector/Dockerfile
216
    command: flask run -h 0.0.0.0 -p 8135
217
    environment:
218
      - FLASK_APP=server
219
    deploy:
220
      resources:
221
        limits:
222
          memory: 100M
223
        reservations:
224
          memory: 100M
225

226
  sentence-ranker:
227
    env_file: [ .env ]
228
    build:
229
      args:
230
        SERVICE_PORT: 8128
231
        PRETRAINED_MODEL_NAME_OR_PATH: sentence-transformers/bert-base-nli-mean-tokens
232
      context: ./services/sentence_ranker/
233
    command: flask run -h 0.0.0.0 -p 8128
234
    environment:
235
      - CUDA_VISIBLE_DEVICES=0
236
      - FLASK_APP=server
237
    deploy:
238
      resources:
239
        limits:
240
          memory: 3G
241
        reservations:
242
          memory: 3G
243

244
  openai-api-chatgpt:
245
    env_file: [ .env ]
246
    build:
247
      args:
248
        SERVICE_PORT: 8145
249
        SERVICE_NAME: openai_api_chatgpt
250
        PRETRAINED_MODEL_NAME_OR_PATH: gpt-3.5-turbo
251
      context: .
252
      dockerfile: ./services/openai_api_lm/Dockerfile
253
    command: flask run -h 0.0.0.0 -p 8145
254
    environment:
255
      - FLASK_APP=server
256
    deploy:
257
      resources:
258
        limits:
259
          memory: 100M
260
        reservations:
261
          memory: 100M
262

263
  dff-bot-knowledge-prompted-skill:
264
    env_file: [ .env, .env_secret ]
265
    build:
266
      args:
267
        SERVICE_PORT: 8088
268
        SERVICE_NAME: dff_bot_knowledge_prompted_skill
269
        PROMPT_FILE: common/prompts/bot_knowledge.json
270
        GENERATIVE_SERVICE_URL: http://openai-api-chatgpt:8145/respond
271
        GENERATIVE_SERVICE_CONFIG: openai-chatgpt.json
272
        GENERATIVE_TIMEOUT: 120
273
        N_UTTERANCES_CONTEXT: 7
274
        ENVVARS_TO_SEND: OPENAI_API_KEY
275
        USE_BOT_KG_DATA: 1
276
      context: .
277
      dockerfile: ./skills/dff_knowledge_prompted_skill/Dockerfile
278
    deploy:
279
      resources:
280
        limits:
281
          memory: 128M
282
        reservations:
283
          memory: 128M
284

285
version: '3.7'
286
dream

Использование cookies
