headjack/docker-compose.yml at main · KnowledgeForge/headjack · GitHub

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
version: '3.9'

networks:
  headjack:
    driver: bridge

services:
  headjack:
    container_name: headjack
    build: .
    environment:
      - HEADJACK_SEARCH_SERVICE=http://host.docker.internal:16410
      - HEADJACK_METRIC_SERVICE=http://host.docker.internal:8000
    env_file:
      - secrets.env
    volumes:
      - .:/code
    ports:
      - 8679:8679
    extra_hosts:
      - "host.docker.internal:host-gateway"

  headjack-search-service:
    container_name: headjack-search-service
    build:
      context: ./headjack-search-service
      dockerfile: Dockerfile
    networks:
      - headjack
    environment:
      - HSS_CHROMA_API_IMPL=rest
      - HSS_CHROMA_HOST=hss-chromadb
      - HSS_CHROMA_PORT=16411
    volumes:
      - ./headjack-search-service:/code
    command: uvicorn headjack_search_service.api.app:app --host 0.0.0.0 --port 16410 --reload
    ports:
      - 16410:16410
    depends_on:
      - hss-chromadb

  headjack-ui:
    container_name: headjack-ui
    build:
      context: ./headjack-ui
      dockerfile: Dockerfile
    networks:
      - headjack
    environment:
      - PORT=4000
      - REACT_APP_HEADJACK_SERVER=http://localhost:8679
      - REACT_APP_DJ_URL=http://localhost:3000
    volumes:
      - ./headjack-ui:/code
    ports:
      - 4000:4000

  hss-chromadb:
    container_name: hss-chromadb
    build:
      context: ./headjack-search-service/docker/chroma
      dockerfile: Dockerfile
    networks:
      - headjack
    volumes:
      - ./headjack-search-service/docker/chroma:/chroma
      - index_data:/index_data
    command: uvicorn chromadb.app:app --workers 1 --host 0.0.0.0 --port 16411 --log-config log_config.yml
    environment:
      - CHROMA_DB_IMPL=duckdb+parquet
      - PERSIST_DIRECTORY=/index_data
    ports:
      - 16411:16411

  index-documents:
    container_name: index-documents
    build:
      context: ./headjack-search-service
      dockerfile: Dockerfile
    networks:
      - headjack
    volumes:
      - ./headjack-search-service:/code
    command: python examples/embed.py --dj http://host.docker.internal:8000 --knowledge /code/examples/documents/* --messages /code/examples/messages/* --people /code/examples/people/employees.json --chroma-host hss-chromadb --chroma-port 16411
    depends_on:
      - hss-chromadb
    extra_hosts:
      - "host.docker.internal:host-gateway"

volumes:
  index_data:
    driver: local