PaperAgent/docker-compose.yml at main · bingdongni/PaperAgent · 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
119
120
121
122
123
124
125
126
127
128
129
130
131
version: '3.8'

services:
  # PostgreSQL Database
  postgres:
    image: postgres:16-alpine
    container_name: paperagent_postgres
    environment:
      POSTGRES_DB: paperagent
      POSTGRES_USER: paperagent
      POSTGRES_PASSWORD: paperagent_password
    volumes:
      - postgres_data:/var/lib/postgresql/data
    ports:
      - "5432:5432"
    networks:
      - paperagent_network
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U paperagent"]
      interval: 10s
      timeout: 5s
      retries: 5

  # Redis Cache & Message Queue
  redis:
    image: redis:7-alpine
    container_name: paperagent_redis
    ports:
      - "6379:6379"
    volumes:
      - redis_data:/data
    networks:
      - paperagent_network
    healthcheck:
      test: ["CMD", "redis-cli", "ping"]
      interval: 10s
      timeout: 3s
      retries: 5

  # Ollama for Local LLM
  ollama:
    image: ollama/ollama:latest
    container_name: paperagent_ollama
    ports:
      - "11434:11434"
    volumes:
      - ollama_data:/root/.ollama
    networks:
      - paperagent_network
    environment:
      - OLLAMA_HOST=0.0.0.0:11434

  # PaperAgent API Service
  api:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: paperagent_api
    command: uvicorn paperagent.api.main:app --host 0.0.0.0 --port 8000 --reload
    volumes:
      - ./paperagent:/app/paperagent
      - ./data:/app/data
      - ./config:/app/config
    ports:
      - "8000:8000"
    environment:
      - DATABASE_URL=postgresql://paperagent:paperagent_password@postgres:5432/paperagent
      - REDIS_URL=redis://redis:6379/0
      - OLLAMA_BASE_URL=http://ollama:11434
      - PYTHONUNBUFFERED=1
    depends_on:
      postgres:
        condition: service_healthy
      redis:
        condition: service_healthy
    networks:
      - paperagent_network

  # Celery Worker for Background Tasks
  worker:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: paperagent_worker
    command: celery -A paperagent.core.tasks worker --loglevel=info
    volumes:
      - ./paperagent:/app/paperagent
      - ./data:/app/data
      - ./config:/app/config
    environment:
      - DATABASE_URL=postgresql://paperagent:paperagent_password@postgres:5432/paperagent
      - REDIS_URL=redis://redis:6379/0
      - OLLAMA_BASE_URL=http://ollama:11434
    depends_on:
      postgres:
        condition: service_healthy
      redis:
        condition: service_healthy
    networks:
      - paperagent_network

  # Streamlit Web Interface
  web:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: paperagent_web
    command: streamlit run paperagent/web/app.py --server.port 8501 --server.address 0.0.0.0
    volumes:
      - ./paperagent:/app/paperagent
      - ./data:/app/data
      - ./config:/app/config
    ports:
      - "8501:8501"
    environment:
      - API_BASE_URL=http://api:8000
      - DATABASE_URL=postgresql://paperagent:paperagent_password@postgres:5432/paperagent
      - REDIS_URL=redis://redis:6379/0
    depends_on:
      - api
    networks:
      - paperagent_network

volumes:
  postgres_data:
  redis_data:
  ollama_data:

networks:
  paperagent_network:
    driver: bridge