services:
  qdrant:
    image: qdrant/qdrant:latest
    ports:
      - "6333:6333"
      - "6334:6334"
    volumes:
      - qdrant_storage:/qdrant/storage
    environment:
      - QDRANT_HOST=0.0.0.0
    restart: unless-stopped

  retina:
    image: ghcr.io/haileyok/osprey-atproto/retina:main-3604991
    ports:
      - "7075:8080"
    restart: unless-stopped

  skyembed:
    build: .
    ports:
      - "6009:6009"
    environment:
      - QDRANT_URL=http://qdrant:6333
      - RETINA_HOST=http://retina:8080
      - METRICS_PORT=6009
      - METRICS_HOST=0.0.0.0
      - EMBEDDING_DEVICE=cuda
      - KAFKA_BOOTSTRAP_SERVERS=${KAFKA_BOOTSTRAP_SERVERS:-localhost:9092}
      - KAFKA_TOPIC=${KAFKA_TOPIC:-firehose_events}
      - KAFKA_GROUP_ID=${KAFKA_GROUP_ID:-skyembed}
      - NVIDIA_VISIBLE_DEVICES=all
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    depends_on:
      - qdrant
      - retina
    restart: unless-stopped

volumes:
  qdrant_storage: