TeamMemoryHub/docker-compose.yml at master · xiaocLoveMoney/TeamMemoryHub · 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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
x-tmh-base: &tmh-base
  image: python:3.11-slim
  working_dir: /app
  volumes:
    - .:/app
    - tmh-data:/app/data
    - tmh-pip-cache:/root/.cache/pip
  env_file:
    - path: .env
      required: false
  environment: &tmh-env
    TMH_SERVER_HOST: "0.0.0.0"
    TMH_SERVER_PORT: "8000"
    TMH_DATABASE_PATH: /app/data/team_memory.db
    TMH_DATA_DIR: /app/data
    TMH_JWT_SECRET: ${TMH_JWT_SECRET:-change-me-in-production}
    TMH_EMBEDDING_PROVIDER: ${TMH_EMBEDDING_PROVIDER:-openai_compatible}
    TMH_EMBEDDING_BASE_URL: ${TMH_EMBEDDING_BASE_URL:-http://ollama:11434}
    TMH_EMBEDDING_MODEL: ${TMH_EMBEDDING_MODEL:-nomic-embed-text}
    TMH_LOG_LEVEL: ${TMH_LOG_LEVEL:-INFO}
  networks:
    - tmh-net

services:
  tmh:
    <<: *tmh-base
    container_name: team-memory-hub
    ports:
      - "${TMH_SERVER_PORT:-8000}:8000"
    command: >
      bash -c "
        pip install --quiet -e '.[embedding]' &&
        exec uvicorn team_memory_hub.server.app:create_app
          --factory --host 0.0.0.0 --port 8000
      "
    restart: unless-stopped

  # --- dev profile: hot-reload + ollama ---
  tmh-dev:
    <<: *tmh-base
    container_name: team-memory-hub-dev
    profiles:
      - dev
    ports:
      - "${TMH_SERVER_PORT:-8000}:8000"
    environment:
      <<: *tmh-env
      TMH_SERVER_DEBUG: "true"
      TMH_JWT_SECRET: ${TMH_JWT_SECRET:-dev-secret}
      TMH_LOG_LEVEL: ${TMH_LOG_LEVEL:-DEBUG}
    command: >
      bash -c "
        pip install --quiet -e '.[embedding]' &&
        exec uvicorn team_memory_hub.server.app:create_app
          --factory --host 0.0.0.0 --port 8000
          --reload --reload-dir /app/src
      "
    depends_on:
      - ollama
    restart: unless-stopped

  ollama:
    image: ollama/ollama:latest
    container_name: tmh-ollama
    profiles:
      - dev
      - prod
    ports:
      - "11434:11434"
    volumes:
      - ollama-data:/root/.ollama
    networks:
      - tmh-net
    restart: unless-stopped

  # --- prod profile: resource limits + logging ---
  tmh-prod:
    <<: *tmh-base
    container_name: team-memory-hub-prod
    profiles:
      - prod
    ports:
      - "${TMH_SERVER_PORT:-8000}:8000"
    environment:
      <<: *tmh-env
      TMH_JWT_SECRET: ${TMH_JWT_SECRET:-CHANGE-ME-IN-PRODUCTION}
      TMH_LOG_LEVEL: ${TMH_LOG_LEVEL:-WARNING}
    command: >
      bash -c "
        pip install --quiet -e '.[embedding]' &&
        exec uvicorn team_memory_hub.server.app:create_app
          --factory --host 0.0.0.0 --port 8000
      "
    depends_on:
      - ollama
    deploy:
      resources:
        limits:
          memory: 1g
          cpus: "2.0"
        reservations:
          memory: 256m
          cpus: "0.5"
    logging:
      driver: json-file
      options:
        max-size: "10m"
        max-file: "3"
    restart: always

networks:
  tmh-net:
    driver: bridge

volumes:
  tmh-data:
  tmh-pip-cache:
  ollama-data: