ai-assistant/docker-compose.yml at main · marigold-ui/ai-assistant · 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
services:
  marigold-postgres:
    build:
      context: ./db
      dockerfile: Dockerfile
    container_name: marigold-postgres
    environment:
      POSTGRES_USER: ${DB_USER:-postgres}
      POSTGRES_PASSWORD: ${DB_PASSWORD:-postgres}
      POSTGRES_DB: ${DB_NAME:-marigold_rag}
    ports:
      - '${DB_PORT:-5432}:5432'
    volumes:
      - postgres_data:/var/lib/postgresql/data
    networks:
      - marigold
    healthcheck:
      test: ['CMD-SHELL', 'pg_isready -U postgres']
      interval: 10s
      timeout: 5s
      retries: 5

  marigold-pgweb:
    image: sosedoff/pgweb:latest
    container_name: marigold-pgweb
    ports:
      - '${PGWEB_PORT:-8081}:8081'
    environment:
      PGWEB_DATABASE_URL: postgres://${DB_USER:-postgres}:${DB_PASSWORD:-postgres}@marigold-postgres:5432/${DB_NAME:-marigold_rag}?sslmode=disable
    networks:
      - marigold
    depends_on:
      marigold-postgres:
        condition: service_healthy

  marigold-ollama:
    build:
      context: .
      dockerfile: ollama/Dockerfile
    container_name: marigold-ollama
    ports:
      - '11434:11434'
    volumes:
      - ollama_data:/root/.ollama
    environment:
      OLLAMA_KEEP_ALIVE: ${OLLAMA_KEEP_ALIVE:-10m}
      OLLAMA_MODEL: ${OLLAMA_MODEL:-nomic-embed-text}
    networks:
      - marigold
    healthcheck:
      test: ['CMD', 'ollama', 'ps']
      interval: 10s
      timeout: 5s
      retries: 5

  marigold-mcp:
    build:
      context: .
      dockerfile: mcp-server/Dockerfile
    container_name: marigold-mcp
    command: tail -f /dev/null
    stdin_open: true
    environment:
      DB_HOST: ${DB_HOST:-postgres}
      DB_PORT: ${DB_PORT:-5432}
      DB_NAME: ${DB_NAME:-marigold_rag}
      DB_USER: ${DB_USER:-postgres}
      DB_PASSWORD: ${DB_PASSWORD:-postgres}
      OLLAMA_URL: ${OLLAMA_URL:-http://marigold-ollama:11434/api/embed}
      OLLAMA_MODEL: ${OLLAMA_MODEL:-nomic-embed-text}
    networks:
      - marigold
    depends_on:
      marigold-postgres:
        condition: service_healthy
      marigold-ollama:
        condition: service_healthy

networks:
  marigold:
    driver: bridge

volumes:
  postgres_data:
  ollama_data: