fix: scope and batch sensitive suggestions

frontend/src/api/catalog-databases.ts

@@ -99,6 +99,11 @@ export interface CatalogColumn {
 
 export interface SensitiveDataSuggestion {
   columnId: string;
+  tableId: string;
+  tableName: string;
+  columnName: string;
+  version: number;
+  currentSensitive: boolean;
   sensitive: boolean;
 }
 
@@ -106,6 +111,11 @@ export interface SensitiveDataSuggestions {
   suggestions: SensitiveDataSuggestion[];
 }
 
+export type SensitiveDataSuggestionRequest =
+  | { scope: "all" }
+  | { scope: "selected_tables"; targetIds: string[] }
+  | { scope: "selected_columns"; targetIds: string[] };
+
 export interface CatalogRelationshipColumn {
   position: number;
   sourceColumnId: string;
@@ -357,10 +367,25 @@ export const updateCatalogColumnMetadata = (
   { method: "PATCH", body: JSON.stringify({ version, description, generatedDescription, sensitive }) },
 );
 
-export const suggestSensitiveFields = (databaseId: string, modelId: string) =>
+export const updateCatalogColumnSensitive = (
+  databaseId: string,
+  tableId: string,
+  columnId: string,
+  version: number,
+  sensitive: boolean,
+) => apiFetch<CatalogColumn>(
+  `/catalog/databases/${encodeURIComponent(databaseId)}/tables/${encodeURIComponent(tableId)}/columns/${encodeURIComponent(columnId)}`,
+  { method: "PATCH", body: JSON.stringify({ version, sensitive }) },
+);
+
+export const suggestSensitiveFields = (
+  databaseId: string,
+  modelId: string,
+  selection: SensitiveDataSuggestionRequest,
+) =>
   apiFetch<SensitiveDataSuggestions>(
     `/catalog/databases/${encodeURIComponent(databaseId)}/sensitive-data-suggestions`,
-    { method: "POST", body: JSON.stringify({ modelId }) },
+    { method: "POST", body: JSON.stringify({ modelId, ...selection }) },
   );
 
 export const listCatalogRelationships = (databaseId: string) =>