@mui/x-data-grid-premium 7.22.3 → 8.0.0-alpha.1

package/components/promptControl/GridToolbarPromptControl.js

@@ -0,0 +1,209 @@
+"use strict";
+
+var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.GridToolbarPromptControl = GridToolbarPromptControl;
+var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
+var React = _interopRequireWildcard(require("react"));
+var _styles = require("@mui/material/styles");
+var _composeClasses = _interopRequireDefault(require("@mui/utils/composeClasses"));
+var _useEventCallback = _interopRequireDefault(require("@mui/utils/useEventCallback"));
+var _xDataGrid = require("@mui/x-data-grid");
+var _internals = require("@mui/x-data-grid/internals");
+var _useGridApiContext = require("../../hooks/utils/useGridApiContext");
+var _useGridRootProps = require("../../hooks/utils/useGridRootProps");
+var _RecordButton = require("./RecordButton");
+var _jsxRuntime = require("react/jsx-runtime");
+const supportsSpeechRecognition = !!_RecordButton.BrowserSpeechRecognition;
+const useUtilityClasses = (ownerState, recording) => {
+  const {
+    classes
+  } = ownerState;
+  const slots = {
+    root: ['toolbarPromptControl', recording && 'toolbarPromptControl--recording'],
+    recordingIndicator: ['toolbarPromptControlRecordingIndicator'],
+    recordButton: ['toolbarPromptControlRecordButton'],
+    sendButton: ['toolbarPromptControlSendButton']
+  };
+  return (0, _composeClasses.default)(slots, _xDataGrid.getDataGridUtilityClass, classes);
+};
+const GridToolbarPromptControlRoot = (0, _styles.styled)('div', {
+  name: 'MuiDataGrid',
+  slot: 'ToolbarPromptControl',
+  overridesResolver: (_, styles) => styles.toolbarPromptControl
+})({
+  flex: 1,
+  display: 'flex',
+  flexDirection: 'row'
+});
+function sampleData(apiRef) {
+  const columns = (0, _xDataGrid.gridColumnDefinitionsSelector)(apiRef);
+  const rows = Object.values((0, _xDataGrid.gridRowsLookupSelector)(apiRef));
+  const columnExamples = {};
+  columns.forEach(column => {
+    columnExamples[column.field] = Array.from({
+      length: Math.min(5, rows.length)
+    }).map(() => {
+      const row = rows[Math.floor(Math.random() * rows.length)];
+      if (column.valueGetter) {
+        return column.valueGetter(row[column.field], row, column, apiRef);
+      }
+      return row[column.field];
+    });
+  });
+  return columnExamples;
+}
+function generateContext(apiRef, examples) {
+  const columns = (0, _xDataGrid.gridColumnDefinitionsSelector)(apiRef);
+  const columnsContext = columns.map(column => ({
+    field: column.field,
+    description: column.description ?? null,
+    examples: examples?.[column.field] ?? column.unstable_examples ?? [],
+    type: column.type ?? 'string',
+    allowedOperators: column.filterOperators?.map(operator => operator.value) ?? []
+  }));
+  return `The columns are described by the following JSON:\n${JSON.stringify(columnsContext)}`;
+}
+function GridToolbarPromptControl(props) {
+  const apiRef = (0, _useGridApiContext.useGridApiContext)();
+  const rootProps = (0, _useGridRootProps.useGridRootProps)();
+  const {
+    onPrompt,
+    lang,
+    allowDataSampling = false
+  } = props;
+  const [isLoading, setLoading] = React.useState(false);
+  const [error, setError] = React.useState(null);
+  const [isRecording, setRecording] = React.useState(false);
+  const [query, setQuery] = React.useState('');
+  const classes = useUtilityClasses(rootProps, isRecording);
+  const examplesFromData = React.useMemo(() => allowDataSampling ? sampleData(apiRef) : undefined, [apiRef, allowDataSampling]);
+  const processPrompt = React.useCallback(() => {
+    const context = generateContext(apiRef, examplesFromData);
+    const columnsByField = (0, _xDataGrid.gridColumnLookupSelector)(apiRef);
+    setLoading(true);
+    setError(null);
+    apiRef.current.setLoading(true);
+    onPrompt(context, query).then(result => {
+      const interestColumns = [];
+      apiRef.current.setFilterModel({
+        items: result.filters.map((filter, index) => {
+          const item = {
+            id: index,
+            field: filter.column,
+            operator: filter.operator,
+            value: filter.value
+          };
+          const column = columnsByField[filter.column];
+          if (column.type === 'singleSelect') {
+            const options = (0, _internals.getValueOptions)(column) ?? [];
+            const found = options.find(option => typeof option === 'object' && option.label === filter.value);
+            if (found) {
+              item.value = found.value;
+            }
+          }
+          return item;
+        }),
+        logicOperator: result.filterOperator ?? _xDataGrid.GridLogicOperator.And,
+        quickFilterValues: []
+      });
+      apiRef.current.setRowGroupingModel(result.grouping.map(g => g.column));
+      apiRef.current.setAggregationModel(result.aggregation);
+      apiRef.current.setSortModel(result.sorting.map(s => ({
+        field: s.column,
+        sort: s.direction
+      })));
+      const rows = (0, _internals.getVisibleRows)(apiRef, rootProps);
+      const selectedRowIds = result.select === -1 ? [] : rows.rows.slice(0, result.select).map(r => {
+        return apiRef.current.getRowId(r);
+      });
+      apiRef.current.setRowSelectionModel(selectedRowIds);
+      const columns = apiRef.current.getAllColumns();
+      const targetIndex = Number(columns.find(c => c.field === _xDataGrid.GRID_CHECKBOX_SELECTION_FIELD) !== undefined) + Number(result.grouping.length);
+      interestColumns.push(...Object.keys(result.aggregation));
+      interestColumns.push(...result.filters.map(f => f.column));
+      interestColumns.reverse().forEach(c => apiRef.current.setColumnIndex(c, targetIndex));
+    }).catch(_ => {
+      setError(apiRef.current.getLocaleText('toolbarPromptControlErrorMessage'));
+    }).finally(() => {
+      setLoading(false);
+      apiRef.current.setState(state => (0, _extends2.default)({}, state, {
+        rows: (0, _extends2.default)({}, state.rows, {
+          loading: false
+        })
+      }));
+    });
+  }, [apiRef, rootProps, onPrompt, examplesFromData, query]);
+  const handleChange = (0, _useEventCallback.default)(event => {
+    setQuery(event.target.value);
+  });
+  const handleKeyDown = (0, _useEventCallback.default)(event => {
+    if (event.code === 'Enter') {
+      processPrompt();
+    }
+  });
+  const handleDone = (0, _useEventCallback.default)(value => {
+    setQuery(value);
+    if (value) {
+      processPrompt();
+    }
+  });
+  const placeholder = supportsSpeechRecognition ? apiRef.current.getLocaleText('toolbarPromptControlWithRecordingPlaceholder') : apiRef.current.getLocaleText('toolbarPromptControlPlaceholder');
+  return /*#__PURE__*/(0, _jsxRuntime.jsx)(GridToolbarPromptControlRoot, {
+    ownerState: rootProps,
+    className: classes.root,
+    children: /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.baseTextField, {
+      variant: "outlined",
+      placeholder: isRecording ? apiRef.current.getLocaleText('toolbarPromptControlRecordingPlaceholder') : placeholder,
+      "aria-label": apiRef.current.getLocaleText('toolbarPromptControlLabel'),
+      disabled: isLoading,
+      value: query,
+      style: {
+        flex: 1
+      },
+      onChange: handleChange,
+      size: "small",
+      onKeyDown: handleKeyDown,
+      error: !!error,
+      helperText: error,
+      InputProps: {
+        startAdornment: supportsSpeechRecognition && /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.baseInputAdornment, {
+          position: "start",
+          children: /*#__PURE__*/(0, _jsxRuntime.jsx)(_RecordButton.RecordButton, {
+            className: classes.recordButton,
+            lang: lang,
+            recording: isRecording,
+            setRecording: setRecording,
+            disabled: isLoading,
+            onUpdate: setQuery,
+            onDone: handleDone,
+            onError: setError
+          })
+        }),
+        endAdornment: /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.baseInputAdornment, {
+          position: "end",
+          children: /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.baseTooltip, {
+            title: apiRef.current.getLocaleText('toolbarPromptControlSendActionLabel'),
+            children: /*#__PURE__*/(0, _jsxRuntime.jsx)("div", {
+              children: /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.baseIconButton, {
+                className: classes.sendButton,
+                disabled: isLoading || isRecording || query === '',
+                color: "primary",
+                onClick: processPrompt,
+                size: "small",
+                "aria-label": apiRef.current.getLocaleText('toolbarPromptControlSendActionAriaLabel'),
+                edge: "end",
+                children: /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.toolbarPromptSendIcon, {
+                  fontSize: "small"
+                })
+              })
+            })
+          })
+        })
+      }
+    })
+  });
+}

package/components/promptControl/RecordButton.d.ts

@@ -0,0 +1,16 @@
+import * as React from 'react';
+export declare const BrowserSpeechRecognition: any;
+type SpeechRecognitionOptions = {
+    onUpdate: (value: string) => void;
+    onDone: (value: string) => void;
+    onError: (error: string) => void;
+};
+interface RecordButtonProps extends SpeechRecognitionOptions {
+    disabled: boolean;
+    lang?: string;
+    recording: boolean;
+    setRecording: (value: boolean) => void;
+    className: string;
+}
+declare function RecordButton(props: RecordButtonProps): React.JSX.Element;
+export { RecordButton };

package/components/promptControl/RecordButton.js

@@ -0,0 +1,119 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
+var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.BrowserSpeechRecognition = void 0;
+exports.RecordButton = RecordButton;
+var React = _interopRequireWildcard(require("react"));
+var _useTimeout = require("@mui/utils/useTimeout");
+var _useLazyRef = _interopRequireDefault(require("@mui/utils/useLazyRef"));
+var _useGridApiContext = require("../../hooks/utils/useGridApiContext");
+var _useGridRootProps = require("../../hooks/utils/useGridRootProps");
+var _jsxRuntime = require("react/jsx-runtime");
+const BrowserSpeechRecognition = exports.BrowserSpeechRecognition = globalThis.SpeechRecognition || globalThis.webkitSpeechRecognition;
+function RecordButton(props) {
+  const apiRef = (0, _useGridApiContext.useGridApiContext)();
+  const rootProps = (0, _useGridRootProps.useGridRootProps)();
+  const {
+    lang,
+    recording,
+    setRecording,
+    disabled,
+    className,
+    onDone,
+    onUpdate,
+    onError
+  } = props;
+  const buttonRef = React.useRef(null);
+  const recognition = (0, _useLazyRef.default)(() => {
+    if (!BrowserSpeechRecognition) {
+      return {
+        start: () => {},
+        abort: () => {}
+      };
+    }
+    const timeout = new _useTimeout.Timeout();
+    const instance = new BrowserSpeechRecognition();
+    instance.continuous = true;
+    instance.interimResults = true;
+    instance.lang = lang;
+    let finalResult = '';
+    let interimResult = '';
+    function start(options) {
+      if (recording) {
+        return;
+      }
+      setRecording(true);
+      instance.onresult = event => {
+        finalResult = '';
+        interimResult = '';
+        if (typeof event.results === 'undefined') {
+          instance.stop();
+          return;
+        }
+        for (let i = event.resultIndex; i < event.results.length; i += 1) {
+          if (event.results[i].isFinal) {
+            finalResult += event.results[i][0].transcript;
+          } else {
+            interimResult += event.results[i][0].transcript;
+          }
+        }
+        if (finalResult === '') {
+          options.onUpdate(interimResult);
+        }
+        timeout.start(1000, () => instance.stop());
+      };
+      instance.onsoundend = () => {
+        instance.stop();
+      };
+      instance.onend = () => {
+        options.onDone(finalResult);
+        setRecording(false);
+      };
+      instance.onerror = error => {
+        options.onError(error.message);
+        instance.stop();
+        setRecording(false);
+      };
+      instance.start();
+    }
+    function abort() {
+      instance.abort();
+    }
+    return {
+      start,
+      abort
+    };
+  }).current;
+  const handleClick = () => {
+    if (!recording) {
+      recognition.start({
+        onDone,
+        onUpdate,
+        onError
+      });
+      return;
+    }
+    recognition.abort();
+  };
+  return /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.baseTooltip, {
+    title: recording ? apiRef.current.getLocaleText('toolbarPromptControlRecordButtonActiveLabel') : apiRef.current.getLocaleText('toolbarPromptControlRecordButtonDefaultLabel'),
+    children: /*#__PURE__*/(0, _jsxRuntime.jsx)("div", {
+      children: /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.baseIconButton, {
+        color: recording ? 'primary' : 'default',
+        className: className,
+        disabled: disabled,
+        onClick: handleClick,
+        ref: buttonRef,
+        size: "small",
+        edge: "start",
+        children: /*#__PURE__*/(0, _jsxRuntime.jsx)(rootProps.slots.toolbarPromptRecordIcon, {
+          fontSize: "small"
+        })
+      })
+    })
+  });
+}

package/components/promptControl/index.d.ts

@@ -0,0 +1 @@
+export { GridToolbarPromptControl as Unstable_GridToolbarPromptControl } from './GridToolbarPromptControl';

package/components/promptControl/index.js

@@ -0,0 +1,12 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+Object.defineProperty(exports, "Unstable_GridToolbarPromptControl", {
+  enumerable: true,
+  get: function () {
+    return _GridToolbarPromptControl.GridToolbarPromptControl;
+  }
+});
+var _GridToolbarPromptControl = require("./GridToolbarPromptControl");

package/esm/components/GridColumnMenuAggregationItem.js

@@ -4,13 +4,11 @@ import _toPropertyKey from "@babel/runtime/helpers/esm/toPropertyKey";
 import * as React from 'react';
 import PropTypes from 'prop-types';
 import { useGridSelector } from '@mui/x-data-grid-pro';
-import MenuItem from '@mui/material/MenuItem';
 import ListItemIcon from '@mui/material/ListItemIcon';
 import ListItemText from '@mui/material/ListItemText';
 import FormControl from '@mui/material/FormControl';
 import InputLabel from '@mui/material/InputLabel';
 import { unstable_useId as useId } from '@mui/utils';
-import Select from '@mui/material/Select';
 import { useGridApiContext } from "../hooks/utils/useGridApiContext.js";
 import { useGridRootProps } from "../hooks/utils/useGridRootProps.js";
 import { canColumnHaveAggregationFunction, getAggregationFunctionLabel, getAvailableAggregationFunctions } from "../hooks/features/aggregation/gridAggregationUtils.js";
@@ -54,7 +52,7 @@ function GridColumnMenuAggregationItem(props) {
     apiRef.current.hideColumnMenu();
   };
   const label = apiRef.current.getLocaleText('aggregationMenuItemHeader');
-  return /*#__PURE__*/_jsxs(MenuItem, {
+  return /*#__PURE__*/_jsxs(rootProps.slots.baseMenuItem, {
     disableRipple: true,
     children: [/*#__PURE__*/_jsx(ListItemIcon, {
       children: /*#__PURE__*/_jsx(rootProps.slots.columnMenuAggregationIcon, {
@@ -70,7 +68,7 @@ function GridColumnMenuAggregationItem(props) {
         children: [/*#__PURE__*/_jsx(InputLabel, {
           id: `${id}-label`,
           children: label
-        }), /*#__PURE__*/_jsxs(Select, {
+        }), /*#__PURE__*/_jsxs(rootProps.slots.baseSelect, {
           labelId: `${id}-label`,
           id: `${id}-input`,
           value: selectedAggregationRule,
@@ -79,10 +77,10 @@ function GridColumnMenuAggregationItem(props) {
           onChange: handleAggregationItemChange,
           onBlur: event => event.stopPropagation(),
           fullWidth: true,
-          children: [/*#__PURE__*/_jsx(MenuItem, {
+          children: [/*#__PURE__*/_jsx(rootProps.slots.baseMenuItem, {
             value: "",
             children: "..."
-          }), availableAggregationFunctions.map(aggFunc => /*#__PURE__*/_jsx(MenuItem, {
+          }), availableAggregationFunctions.map(aggFunc => /*#__PURE__*/_jsx(rootProps.slots.baseMenuItem, {
             value: aggFunc,
             children: getAggregationFunctionLabel({
               apiRef,

package/esm/components/GridColumnMenuRowGroupItem.js

@@ -1,5 +1,4 @@
 import * as React from 'react';
-import MenuItem from '@mui/material/MenuItem';
 import ListItemIcon from '@mui/material/ListItemIcon';
 import ListItemText from '@mui/material/ListItemText';
 import { useGridSelector, gridColumnLookupSelector } from '@mui/x-data-grid-pro';
@@ -24,7 +23,7 @@ export function GridColumnMenuRowGroupItem(props) {
     };
     const groupedColumn = columnsLookup[field];
     const name = groupedColumn.headerName ?? field;
-    return /*#__PURE__*/_jsxs(MenuItem, {
+    return /*#__PURE__*/_jsxs(rootProps.slots.baseMenuItem, {
       onClick: ungroupColumn,
       disabled: !groupedColumn.groupable,
       children: [/*#__PURE__*/_jsx(ListItemIcon, {

package/esm/components/GridColumnMenuRowUngroupItem.js

@@ -1,5 +1,4 @@
 import * as React from 'react';
-import MenuItem from '@mui/material/MenuItem';
 import ListItemIcon from '@mui/material/ListItemIcon';
 import ListItemText from '@mui/material/ListItemText';
 import { gridColumnLookupSelector, useGridSelector } from '@mui/x-data-grid-pro';
@@ -29,7 +28,7 @@ export function GridColumnMenuRowUngroupItem(props) {
   };
   const name = columnsLookup[colDef.field].headerName ?? colDef.field;
   if (rowGroupingModel.includes(colDef.field)) {
-    return /*#__PURE__*/_jsxs(MenuItem, {
+    return /*#__PURE__*/_jsxs(rootProps.slots.baseMenuItem, {
       onClick: ungroupColumn,
       children: [/*#__PURE__*/_jsx(ListItemIcon, {
         children: /*#__PURE__*/_jsx(rootProps.slots.columnMenuUngroupIcon, {
@@ -40,7 +39,7 @@ export function GridColumnMenuRowUngroupItem(props) {
       })]
     });
   }
-  return /*#__PURE__*/_jsxs(MenuItem, {
+  return /*#__PURE__*/_jsxs(rootProps.slots.baseMenuItem, {
     onClick: groupColumn,
     children: [/*#__PURE__*/_jsx(ListItemIcon, {
       children: /*#__PURE__*/_jsx(rootProps.slots.columnMenuGroupIcon, {

package/esm/components/GridExcelExportMenuItem.js

@@ -3,17 +3,18 @@ import _objectWithoutPropertiesLoose from "@babel/runtime/helpers/esm/objectWith
 const _excluded = ["hideMenu", "options"];
 import * as React from 'react';
 import PropTypes from 'prop-types';
-import MenuItem from '@mui/material/MenuItem';
 import { useGridApiContext } from "../hooks/utils/useGridApiContext.js";
+import { useGridRootProps } from "../hooks/utils/useGridRootProps.js";
 import { jsx as _jsx } from "react/jsx-runtime";
 function GridExcelExportMenuItem(props) {
   const apiRef = useGridApiContext();
+  const rootProps = useGridRootProps();
   const {
       hideMenu,
       options
     } = props,
     other = _objectWithoutPropertiesLoose(props, _excluded);
-  return /*#__PURE__*/_jsx(MenuItem, _extends({
+  return /*#__PURE__*/_jsx(rootProps.slots.baseMenuItem, _extends({
     onClick: () => {
       apiRef.current.exportDataAsExcel(options);
       hideMenu?.();

package/esm/components/index.js

@@ -1,4 +1,5 @@
 export * from "./GridExcelExportMenuItem.js";
 export * from "../material/icons.js";
 export * from "./GridColumnMenuAggregationItem.js";
+export * from "./promptControl/index.js";
 export { GridColumnMenuGroupingItem } from "./GridPremiumColumnMenu.js";

package/esm/components/promptControl/GridToolbarPromptControl.js

@@ -0,0 +1,202 @@
+import _extends from "@babel/runtime/helpers/esm/extends";
+import * as React from 'react';
+import { styled } from '@mui/material/styles';
+import composeClasses from '@mui/utils/composeClasses';
+import useEventCallback from '@mui/utils/useEventCallback';
+import { getDataGridUtilityClass, GRID_CHECKBOX_SELECTION_FIELD, gridColumnDefinitionsSelector, gridColumnLookupSelector, GridLogicOperator, gridRowsLookupSelector } from '@mui/x-data-grid';
+import { getValueOptions, getVisibleRows } from '@mui/x-data-grid/internals';
+import { useGridApiContext } from "../../hooks/utils/useGridApiContext.js";
+import { useGridRootProps } from "../../hooks/utils/useGridRootProps.js";
+import { RecordButton, BrowserSpeechRecognition } from "./RecordButton.js";
+import { jsx as _jsx } from "react/jsx-runtime";
+const supportsSpeechRecognition = !!BrowserSpeechRecognition;
+const useUtilityClasses = (ownerState, recording) => {
+  const {
+    classes
+  } = ownerState;
+  const slots = {
+    root: ['toolbarPromptControl', recording && 'toolbarPromptControl--recording'],
+    recordingIndicator: ['toolbarPromptControlRecordingIndicator'],
+    recordButton: ['toolbarPromptControlRecordButton'],
+    sendButton: ['toolbarPromptControlSendButton']
+  };
+  return composeClasses(slots, getDataGridUtilityClass, classes);
+};
+const GridToolbarPromptControlRoot = styled('div', {
+  name: 'MuiDataGrid',
+  slot: 'ToolbarPromptControl',
+  overridesResolver: (_, styles) => styles.toolbarPromptControl
+})({
+  flex: 1,
+  display: 'flex',
+  flexDirection: 'row'
+});
+function sampleData(apiRef) {
+  const columns = gridColumnDefinitionsSelector(apiRef);
+  const rows = Object.values(gridRowsLookupSelector(apiRef));
+  const columnExamples = {};
+  columns.forEach(column => {
+    columnExamples[column.field] = Array.from({
+      length: Math.min(5, rows.length)
+    }).map(() => {
+      const row = rows[Math.floor(Math.random() * rows.length)];
+      if (column.valueGetter) {
+        return column.valueGetter(row[column.field], row, column, apiRef);
+      }
+      return row[column.field];
+    });
+  });
+  return columnExamples;
+}
+function generateContext(apiRef, examples) {
+  const columns = gridColumnDefinitionsSelector(apiRef);
+  const columnsContext = columns.map(column => ({
+    field: column.field,
+    description: column.description ?? null,
+    examples: examples?.[column.field] ?? column.unstable_examples ?? [],
+    type: column.type ?? 'string',
+    allowedOperators: column.filterOperators?.map(operator => operator.value) ?? []
+  }));
+  return `The columns are described by the following JSON:\n${JSON.stringify(columnsContext)}`;
+}
+function GridToolbarPromptControl(props) {
+  const apiRef = useGridApiContext();
+  const rootProps = useGridRootProps();
+  const {
+    onPrompt,
+    lang,
+    allowDataSampling = false
+  } = props;
+  const [isLoading, setLoading] = React.useState(false);
+  const [error, setError] = React.useState(null);
+  const [isRecording, setRecording] = React.useState(false);
+  const [query, setQuery] = React.useState('');
+  const classes = useUtilityClasses(rootProps, isRecording);
+  const examplesFromData = React.useMemo(() => allowDataSampling ? sampleData(apiRef) : undefined, [apiRef, allowDataSampling]);
+  const processPrompt = React.useCallback(() => {
+    const context = generateContext(apiRef, examplesFromData);
+    const columnsByField = gridColumnLookupSelector(apiRef);
+    setLoading(true);
+    setError(null);
+    apiRef.current.setLoading(true);
+    onPrompt(context, query).then(result => {
+      const interestColumns = [];
+      apiRef.current.setFilterModel({
+        items: result.filters.map((filter, index) => {
+          const item = {
+            id: index,
+            field: filter.column,
+            operator: filter.operator,
+            value: filter.value
+          };
+          const column = columnsByField[filter.column];
+          if (column.type === 'singleSelect') {
+            const options = getValueOptions(column) ?? [];
+            const found = options.find(option => typeof option === 'object' && option.label === filter.value);
+            if (found) {
+              item.value = found.value;
+            }
+          }
+          return item;
+        }),
+        logicOperator: result.filterOperator ?? GridLogicOperator.And,
+        quickFilterValues: []
+      });
+      apiRef.current.setRowGroupingModel(result.grouping.map(g => g.column));
+      apiRef.current.setAggregationModel(result.aggregation);
+      apiRef.current.setSortModel(result.sorting.map(s => ({
+        field: s.column,
+        sort: s.direction
+      })));
+      const rows = getVisibleRows(apiRef, rootProps);
+      const selectedRowIds = result.select === -1 ? [] : rows.rows.slice(0, result.select).map(r => {
+        return apiRef.current.getRowId(r);
+      });
+      apiRef.current.setRowSelectionModel(selectedRowIds);
+      const columns = apiRef.current.getAllColumns();
+      const targetIndex = Number(columns.find(c => c.field === GRID_CHECKBOX_SELECTION_FIELD) !== undefined) + Number(result.grouping.length);
+      interestColumns.push(...Object.keys(result.aggregation));
+      interestColumns.push(...result.filters.map(f => f.column));
+      interestColumns.reverse().forEach(c => apiRef.current.setColumnIndex(c, targetIndex));
+    }).catch(_ => {
+      setError(apiRef.current.getLocaleText('toolbarPromptControlErrorMessage'));
+    }).finally(() => {
+      setLoading(false);
+      apiRef.current.setState(state => _extends({}, state, {
+        rows: _extends({}, state.rows, {
+          loading: false
+        })
+      }));
+    });
+  }, [apiRef, rootProps, onPrompt, examplesFromData, query]);
+  const handleChange = useEventCallback(event => {
+    setQuery(event.target.value);
+  });
+  const handleKeyDown = useEventCallback(event => {
+    if (event.code === 'Enter') {
+      processPrompt();
+    }
+  });
+  const handleDone = useEventCallback(value => {
+    setQuery(value);
+    if (value) {
+      processPrompt();
+    }
+  });
+  const placeholder = supportsSpeechRecognition ? apiRef.current.getLocaleText('toolbarPromptControlWithRecordingPlaceholder') : apiRef.current.getLocaleText('toolbarPromptControlPlaceholder');
+  return /*#__PURE__*/_jsx(GridToolbarPromptControlRoot, {
+    ownerState: rootProps,
+    className: classes.root,
+    children: /*#__PURE__*/_jsx(rootProps.slots.baseTextField, {
+      variant: "outlined",
+      placeholder: isRecording ? apiRef.current.getLocaleText('toolbarPromptControlRecordingPlaceholder') : placeholder,
+      "aria-label": apiRef.current.getLocaleText('toolbarPromptControlLabel'),
+      disabled: isLoading,
+      value: query,
+      style: {
+        flex: 1
+      },
+      onChange: handleChange,
+      size: "small",
+      onKeyDown: handleKeyDown,
+      error: !!error,
+      helperText: error,
+      InputProps: {
+        startAdornment: supportsSpeechRecognition && /*#__PURE__*/_jsx(rootProps.slots.baseInputAdornment, {
+          position: "start",
+          children: /*#__PURE__*/_jsx(RecordButton, {
+            className: classes.recordButton,
+            lang: lang,
+            recording: isRecording,
+            setRecording: setRecording,
+            disabled: isLoading,
+            onUpdate: setQuery,
+            onDone: handleDone,
+            onError: setError
+          })
+        }),
+        endAdornment: /*#__PURE__*/_jsx(rootProps.slots.baseInputAdornment, {
+          position: "end",
+          children: /*#__PURE__*/_jsx(rootProps.slots.baseTooltip, {
+            title: apiRef.current.getLocaleText('toolbarPromptControlSendActionLabel'),
+            children: /*#__PURE__*/_jsx("div", {
+              children: /*#__PURE__*/_jsx(rootProps.slots.baseIconButton, {
+                className: classes.sendButton,
+                disabled: isLoading || isRecording || query === '',
+                color: "primary",
+                onClick: processPrompt,
+                size: "small",
+                "aria-label": apiRef.current.getLocaleText('toolbarPromptControlSendActionAriaLabel'),
+                edge: "end",
+                children: /*#__PURE__*/_jsx(rootProps.slots.toolbarPromptSendIcon, {
+                  fontSize: "small"
+                })
+              })
+            })
+          })
+        })
+      }
+    })
+  });
+}
+export { GridToolbarPromptControl };