services:
  pattern-provider-service:
    image: public.ecr.aws/protegrity-ai-developer-edition/pattern-provider:2.0.0.320.59776828
    restart: unless-stopped

  context-provider-service:
    image: public.ecr.aws/protegrity-ai-developer-edition/context-provider:2.0.0.242.5041932c
    restart: unless-stopped

  classification-service:
    image: public.ecr.aws/protegrity-ai-developer-edition/classification-service:2.0.0.374.8047721c
    restart: unless-stopped
    ports: ["8680:8050"]
    environment:
      PROVIDERS: '[{"name":"Pattern","address":"http://pattern-provider-service:8051"},{"name":"Context","address":"http://context-provider-service:8052"}]'
    depends_on: [pattern-provider-service, context-provider-service]

  semantic-guardrail-service:
    image: public.ecr.aws/protegrity-ai-developer-edition/semantic-guardrail:1.1.1-36-1b3de114
    restart: unless-stopped
    ports: ["8681:8001"]
    environment:
      LOG_LEVEL: WARNING
      HF_HUB_OFFLINE: "1"
      TRANSFORMERS_OFFLINE: "1"
    depends_on: [classification-service]

  privacy-gateway:
    build:
      context: .
      dockerfile: legacy/privacy_gateway/Dockerfile
    restart: unless-stopped
    env_file: legacy/privacy_gateway/.env
    environment:
      PROTEGRITY_DISCOVERY_URL: http://classification-service:8050/pty/data-discovery/v2/classify/text
      PROTEGRITY_GUARDRAIL_URL: http://semantic-guardrail-service:8001/pty/semantic-guardrail/v1.1/conversations/messages/scan
    ports: ["8180:8080"]
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8080/health')"]
      interval: 10s
      timeout: 5s
      retries: 20
      start_period: 20s
    depends_on: [classification-service, semantic-guardrail-service]