fixed build

Author: aadamgough

apps/sim/tools/snowflake/describe_table.ts

@@ -3,7 +3,7 @@ import type {
   SnowflakeDescribeTableParams,
   SnowflakeDescribeTableResponse,
 } from '@/tools/snowflake/types'
-import { extractResponseData, parseAccountUrl } from '@/tools/snowflake/utils'
+import { extractResponseData, parseAccountUrl, sanitizeIdentifier } from '@/tools/snowflake/utils'
 import type { ToolConfig } from '@/tools/types'
 
 const logger = createLogger('SnowflakeDescribeTableTool')
@@ -79,9 +79,16 @@ export const snowflakeDescribeTableTool: ToolConfig<
       'X-Snowflake-Authorization-Token-Type': 'OAUTH',
     }),
     body: (params: SnowflakeDescribeTableParams) => {
-      const requestBody: any = {
-        statement: `DESCRIBE TABLE ${params.database}.${params.schema}.${params.table}`,
+      const sanitizedDatabase = sanitizeIdentifier(params.database)
+      const sanitizedSchema = sanitizeIdentifier(params.schema)
+      const sanitizedTable = sanitizeIdentifier(params.table)
+      const fullTableName = `${sanitizedDatabase}.${sanitizedSchema}.${sanitizedTable}`
+
+      const requestBody: Record<string, any> = {
+        statement: `DESCRIBE TABLE ${fullTableName}`,
         timeout: 60,
+        database: params.database,
+        schema: params.schema,
       }
 
       if (params.warehouse) {
@@ -92,7 +99,7 @@ export const snowflakeDescribeTableTool: ToolConfig<
         requestBody.role = params.role
       }
 
-      return JSON.stringify(requestBody)
+      return requestBody
     },
   },
 

apps/sim/tools/snowflake/execute_query.ts

@@ -89,7 +89,7 @@ export const snowflakeExecuteQueryTool: ToolConfig<
       'X-Snowflake-Authorization-Token-Type': 'OAUTH',
     }),
     body: (params: SnowflakeExecuteQueryParams) => {
-      const requestBody: any = {
+      const requestBody: Record<string, any> = {
         statement: params.query,
         timeout: params.timeout || 60,
       }
@@ -110,7 +110,7 @@ export const snowflakeExecuteQueryTool: ToolConfig<
         requestBody.role = params.role
       }
 
-      return JSON.stringify(requestBody)
+      return requestBody
     },
   },
 

apps/sim/tools/snowflake/list_databases.ts

@@ -61,7 +61,7 @@ export const snowflakeListDatabasesTool: ToolConfig<
       'X-Snowflake-Authorization-Token-Type': 'OAUTH',
     }),
     body: (params: SnowflakeListDatabasesParams) => {
-      const requestBody: any = {
+      const requestBody: Record<string, any> = {
         statement: 'SHOW DATABASES',
         timeout: 60,
       }
@@ -74,7 +74,7 @@ export const snowflakeListDatabasesTool: ToolConfig<
         requestBody.role = params.role
       }
 
-      return JSON.stringify(requestBody)
+      return requestBody
     },
   },
 

apps/sim/tools/snowflake/list_schemas.ts

@@ -3,7 +3,7 @@ import type {
   SnowflakeListSchemasParams,
   SnowflakeListSchemasResponse,
 } from '@/tools/snowflake/types'
-import { extractResponseData, parseAccountUrl } from '@/tools/snowflake/utils'
+import { extractResponseData, parseAccountUrl, sanitizeIdentifier } from '@/tools/snowflake/utils'
 import type { ToolConfig } from '@/tools/types'
 
 const logger = createLogger('SnowflakeListSchemasTool')
@@ -67,9 +67,12 @@ export const snowflakeListSchemasTool: ToolConfig<
       'X-Snowflake-Authorization-Token-Type': 'OAUTH',
     }),
     body: (params: SnowflakeListSchemasParams) => {
-      const requestBody: any = {
-        statement: `SHOW SCHEMAS IN DATABASE ${params.database}`,
+      const sanitizedDatabase = sanitizeIdentifier(params.database)
+
+      const requestBody: Record<string, any> = {
+        statement: `SHOW SCHEMAS IN DATABASE ${sanitizedDatabase}`,
         timeout: 60,
+        database: params.database,
       }
 
       if (params.warehouse) {
@@ -80,7 +83,7 @@ export const snowflakeListSchemasTool: ToolConfig<
         requestBody.role = params.role
       }
 
-      return JSON.stringify(requestBody)
+      return requestBody
     },
   },
 

apps/sim/tools/snowflake/list_tables.ts

@@ -3,7 +3,7 @@ import type {
   SnowflakeListTablesParams,
   SnowflakeListTablesResponse,
 } from '@/tools/snowflake/types'
-import { extractResponseData, parseAccountUrl } from '@/tools/snowflake/utils'
+import { extractResponseData, parseAccountUrl, sanitizeIdentifier } from '@/tools/snowflake/utils'
 import type { ToolConfig } from '@/tools/types'
 
 const logger = createLogger('SnowflakeListTablesTool')
@@ -73,9 +73,14 @@ export const snowflakeListTablesTool: ToolConfig<
       'X-Snowflake-Authorization-Token-Type': 'OAUTH',
     }),
     body: (params: SnowflakeListTablesParams) => {
-      const requestBody: any = {
-        statement: `SHOW TABLES IN ${params.database}.${params.schema}`,
+      const sanitizedDatabase = sanitizeIdentifier(params.database)
+      const sanitizedSchema = sanitizeIdentifier(params.schema)
+
+      const requestBody: Record<string, any> = {
+        statement: `SHOW TABLES IN ${sanitizedDatabase}.${sanitizedSchema}`,
         timeout: 60,
+        database: params.database,
+        schema: params.schema,
       }
 
       if (params.warehouse) {
@@ -86,7 +91,7 @@ export const snowflakeListTablesTool: ToolConfig<
         requestBody.role = params.role
       }
 
-      return JSON.stringify(requestBody)
+      return requestBody
     },
   },