ALTER TABLE notifications
  ADD COLUMN IF NOT EXISTS context_id UUID,
  ADD COLUMN IF NOT EXISTS context_type VARCHAR(50);

CREATE INDEX IF NOT EXISTS idx_notifications_context ON notifications(context_type, context_id);