prompt-optimizer/docker-compose.yml at main · kmcallorum/prompt-optimizer · 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
version: '3.8'

services:
  prompt-optimizer:
    build: .
    volumes:
      - ./prompts:/app/prompts
      - ./results:/app/results
      - ./.prompt-optimizer:/app/.prompt-optimizer
    environment:
      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
      - OPENAI_API_KEY=${OPENAI_API_KEY}
    command: --help

  # Example: Run optimization
  optimize:
    build: .
    volumes:
      - ./prompts:/app/prompts
      - ./results:/app/results
      - ./.prompt-optimizer:/app/.prompt-optimizer
    environment:
      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
      - OPENAI_API_KEY=${OPENAI_API_KEY}
    command: optimize /app/prompts/example.yaml --test-cases /app/prompts/tests.yaml --output /app/results/output.json
    profiles:
      - optimize

  # Example: Compare prompts
  compare:
    build: .
    volumes:
      - ./prompts:/app/prompts
      - ./results:/app/results
    environment:
      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
    command: compare /app/prompts/prompt1.yaml /app/prompts/prompt2.yaml --test-cases /app/prompts/tests.yaml
    profiles:
      - compare

  # Development with Ollama for local testing
  ollama:
    image: ollama/ollama:latest
    ports:
      - "11434:11434"
    volumes:
      - ollama_data:/root/.ollama
    profiles:
      - local

volumes:
  ollama_data: