From ec4a1f8590412a8debb7a7404264534edcb7df47 Mon Sep 17 00:00:00 2001
From: Andrey Avtomonov <andreybavt@gmail.com>
Date: Tue, 12 May 2026 23:39:05 +0200
Subject: [PATCH] chore(context): verify warehouse verification tools

---
 .../tools/warehouse-verification/discover-data.tool.ts      | 2 +-
 .../tools/warehouse-verification/sql-execution.tool.ts      | 2 +-
 .../warehouse-verification/warehouse-catalog.service.ts     | 6 +++++-
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/packages/context/src/ingest/tools/warehouse-verification/discover-data.tool.ts b/packages/context/src/ingest/tools/warehouse-verification/discover-data.tool.ts
index ec05f03f..cb5a817b 100644
--- a/packages/context/src/ingest/tools/warehouse-verification/discover-data.tool.ts
+++ b/packages/context/src/ingest/tools/warehouse-verification/discover-data.tool.ts
@@ -9,7 +9,7 @@ const discoverDataInputSchema = z.object({
   sourceName: z.string().optional(),
 });
 
-type DiscoverDataInput = z.infer<typeof discoverDataInputSchema>;
+type DiscoverDataInput = z.input<typeof discoverDataInputSchema>;
 
 export interface DiscoverDataStructured {
   wiki: unknown | null;
diff --git a/packages/context/src/ingest/tools/warehouse-verification/sql-execution.tool.ts b/packages/context/src/ingest/tools/warehouse-verification/sql-execution.tool.ts
index 459e480f..03375938 100644
--- a/packages/context/src/ingest/tools/warehouse-verification/sql-execution.tool.ts
+++ b/packages/context/src/ingest/tools/warehouse-verification/sql-execution.tool.ts
@@ -9,7 +9,7 @@ const sqlExecutionInputSchema = z.object({
   rowLimit: z.number().int().positive().max(1000).optional().default(100),
 });
 
-type SqlExecutionInput = z.infer<typeof sqlExecutionInputSchema>;
+type SqlExecutionInput = z.input<typeof sqlExecutionInputSchema>;
 
 export interface SqlExecutionStructured {
   headers: string[];
diff --git a/packages/context/src/ingest/tools/warehouse-verification/warehouse-catalog.service.ts b/packages/context/src/ingest/tools/warehouse-verification/warehouse-catalog.service.ts
index 1589d61b..f935f0ca 100644
--- a/packages/context/src/ingest/tools/warehouse-verification/warehouse-catalog.service.ts
+++ b/packages/context/src/ingest/tools/warehouse-verification/warehouse-catalog.service.ts
@@ -245,7 +245,11 @@ export class WarehouseCatalogService {
       comment: table.comment,
       description: table.description ?? firstDescription(table.descriptions),
       rowCount: profileTable?.rowCount ?? table.estimatedRows ?? null,
-      columns: table.columns.map((column) => {
+      columns: table.columns.map((rawColumn) => {
+        const column = rawColumn as KtxSchemaColumn & {
+          description?: string | null;
+          descriptions?: Record<string, string>;
+        };
         const profileColumn =
           profileColumns[columnKey(table, column.name)] ??
           Object.entries(profileColumns).find(