@quillsql/react 2.11.9 → 2.11.13

package/dist/esm/ReportBuilder.js

@@ -1,17 +1,17 @@
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
-import { useContext, useEffect, useRef, useState } from 'react';
+import { useContext, useEffect, useRef, useState, } from 'react';
 import { MemoizedButton, MemoizedCheckbox, MemoizedHeader, MemoizedLabel, MemoizedPopover, MemoizedSecondaryButton, MemoizedText, } from './components/UiComponents';
 import { DndContext, closestCenter, KeyboardSensor, PointerSensor, useSensor, useSensors, } from '@dnd-kit/core';
 import { arrayMove, SortableContext, sortableKeyboardCoordinates, verticalListSortingStrategy, useSortable, } from '@dnd-kit/sortable';
 import { CSS as DND_CSS } from '@dnd-kit/utilities';
 import { getQuarter } from 'date-fns';
 import { ClientContext, ThemeContext } from './Context';
-import { getTableNames, isDateishColumnType, isNumericColumnType, isTextColumnType, } from './components/ReportBuilder/ast';
+import { getTableAliases, getTableNames, isDateishColumnType, isNumericColumnType, isTextColumnType, } from './components/ReportBuilder/ast';
 import { ChartBuilderWithModal } from './ChartBuilder';
-import { QuillPopover, QuillSelect, QuillTabs, QuillTextInput, QuillSidebar, CustomContainer, QuillHandleButton, QuillSelectColumn, QuillDraggableColumn, QuillButtonLoadingState, QuillTableLoadingState, QuillSidebarHeading, QuillSidebarSubHeading, QuillFilterPopover, QuillSortPopover, DEFAULT_TAB_OPTIONS, TagWrapper, EditPopover, AddFilterPopover, } from './components/ReportBuilder/ui';
+import { QuillPopover, QuillTabs, QuillTextInput, QuillSidebar, CustomContainer, QuillHandleButton, QuillSelectColumn, QuillDraggableColumn, QuillButtonLoadingState, QuillTableLoadingState, QuillSidebarHeading, QuillSidebarSubHeading, QuillFilterPopover, QuillSortPopover, DEFAULT_TAB_OPTIONS, TagWrapper, EditPopover, AddFilterPopover, } from './components/ReportBuilder/ui';
 import { generateCurrentPeriodPostgres, generateEqualsPostgres, generateLastNPeriodsPostgres, generatePreviousPeriodPostgres, } from './components/ReportBuilder/postgres';
 import { convertBigQuery, convertGroupBy, convertRemoveSimpleParentheses, convertStringComparison, convertWildcardColumns, } from './components/ReportBuilder/convert';
-import { deepCopy, formatDateComparisonNode, getDateFilterInfo, getInTheCurrentIntervalSentence, getInTheLastIntervalSentence, getInThePreviousIntervalSentence, getPostgresBasicType, isColumnComparison, isDateTruncEquals, isInTheLastInterval, isNodeEmptyCollection, isTheCurrentInterval, isThePreviousInterval, isTopLevelBoolean, showNodeAsRow, tryConvertDateEquality, } from './components/ReportBuilder/util';
+import { deepCopy, formatDateComparisonNode, getDateFilterInfo, getInTheCurrentIntervalSentence, getInTheLastIntervalSentence, getInThePreviousIntervalSentence, getPostgresBasicType, isColumnComparison, isDateTruncEquals, isInTheLastInterval, isNodeEmptyCollection, isTheCurrentInterval, isThePreviousInterval, isTopLevelBoolean, showNodeAsRow, tryConvertDateEquality, removeNonSelectedTableReferences, } from './components/ReportBuilder/util';
 import { getDefaultOperatorSubtrees, OPERATOR_GROUPS, } from './components/ReportBuilder/operators';
 import { hashCode } from './utils/crypto';
 import { DATE_FMT, DAY_OF_WEEK, defaultAST, defaultColumn, defaultEntry, defaultNumericComparison, defaultTable, defaultVariant, MONTH_OF_YEAR, } from './components/ReportBuilder/constants';
@@ -20,6 +20,9 @@ import { AddSortPopover, SortSentence, } from './components/ReportBuilder/AddSor
 import { PivotModal, generatePivotTable, } from './internals/ReportBuilder/PivotModal';
 import { PivotCard } from './internals/ReportBuilder/PivotList';
 import QuillTable from './components/QuillTable';
+import { QuillSelectComponent } from './components/QuillSelect';
+import { snakeCaseToTitleCase } from './utils/textProcessing';
+import { AddLimitPopover, LimitSentence, } from './components/ReportBuilder/AddLimitPopover';
 /**
  * Quill Report Builder
  *
@@ -28,7 +31,7 @@ import QuillTable from './components/QuillTable';
  * they can click a button and add that report to their dashboard or export it
  * as a CSV.
  */
-export default function ReportBuilder({ initialTableName = '', onAddToDashboardComplete = () => void null, destinationDashboard = undefined, organizationName = '', Button = MemoizedButton, SecondaryButton = MemoizedSecondaryButton, TextInput = QuillTextInput, Select = QuillSelect, Table = QuillTable, Popover = QuillPopover, Tabs = QuillTabs, Checkbox = MemoizedCheckbox, Sidebar = QuillSidebar, Container = CustomContainer, HandleButton = QuillHandleButton, SelectColumn = QuillSelectColumn, DraggableColumn = QuillDraggableColumn, ButtonLoadingState = QuillButtonLoadingState, TableLoadingState = QuillTableLoadingState, SidebarHeading = QuillSidebarHeading, SidebarSubHeading = QuillSidebarSubHeading, FilterPopover = QuillFilterPopover, SortPopover = QuillSortPopover, Label = MemoizedLabel, Header = MemoizedHeader, Text = MemoizedText, PivotPopover = MemoizedPopover, admin = false, hideAi = false, }) {
+export default function ReportBuilder({ initialTableName = '', onAddToDashboardComplete = () => void null, destinationDashboard = undefined, organizationName = '', Button = MemoizedButton, SecondaryButton = MemoizedSecondaryButton, TextInput = QuillTextInput, Select = QuillSelectComponent, Table = QuillTable, Popover = QuillPopover, Tabs = QuillTabs, Checkbox = MemoizedCheckbox, Sidebar = QuillSidebar, Container = CustomContainer, HandleButton = QuillHandleButton, SelectColumn = QuillSelectColumn, DraggableColumn = QuillDraggableColumn, ButtonLoadingState = QuillButtonLoadingState, TableLoadingState = QuillTableLoadingState, SidebarHeading = QuillSidebarHeading, SidebarSubHeading = QuillSidebarSubHeading, FilterPopover = QuillFilterPopover, SortPopover = QuillSortPopover, Label = MemoizedLabel, Header = MemoizedHeader, Text = MemoizedText, PivotPopover = MemoizedPopover, admin = false, hideAi = false, containerStyle, }) {
     const [aiPrompt, setAiPrompt] = useState('');
     const [errorMessage, setErrorMessage] = useState('');
     const [baseAst, setBaseAst] = useState(null);
@@ -84,7 +87,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
         return columns.map((col) => ({
             label: col,
             name: col,
-            displayName: col,
+            displayName: snakeCaseToTitleCase(col),
             field: col,
             format: getPostgresBasicType(fields.find((f) => f.name === col))?.replace('number', 'whole_number') || 'string',
             fieldType: schemaTables
@@ -116,6 +119,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
             // setUniqueValues({});
             setPivot(null);
             setPivotData(null);
+            setRecommendedPivots([]);
         }, 0);
     };
     useEffect(() => {
@@ -243,14 +247,14 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                         'Content-Type': 'application/json',
                     },
                     body: JSON.stringify({
-                        orgId: 2,
+                        orgId: client.customerId,
                         publicKey: client.publicKey,
                         query: query,
                     }),
                 });
                 const data = await response.json();
                 if (data.errorMessage) {
-                    console.error(data.errorMessage);
+                    // console.error(data.errorMessage);
                     return null;
                 }
                 const options = data.rows.map((r) => r[column]);
@@ -267,19 +271,36 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
         };
         const fetchSchema = async () => {
             try {
-                const response = await fetch(`https://quill-344421.uc.r.appspot.com/schema2/${client.publicKey}`).then((res) => res.json());
+                const response = await fetch(`${client.queryEndpoint}`, {
+                    method: 'POST',
+                    headers: {
+                        ...client.queryHeaders,
+                        'Content-Type': 'application/json',
+                    },
+                    body: JSON.stringify({
+                        metadata: {
+                            clientId: client.publicKey,
+                            publicKey: client.publicKey,
+                            task: 'schema',
+                            removeCustomerField: true,
+                        },
+                    }),
+                });
+                const results = await response.json();
                 // Filter out hidden columns on tables back from schema2.
-                const tables = response?.tables;
-                for (const table of tables) {
-                    table.columns = table.columns.filter((column) => 
-                    // Quick and dirty fix for removing org ids from response.
-                    // TODO: Fix this on the backend or something.
-                    column.isVisible && column.displayName !== 'pm_company_id');
-                }
+                const tables = results.data.tables || results.data.data.tables;
                 setSchemaTables(tables ?? []);
-                setOrderedColumnNames((tables ?? [])
-                    // .filter((t: any) => t.displayName === initialTableName)
-                    .flatMap((table) => table.columns.map((c) => `${table.displayName}.${c.displayName}`)));
+                setOrderedColumnNames((tables ?? []).flatMap((table) => table.columns
+                    .map((c) => `${table.name}.${c.name}`)
+                    .sort((a, b) => {
+                    const aIsId = a.endsWith('.id') || a.endsWith('_id');
+                    const bIsId = b.endsWith('.id') || b.endsWith('_id');
+                    if (aIsId && !bIsId)
+                        return 1;
+                    if (bIsId && !aIsId)
+                        return -1;
+                    return 0;
+                })));
                 // Fetch all the unique values in parallel
                 const pendingFetches = [];
                 for (let table of tables ?? []) {
@@ -304,6 +325,12 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                 if (hashCode(uniqueValues) !== hashCode(newUniqueValues)) {
                     setUniqueValues(newUniqueValues);
                 }
+                if (initialTableName) {
+                    const columnsForTable = tables
+                        .find((t) => t.name === initialTableName)
+                        ?.columns.map((c) => c.name);
+                    await handleAsk(`get ${columnsForTable} from ${initialTableName}`);
+                }
             }
             catch (error) {
                 console.error(error);
@@ -606,6 +633,9 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
     };
     // Function to handle operator changes
     const handleOperatorChange = (value, node, keyPrefix, column = null) => {
+        if (!keyPrefix) {
+            setTopLevelBinaryOperator(value);
+        }
         if (isPending) {
             updateActiveItem([{ path: keyPrefix + 'operator', value }], { column });
         }
@@ -703,7 +733,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                             'node.right.args.value.1.column') ??
                         (node.right?.args?.value[1]?.value &&
                             'node.right.args.value.1.value');
-                    return (_jsxs("div", { style: { display: 'flex', gap: 20 }, children: [_jsx(Select, { value: dateColumn, onChange: (value) => {
+                    return (_jsxs("div", { style: { display: 'flex', gap: 20 }, children: [_jsx(Select, { theme: theme, value: dateColumn, onChange: (value) => {
                                     const columnType = getColumnTypeByName(value);
                                     if (isDateishColumnType(columnType)) {
                                         // handleChange(value, keyPrefix + dateColumnPath, "text");
@@ -720,9 +750,9 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                         handleReplaceSubtree(keyPrefix, newSubtree);
                                     }
                                 }, options: getAllPossibleColumns().map((column) => ({
-                                    label: column.displayName,
+                                    label: snakeCaseToTitleCase(column.displayName),
                                     value: column.name,
-                                })) }), _jsx(Select, { value: dateFilterType, onChange: (value) => {
+                                })) }), _jsx(Select, { theme: theme, value: dateFilterType, onChange: (value) => {
                                     if (value === dateFilterType)
                                         return null;
                                     let newSubtree = {};
@@ -768,7 +798,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                             path: keyPrefix + intervalPath,
                                         },
                                     ]));
-                                } })), _jsx(Select, { value: intervalType, onChange: (value) => {
+                                } })), _jsx(Select, { theme: theme, value: intervalType, onChange: (value) => {
                                     if (intervalPaths.length === 1) {
                                         handleChangeText([
                                             {
@@ -814,7 +844,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                 else if (isInTheLastInterval(node, client.databaseType)) {
                     const { dateColumn, dateFilterType, intervalCount, intervalType } = getDateFilterInfo(node);
                     const options = getAllPossibleColumns().map((column) => ({
-                        label: column.displayName,
+                        label: snakeCaseToTitleCase(column.displayName),
                         value: column.name,
                     }));
                     const plural = node.right.args.value[1].expr.value > 1 ? 's' : '';
@@ -823,7 +853,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                             flexDirection: 'row',
                             alignItems: 'center',
                             gap: 20,
-                        }, children: [_jsx("div", { children: _jsx(Select, { value: node.left.column, onChange: (value) => {
+                        }, children: [_jsx("div", { children: _jsx(Select, { theme: theme, value: node.left.column, onChange: (value) => {
                                         const columnType = getColumnTypeByName(value);
                                         if (isDateishColumnType(columnType)) {
                                             // handleChange(value, keyPrefix + dateColumnPath, "text");
@@ -839,7 +869,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                             newSubtree.left.args.value[0].column = value;
                                             handleReplaceSubtree(keyPrefix, newSubtree);
                                         }
-                                    }, options: options }) }), _jsx(Select, { value: dateFilterType, onChange: (value) => {
+                                    }, options: options }) }), _jsx(Select, { theme: theme, value: dateFilterType, onChange: (value) => {
                                     handleOperatorChange(value, node, keyPrefix, dateColumn);
                                 }, options: [
                                     { label: 'in the last', value: 'IN_THE_LAST' },
@@ -853,7 +883,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                             path: keyPrefix + 'right.args.value||1.expr.value',
                                         },
                                     ]);
-                                } }), _jsx("div", { children: _jsx(Select, { value: node.right.args.value[1].unit, onChange: (value) => handleChange([
+                                } }), _jsx("div", { children: _jsx(Select, { theme: theme, value: node.right.args.value[1].unit, onChange: (value) => handleChange([
                                         { value, path: keyPrefix + 'right.args.value||1.unit' },
                                     ]), options: [
                                         { label: `year${plural}`, value: '* 365 DAY' },
@@ -865,7 +895,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                 else if (isTheCurrentInterval(node, client.databaseType)) {
                     const { dateFilterType } = getDateFilterInfo(node);
                     const options = getAllPossibleColumns().map((column) => ({
-                        label: column.displayName,
+                        label: snakeCaseToTitleCase(column.displayName),
                         value: column.name,
                     }));
                     return (_jsxs("div", { style: {
@@ -873,7 +903,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                             flexDirection: 'row',
                             alignItems: 'center',
                             gap: 20,
-                        }, children: [_jsx(Select, { value: node.left.column, onChange: (value) => {
+                        }, children: [_jsx(Select, { theme: theme, value: node.left.column, onChange: (value) => {
                                     const columnType = getColumnTypeByName(value);
                                     if (isDateishColumnType(columnType)) {
                                         // handleChange(value, keyPrefix + dateColumnPath, "text");
@@ -889,14 +919,14 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                         newSubtree.left.args.value[0].column = value;
                                         handleReplaceSubtree(keyPrefix, newSubtree);
                                     }
-                                }, options: options }), _jsx(Select, { value: 'IN_THE_CURRENT', onChange: (value) => {
+                                }, options: options }), _jsx(Select, { theme: theme, value: 'IN_THE_CURRENT', onChange: (value) => {
                                     handleOperatorChange(value, node, keyPrefix, node.left.column);
                                 }, options: [
                                     { label: 'in the last', value: 'IN_THE_LAST' },
                                     { label: 'in the previous', value: 'IN_THE_PREVIOUS' },
                                     { label: 'in the current', value: 'IN_THE_CURRENT' },
                                     // { label: 'equals', value: 'equals' },
-                                ] }), _jsx(Select, { value: node.left.args.value[1].column, onChange: (value) => {
+                                ] }), _jsx(Select, { theme: theme, value: node.left.args.value[1].column, onChange: (value) => {
                                     handleChange([
                                         { value, path: 'right.args.value||1.column' },
                                         { value, path: 'left.args.value||1.column' },
@@ -910,7 +940,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                 }
                 else if (isThePreviousInterval(node, client.databaseType)) {
                     const options = getAllPossibleColumns().map((column) => ({
-                        label: column.displayName,
+                        label: snakeCaseToTitleCase(column.displayName),
                         value: column.name,
                     }));
                     return (_jsxs("div", { style: {
@@ -918,7 +948,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                             flexDirection: 'row',
                             alignItems: 'center',
                             gap: 20,
-                        }, children: [_jsx(Select, { value: node.left.column, onChange: (value) => {
+                        }, children: [_jsx(Select, { theme: theme, value: node.left.column, onChange: (value) => {
                                     const columnType = getColumnTypeByName(value);
                                     if (isDateishColumnType(columnType)) {
                                         // handleChange(value, keyPrefix + dateColumnPath, "text");
@@ -934,14 +964,14 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                         newSubtree.left.args.value[0].column = value;
                                         handleReplaceSubtree(keyPrefix, newSubtree);
                                     }
-                                }, opt: true, options: options }), _jsx(Select, { value: 'IN_THE_PREVIOUS', onChange: (value) => {
+                                }, opt: true, options: options }), _jsx(Select, { theme: theme, value: 'IN_THE_PREVIOUS', onChange: (value) => {
                                     handleOperatorChange(value, node, keyPrefix, node.left.column);
                                 }, options: [
                                     { label: 'in the last', value: 'IN_THE_LAST' },
                                     { label: 'in the previous', value: 'IN_THE_PREVIOUS' },
                                     { label: 'in the current', value: 'IN_THE_CURRENT' },
                                     // { label: 'equals', value: 'equals' },
-                                ] }), _jsx(Select, { value: node.left.args.value[1].column, onChange: (value) => {
+                                ] }), _jsx(Select, { theme: theme, value: node.left.args.value[1].column, onChange: (value) => {
                                     const dayConversion = {
                                         YEAR: 365,
                                         QUARTER: 90,
@@ -971,7 +1001,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                 }
                 else if (isColumnComparison(node)) {
                     const options = getAllPossibleColumns().map((column) => ({
-                        label: column.displayName,
+                        label: snakeCaseToTitleCase(column.displayName),
                         value: column.name,
                     }));
                     // grab the value of the left child of the column comparison
@@ -1025,6 +1055,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                             gap: 12,
                             flexDirection: 'column',
                             width: '100%',
+                            padding: '6px 0px',
                         }, children: [_jsxs("div", { style: {
                                     display: 'flex',
                                     gap: 20,
@@ -1033,7 +1064,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                         ? 'row'
                                         : 'column',
                                     width: '100%',
-                                }, children: [_jsx(Select, { style: { width: 'min-content' }, value: leftChildValue, onChange: (value) => {
+                                }, children: [_jsx(Select, { theme: theme, style: { width: 'min-content' }, value: leftChildValue, onChange: (value) => {
                                             const columnType = getColumnTypeByName(value);
                                             if (isDateishColumnType(columnType)) {
                                                 handleOperatorChange('IN_THE_LAST', node, keyPrefix, value);
@@ -1048,7 +1079,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                                 newSubtree.left.args.value[0].column = value;
                                                 handleReplaceSubtree(keyPrefix, newSubtree);
                                             }
-                                        }, options: options }), operatorOptions.length > 0 && (_jsx(Select, { value: node.operator, onChange: (value) => {
+                                        }, options: options }), operatorOptions.length > 0 && (_jsx(Select, { theme: theme, value: node.operator, onChange: (value) => {
                                             handleOperatorChange(value, node, keyPrefix, leftChildValue);
                                         }, style: { width: 'min-content' }, options: operatorOptions })), node.right &&
                                         node.right.type !== 'expr_list' &&
@@ -1078,7 +1109,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                             justifyContent: 'space-between',
                             flexDirection: showNodeAsRow(node, formData) ? 'row' : 'column',
                             width: '100%',
-                        }, children: [node.left && renderNode(node.left, keyPrefix + 'left.'), _jsx(Select, { value: node.operator, onChange: (value) => {
+                        }, children: [node.left && renderNode(node.left, keyPrefix + 'left.'), _jsx(Select, { theme: theme, value: node.operator, onChange: (value) => {
                                     handleOperatorChange(value, node, keyPrefix);
                                 }, style: { width: `100%` }, options: [
                                     // { label: `and`, value: "AND" },
@@ -1120,10 +1151,10 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                 }
             case 'column_ref': {
                 const options = getAllPossibleColumns().map((column) => ({
-                    label: column.displayName,
+                    label: snakeCaseToTitleCase(column.displayName),
                     value: column.name,
                 }));
-                return (_jsx(Select, { style: { width: '120px' }, value: node.column ?? options[0]?.value, onChange: (value) => {
+                return (_jsx(Select, { theme: theme, style: { width: '120px' }, value: node.column ?? options[0]?.value, onChange: (value) => {
                         handleChange([{ value, path: keyPrefix + 'column' }]);
                     }, options: options }));
             }
@@ -1247,10 +1278,10 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                         isInTheCurrentIntervalSentence ??
                         isInTheLastIntervalSentence ??
                         isInThePreviousIntervalSentence ?? (_jsxs(_Fragment, { children: [node.left &&
-                                renderSentence(formData, node.left, keyPrefix + 'left.', false, false, isRow), isRow ? (' ' + OPS[node.operator] + ' ') : topLevelBinaryOperator === 'OR' ? (_jsx(TopLevelBooleanSwitch, { node: node, keyPrefix: keyPrefix, handleOperatorChange: handleOperatorChange, Select: Select })) : null, node.right &&
+                                renderSentence(formData, node.left, keyPrefix + 'left.', false, false, isRow), isRow ? (' ' + OPS[node.operator] + ' ') : isTopLevel || topLevelBinaryOperator === 'OR' ? (_jsx(TopLevelBooleanSwitch, { node: node, keyPrefix: keyPrefix, handleOperatorChange: handleOperatorChange, Select: Select })) : null, node.right &&
                                 renderSentence(formData, node.right, keyPrefix + 'right.', false, false, isRow)] })) }));
             case 'column_ref':
-                return node.column;
+                return snakeCaseToTitleCase(node.column);
             case 'expr_list':
                 if (node.value.length === 1) {
                     const subQuery = renderSentence(formData, node.value[0]);
@@ -1282,10 +1313,10 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                     if (node.args.value.length < 1)
                         return null;
                     if (node.args.value[0].value) {
-                        return node.args.value[0].value.replaceAll('%', '');
+                        return snakeCaseToTitleCase(node.args.value[0].value.replaceAll('%', ''));
                     }
                     if (node.args.value[0].column)
-                        return node.args.value[0].column.replaceAll('%', '');
+                        return snakeCaseToTitleCase(node.args.value[0].column.replaceAll('%', ''));
                     return null;
                 }
                 if (node.name.toLowerCase() === 'current_date' ||
@@ -1322,10 +1353,22 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
         const tableNamesInQuery = baseAst.from.map((tbl) => tbl.table);
         return schemaTables
             .filter((t) => tableNamesInQuery.includes(t.displayName))
-            .flatMap((table) => table.columns.map((c) => ({
+            .flatMap((table) => table.columns
+            .map((c) => ({
             ...c,
             table: table.displayName,
-        })));
+        }))
+            .sort((a, b) => {
+            const aIsId = a.name.toLowerCase() === 'id' ||
+                a.name.toLowerCase().endsWith('_id');
+            const bIsId = b.name.toLowerCase() === 'id' ||
+                b.name.toLowerCase().endsWith('_id');
+            if (aIsId && !bIsId)
+                return 1;
+            if (bIsId && !aIsId)
+                return -1;
+            return 0;
+        }));
     };
     const getDateColumns = () => {
         const allColumns = getAllPossibleColumns();
@@ -1376,26 +1419,6 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
         },
         as: null,
     });
-    const SortableItem = ({ id, label, setSelectedColumns, selectedColumns, }) => {
-        const { attributes, listeners, setNodeRef, transform, transition } = useSortable({ id: id });
-        const style = {
-            transform: DND_CSS.Transform.toString(transform),
-            transition,
-        };
-        const handleSelect = () => {
-            setSelectedColumns((selectedColumns) => {
-                if (selectedColumns.includes(id)) {
-                    return selectedColumns.filter((column) => column !== id);
-                }
-                else {
-                    return [...selectedColumns, id];
-                }
-            });
-        };
-        return (_jsx("div", { style: { userSelect: 'none', ...style }, ref: setNodeRef, children: _jsx(SelectColumn, { selected: selectedColumns?.includes(id), setSelected: handleSelect, label: label, children: _jsx("div", { style: {
-                        cursor: 'grab',
-                    }, ...attributes, ...listeners, children: _jsx(HandleButton, {}) }) }) }));
-    };
     const AddConditionPopover = ({ onSave }) => {
         return (_jsxs("div", { style: { display: 'flex', flexDirection: 'column', gap: 2 }, children: [_jsx("h1", { style: {
                         fontWeight: '600',
@@ -1421,13 +1444,24 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                     body: JSON.stringify({
                         ast: { ...baseAst, where: formData },
                         publicKey: client.publicKey,
-                        orgId: '2',
+                        orgId: client.customerId,
                     }),
                 });
                 const data2 = await res2.json();
                 if (data2.rows && data2.rows.length) {
                     const tables = getTableNames(baseAst);
                     const table = tables.length >= 1 ? tables[0] : initialTableName;
+                    const sortedFields = data2.fields.sort((a, b) => {
+                        const aIsId = a.name.toLowerCase() === 'id' ||
+                            a.name.toLowerCase().endsWith('_id');
+                        const bIsId = b.name.toLowerCase() === 'id' ||
+                            b.name.toLowerCase().endsWith('_id');
+                        if (aIsId && !bIsId)
+                            return 1;
+                        if (bIsId && !aIsId)
+                            return -1;
+                        return 0;
+                    });
                     if (pivot) {
                         // Do all of this to make sure we have the right unique columns when applying a pivot
                         let uniqueFormatted = {};
@@ -1439,6 +1473,8 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                         const pivotedData = generatePivotTable(pivot, data2.rows, [null, null, null], false);
                         console.info(`%c[Pivot]: ${JSON.stringify(pivot)}`, 'color: dimgray');
                         setPivotData(pivotedData);
+                        setRows(data2.rows);
+                        setFields(data2.fields);
                     }
                     else {
                         setRows(data2.rows);
@@ -1559,8 +1595,8 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
         }
         return false;
     };
-    const handleAsk = async () => {
-        if (!aiPrompt) {
+    const handleAsk = async (overridePrompt = '') => {
+        if (!aiPrompt && !overridePrompt) {
             return;
         }
         try {
@@ -1579,7 +1615,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                         method: 'POST',
                         headers: { 'Content-Type': 'application/json' },
                         body: JSON.stringify({
-                            initialQuestion: aiPrompt,
+                            initialQuestion: aiPrompt || overridePrompt,
                             publicKey: client.publicKey,
                         }),
                     });
@@ -1616,11 +1652,21 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                 newAst = convertBigQuery(ast);
                 newAst = convertWildcardColumns(newAst, schemaTables); // must go before groupby
                 ({ ast: newAst, pivot: groupByPivot } = convertGroupBy(newAst, pivot, schemaTables));
+                if (groupByPivot && !groupByPivot?.valueField) {
+                    setErrorMessage("Error: Couldn't process your request, please re-word your prompt.");
+                    return;
+                }
                 newAst = convertStringComparison(newAst, client.databaseType);
                 newAst = convertRemoveSimpleParentheses(newAst);
+                const table = getTableNames(newAst)[0] ?? initialTableName;
+                const tableAlias = getTableAliases(newAst)[0] ?? initialTableName;
+                newAst = removeNonSelectedTableReferences(newAst, tableAlias ?? table);
                 // newAst = convertDateComparison(newAst); // TODO
                 ast = newAst; // so we fetch data for newAst later.
-                const table = getTableNames(newAst)[0] ?? initialTableName;
+                setPivotRowField(groupByPivot?.rowField);
+                setPivotColumnField(groupByPivot?.columnField);
+                setPivotValueField(groupByPivot?.valueField);
+                setPivotAggregation(groupByPivot?.aggregationType);
                 setPivot(groupByPivot);
                 setSelectedColumns(deepCopy(newAst).columns?.map((column) => {
                     if (column.expr.type === 'column_ref') {
@@ -1652,7 +1698,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                 body: JSON.stringify({
                     ast: ast,
                     publicKey: client.publicKey,
-                    orgId: '2',
+                    orgId: client.customerId,
                 }),
             });
             const data2 = await res2.json();
@@ -1663,6 +1709,8 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                     const pivotedData = generatePivotTable(pivot, data2.rows, [null, null, null], false);
                     console.info(`%c[Pivot]: ${JSON.stringify(groupByPivot)}`, 'color: dimgray');
                     setPivotData(pivotedData);
+                    setRows(data2.rows);
+                    setFields(data2.fields);
                 }
                 else {
                     setRows(data2.rows);
@@ -1726,7 +1774,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
             transform: DND_CSS.Transform.toString(transform),
             transition,
         };
-        return (_jsx("div", { style: { ...style }, ref: setNodeRef, children: _jsx(DraggableColumn, { label: label, onDelete: onDelete, children: _jsx("div", { style: {
+        return (_jsx("div", { style: { ...style }, ref: setNodeRef, children: _jsx(DraggableColumn, { label: snakeCaseToTitleCase(label), onDelete: onDelete, children: _jsx("div", { style: {
                         cursor: 'grab',
                     }, ...attributes, ...listeners, children: _jsx(HandleButton, {}) }) }) }));
     };
@@ -1787,24 +1835,17 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                         gap: 8,
                     }, children: [columnNamesInAst.map((name) => (_jsx(DraggableItem, { id: name, label: name, onDelete: () => handleDeleteColumn(name) }, name))), columnNamesInAst?.length > 0 && _jsx("div", { style: { height: 6 } })] }) }) }));
     }
-    const allNumericColumns = getNumericColumns().map((column) => ({
-        label: column.displayName,
-        value: column.name,
-    }));
-    const allNonNumericColumns = getNonNumericColumns().map((column) => ({
-        label: column.displayName,
-        value: column.name,
-    }));
-    const allStringColumns = getStringColumns().map((column) => ({
-        label: column.displayName,
-        value: column.name,
-    }));
     if (loading) {
-        return (_jsxs("div", { style: { display: 'flex', flexDirection: 'row', height: '100%' }, ref: parentRef, children: [_jsxs(Sidebar, { children: [_jsx(SidebarHeading, { label: "Columns" }), _jsx("div", { style: { height: 4, width: '100%' } }), _jsx(DraggableColumns, {}), _jsx(Popover, { isOpen: openPopover === 'AddColumnPopover', trigger: _jsx(SecondaryButton, { onClick: () => {
+        return (_jsxs("div", { style: {
+                display: 'flex',
+                flexDirection: 'row',
+                height: '100%',
+                ...containerStyle,
+            }, ref: parentRef, children: [_jsxs(Sidebar, { children: [_jsx(SidebarHeading, { label: "Columns" }), _jsx("div", { style: { height: 4, width: '100%' } }), _jsx(DraggableColumns, {}), _jsx(Popover, { isOpen: openPopover === 'AddColumnPopover', trigger: _jsx(SecondaryButton, { onClick: () => {
                                     if (!openPopover) {
                                         setOpenPopover('AddColumnPopover');
                                     }
-                                }, label: 'Select columns' }), title: "Select columns", onClose: () => {
+                                }, label: 'Select columns' }), label: "Select columns", onClose: () => {
                                 setIsPending(false);
                                 setActiveEditItem(null);
                                 setActivePath(null);
@@ -1917,7 +1958,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                 newAst.orderby = null;
                                 setBaseAst(newAst); // trigger refetch
                                 setPivot(pivot);
-                            }, selectPivotOnEdit: true, showTrigger: !pivot, theme: theme, LabelComponent: Label, HeaderComponent: Header, dateRange: [null, null, null], recommendPivotCount: 3 }), pivot && (_jsx(PivotCard, { pivotTable: {
+                            }, selectPivotOnEdit: true, showTrigger: !pivot || !pivotData, theme: theme, LabelComponent: Label, HeaderComponent: Header, dateRange: [null, null, null], recommendPivotCount: 3, SecondaryButtonComponent: SecondaryButton }), pivot && pivotData && (_jsx(PivotCard, { pivotTable: {
                                 pivot: pivot,
                                 rows: pivotData?.rows || [],
                                 columns: pivotData?.columns || [],
@@ -1976,7 +2017,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                     const newAst = { ...baseAst };
                                     newAst.orderby.splice(id, 1);
                                     setBaseAst(deepCopy(newAst));
-                                } }, `sort-sentence-${id}`))) })), _jsx(Popover, { isOpen: openPopover === 'AddSortPopover', trigger: _jsx(SecondaryButton, { onClick: () => {
+                                } }, `sort-sentence-${id}`))) })), _jsx(Popover, { isOpen: openPopover === 'AddSortPopover', setIsOpen: () => { }, trigger: _jsx(SecondaryButton, { onClick: () => {
                                     if (!openPopover) {
                                         setOpenPopover('AddSortPopover');
                                     }
@@ -1985,12 +2026,45 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                 setActiveEditItem(null);
                                 setActivePath(null);
                                 setOpenPopover(null);
-                            }, children: _jsx(AddSortPopover, { columns: selectedColumns, Select: Select, Button: Button, onSave: () => { } }) })] }), _jsxs(Container, { children: [_jsxs("form", { style: {
+                            }, children: _jsx(AddSortPopover, { columns: selectedColumns, Select: Select, Button: Button, onSave: () => { } }) }), _jsx("div", { style: { height: 28, width: '100%' } }), _jsx(SidebarHeading, { label: "Limit" }), _jsx("div", { style: { height: 4, width: '100%' } }), baseAst && baseAst.limit ? (_jsx("div", { style: {
+                                display: 'flex',
+                                flexDirection: 'column',
+                                gap: 8,
+                                marginBottom: 12,
+                            }, children: _jsx(LimitSentence, { limit: baseAst.limit, setOpenPopover: setOpenPopover, LimitPopover: SortPopover, EditPopover: AddLimitPopover, handleDelete: () => {
+                                    const newAst = { ...baseAst };
+                                    newAst.limit = null;
+                                    setBaseAst(deepCopy(newAst));
+                                }, onSave: (limit) => {
+                                    const newAst = { ...baseAst };
+                                    newAst.limit = {
+                                        seperator: '',
+                                        value: [
+                                            {
+                                                type: 'number',
+                                                value: limit,
+                                            },
+                                        ],
+                                    };
+                                    setOpenPopover(null);
+                                    setBaseAst(deepCopy(newAst));
+                                } }) })) : (_jsx(Popover, { isOpen: openPopover === 'AddLimitPopover', setIsOpen: () => { }, trigger: _jsx(SecondaryButton, { onClick: () => {
+                                    if (!openPopover) {
+                                        setOpenPopover('AddLimitPopover');
+                                    }
+                                }, label: 'Add limit' }), title: "Limit", onClose: () => {
+                                setIsPending(false);
+                                setActiveEditItem(null);
+                                setActivePath(null);
+                                setOpenPopover(null);
+                            }, children: _jsx(TextInput, { value: 0, type: "number", style: { width: 120, minHeight: 32 }, onChange: (e) => { } }) }))] }), _jsxs(Container, { children: [!hideAi && (_jsxs("form", { onSubmit: (event) => {
+                                event.preventDefault();
+                            }, style: {
                                 display: 'flex',
                                 flexDirection: 'row',
                                 gap: 12,
                                 padding: 1,
-                            }, children: [_jsx(TextInput, { placeholder: "Ask a question...", type: "text", style: { width: '100%', fontSize: 14 }, value: aiPrompt }), _jsx(ButtonLoadingState, {}), baseAst && (_jsx(SecondaryButton, { onClick: clearAllState, label: "New report" }))] }), baseAst && (_jsxs(_Fragment, { children: [_jsx(TableLoadingState, {}), _jsxs("div", { style: {
+                            }, children: [_jsx(TextInput, { placeholder: baseAst ? 'Ask a follow-up question...' : 'Ask a question...', type: "text", style: { width: '100%', fontSize: 14 }, value: aiPrompt }), _jsx(Button, { onClick: () => { }, label: 'Ask AI' }), baseAst && (_jsx(SecondaryButton, { onClick: clearAllState, label: "New report" }))] })), baseAst && (_jsxs(_Fragment, { children: [_jsx(TableLoadingState, {}), _jsxs("div", { style: {
                                         display: 'flex',
                                         flexDirection: 'row',
                                         gap: '12px',
@@ -2001,6 +2075,8 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
             flexDirection: 'row',
             height: '100%',
             overflowY: 'auto',
+            boxSizing: 'border-box',
+            ...containerStyle,
         }, children: [_jsxs(Sidebar, { children: [_jsx(SidebarHeading, { label: "Columns" }), _jsx("div", { style: { height: 4, width: '100%' } }), _jsx(DraggableColumns, {}), _jsx(Popover, { isOpen: openPopover === 'AddColumnPopover', title: "Select columns", trigger: _jsx(SecondaryButton, { onClick: () => {
                                 if (!openPopover) {
                                     setOpenPopover('AddColumnPopover');
@@ -2113,7 +2189,7 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                             setOpenPopover(null);
                                             clearCheckboxes();
                                         }
-                                    } }) }))] }), _jsx("div", { style: { height: 28, width: '100%' } }), _jsx(SidebarHeading, { label: "Pivot" }), _jsx("div", { style: { height: 4, width: '100%' } }), _jsx(PivotModal, { pivotRowField: pivotRowField, setPivotRowField: setPivotRowField, pivotColumnField: pivotColumnField, setPivotColumnField: setPivotColumnField, pivotValueField: pivotValueField, setPivotValueField: setPivotValueField, pivotAggregation: pivotAggregation, setPivotAggregation: setPivotAggregation, createdPivots: createdPivots, setCreatedPivots: setCreatedPivots, recommendedPivots: recommendedPivots, setRecommendedPivots: setRecommendedPivots, popUpTitle: pivotPopUpTitle, setPopUpTitle: setPivotPopUpTitle, selectedTable: initialTableName, SelectComponent: Select, ButtonComponent: Button, PopoverComponent: PivotPopover, TextComponent: Text, isOpen: showPivotPopover, setIsOpen: setShowPivotPopover, showUpdatePivot: isEdittingPivot, setShowUpdatePivot: setIsEdittingPivot, parentRef: parentRef, data: rows, columns: processColumnsForChartBuilder(Object.keys(rows[0] ?? {})), triggerButtonText: 'Add pivot', selectedPivotIndex: selectedPivotIndex, setSelectedPivotIndex: setSelectedPivotIndex, removePivot: () => {
+                                    } }) }))] }), _jsx("div", { style: { height: 28, width: '100%' } }), _jsx(SidebarHeading, { label: "Pivot" }), _jsx("div", { style: { height: 4, width: '100%' } }), _jsx(PivotModal, { pivotRowField: pivotRowField, setPivotRowField: setPivotRowField, pivotColumnField: pivotColumnField, setPivotColumnField: setPivotColumnField, pivotValueField: pivotValueField, setPivotValueField: setPivotValueField, pivotAggregation: pivotAggregation, setPivotAggregation: setPivotAggregation, createdPivots: createdPivots, setCreatedPivots: setCreatedPivots, recommendedPivots: recommendedPivots, setRecommendedPivots: setRecommendedPivots, popUpTitle: pivotPopUpTitle, setPopUpTitle: setPivotPopUpTitle, selectedTable: initialTableName, SelectComponent: Select, ButtonComponent: Button, SecondaryButtonComponent: SecondaryButton, PopoverComponent: PivotPopover, TextComponent: Text, isOpen: showPivotPopover, setIsOpen: setShowPivotPopover, showUpdatePivot: isEdittingPivot, setShowUpdatePivot: setIsEdittingPivot, parentRef: parentRef, data: rows, columns: processColumnsForChartBuilder(Object.keys(rows[0] ?? {})), triggerButtonText: 'Add pivot', selectedPivotIndex: selectedPivotIndex, setSelectedPivotIndex: setSelectedPivotIndex, removePivot: () => {
                             setPivot(null);
                             setPivotData(null);
                         }, selectPivot: (pivot) => {
@@ -2121,9 +2197,13 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                 return;
                             const newAst = { ...baseAst };
                             newAst.orderby = null;
+                            if (pivot.rowFieldType === 'date') {
+                                pivot['sort'] = true;
+                                pivot['sortDirection'] = 'ASC';
+                            }
                             setBaseAst(newAst); // trigger refetch
                             setPivot(pivot);
-                        }, selectPivotOnEdit: true, showTrigger: !pivot, theme: theme, LabelComponent: Label, HeaderComponent: Header, dateRange: [null, null, null], recommendPivotCount: 3 }), pivot && (_jsx(PivotCard, { pivotTable: {
+                        }, selectPivotOnEdit: true, showTrigger: !pivot || !pivotData, theme: theme, LabelComponent: Label, HeaderComponent: Header, dateRange: [null, null, null], recommendPivotCount: 3 }), pivot && pivotData && (_jsx(PivotCard, { pivotTable: {
                             pivot: pivot,
                             rows: pivotData?.rows || [],
                             columns: pivotData?.columns || [],
@@ -2224,27 +2304,77 @@ export default function ReportBuilder({ initialTableName = '', onAddToDashboardC
                                 setActivePath(null);
                                 setOpenPopover(null);
                                 setBaseAst(deepCopy(newAst));
-                            } }) })] }), _jsxs(Container, { children: [!hideAi && (_jsxs("form", { onSubmit: handleAsk, style: {
+                            } }) }), _jsx("div", { style: { height: 28, width: '100%' } }), _jsx(SidebarHeading, { label: "Limit" }), _jsx("div", { style: { height: 4, width: '100%' } }), baseAst && baseAst.limit ? (_jsx("div", { style: {
+                            display: 'flex',
+                            flexDirection: 'column',
+                            gap: 8,
+                            marginBottom: 12,
+                        }, children: _jsx(LimitSentence, { limit: baseAst.limit, setOpenPopover: setOpenPopover, LimitPopover: SortPopover, EditPopover: AddLimitPopover, handleDelete: () => {
+                                const newAst = { ...baseAst };
+                                newAst.limit = null;
+                                setBaseAst(deepCopy(newAst));
+                            }, onSave: (limit) => {
+                                const newAst = { ...baseAst };
+                                newAst.limit = {
+                                    seperator: '',
+                                    value: [
+                                        {
+                                            type: 'number',
+                                            value: limit,
+                                        },
+                                    ],
+                                };
+                                setOpenPopover(null);
+                                setBaseAst(deepCopy(newAst));
+                            } }) })) : (_jsx(Popover, { isOpen: openPopover === 'AddLimitPopover', setIsOpen: () => { }, trigger: _jsx(SecondaryButton, { onClick: () => {
+                                if (!baseAst) {
+                                    return;
+                                }
+                                if (!openPopover) {
+                                    setOpenPopover('AddLimitPopover');
+                                }
+                            }, label: 'Add limit' }), title: "Limit", onClose: () => {
+                            setIsPending(false);
+                            setActiveEditItem(null);
+                            setActivePath(null);
+                            setOpenPopover(null);
+                        }, children: _jsx(AddLimitPopover, { TextInput: TextInput, onSave: (limit) => {
+                                const newAst = { ...baseAst };
+                                newAst.limit = {
+                                    seperator: '',
+                                    value: [
+                                        {
+                                            type: 'number',
+                                            value: Number(limit),
+                                        },
+                                    ],
+                                };
+                                setOpenPopover(null);
+                                setBaseAst(deepCopy(newAst));
+                            } }) }))] }), _jsxs(Container, { children: [!hideAi && (_jsxs("form", { onSubmit: (event) => {
+                            event.preventDefault();
+                            handleAsk();
+                        }, style: {
                             display: 'flex',
                             flexDirection: 'row',
                             gap: 12,
                             padding: 1,
-                        }, children: [_jsx(TextInput, { type: "text", value: aiPrompt, style: { width: '100%', fontSize: 14 }, onChange: (e) => setAiPrompt(e.target.value), placeholder: baseAst ? 'Ask a follow-up question...' : 'Ask a question...' }), _jsx(Button, { onClick: handleAsk, label: 'Ask AI' }), baseAst && (_jsx(SecondaryButton, { label: 'New report', onClick: clearAllState }))] })), baseAst && (_jsxs(_Fragment, { children: [loading && errorMessage.length === 0 ? (_jsx(TableLoadingState, {})) : (_jsx(Table, { rows: applyFormatting({
-                                    rows: pivotData?.rows || rows,
-                                    fields: pivotData?.fields || fields,
-                                }, baseAst?.columns ?? []), columns: pivotData?.columns ||
-                                    enforceOrderOnColumns(Object.keys(rows[0] ?? {})).map((c) => {
-                                        return { label: c, field: c };
-                                    }), error: errorMessage, rowsPerPage: 20 })), _jsxs("div", { style: {
-                                    display: 'flex',
-                                    flexDirection: 'row',
-                                    gap: '12px',
-                                }, children: [errorMessage && (_jsx("div", { style: {
-                                            color: 'red',
-                                            fontSize: 14,
-                                            padding: '12px',
-                                            whiteSpace: 'nowrap',
-                                        }, children: errorMessage })), _jsx("div", { style: { width: '100%' } }), _jsx(SecondaryButton, { label: isCopying ? '✅ Copied' : 'Copy SQL', onClick: () => copyToClipboard(activeQuery) }), _jsx(Button, { onClick: () => {
+                        }, children: [_jsx(TextInput, { type: "text", value: aiPrompt, style: { width: '100%', fontSize: 14 }, onChange: (e) => setAiPrompt(e.target.value), placeholder: baseAst ? 'Ask a follow-up question...' : 'Ask a question...' }), _jsx(Button, { onClick: handleAsk, label: 'Ask AI' }), baseAst && (_jsx(SecondaryButton, { label: 'New report', onClick: clearAllState }))] })), baseAst && (_jsx(_Fragment, { children: loading && errorMessage.length === 0 ? (_jsx(TableLoadingState, {})) : (_jsx(Table, { rows: applyFormatting({
+                                rows: pivotData?.rows || rows,
+                                fields: pivotData?.fields || fields,
+                            }, baseAst?.columns ?? []), columns: pivotData?.columns ||
+                                enforceOrderOnColumns(Object.keys(rows[0] ?? {})).map((c) => {
+                                    return { label: snakeCaseToTitleCase(c), field: c };
+                                }), error: errorMessage, rowsPerPage: 20 })) })), _jsxs("div", { style: {
+                            display: 'flex',
+                            flexDirection: 'row',
+                            gap: '12px',
+                        }, children: [errorMessage && (_jsx("div", { style: {
+                                    color: 'red',
+                                    fontSize: 14,
+                                    padding: '12px',
+                                    whiteSpace: 'nowrap',
+                                }, children: errorMessage })), _jsx("div", { style: { width: '100%' } }), baseAst && (_jsxs(_Fragment, { children: [_jsx(SecondaryButton, { label: isCopying ? '✅ Copied' : 'Copy SQL', onClick: () => copyToClipboard(activeQuery) }), _jsx(Button, { onClick: () => {
                                             setIsChartBuilderOpen(true);
-                                        }, label: 'Add to dashboard' })] })] }))] }), _jsx("style", { children: `body{margin:0;}` }), _jsx(ChartBuilderWithModal, { rows: applyFormatting({ rows, fields }, baseAst?.columns ?? []), columns: processColumnsForChartBuilder(Object.keys(rows[0] ?? {})), fields: fields, pivot: pivot, query: activeQuery, showTableFormatOptions: admin ? true : false, showDateFieldOptions: admin ? true : false, showAccessControlOptions: admin ? true : false, title: "Add to dashboard", isEditMode: true, isOpen: isChartBuilderOpen, setIsOpen: setIsChartBuilderOpen, onAddToDashboardComplete: onAddToDashboardComplete, destinationDashboard: destinationDashboard, organizationName: organizationName })] }));
+                                        }, label: 'Add to dashboard' })] }))] })] }), _jsx("style", { children: `body{margin:0;}` }), _jsx(ChartBuilderWithModal, { rows: applyFormatting({ rows, fields }, baseAst?.columns ?? []), columns: processColumnsForChartBuilder(Object.keys(rows[0] ?? {})), fields: fields, pivot: pivot, query: activeQuery, showTableFormatOptions: admin ? true : false, showDateFieldOptions: admin ? true : false, showAccessControlOptions: admin ? true : false, title: "Add to dashboard", isEditMode: true, isOpen: isChartBuilderOpen, setIsOpen: setIsChartBuilderOpen, onAddToDashboardComplete: onAddToDashboardComplete, destinationDashboard: destinationDashboard, organizationName: organizationName, pivotData: pivotData })] }));
 }