Merge pull request #1743 from ashrafchowdury/fix/1737-testset-columns…

…-name-style fix: fixed testset columns naming conventions
Agenta-AI · Jun 4, 2024 · 8df17b2 · 8df17b2
2 parents 01a8952 + 42c75cc
commit 8df17b2
Showing 1 changed file with 22 additions and 6 deletions.
diff --git a/agenta-web/src/components/TestSetTable/TestsetTable.tsx b/agenta-web/src/components/TestSetTable/TestsetTable.tsx
@@ -192,7 +192,14 @@ const TestsetTable: React.FC<testsetTableProps> = ({mode}) => {
 
     useEffect(() => {
         async function applyColData(colData: {field: string}[] = []) {
-            const newColDefs = [CHECKBOX_COL, ...colData, ADD_BUTTON_COL]
+            const newColDefs = [
+                CHECKBOX_COL,
+                ...colData.map((col) => ({
+                    ...col,
+                    headerName: col.field,
+                })),
+                ADD_BUTTON_COL,
+            ]
             setColumnDefs(newColDefs)
             if (mode === "create") {
                 const initialRowData = Array(3).fill({})
@@ -241,7 +248,14 @@ const TestsetTable: React.FC<testsetTableProps> = ({mode}) => {
             }
         })
 
-        const newColumnDefs = [CHECKBOX_COL, ...newDataColumns, ADD_BUTTON_COL]
+        const newColumnDefs = [
+            CHECKBOX_COL,
+            ...newDataColumns.map((col) => ({
+                ...col,
+                headerName: col.field,
+            })),
+            ADD_BUTTON_COL,
+        ]
 
         const keyMap = dataColumns.reduce((acc: KeyValuePair, colDef, index) => {
             acc[colDef.field] = newDataColumns[index].field
@@ -285,9 +299,7 @@ const TestsetTable: React.FC<testsetTableProps> = ({mode}) => {
             }
 
             if (
-                inputValues.some(
-                    (input) => input.toLowerCase() === scopedInputValues[index].toLowerCase(),
-                ) ||
+                inputValues.some((input) => input === scopedInputValues[index]) ||
                 scopedInputValues[index] == ""
             ) {
                 message.error(
@@ -320,7 +332,11 @@ const TestsetTable: React.FC<testsetTableProps> = ({mode}) => {
             newColmnDef.pop()
 
             setInputValues([...inputValues, newColumnName])
-            setColumnDefs([...columnDefs, {field: newColumnName}, ADD_BUTTON_COL])
+            setColumnDefs([
+                ...columnDefs,
+                {field: newColumnName, headerName: newColumnName},
+                ADD_BUTTON_COL,
+            ])
             setRowData(updatedRowData)
             setLoading(false)
         }