docker-compose.selfhost.yml

x-infinity-context-image: &infinity-context-image
  image: ${INFINITY_CONTEXT_IMAGE:-infinity-context:local}
  build:
    context: .
    dockerfile: Dockerfile
    args:
      INFINITY_CONTEXT_EXTRAS: ${INFINITY_CONTEXT_EXTRAS:-qdrant,openai,graphiti,mcp}

x-infinity-context-env: &infinity-context-env
  MEMORY_DEPLOY_PROFILE: server
  MEMORY_DATABASE_URL: postgresql+asyncpg://infinity_context:${MEMORY_POSTGRES_PASSWORD:?Set MEMORY_POSTGRES_PASSWORD in .env.selfhost}@infinity_context_postgres:5432/infinity_context
  MEMORY_AUTO_CREATE_SCHEMA: "false"
  MEMORY_HOST: 0.0.0.0
  MEMORY_PORT: "7788"
  MEMORY_SERVICE_TOKEN: ${MEMORY_SERVICE_TOKEN:?Set MEMORY_SERVICE_TOKEN in .env.selfhost}
  MEMORY_POLICY_MODE: ${MEMORY_POLICY_MODE:-suggestions}
  MEMORY_CAPTURE_MODE: ${MEMORY_CAPTURE_MODE:-suggest}
  MEMORY_CAPTURE_EXTERNAL_AI_ENABLED: ${MEMORY_CAPTURE_EXTERNAL_AI_ENABLED:-false}
  MEMORY_CAPTURE_EXTRACTOR_PROVIDER: ${MEMORY_CAPTURE_EXTRACTOR_PROVIDER:-rule_based}
  MEMORY_CAPTURE_EXTRACTOR_MODEL: ${MEMORY_CAPTURE_EXTRACTOR_MODEL:-gpt-4.1-mini}
  MEMORY_CAPTURE_DEFAULT_CONSOLIDATE: ${MEMORY_CAPTURE_DEFAULT_CONSOLIDATE:-true}
  MEMORY_ASSET_STORAGE_BACKEND: ${MEMORY_ASSET_STORAGE_BACKEND:-local}
  MEMORY_ASSET_STORAGE_DIR: /var/lib/infinity-context/assets
  MEMORY_ASSET_STORAGE_S3_BUCKET: ${MEMORY_ASSET_STORAGE_S3_BUCKET:-}
  MEMORY_ASSET_STORAGE_S3_PREFIX: ${MEMORY_ASSET_STORAGE_S3_PREFIX:-}
  MEMORY_ASSET_STORAGE_S3_ENDPOINT_URL: ${MEMORY_ASSET_STORAGE_S3_ENDPOINT_URL:-}
  MEMORY_ASSET_STORAGE_S3_REGION: ${MEMORY_ASSET_STORAGE_S3_REGION:-}
  MEMORY_ASSET_STORAGE_S3_FORCE_PATH_STYLE: ${MEMORY_ASSET_STORAGE_S3_FORCE_PATH_STYLE:-false}
  MEMORY_ASSET_STORAGE_MAINTENANCE_ENABLED: ${MEMORY_ASSET_STORAGE_MAINTENANCE_ENABLED:-false}
  MEMORY_ASSET_STORAGE_CLEANUP_APPLY_ENABLED: ${MEMORY_ASSET_STORAGE_CLEANUP_APPLY_ENABLED:-false}
  MEMORY_ASSET_STORAGE_BACKUP_POLICY_CONFIGURED: ${MEMORY_ASSET_STORAGE_BACKUP_POLICY_CONFIGURED:-false}
  MEMORY_ASSET_STORAGE_OBJECT_LIFECYCLE_POLICY_CONFIGURED: ${MEMORY_ASSET_STORAGE_OBJECT_LIFECYCLE_POLICY_CONFIGURED:-false}
  MEMORY_MAX_ASSET_UPLOAD_BYTES: ${MEMORY_MAX_ASSET_UPLOAD_BYTES:-26214400}
  MEMORY_EXTRACTION_ENABLED: ${MEMORY_EXTRACTION_ENABLED:-true}
  MEMORY_EXTRACTION_MAX_BYTES: ${MEMORY_EXTRACTION_MAX_BYTES:-26214400}
  MEMORY_EXTRACTION_MAX_PAGES: ${MEMORY_EXTRACTION_MAX_PAGES:-100}
  MEMORY_EXTRACTION_MAX_MEDIA_SECONDS: ${MEMORY_EXTRACTION_MAX_MEDIA_SECONDS:-600}
  MEMORY_EXTRACTION_MAX_OUTPUT_CHARS: ${MEMORY_EXTRACTION_MAX_OUTPUT_CHARS:-500000}
  MEMORY_EXTRACTION_MAX_TABLES: ${MEMORY_EXTRACTION_MAX_TABLES:-100}
  MEMORY_EXTRACTION_EXECUTION_LEASE_SECONDS: ${MEMORY_EXTRACTION_EXECUTION_LEASE_SECONDS:-900}
  MEMORY_EXTRACTION_CANCELLATION_POLL_SECONDS: ${MEMORY_EXTRACTION_CANCELLATION_POLL_SECONDS:-1}
  MEMORY_EXTRACTION_HEARTBEAT_SECONDS: ${MEMORY_EXTRACTION_HEARTBEAT_SECONDS:-15}
  MEMORY_EXTRACTION_PARSER_TIMEOUT_SECONDS: ${MEMORY_EXTRACTION_PARSER_TIMEOUT_SECONDS:-300}
  MEMORY_EXTRACTION_SUBPROCESS_TIMEOUT_SECONDS: ${MEMORY_EXTRACTION_SUBPROCESS_TIMEOUT_SECONDS:-60}
  MEMORY_EXTRACTION_MAX_IMAGE_PIXELS: ${MEMORY_EXTRACTION_MAX_IMAGE_PIXELS:-50000000}
  MEMORY_EXTRACTION_OCR_ENABLED: ${MEMORY_EXTRACTION_OCR_ENABLED:-true}
  MEMORY_EXTRACTION_EXTERNAL_AI_ENABLED: ${MEMORY_EXTRACTION_EXTERNAL_AI_ENABLED:-false}
  MEMORY_EXTRACTION_DEFAULT_PROFILE: ${MEMORY_EXTRACTION_DEFAULT_PROFILE:-standard_local}
  MEMORY_LEGACY_CLIENT_ENABLED: ${MEMORY_LEGACY_CLIENT_ENABLED:-false}
  MEMORY_DEFAULT_SPACE_SLUG: ${MEMORY_DEFAULT_SPACE_SLUG:-team}
  MEMORY_DEFAULT_MEMORY_SCOPE_EXTERNAL_REF: ${MEMORY_DEFAULT_MEMORY_SCOPE_EXTERNAL_REF:-default}
  MEMORY_QDRANT_ENABLED: ${MEMORY_QDRANT_ENABLED:-false}
  MEMORY_QDRANT_URL: http://infinity_context_qdrant:6333
  MEMORY_QDRANT_COLLECTION: ${MEMORY_QDRANT_COLLECTION:-infinity_context_chunks_v1}
  MEMORY_QDRANT_HYBRID_SPARSE_ENABLED: ${MEMORY_QDRANT_HYBRID_SPARSE_ENABLED:-false}
  MEMORY_EMBEDDINGS_ENABLED: ${MEMORY_EMBEDDINGS_ENABLED:-false}
  MEMORY_EMBEDDINGS_PROVIDER: ${MEMORY_EMBEDDINGS_PROVIDER:-noop}
  MEMORY_EMBEDDINGS_MODEL: ${MEMORY_EMBEDDINGS_MODEL:-text-embedding-3-small}
  MEMORY_EMBEDDINGS_DIMENSIONS: ${MEMORY_EMBEDDINGS_DIMENSIONS:-1536}
  MEMORY_EMBEDDINGS_MODEL_REVISION: ${MEMORY_EMBEDDINGS_MODEL_REVISION:-}
  MEMORY_EMBEDDINGS_RUNTIME_BUILD_REVISION: ${MEMORY_EMBEDDINGS_RUNTIME_BUILD_REVISION:-}
  MEMORY_EMBEDDINGS_RUNTIME_INFO_URL: ${MEMORY_EMBEDDINGS_RUNTIME_INFO_URL:-}
  MEMORY_EMBEDDINGS_BASE_URL: ${MEMORY_EMBEDDINGS_BASE_URL:-${OPENAI_BASE_URL:-}}
  MEMORY_OPENAI_API_KEY: ${MEMORY_OPENAI_API_KEY:-}
  OPENAI_API_KEY: ${OPENAI_API_KEY:-}
  MEMORY_GRAPHITI_ENABLED: ${MEMORY_GRAPHITI_ENABLED:-false}
  MEMORY_GRAPHITI_NEO4J_URI: bolt://infinity_context_neo4j:7687
  OPENAI_BASE_URL: ${OPENAI_BASE_URL:-}
  MEMORY_GRAPHITI_NEO4J_USER: ${MEMORY_GRAPHITI_NEO4J_USER:-neo4j}
  MEMORY_GRAPHITI_NEO4J_PASSWORD: ${MEMORY_GRAPHITI_NEO4J_PASSWORD:-}
  MEMORY_GRAPHITI_BUILD_INDICES: ${MEMORY_GRAPHITI_BUILD_INDICES:-false}
  MEMORY_PROVIDER_CIRCUIT_FAILURE_THRESHOLD: ${MEMORY_PROVIDER_CIRCUIT_FAILURE_THRESHOLD:-3}
  MEMORY_PROVIDER_CIRCUIT_RESET_AFTER_SECONDS: ${MEMORY_PROVIDER_CIRCUIT_RESET_AFTER_SECONDS:-60}

x-infinity-context-service: &infinity-context-service
  <<: *infinity-context-image
  restart: unless-stopped
  environment:
    <<: *infinity-context-env
  volumes:
    - infinity_context_assets:/var/lib/infinity-context/assets
  depends_on:
    infinity_context_postgres:
      condition: service_healthy

x-infinity-context-healthcheck: &infinity-context-healthcheck
  test:
    [
      "CMD",
      "python",
      "-c",
      "import urllib.request; urllib.request.urlopen('http://127.0.0.1:7788/v1/health', timeout=2)",
    ]
  interval: 10s
  timeout: 3s
  retries: 30
  start_period: 20s

services:
  infinity_context_postgres:
    image: postgres:16
    restart: unless-stopped
    environment:
      POSTGRES_USER: infinity_context
      POSTGRES_PASSWORD: ${MEMORY_POSTGRES_PASSWORD:?Set MEMORY_POSTGRES_PASSWORD in .env.selfhost}
      POSTGRES_DB: infinity_context
    volumes:
      - infinity_context_postgres_data:/var/lib/postgresql/data
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U infinity_context -d infinity_context"]
      interval: 5s
      timeout: 3s
      retries: 30

  infinity_context_qdrant:
    image: qdrant/qdrant:v1.18.0
    profiles: ["full"]
    restart: unless-stopped
    volumes:
      - infinity_context_qdrant_data:/qdrant/storage

  infinity_context_neo4j:
    image: neo4j:5.26-community
    profiles: ["full"]
    restart: unless-stopped
    environment:
      NEO4J_AUTH: ${MEMORY_GRAPHITI_NEO4J_USER:-neo4j}/${MEMORY_GRAPHITI_NEO4J_PASSWORD:-infinitycontextgraph}
    volumes:
      - infinity_context_neo4j_data:/data

  infinity_context_migrate:
    <<: *infinity-context-service
    restart: "no"
    command: >
      sh -c "python -m infinity_context_server.db upgrade
      && python -m infinity_context_server.admin seed-defaults"

  infinity_context_server:
    <<: *infinity-context-service
    command: python -m infinity_context_server.main
    ports:
      - "${MEMORY_SERVER_PORT:-7788}:7788"
    depends_on:
      infinity_context_postgres:
        condition: service_healthy
      infinity_context_migrate:
        condition: service_completed_successfully
    healthcheck: *infinity-context-healthcheck

  infinity_context_projection_worker:
    <<: *infinity-context-service
    command: >
      python -m infinity_context_server.worker
      --loop
      --role projection
      --limit ${MEMORY_PROJECTION_WORKER_LIMIT:-25}
      --sleep-seconds ${MEMORY_PROJECTION_WORKER_SLEEP_SECONDS:-2}
    depends_on:
      infinity_context_server:
        condition: service_healthy

  infinity_context_extraction_worker:
    <<: *infinity-context-service
    command: >
      python -m infinity_context_server.worker
      --loop
      --role extraction
      --limit ${MEMORY_EXTRACTION_WORKER_LIMIT:-2}
      --concurrency ${MEMORY_EXTRACTION_WORKER_CONCURRENCY:-1}
      --sleep-seconds ${MEMORY_EXTRACTION_WORKER_SLEEP_SECONDS:-2}
    depends_on:
      infinity_context_server:
        condition: service_healthy

volumes:
  infinity_context_postgres_data:
  infinity_context_assets:
  infinity_context_qdrant_data:
  infinity_context_neo4j_data: