feat: complete catalog fleet management workflow

backend/src/catalog/sensitive-data-suggester.ts

@@ -1,7 +1,14 @@
 import { z } from "zod";
 import type { MetadataGenerationModels } from "./metadata-generation-models.js";
 import type { ModelCompleter, ModelCompletionMessage } from "./model-completer.js";
-import type { CatalogColumn, CatalogRepository, CatalogTable } from "./types.js";
+import type {
+  CatalogColumn,
+  CatalogRepository,
+  CatalogTable,
+  SensitiveDataSuggestionScope,
+} from "./types.js";
+
+export type { SensitiveDataSuggestionScope } from "./types.js";
 
 // The helper accepts at most 64 KiB per message. Keep the same safety margin used by
 // Description Generation so UTF-8 structural metadata never reaches that hard limit.
@@ -16,8 +23,6 @@ const responseSchema = z.object({
   }).strict()),
 }).strict();
 
-export type SensitiveDataSuggestionScope = "all" | "selected_tables" | "selected_columns";
-
 interface StructuralColumn {
   columnId: string;
   tableId: string;
@@ -195,10 +200,12 @@ export class SensitiveDataSuggester {
     scope: SensitiveDataSuggestionScope,
     targetIds: readonly string[],
     signal: AbortSignal,
+    onPrepared?: (total: number) => void | Promise<void>,
   ): Promise<readonly SensitiveDataSuggestion[]> {
     const database = await this.repository.get(databaseId);
     if (!database) throw new SensitiveDataSuggestionTargetNotFoundError("database");
     const columns = await this.selectColumns(databaseId, scope, targetIds);
+    await onPrepared?.(columns.length);
     const model = this.models.resolve(modelId);
     const suggestions: SensitiveDataSuggestion[] = [];