<DataGridPremium
  columns={columns}
  unstable_dataSource={dataSource}
  pagination
  initialState={initialStateWithPagination}
  pageSizeOptions={[10, 20, 50]}
  slots={{
    toolbar: ToolbarWithPromptInput,
  }}
/>