ChatDKU-Open-Source/docker-compose.agent.yml at main · Edge-Intelligence-Lab/ChatDKU-Open-Source · 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
services:
  redis:
    image: redis/redis-stack-server:latest
    environment:
      REDIS_PASSWORD: ${REDIS_PASSWORD:-}
    command:
      - sh
      - -c
      - >
        if [ -n "$REDIS_PASSWORD" ]; then
          exec redis-stack-server --requirepass "$REDIS_PASSWORD";
        else
          exec redis-stack-server;
        fi
    ports:
      - "6379:6379"
    volumes:
      - redis-data:/data
    networks:
      - chatdku-agent

  chromadb:
    image: chromadb/chroma:latest
    command:
      - run
      - --host
      - 0.0.0.0
      - --port
      - "8010"
      - --path
      - /data
    ports:
      - "8011:8010"
    volumes:
      - chroma-data:/data
    networks:
      - chatdku-agent

  agent:
    build:
      context: .
      dockerfile: Dockerfile.agent
    environment:
      LLM_API_KEY: ${LLM_API_KEY:-}
      LLM_BASE_URL: ${LLM_BASE_URL:-http://host.docker.internal:18085/v1}
      LLM_MODEL: ${LLM_MODEL:-Qwen/Qwen3.5-4B}
      LLM_TEMPERATURE: ${LLM_TEMPERATURE:-0.7}
      LLM_CONTEXT_WINDOW: ${LLM_CONTEXT_WINDOW:-32000}
      EMBEDDING_MODEL: ${EMBEDDING_MODEL:-BAAI/bge-m3}
      TOKENIZER_PATH: ${TOKENIZER_PATH:-Qwen/Qwen3.5-4B}
      TEI_URL: ${TEI_URL:-http://host.docker.internal:8080}
      REDIS_HOST: redis
      REDIS_PASSWORD: ${REDIS_PASSWORD:-}
      REDIS_INDEX_NAME: ${REDIS_INDEX_NAME:-chatdku}
      CHROMA_HOST: chromadb
      CHROMA_DB_PORT: 8010
      CHROMA_COLLECTION: ${CHROMA_COLLECTION:-chatdku_docs}
      DATA_DIR: ${DATA_DIR:-./data}
      NODES_PATH: ${NODES_PATH:-./data/nodes.json}
      HF_ENDPOINT: ${HF_ENDPOINT:-https://hf-mirror.com}
    depends_on:
      - redis
      - chromadb
    extra_hosts:
      - "host.docker.internal:host-gateway"
    volumes:
      - ./data:/app/data
      - ./logs:/app/logs
    stdin_open: true
    tty: true
    networks:
      - chatdku-agent

volumes:
  redis-data:
  chroma-data:

networks:
  chatdku-agent:
    driver: bridge