Fix checks

Author: tsck

scripts/generate/generateToolDocumentation.ts

@@ -9,6 +9,7 @@ import { readFileSync, writeFileSync } from "fs";
 import { join, dirname } from "path";
 import { fileURLToPath } from "url";
 import { AllTools } from "../../src/tools/index.js";
+import { UIRegistry } from "../../src/ui/registry/index.js";
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = dirname(__filename);
@@ -53,6 +54,7 @@ function extractToolInformation(): ToolInfo[] {
             elicitation: {
                 requestConfirmation: () => Promise.resolve(false),
             } as never,
+            uiRegistry: new UIRegistry(),
         };
 
         try {

tests/unit/toolBase.test.ts

@@ -15,6 +15,7 @@ import type { Server } from "../../src/server.js";
 import type { TelemetryToolMetadata, ToolEvent } from "../../src/telemetry/types.js";
 import { expectDefined } from "../integration/helpers.js";
 import type { PreviewFeature } from "../../src/common/schemas.js";
+import { UIRegistry } from "../../src/ui/registry/index.js";
 
 describe("ToolBase", () => {
     let mockSession: Session;
@@ -60,6 +61,7 @@ describe("ToolBase", () => {
             config: mockConfig,
             telemetry: mockTelemetry,
             elicitation: mockElicitation,
+            uiRegistry: new UIRegistry(),
         };
 
         testTool = new TestTool(constructorParams);