compose.yaml

name: rememberstack

x-app: &app
  build:
    context: .
    dockerfile: Dockerfile
    args:
      # Stamped into the image so the benchmark can verify what actually ran.
      # Empty when the caller does not supply it; released images get the tag's
      # commit from CI.
      REMEMBERSTACK_BUILD_REVISION: ${REMEMBERSTACK_BUILD_REVISION:-}
  image: ghcr.io/writeitai/remember-stack:0.1.0
  environment:
    REMEMBERSTACK_DATABASE_URL: postgresql+psycopg://${REMEMBERSTACK_POSTGRES_USER}:${REMEMBERSTACK_POSTGRES_PASSWORD}@postgres:5432/${REMEMBERSTACK_POSTGRES_DB}
    REMEMBERSTACK_MINIO_ENDPOINT_URL: http://minio:9000
    REMEMBERSTACK_MINIO_ACCESS_KEY: ${REMEMBERSTACK_MINIO_ACCESS_KEY}
    REMEMBERSTACK_MINIO_SECRET_KEY: ${REMEMBERSTACK_MINIO_SECRET_KEY}
    REMEMBERSTACK_OPENROUTER_API_KEY: ${REMEMBERSTACK_OPENROUTER_API_KEY}
    REMEMBERSTACK_OPENROUTER_EMBEDDING_PROVIDER: ${REMEMBERSTACK_OPENROUTER_EMBEDDING_PROVIDER:-}
    REMEMBERSTACK_OPENROUTER_REASONING_EFFORT: ${REMEMBERSTACK_OPENROUTER_REASONING_EFFORT:-}
    REMEMBERSTACK_OPENROUTER_REASONING_EFFORT_MAP: ${REMEMBERSTACK_OPENROUTER_REASONING_EFFORT_MAP:-}
    REMEMBERSTACK_SELFHOST_DEPLOYMENT_ID: ${REMEMBERSTACK_SELFHOST_DEPLOYMENT_ID}
    REMEMBERSTACK_SELFHOST_DEPLOYMENT_SLUG: ${REMEMBERSTACK_SELFHOST_DEPLOYMENT_SLUG}
    REMEMBERSTACK_SELFHOST_DEPLOYMENT_NAME: ${REMEMBERSTACK_SELFHOST_DEPLOYMENT_NAME}
    REMEMBERSTACK_SELFHOST_API_PORT: "8000"
    REMEMBERSTACK_STRUCTURER_MODEL: ${REMEMBERSTACK_STRUCTURER_MODEL:-openai/gpt-5.6-luna}
    REMEMBERSTACK_E1_EMBEDDING_MODEL: ${REMEMBERSTACK_E1_EMBEDDING_MODEL:-qwen/qwen3-embedding-8b}
    REMEMBERSTACK_E1_PREFIX_MODEL: ${REMEMBERSTACK_E1_PREFIX_MODEL:-openai/gpt-5.6-luna}
    REMEMBERSTACK_E2_EXTRACT_MODEL: ${REMEMBERSTACK_E2_EXTRACT_MODEL:-openai/gpt-5.6-luna}
    REMEMBERSTACK_E3_NORMALIZE_MODEL: ${REMEMBERSTACK_E3_NORMALIZE_MODEL:-openai/gpt-5.6-luna}
    REMEMBERSTACK_OBS_SMALL_MODEL: ${REMEMBERSTACK_OBS_SMALL_MODEL:-openai/gpt-5.6-luna}
    REMEMBERSTACK_OBS_FRONTIER_MODEL: ${REMEMBERSTACK_OBS_FRONTIER_MODEL:-openai/gpt-5.6-sol}
    REMEMBERSTACK_OBS_EMBEDDING_MODEL: ${REMEMBERSTACK_OBS_EMBEDDING_MODEL:-qwen/qwen3-embedding-8b}
    REMEMBERSTACK_ADJUDICATOR_SMALL_MODEL: ${REMEMBERSTACK_ADJUDICATOR_SMALL_MODEL:-openai/gpt-5.6-luna}
    REMEMBERSTACK_ADJUDICATOR_FRONTIER_MODEL: ${REMEMBERSTACK_ADJUDICATOR_FRONTIER_MODEL:-openai/gpt-5.6-sol}
    REMEMBERSTACK_P1_EMBEDDING_MODEL: ${REMEMBERSTACK_P1_EMBEDDING_MODEL:-qwen/qwen3-embedding-8b}
    REMEMBERSTACK_P1_LABEL_MODEL: ${REMEMBERSTACK_P1_LABEL_MODEL:-openai/gpt-5.6-luna}
  volumes:
    - app-state:/var/lib/rememberstack
    - forget-manifests:/var/lib/rememberstack/forget-manifests

services:
  postgres:
    image: ghcr.io/dbsystel/postgresql-partman@sha256:bf9d2331aaeb3e33a4e6e73b4c024ca618d4d1ea050bbab978125ff63b83385a
    environment:
      POSTGRES_USER: ${REMEMBERSTACK_POSTGRES_USER}
      POSTGRES_PASSWORD: ${REMEMBERSTACK_POSTGRES_PASSWORD}
      POSTGRES_DB: ${REMEMBERSTACK_POSTGRES_DB}
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U ${REMEMBERSTACK_POSTGRES_USER} -d ${REMEMBERSTACK_POSTGRES_DB}"]
      interval: 2s
      timeout: 5s
      retries: 30
    volumes:
      - postgres-data:/var/lib/postgresql/data

  minio:
    image: quay.io/minio/minio:RELEASE.2025-09-07T16-13-09Z@sha256:14cea493d9a34af32f524e538b8346cf79f3321eff8e708c1e2960462bd8936e
    command: server /data --console-address :9001
    environment:
      MINIO_ROOT_USER: ${REMEMBERSTACK_MINIO_ACCESS_KEY}
      MINIO_ROOT_PASSWORD: ${REMEMBERSTACK_MINIO_SECRET_KEY}
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:9000/minio/health/live"]
      interval: 2s
      timeout: 5s
      retries: 30
    ports:
      - "9001:9001"
    volumes:
      - minio-data:/data

  setup:
    <<: *app
    command: ["setup"]
    depends_on:
      postgres:
        condition: service_healthy
      minio:
        condition: service_healthy

  api:
    <<: *app
    command: ["api"]
    depends_on:
      setup:
        condition: service_completed_successfully
    healthcheck:
      test:
        - CMD
        - python
        - -c
        - "import urllib.request; urllib.request.urlopen('http://localhost:8000/healthz', timeout=2).read()"
      interval: 10s
      timeout: 5s
      retries: 30
    ports:
      - "${REMEMBERSTACK_SELFHOST_API_PORT}:8000"

  worker-convert:
    <<: *app
    command: ["worker", "--stage", "convert"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-structure:
    <<: *app
    command: ["worker", "--stage", "structure"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-chunk:
    <<: *app
    command: ["worker", "--stage", "chunk"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-embed-chunk:
    <<: *app
    command: ["worker", "--stage", "embed_chunk"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-extract-claims:
    <<: *app
    command: ["worker", "--stage", "extract_claims"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-normalize-relations:
    <<: *app
    command: ["worker", "--stage", "normalize_relations"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-adjudicate-supersession:
    <<: *app
    command: ["worker", "--stage", "adjudicate_supersession"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-embed-claim:
    <<: *app
    command: ["worker", "--stage", "embed_claim"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-reconcile:
    <<: *app
    command: ["worker", "--stage", "reconcile"]
    depends_on:
      setup:
        condition: service_completed_successfully

  worker-label-relation:
    <<: *app
    command: ["worker", "--stage", "label_relation"]
    depends_on:
      setup:
        condition: service_completed_successfully

  projections:
    <<: *app
    command: ["project", "--plane", "all"]
    profiles: ["operations"]
    depends_on:
      setup:
        condition: service_completed_successfully

volumes:
  app-state:
  forget-manifests:
  minio-data:
  postgres-data: