delete table if rows upsert fails

dust-tt · Jan 24, 2024 · aa47670 · aa47670
1 parent e4b9da7
commit aa47670
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/front/pages/api/w/[wId]/data_sources/[name]/tables/csv.ts b/front/pages/api/w/[wId]/data_sources/[name]/tables/csv.ts
@@ -175,6 +175,25 @@ async function handler(
             "Failed to upsert rows."
           );
 
+          const delRes = await coreAPI.deleteTable({
+            projectId: dataSource.dustAPIProjectId,
+            dataSourceName: dataSource.name,
+            tableId,
+          });
+
+          if (delRes.isErr()) {
+            logger.error(
+              {
+                dataSourceName: dataSource.name,
+                workspaceId: owner.id,
+                tableId,
+                tableName: name,
+                error: delRes.error,
+              },
+              "Failed to delete table after failed upsert."
+            );
+          }
+
           return apiError(req, res, {
             status_code: 500,
             api_error: {