@squadbase/connectors 0.0.4 → 0.0.5

package/dist/index.d.ts

@@ -52,6 +52,12 @@ interface ProxyPolicyRule {
 interface ProxyPolicy {
     allowlist: ProxyPolicyRule[];
 }
+type ConnectionCheckResult = {
+    success: true;
+} | {
+    success: false;
+    error: string;
+};
 interface ConnectorToolsConfig {
     oauthProxy: {
         appApiKey: string;
@@ -97,6 +103,9 @@ declare class ConnectorPlugin<P extends Record<string, ParameterDefinition> = Re
     }) => Promise<{
         rows: Record<string, unknown>[];
     }>;
+    readonly checkConnection?: (params: Record<string, string>, config: {
+        proxyFetch: typeof fetch;
+    }) => Promise<ConnectionCheckResult>;
     constructor(config: {
         slug: string;
         authType: string | null;
@@ -115,6 +124,9 @@ declare class ConnectorPlugin<P extends Record<string, ParameterDefinition> = Re
         }) => Promise<{
             rows: Record<string, unknown>[];
         }>;
+        checkConnection?: (params: Record<string, string>, config: {
+            proxyFetch: typeof fetch;
+        }) => Promise<ConnectionCheckResult>;
     });
     get connectorKey(): string;
     /**
@@ -790,4 +802,4 @@ declare const squadbaseDbConnector: ConnectorPlugin<{
     }>;
 }>;
 
-export { AUTH_TYPES, ConnectorPlugin, ConnectorSetup, ConnectorTool, type ConnectorToolsConfig, ParameterDefinition, type ProxyPolicy, type ProxyPolicyRule, type ReleaseFlag, type SetupLanguage, airtableConnector, awsAthenaConnector, bigqueryConnector, bigqueryOauthConnector, connectors, databricksConnector, dbtConnector, googleAnalyticsConnector, kintoneConnector, mysqlConnector, postgresqlConnector, redshiftConnector, snowflakeConnector, snowflakePatConnector, squadbaseDbConnector, wixStoreConnector };
+export { AUTH_TYPES, type ConnectionCheckResult, ConnectorPlugin, ConnectorSetup, ConnectorTool, type ConnectorToolsConfig, ParameterDefinition, type ProxyPolicy, type ProxyPolicyRule, type ReleaseFlag, type SetupLanguage, airtableConnector, awsAthenaConnector, bigqueryConnector, bigqueryOauthConnector, connectors, databricksConnector, dbtConnector, googleAnalyticsConnector, kintoneConnector, mysqlConnector, postgresqlConnector, redshiftConnector, snowflakeConnector, snowflakePatConnector, squadbaseDbConnector, wixStoreConnector };

package/dist/index.js

@@ -92,6 +92,7 @@ var ConnectorPlugin = class _ConnectorPlugin {
   systemPrompt;
   tools;
   query;
+  checkConnection;
   constructor(config) {
     this.slug = config.slug;
     this.authType = config.authType;
@@ -106,6 +107,7 @@ var ConnectorPlugin = class _ConnectorPlugin {
     this.systemPrompt = config.systemPrompt;
     this.tools = config.tools;
     this.query = config.query;
+    this.checkConnection = config.checkConnection;
   }
   get connectorKey() {
     return _ConnectorPlugin.deriveKey(this.slug, this.authType);
@@ -439,6 +441,45 @@ var snowflakeConnector = new ConnectorPlugin({
   - List columns: \`DESCRIBE TABLE db_name.schema_name.table_name\`
 - INFORMATION_SCHEMA is also available: \`SELECT * FROM db_name.INFORMATION_SCHEMA.TABLES\``,
   tools,
+  async checkConnection(params, _config) {
+    try {
+      const snowflake = (await import("snowflake-sdk")).default;
+      snowflake.configure({ logLevel: "ERROR" });
+      const privateKey = Buffer.from(
+        params[parameters.privateKeyBase64.slug],
+        "base64"
+      ).toString("utf-8");
+      const conn = snowflake.createConnection({
+        account: params[parameters.account.slug],
+        username: params[parameters.user.slug],
+        role: params[parameters.role.slug],
+        warehouse: params[parameters.warehouse.slug],
+        authenticator: "SNOWFLAKE_JWT",
+        privateKey
+      });
+      await new Promise((resolve, reject) => {
+        conn.connect((err) => {
+          if (err) reject(new Error(`Snowflake connect failed: ${err.message}`));
+          else resolve();
+        });
+      });
+      await new Promise((resolve, reject) => {
+        conn.execute({
+          sqlText: "SELECT 1",
+          complete: (err) => {
+            if (err) reject(new Error(`Snowflake query failed: ${err.message}`));
+            else resolve();
+          }
+        });
+      });
+      conn.destroy((err) => {
+        if (err) console.warn(`[connector-client] Snowflake destroy error: ${err.message}`);
+      });
+      return { success: true };
+    } catch (e) {
+      return { success: false, error: e instanceof Error ? e.message : String(e) };
+    }
+  },
   async query(params, sql, namedParams) {
     const resolvedSql = replaceLiteralParams(sql, namedParams);
     const snowflake = (await import("snowflake-sdk")).default;
@@ -653,6 +694,45 @@ var snowflakePatConnector = new ConnectorPlugin({
   - List columns: \`DESCRIBE TABLE db_name.schema_name.table_name\`
 - INFORMATION_SCHEMA is also available: \`SELECT * FROM db_name.INFORMATION_SCHEMA.TABLES\``,
   tools: tools2,
+  async checkConnection(params, _config) {
+    try {
+      const snowflake = (await import("snowflake-sdk")).default;
+      snowflake.configure({ logLevel: "ERROR" });
+      const conn = snowflake.createConnection({
+        account: params[parameters2.account.slug],
+        username: params[parameters2.user.slug],
+        role: params[parameters2.role.slug],
+        warehouse: params[parameters2.warehouse.slug],
+        password: params[parameters2.pat.slug]
+      });
+      await new Promise((resolve, reject) => {
+        conn.connect((err) => {
+          if (err)
+            reject(new Error(`Snowflake connect failed: ${err.message}`));
+          else resolve();
+        });
+      });
+      await new Promise((resolve, reject) => {
+        conn.execute({
+          sqlText: "SELECT 1",
+          complete: (err) => {
+            if (err)
+              reject(new Error(`Snowflake query failed: ${err.message}`));
+            else resolve();
+          }
+        });
+      });
+      conn.destroy((err) => {
+        if (err)
+          console.warn(
+            `[connector-client] Snowflake destroy error: ${err.message}`
+          );
+      });
+      return { success: true };
+    } catch (e) {
+      return { success: false, error: e instanceof Error ? e.message : String(e) };
+    }
+  },
   async query(params, sql, namedParams) {
     const resolvedSql = replaceLiteralParams(sql, namedParams);
     const snowflake = (await import("snowflake-sdk")).default;
@@ -797,6 +877,22 @@ var postgresqlConnector = new ConnectorPlugin({
   - List columns: \`SELECT column_name, data_type FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'xxx'\`
 - Always include LIMIT in queries`,
   tools: tools3,
+  async checkConnection(params, _config) {
+    const { Pool } = await import("pg");
+    const pool = new Pool({
+      connectionString: params[parameters3.connectionUrl.slug],
+      ssl: { rejectUnauthorized: false },
+      connectionTimeoutMillis: 1e4
+    });
+    try {
+      await pool.query("SELECT 1");
+      return { success: true };
+    } catch (error) {
+      return { success: false, error: error instanceof Error ? error.message : String(error) };
+    } finally {
+      await pool.end();
+    }
+  },
   async query(params, sql, namedParams) {
     const { Pool } = await import("pg");
     const { text, values } = buildPositionalParams(sql, namedParams);
@@ -921,6 +1017,21 @@ var mysqlConnector = new ConnectorPlugin({
   - List columns: \`SELECT COLUMN_NAME, DATA_TYPE FROM information_schema.COLUMNS WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'xxx'\`
 - Always include LIMIT in queries`,
   tools: tools4,
+  async checkConnection(params, _config) {
+    const mysql = await import("mysql2/promise");
+    const pool = mysql.createPool({
+      uri: params[parameters4.connectionUrl.slug],
+      connectTimeout: 1e4
+    });
+    try {
+      await pool.query("SELECT 1");
+      return { success: true };
+    } catch (e) {
+      return { success: false, error: e instanceof Error ? e.message : String(e) };
+    } finally {
+      await pool.end();
+    }
+  },
   async query(params, sql, namedParams) {
     const mysql = await import("mysql2/promise");
     const { text, values } = buildQuestionmarkParams(sql, namedParams);
@@ -1039,14 +1150,10 @@ var bigquerySetup = new ConnectorSetup({
 
 #### \u30B9\u30C6\u30C3\u30D71: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u9078\u629E
 1. \`${listProjectsToolName}\` \u3092\u547C\u3073\u51FA\u3057\u3066\u3001\u30B5\u30FC\u30D3\u30B9\u30A2\u30AB\u30A6\u30F3\u30C8\u304C\u30A2\u30AF\u30BB\u30B9\u53EF\u80FD\u306AGCP\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B
-2. \u7D50\u679C\u306B\u5FDC\u3058\u3066\u5206\u5C90:
-   - **\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u304C2\u3064\u4EE5\u4E0A**: \`askUserQuestion\`\uFF08multiSelect: false\uFF09\u3067\u30E6\u30FC\u30B6\u30FC\u306B\u63D0\u793A\u3057\u3001\u4F7F\u7528\u3059\u308B\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u3092\u9078\u629E\u3055\u305B\u308B\u3002\u9078\u629E\u80A2\u306E label \u306F \`\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u540D (id: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID)\` \u306E\u5F62\u5F0F\u306B\u3059\u308B
-   - **\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u304C1\u3064\u3060\u3051**: askUserQuestion \u306F\u4F7F\u308F\u305A\u81EA\u52D5\u63A1\u7528\u3002\u300C\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8 X \u3092\u81EA\u52D5\u9078\u629E\u3057\u307E\u3057\u305F\u300D\u30681\u6587\u3060\u3051\u66F8\u304F
-3. \`updateConnectionParameters\` \u3067\u9078\u629E\u3055\u308C\u305F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u3092\u4FDD\u5B58\u3059\u308B:
+2. \`updateConnectionParameters\` \u3092\u547C\u3073\u51FA\u3057\u3066\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u3092\u4FDD\u5B58\u3059\u308B:
    - \`parameterSlug\`: \`"project-id"\`
-   - \`value\`: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID
-   - \`displayValue\`: \`"\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u540D (id: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID)"\`
-4. \u7D9A\u884C\u30E1\u30C3\u30BB\u30FC\u30B8\u3092\u53D7\u3051\u53D6\u3063\u305F\u3089\u3001\u30B9\u30C6\u30C3\u30D72\u306B\u9032\u3080
+   - \`options\`: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u4E00\u89A7\u3002\u5404 option \u306E \`label\` \u306F \`\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u540D (id: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID)\` \u306E\u5F62\u5F0F\u3001\`value\` \u306F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID
+3. \u7D9A\u884C\u30E1\u30C3\u30BB\u30FC\u30B8\u3092\u53D7\u3051\u53D6\u3063\u305F\u3089\u3001\u30B9\u30C6\u30C3\u30D72\u306B\u9032\u3080
 
 #### \u30B9\u30C6\u30C3\u30D72: \u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u9078\u629E
 1. \`SELECT schema_name FROM INFORMATION_SCHEMA.SCHEMATA\` \u3092\u5B9F\u884C\u3057\u3066\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B
@@ -1081,14 +1188,10 @@ Follow these steps to set up the BigQuery connection.
 
 #### Step 1: Project Selection
 1. Call \`${listProjectsToolName}\` to get the list of GCP projects accessible with the service account credentials
-2. Branch based on results:
-   - **2 or more projects**: Present them to the user via \`askUserQuestion\` (multiSelect: false) and let them select a project. Format option labels as \`Project Name (id: project-id)\`
-   - **Exactly 1 project**: Do NOT call askUserQuestion. Auto-select it. Just write "Auto-selected project X" in one sentence
-3. Call \`updateConnectionParameters\` to save the selected project:
+2. Call \`updateConnectionParameters\` to save the project:
    - \`parameterSlug\`: \`"project-id"\`
-   - \`value\`: the project ID
-   - \`displayValue\`: \`"Project Name (id: project-id)"\`
-4. After receiving the continuation message, proceed to Step 2
+   - \`options\`: The project list. Each option's \`label\` should be \`Project Name (id: project-id)\`, \`value\` should be the project ID
+3. After receiving the continuation message, proceed to Step 2
 
 #### Step 2: Dataset Selection
 1. Run \`SELECT schema_name FROM INFORMATION_SCHEMA.SCHEMATA\` to get the list of datasets
@@ -1203,6 +1306,23 @@ var bigqueryConnector = new ConnectorPlugin({
   - List columns: \`SELECT column_name, data_type FROM \\\`project_id.dataset\\\`.INFORMATION_SCHEMA.COLUMNS WHERE table_name = 'xxx'\`
 - Always specify project_id explicitly in queries`,
   tools: tools5,
+  async checkConnection(params, _config) {
+    const { BigQuery } = await import("@google-cloud/bigquery");
+    const credentials = JSON.parse(
+      Buffer.from(params[parameters5.serviceAccountKeyJsonBase64.slug], "base64").toString("utf-8")
+    );
+    const bq = new BigQuery({
+      projectId: params[parameters5.projectId.slug],
+      credentials
+    });
+    try {
+      const [job] = await bq.createQueryJob({ query: "SELECT 1" });
+      await job.getQueryResults();
+      return { success: true };
+    } catch (error) {
+      return { success: false, error: error instanceof Error ? error.message : String(error) };
+    }
+  },
   async query(params, sql, namedParams) {
     const { BigQuery } = await import("@google-cloud/bigquery");
     const resolvedSql = replaceLiteralParams(sql, namedParams);
@@ -1337,14 +1457,10 @@ var bigquerySetup2 = new ConnectorSetup({
 
 #### \u30B9\u30C6\u30C3\u30D71: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u9078\u629E
 1. \`${listProjectsToolName2}\` \u3092\u547C\u3073\u51FA\u3057\u3066\u3001OAuth\u3067\u30A2\u30AF\u30BB\u30B9\u53EF\u80FD\u306AGCP\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B
-2. \u7D50\u679C\u306B\u5FDC\u3058\u3066\u5206\u5C90:
-   - **\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u304C2\u3064\u4EE5\u4E0A**: \`askUserQuestion\`\uFF08multiSelect: false\uFF09\u3067\u30E6\u30FC\u30B6\u30FC\u306B\u63D0\u793A\u3057\u3001\u4F7F\u7528\u3059\u308B\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u3092\u9078\u629E\u3055\u305B\u308B\u3002\u9078\u629E\u80A2\u306E label \u306F \`\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u540D (id: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID)\` \u306E\u5F62\u5F0F\u306B\u3059\u308B
-   - **\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u304C1\u3064\u3060\u3051**: askUserQuestion \u306F\u4F7F\u308F\u305A\u81EA\u52D5\u63A1\u7528\u3002\u300C\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8 X \u3092\u81EA\u52D5\u9078\u629E\u3057\u307E\u3057\u305F\u300D\u30681\u6587\u3060\u3051\u66F8\u304F
-3. \`updateConnectionParameters\` \u3067\u9078\u629E\u3055\u308C\u305F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u3092\u4FDD\u5B58\u3059\u308B:
+2. \`updateConnectionParameters\` \u3092\u547C\u3073\u51FA\u3057\u3066\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u3092\u4FDD\u5B58\u3059\u308B:
    - \`parameterSlug\`: \`"project-id"\`
-   - \`value\`: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID
-   - \`displayValue\`: \`"\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u540D (id: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID)"\`
-4. \u7D9A\u884C\u30E1\u30C3\u30BB\u30FC\u30B8\u3092\u53D7\u3051\u53D6\u3063\u305F\u3089\u3001\u30B9\u30C6\u30C3\u30D72\u306B\u9032\u3080
+   - \`options\`: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u4E00\u89A7\u3002\u5404 option \u306E \`label\` \u306F \`\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u540D (id: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID)\` \u306E\u5F62\u5F0F\u3001\`value\` \u306F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID
+3. \u7D9A\u884C\u30E1\u30C3\u30BB\u30FC\u30B8\u3092\u53D7\u3051\u53D6\u3063\u305F\u3089\u3001\u30B9\u30C6\u30C3\u30D72\u306B\u9032\u3080
 
 #### \u30B9\u30C6\u30C3\u30D72: \u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u9078\u629E
 1. \`SELECT schema_name FROM INFORMATION_SCHEMA.SCHEMATA\` \u3092\u5B9F\u884C\u3057\u3066\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B
@@ -1379,14 +1495,10 @@ Follow these steps to set up the BigQuery connection.
 
 #### Step 1: Project Selection
 1. Call \`${listProjectsToolName2}\` to get the list of GCP projects accessible with the OAuth credentials
-2. Branch based on results:
-   - **2 or more projects**: Present them to the user via \`askUserQuestion\` (multiSelect: false) and let them select a project. Format option labels as \`Project Name (id: project-id)\`
-   - **Exactly 1 project**: Do NOT call askUserQuestion. Auto-select it. Just write "Auto-selected project X" in one sentence
-3. Call \`updateConnectionParameters\` to save the selected project:
+2. Call \`updateConnectionParameters\` to save the project:
    - \`parameterSlug\`: \`"project-id"\`
-   - \`value\`: the project ID
-   - \`displayValue\`: \`"Project Name (id: project-id)"\`
-4. After receiving the continuation message, proceed to Step 2
+   - \`options\`: The project list. Each option's \`label\` should be \`Project Name (id: project-id)\`, \`value\` should be the project ID
+3. After receiving the continuation message, proceed to Step 2
 
 #### Step 2: Dataset Selection
 1. Run \`SELECT schema_name FROM INFORMATION_SCHEMA.SCHEMATA\` to get the list of datasets
@@ -1594,6 +1706,25 @@ var bigqueryOauthConnector = new ConnectorPlugin({
   - List columns: \`SELECT column_name, data_type FROM \\\`project_id.dataset\\\`.INFORMATION_SCHEMA.COLUMNS WHERE table_name = 'xxx'\`
 - Always specify project_id explicitly in queries`,
   tools: tools6,
+  async checkConnection(params, config) {
+    const { proxyFetch } = config;
+    const projectId = params[parameters6.projectId.slug];
+    const url = `https://bigquery.googleapis.com/bigquery/v2/projects/${projectId}/queries`;
+    try {
+      const res = await proxyFetch(url, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ query: "SELECT 1", useLegacySql: false })
+      });
+      if (!res.ok) {
+        const errorText = await res.text().catch(() => res.statusText);
+        return { success: false, error: `BigQuery query failed: HTTP ${res.status} ${errorText}` };
+      }
+      return { success: true };
+    } catch (error) {
+      return { success: false, error: error instanceof Error ? error.message : String(error) };
+    }
+  },
   async query(params, sql, namedParams, context) {
     const { proxyFetch } = context;
     const projectId = params[parameters6.projectId.slug];
@@ -1798,6 +1929,62 @@ var awsAthenaConnector = new ConnectorPlugin({
 - Always include LIMIT in queries
 - Query execution is asynchronous and results may take time to retrieve`,
   tools: tools7,
+  async checkConnection(params, _config) {
+    try {
+      const {
+        AthenaClient,
+        StartQueryExecutionCommand,
+        GetQueryExecutionCommand
+      } = await import("@aws-sdk/client-athena");
+      const workgroup = params[parameters7.workgroup.slug];
+      const outputLocation = params[parameters7.outputLocation.slug];
+      if (!workgroup && !outputLocation) {
+        return {
+          success: false,
+          error: "Either workgroup or output-location is required"
+        };
+      }
+      const client = new AthenaClient({
+        region: params[parameters7.awsRegion.slug],
+        credentials: {
+          accessKeyId: params[parameters7.awsAccessKeyId.slug],
+          secretAccessKey: params[parameters7.awsSecretAccessKey.slug]
+        }
+      });
+      const startParams = { QueryString: "SELECT 1" };
+      if (workgroup) startParams.WorkGroup = workgroup;
+      if (outputLocation) {
+        startParams.ResultConfiguration = { OutputLocation: outputLocation };
+      }
+      const { QueryExecutionId } = await client.send(
+        new StartQueryExecutionCommand(startParams)
+      );
+      const startTime = Date.now();
+      while (true) {
+        const exec = await client.send(
+          new GetQueryExecutionCommand({ QueryExecutionId })
+        );
+        const state = exec.QueryExecution?.Status?.State;
+        if (state === "SUCCEEDED") break;
+        if (state === "FAILED" || state === "CANCELLED") {
+          return {
+            success: false,
+            error: exec.QueryExecution?.Status?.StateChangeReason || `Query ${state}`
+          };
+        }
+        if (Date.now() - startTime > 12e4) {
+          return { success: false, error: "Query timed out after 120 seconds" };
+        }
+        await new Promise((r) => setTimeout(r, 1e3));
+      }
+      return { success: true };
+    } catch (error) {
+      return {
+        success: false,
+        error: error instanceof Error ? error.message : String(error)
+      };
+    }
+  },
   async query(params, sql, namedParams) {
     const {
       AthenaClient,
@@ -2075,6 +2262,65 @@ var redshiftConnector = new ConnectorPlugin({
 - Always include LIMIT in queries
 - Query execution is asynchronous and results may take time to retrieve`,
   tools: tools8,
+  async checkConnection(params, _config) {
+    try {
+      const {
+        RedshiftDataClient,
+        ExecuteStatementCommand,
+        DescribeStatementCommand
+      } = await import("@aws-sdk/client-redshift-data");
+      const clusterIdentifier = params[parameters8.clusterIdentifier.slug];
+      const workgroupName = params[parameters8.workgroupName.slug];
+      const secretArn = params[parameters8.secretArn.slug];
+      const dbUser = params[parameters8.dbUser.slug];
+      if (!clusterIdentifier && !workgroupName) {
+        return {
+          success: false,
+          error: "Either cluster-identifier or workgroup-name is required"
+        };
+      }
+      const client = new RedshiftDataClient({
+        region: params[parameters8.awsRegion.slug],
+        credentials: {
+          accessKeyId: params[parameters8.awsAccessKeyId.slug],
+          secretAccessKey: params[parameters8.awsSecretAccessKey.slug]
+        }
+      });
+      const { Id: statementId } = await client.send(
+        new ExecuteStatementCommand({
+          Database: params[parameters8.database.slug],
+          Sql: "SELECT 1",
+          ...clusterIdentifier && { ClusterIdentifier: clusterIdentifier },
+          ...workgroupName && { WorkgroupName: workgroupName },
+          ...secretArn && { SecretArn: secretArn },
+          ...dbUser && { DbUser: dbUser }
+        })
+      );
+      const startTime = Date.now();
+      while (true) {
+        const desc = await client.send(
+          new DescribeStatementCommand({ Id: statementId })
+        );
+        if (desc.Status === "FINISHED") break;
+        if (desc.Status === "FAILED" || desc.Status === "ABORTED") {
+          return {
+            success: false,
+            error: desc.Error || `Statement ${desc.Status}`
+          };
+        }
+        if (Date.now() - startTime > 12e4) {
+          return { success: false, error: "Query timed out after 120 seconds" };
+        }
+        await new Promise((r) => setTimeout(r, 1e3));
+      }
+      return { success: true };
+    } catch (error) {
+      return {
+        success: false,
+        error: error instanceof Error ? error.message : String(error)
+      };
+    }
+  },
   async query(params, sql, namedParams) {
     const {
       RedshiftDataClient,
@@ -2263,6 +2509,34 @@ var databricksConnector = new ConnectorPlugin({
   - List columns: \`DESCRIBE TABLE table_name\`
 - Always include LIMIT in queries`,
   tools: tools9,
+  async checkConnection(params, _config) {
+    const { DBSQLClient } = await import("@databricks/sql");
+    const client = new DBSQLClient();
+    await client.connect({
+      host: params[parameters9.host.slug],
+      path: params[parameters9.httpPath.slug],
+      token: params[parameters9.token.slug]
+    });
+    let session;
+    let operation;
+    try {
+      session = await client.openSession();
+      operation = await session.executeStatement("SELECT 1", {
+        runAsync: true
+      });
+      await operation.fetchAll();
+      return { success: true };
+    } catch (e) {
+      return { success: false, error: e instanceof Error ? e.message : String(e) };
+    } finally {
+      if (operation) await operation.close().catch(() => {
+      });
+      if (session) await session.close().catch(() => {
+      });
+      await client.close().catch(() => {
+      });
+    }
+  },
   async query(params, sql, namedParams) {
     const { DBSQLClient } = await import("@databricks/sql");
     const resolvedSql = replaceLiteralParams(sql, namedParams);
@@ -3124,6 +3398,22 @@ var squadbaseDbConnector = new ConnectorPlugin({
   - List columns: \`SELECT column_name, data_type FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'xxx'\`
 - Always include LIMIT in queries`,
   tools: tools15,
+  async checkConnection(params, _config) {
+    const { Pool } = await import("pg");
+    const pool = new Pool({
+      connectionString: params[parameters15.connectionUrl.slug],
+      ssl: { rejectUnauthorized: false },
+      connectionTimeoutMillis: 1e4
+    });
+    try {
+      await pool.query("SELECT 1");
+      return { success: true };
+    } catch (error) {
+      return { success: false, error: error instanceof Error ? error.message : String(error) };
+    } finally {
+      await pool.end();
+    }
+  },
   async query(params, sql, namedParams) {
     const { Pool } = await import("pg");
     const { text, values } = buildPositionalParams(sql, namedParams);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@squadbase/connectors",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "description": "Squadbase Connectors",
   "type": "module",
   "main": "dist/index.js",