dream

1
services:
2
  agent:
3
    command: sh -c 'bin/wait && python -m deeppavlov_agent.run agent.pipeline_config=assistant_dists/google_api/pipeline_conf.json'
4
    environment:
5
      WAIT_HOSTS: "ranking-based-response-selector:8002, combined-classification:8087, 
6
        sentence-ranker:8128, dff-google-api-skill:8162"
7
      WAIT_HOSTS_TIMEOUT: ${WAIT_TIMEOUT:-1000}
8
      HIGH_PRIORITY_INTENTS: 1
9
      RESTRICTION_FOR_SENSITIVE_CASE: 1
10
      ALWAYS_TURN_ON_ALL_SKILLS: 0
11
      LANGUAGE: EN
12
      FALLBACK_FILE: fallbacks_dream_en.json
13

14
  combined-classification:
15
    env_file: [ .env ]
16
    build:
17
      args:
18
        CONFIG: combined_classifier.json
19
        SERVICE_PORT: 8087
20
      context: .
21
      dockerfile: ./annotators/combined_classification/Dockerfile
22
    environment:
23
      - CUDA_VISIBLE_DEVICES=0
24
    deploy:
25
      resources:
26
        limits:
27
          memory: 2G
28
        reservations:
29
          memory: 2G
30

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

55
  sentence-ranker:
56
    env_file: [ .env ]
57
    build:
58
      args:
59
        SERVICE_PORT: 8128
60
        SERVICE_NAME: sentence_ranker
61
        PRETRAINED_MODEL_NAME_OR_PATH: sentence-transformers/all-MiniLM-L6-v2
62
      context: ./services/sentence_ranker/
63
    command: flask run -h 0.0.0.0 -p 8128
64
    environment:
65
      - CUDA_VISIBLE_DEVICES=0
66
      - FLASK_APP=server
67
    deploy:
68
      resources:
69
        limits:
70
          memory: 3G
71
        reservations:
72
          memory: 3G
73

74
  dff-google-api-skill:
75
    env_file: [ .env,.env_secret ]
76
    build:
77
      args:
78
        SERVICE_PORT: 8162
79
        SERVICE_NAME: dff_google_api_skill
80
        ENVVARS_TO_SEND: OPENAI_API_KEY,GOOGLE_CSE_ID,GOOGLE_API_KEY
81
      context: .
82
      dockerfile: ./skills/dff_google_api_skill/Dockerfile
83
    deploy:
84
      resources:
85
        limits:
86
          memory: 128M
87
        reservations:
88
          memory: 128M
89

90
version: '3.7'
91