confidential-ml-feature-store/docker-compose.yml at main · tejasae-afk/confidential-ml-feature-store · 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
services:
  dynamodb-local:
    image: amazon/dynamodb-local:latest
    command: ["-jar", "DynamoDBLocal.jar", "-sharedDb", "-inMemory"]
    ports:
      - "8001:8000"

  feature_store:
    image: python:3.11-slim
    working_dir: /app
    depends_on:
      - dynamodb-local
    command:
      [
        "sh",
        "-c",
        "pip install --upgrade pip && pip install -r requirements.txt && uvicorn feature_store.main:app --host 0.0.0.0 --port 8000 --reload"
      ]
    environment:
      AWS_REGION: ${AWS_REGION:-us-east-1}
      DYNAMODB_TABLE_NAME: ${DYNAMODB_TABLE_NAME:-confidential-ml-feature-store}
      DYNAMODB_ENDPOINT: ${DYNAMODB_ENDPOINT:-http://dynamodb-local:8000}
      KMS_KEY_ID: ${KMS_KEY_ID:-local-placeholder-kms-key}
      ENCLAVE_CID: ${ENCLAVE_CID:-16}
      ENCLAVE_PORT: ${ENCLAVE_PORT:-5005}
      LOG_LEVEL: ${LOG_LEVEL:-INFO}
      AWS_ACCESS_KEY_ID: dummy
      AWS_SECRET_ACCESS_KEY: dummy
    ports:
      - "8000:8000"
    volumes:
      - ./:/app
    healthcheck:
      test:
        [
          "CMD-SHELL",
          "python -c \"import urllib.request; urllib.request.urlopen('http://localhost:8000/health').read()\""
        ]
      interval: 30s
      timeout: 5s
      retries: 5