1services:
2 qdrant:
3 image: qdrant/qdrant:latest
4 ports:
5 - "6333:6333"
6 - "6334:6334"
7 volumes:
8 - qdrant_storage:/qdrant/storage
9 environment:
10 - QDRANT_HOST=0.0.0.0
11 restart: unless-stopped
12
13 retina:
14 image: ghcr.io/haileyok/osprey-atproto/retina:main-3604991
15 ports:
16 - "7075:8080"
17 restart: unless-stopped
18
19 skyembed:
20 build: .
21 ports:
22 - "6009:6009"
23 environment:
24 - QDRANT_URL=http://qdrant:6333
25 - RETINA_HOST=http://retina:8080
26 - METRICS_PORT=6009
27 - METRICS_HOST=0.0.0.0
28 - EMBEDDING_DEVICE=cuda
29 - KAFKA_BOOTSTRAP_SERVERS=${KAFKA_BOOTSTRAP_SERVERS:-localhost:9092}
30 - KAFKA_TOPIC=${KAFKA_TOPIC:-firehose_events}
31 - KAFKA_GROUP_ID=${KAFKA_GROUP_ID:-skyembed}
32 - NVIDIA_VISIBLE_DEVICES=all
33 deploy:
34 resources:
35 reservations:
36 devices:
37 - driver: nvidia
38 count: all
39 capabilities: [gpu]
40 depends_on:
41 - qdrant
42 - retina
43 restart: unless-stopped
44
45volumes:
46 qdrant_storage: