@squadbase/connectors 0.0.3 → 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);
@@ -942,7 +1053,94 @@ var mysqlConnector = new ConnectorPlugin({
   }
 });
 
+// src/connectors/bigquery/tools/list-projects.ts
+import { z as z5 } from "zod";
+
+// src/connectors/bigquery/parameters.ts
+var parameters5 = {
+  serviceAccountKeyJsonBase64: new ParameterDefinition({
+    slug: "service-account-key-json-base64",
+    name: "Google Cloud Service Account JSON",
+    description: "The service account JSON key used to authenticate with Google Cloud Platform. Ensure that the service account has the necessary permissions to access the required resources.",
+    envVarBaseKey: "BIGQUERY_SERVICE_ACCOUNT_JSON_BASE64",
+    type: "base64EncodedJson",
+    secret: true,
+    required: true
+  }),
+  projectId: new ParameterDefinition({
+    slug: "project-id",
+    name: "Google Cloud Project ID",
+    description: "The ID of the Google Cloud project where resources will be managed.",
+    envVarBaseKey: "BIGQUERY_PROJECT_ID",
+    type: "text",
+    secret: false,
+    required: false
+  })
+};
+
+// src/connectors/bigquery/tools/list-projects.ts
+var inputSchema5 = z5.object({
+  toolUseIntent: z5.string().optional().describe(
+    "Brief description of what you intend to accomplish with this tool call"
+  ),
+  connectionId: z5.string().describe("ID of the BigQuery connection to use")
+});
+var outputSchema5 = z5.discriminatedUnion("success", [
+  z5.object({
+    success: z5.literal(true),
+    projects: z5.array(
+      z5.object({
+        projectId: z5.string(),
+        friendlyName: z5.string()
+      })
+    )
+  }),
+  z5.object({
+    success: z5.literal(false),
+    error: z5.string()
+  })
+]);
+var listProjectsTool = new ConnectorTool({
+  name: "listProjects",
+  description: `List GCP projects accessible with the service account credentials. Returns project IDs and friendly names.`,
+  inputSchema: inputSchema5,
+  outputSchema: outputSchema5,
+  async execute({ connectionId }, connections) {
+    const connection = connections.find((c) => c.id === connectionId);
+    if (!connection) {
+      return {
+        success: false,
+        error: `Connection ${connectionId} not found`
+      };
+    }
+    try {
+      const serviceAccountJsonBase64 = parameters5.serviceAccountKeyJsonBase64.getValue(connection);
+      const credentials = JSON.parse(
+        Buffer.from(serviceAccountJsonBase64, "base64").toString("utf-8")
+      );
+      const { GoogleAuth } = await import("google-auth-library");
+      const auth = new GoogleAuth({
+        credentials,
+        scopes: ["https://www.googleapis.com/auth/bigquery"]
+      });
+      const client = await auth.getClient();
+      const res = await client.request({
+        url: "https://bigquery.googleapis.com/bigquery/v2/projects"
+      });
+      const projects = (res.data.projects ?? []).map((p) => ({
+        projectId: p.projectReference?.projectId ?? "",
+        friendlyName: p.friendlyName ?? p.projectReference?.projectId ?? ""
+      })).filter((p) => p.projectId !== "");
+      return { success: true, projects };
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      return { success: false, error: msg };
+    }
+  }
+});
+
 // src/connectors/bigquery/setup.ts
+var listProjectsToolName = `bigquery_${listProjectsTool.name}`;
 var bigquerySetup = new ConnectorSetup({
   ja: `## BigQuery \u30BB\u30C3\u30C8\u30A2\u30C3\u30D7\u624B\u9806
 
@@ -950,32 +1148,27 @@ var bigquerySetup = new ConnectorSetup({
 
 ### \u624B\u9806
 
-#### \u30B9\u30C6\u30C3\u30D70: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u9078\u629E
-1. \`listProjects\` \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:
+#### \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. \`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\u30D71\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\u30D71: \u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u9078\u629E
+#### \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
 2. \u7D50\u679C\u306B\u5FDC\u3058\u3066\u5206\u5C90:
    - **\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u304C2\u3064\u4EE5\u4E0A**: \`askUserQuestion\`\uFF08multiSelect: true\uFF09\u3067\u30E6\u30FC\u30B6\u30FC\u306B\u63D0\u793A\u3057\u3001\u4F7F\u7528\u3059\u308B\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u3092\u9078\u629E\u3055\u305B\u308B
    - **\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u304C1\u3064\u3060\u3051**: askUserQuestion \u306F\u4F7F\u308F\u305A\u81EA\u52D5\u63A1\u7528\u3002\u300C\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8 X \u3092\u81EA\u52D5\u9078\u629E\u3057\u307E\u3057\u305F\u300D\u30681\u6587\u3060\u3051\u66F8\u304F
 
-#### \u30B9\u30C6\u30C3\u30D72: \u30C6\u30FC\u30D6\u30EB\u9078\u629E
+#### \u30B9\u30C6\u30C3\u30D73: \u30C6\u30FC\u30D6\u30EB\u9078\u629E
 3. \u9078\u629E\u3055\u308C\u305F\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u306B\u5BFE\u3057\u3066 \`SELECT table_name FROM \\\`{project}.{dataset}.INFORMATION_SCHEMA.TABLES\\\`\` \u3092\u5B9F\u884C\u3057\u3066\u30C6\u30FC\u30D6\u30EB\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B\uFF08\u8907\u6570\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u306E\u5834\u5408\u306F UNION ALL \u3067\u4E00\u62EC\u53D6\u5F97\uFF09
 4. \u7D50\u679C\u306B\u5FDC\u3058\u3066\u5206\u5C90:
    - **\u30C6\u30FC\u30D6\u30EB\u304C2\u3064\u4EE5\u4E0A**: \`askUserQuestion\`\uFF08multiSelect: true\uFF09\u3067\u30E6\u30FC\u30B6\u30FC\u306B\u63D0\u793A\u3057\u3001\u4F7F\u7528\u3059\u308B\u30C6\u30FC\u30D6\u30EB\u3092\u9078\u629E\u3055\u305B\u308B\u3002description \u306B\u306F\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u540D\u3092\u8A18\u8F09\u3059\u308B
    - **\u30C6\u30FC\u30D6\u30EB\u304C1\u3064\u3060\u3051**: askUserQuestion \u306F\u4F7F\u308F\u305A\u81EA\u52D5\u63A1\u7528
 
-#### \u30B9\u30C6\u30C3\u30D73: \u4FDD\u5B58
+#### \u30B9\u30C6\u30C3\u30D74: \u4FDD\u5B58
 5. \`updateConnectionContext\` \u3067\u4EE5\u4E0B\u3092\u4FDD\u5B58\u3059\u308B:
-   - \`project\`: BigQuery\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID
    - \`dataset\`: \u9078\u629E\u3055\u308C\u305F\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u540D\uFF08\u8907\u6570\u306E\u5834\u5408\u306F\u30AB\u30F3\u30DE\u533A\u5207\u308A\uFF09
    - \`tables\`: \u9078\u629E\u3055\u308C\u305F\u30C6\u30FC\u30D6\u30EB\u540D\uFF08\u8907\u6570\u306E\u5834\u5408\u306F\u30AB\u30F3\u30DE\u533A\u5207\u308A\uFF09
    - \`note\`: \u30BB\u30C3\u30C8\u30A2\u30C3\u30D7\u5185\u5BB9\u306E\u7C21\u5358\u306A\u8AAC\u660E
@@ -993,32 +1186,27 @@ Follow these steps to set up the BigQuery connection.
 
 ### Steps
 
-#### Step 0: Project Selection
-1. Call \`listProjects\` 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:
+#### Step 1: Project Selection
+1. Call \`${listProjectsToolName}\` to get the list of GCP projects accessible with the service account credentials
+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 1
+   - \`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 1: Dataset Selection
+#### Step 2: Dataset Selection
 1. Run \`SELECT schema_name FROM INFORMATION_SCHEMA.SCHEMATA\` to get the list of datasets
 2. Branch based on results:
    - **2 or more datasets**: Present them to the user via \`askUserQuestion\` (multiSelect: true) and let them select which datasets to use
    - **Exactly 1 dataset**: Do NOT call askUserQuestion. Auto-select it. Just write "Auto-selected dataset X" in one sentence
 
-#### Step 2: Table Selection
+#### Step 3: Table Selection
 3. For the selected dataset(s), run \`SELECT table_name FROM \\\`{project}.{dataset}.INFORMATION_SCHEMA.TABLES\\\`\` to get the table list (use UNION ALL for multiple datasets)
 4. Branch based on results:
    - **2 or more tables**: Present them to the user via \`askUserQuestion\` (multiSelect: true) and let them select which tables to use. Include the dataset name in the description
    - **Exactly 1 table**: Do NOT call askUserQuestion. Auto-select it
 
-#### Step 3: Save
+#### Step 4: Save
 5. Call \`updateConnectionContext\` to save:
-   - \`project\`: BigQuery project ID
    - \`dataset\`: Selected dataset name(s) (comma-separated if multiple)
    - \`tables\`: Selected table name(s) (comma-separated if multiple)
    - \`note\`: Brief description of the setup
@@ -1032,50 +1220,28 @@ Follow these steps to set up the BigQuery connection.
 - Skip unnecessary explanations and proceed efficiently`
 });
 
-// src/connectors/bigquery/parameters.ts
-var parameters5 = {
-  serviceAccountKeyJsonBase64: new ParameterDefinition({
-    slug: "service-account-key-json-base64",
-    name: "Google Cloud Service Account JSON",
-    description: "The service account JSON key used to authenticate with Google Cloud Platform. Ensure that the service account has the necessary permissions to access the required resources.",
-    envVarBaseKey: "BIGQUERY_SERVICE_ACCOUNT_JSON_BASE64",
-    type: "base64EncodedJson",
-    secret: true,
-    required: true
-  }),
-  projectId: new ParameterDefinition({
-    slug: "project-id",
-    name: "Google Cloud Project ID",
-    description: "The ID of the Google Cloud project where resources will be managed.",
-    envVarBaseKey: "BIGQUERY_PROJECT_ID",
-    type: "text",
-    secret: false,
-    required: false
-  })
-};
-
 // src/connectors/bigquery/tools/execute-query.ts
-import { z as z5 } from "zod";
+import { z as z6 } from "zod";
 var MAX_ROWS5 = 500;
-var inputSchema5 = z5.object({
-  toolUseIntent: z5.string().optional().describe(
+var inputSchema6 = z6.object({
+  toolUseIntent: z6.string().optional().describe(
     "Brief description of what you intend to accomplish with this tool call"
   ),
-  connectionId: z5.string().describe("ID of the BigQuery connection to use"),
-  sql: z5.string().describe(
+  connectionId: z6.string().describe("ID of the BigQuery connection to use"),
+  sql: z6.string().describe(
     "BigQuery SQL (GoogleSQL) query. Use backtick-quoted fully qualified names `project.dataset.table` for table references."
   )
 });
-var outputSchema5 = z5.discriminatedUnion("success", [
-  z5.object({
-    success: z5.literal(true),
-    rowCount: z5.number(),
-    truncated: z5.boolean(),
-    rows: z5.array(z5.record(z5.string(), z5.unknown()))
+var outputSchema6 = z6.discriminatedUnion("success", [
+  z6.object({
+    success: z6.literal(true),
+    rowCount: z6.number(),
+    truncated: z6.boolean(),
+    rows: z6.array(z6.record(z6.string(), z6.unknown()))
   }),
-  z5.object({
-    success: z5.literal(false),
-    error: z5.string()
+  z6.object({
+    success: z6.literal(false),
+    error: z6.string()
   })
 ]);
 var executeQueryTool5 = new ConnectorTool({
@@ -1083,8 +1249,8 @@ var executeQueryTool5 = new ConnectorTool({
   description: `Execute SQL against BigQuery. Returns up to ${MAX_ROWS5} rows.
 Use for: schema exploration (INFORMATION_SCHEMA), data sampling, analytical queries.
 Avoid loading large amounts of data; always include LIMIT in queries.`,
-  inputSchema: inputSchema5,
-  outputSchema: outputSchema5,
+  inputSchema: inputSchema6,
+  outputSchema: outputSchema6,
   async execute({ connectionId, sql }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -1121,68 +1287,6 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
   }
 });
 
-// src/connectors/bigquery/tools/list-projects.ts
-import { z as z6 } from "zod";
-var inputSchema6 = z6.object({
-  toolUseIntent: z6.string().optional().describe(
-    "Brief description of what you intend to accomplish with this tool call"
-  ),
-  connectionId: z6.string().describe("ID of the BigQuery connection to use")
-});
-var outputSchema6 = z6.discriminatedUnion("success", [
-  z6.object({
-    success: z6.literal(true),
-    projects: z6.array(
-      z6.object({
-        projectId: z6.string(),
-        friendlyName: z6.string()
-      })
-    )
-  }),
-  z6.object({
-    success: z6.literal(false),
-    error: z6.string()
-  })
-]);
-var listProjectsTool = new ConnectorTool({
-  name: "listProjects",
-  description: `List GCP projects accessible with the service account credentials. Returns project IDs and friendly names.`,
-  inputSchema: inputSchema6,
-  outputSchema: outputSchema6,
-  async execute({ connectionId }, connections) {
-    const connection = connections.find((c) => c.id === connectionId);
-    if (!connection) {
-      return {
-        success: false,
-        error: `Connection ${connectionId} not found`
-      };
-    }
-    try {
-      const serviceAccountJsonBase64 = parameters5.serviceAccountKeyJsonBase64.getValue(connection);
-      const credentials = JSON.parse(
-        Buffer.from(serviceAccountJsonBase64, "base64").toString("utf-8")
-      );
-      const { GoogleAuth } = await import("google-auth-library");
-      const auth = new GoogleAuth({
-        credentials,
-        scopes: ["https://www.googleapis.com/auth/bigquery"]
-      });
-      const client = await auth.getClient();
-      const res = await client.request({
-        url: "https://bigquery.googleapis.com/bigquery/v2/projects"
-      });
-      const projects = (res.data.projects ?? []).map((p) => ({
-        projectId: p.projectReference?.projectId ?? "",
-        friendlyName: p.friendlyName ?? p.projectReference?.projectId ?? ""
-      })).filter((p) => p.projectId !== "");
-      return { success: true, projects };
-    } catch (err) {
-      const msg = err instanceof Error ? err.message : String(err);
-      return { success: false, error: msg };
-    }
-  }
-});
-
 // src/connectors/bigquery/index.ts
 var tools5 = { executeQuery: executeQueryTool5, listProjects: listProjectsTool };
 var bigqueryConnector = new ConnectorPlugin({
@@ -1202,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);
@@ -1218,7 +1339,115 @@ var bigqueryConnector = new ConnectorPlugin({
   }
 });
 
+// src/connectors/bigquery-oauth/tools/list-projects.ts
+import { z as z7 } from "zod";
+var REQUEST_TIMEOUT_MS = 6e4;
+var cachedToken = null;
+async function getProxyToken(config) {
+  if (cachedToken && cachedToken.expiresAt > Date.now() + 6e4) {
+    return cachedToken.token;
+  }
+  const url = `${config.appApiBaseUrl}/v0/database/${config.projectId}/environment/${config.environmentId}/oauth-request-proxy-token`;
+  const res = await fetch(url, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      "x-api-key": config.appApiKey,
+      "project-id": config.projectId
+    },
+    body: JSON.stringify({
+      sandboxId: config.sandboxId,
+      issuedBy: "coding-agent"
+    })
+  });
+  if (!res.ok) {
+    const errorText = await res.text().catch(() => res.statusText);
+    throw new Error(`Failed to get proxy token: HTTP ${res.status} ${errorText}`);
+  }
+  const data = await res.json();
+  cachedToken = {
+    token: data.token,
+    expiresAt: new Date(data.expiresAt).getTime()
+  };
+  return data.token;
+}
+var inputSchema7 = z7.object({
+  toolUseIntent: z7.string().optional().describe(
+    "Brief description of what you intend to accomplish with this tool call"
+  ),
+  connectionId: z7.string().describe("ID of the BigQuery OAuth connection to use")
+});
+var outputSchema7 = z7.discriminatedUnion("success", [
+  z7.object({
+    success: z7.literal(true),
+    projects: z7.array(
+      z7.object({
+        projectId: z7.string(),
+        friendlyName: z7.string()
+      })
+    )
+  }),
+  z7.object({
+    success: z7.literal(false),
+    error: z7.string()
+  })
+]);
+var listProjectsTool2 = new ConnectorTool({
+  name: "listProjects",
+  description: `List GCP projects accessible with the current OAuth credentials. Returns project IDs and friendly names.`,
+  inputSchema: inputSchema7,
+  outputSchema: outputSchema7,
+  async execute({ connectionId }, connections, config) {
+    const connection = connections.find((c) => c.id === connectionId);
+    if (!connection) {
+      return {
+        success: false,
+        error: `Connection ${connectionId} not found`
+      };
+    }
+    console.log(
+      `[connector-query] bigquery-oauth/${connection.name}: listProjects`
+    );
+    try {
+      const token = await getProxyToken(config.oauthProxy);
+      const proxyUrl = `https://${config.oauthProxy.sandboxId}.${config.oauthProxy.previewBaseDomain}/_sqcore/connections/${connectionId}/request`;
+      const controller = new AbortController();
+      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS);
+      try {
+        const response = await fetch(proxyUrl, {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${token}`
+          },
+          body: JSON.stringify({
+            url: "https://bigquery.googleapis.com/bigquery/v2/projects",
+            method: "GET"
+          }),
+          signal: controller.signal
+        });
+        const data = await response.json();
+        if (!response.ok) {
+          const errorMessage = typeof data?.error === "string" ? data.error : typeof data?.message === "string" ? data.message : `HTTP ${response.status} ${response.statusText}`;
+          return { success: false, error: errorMessage };
+        }
+        const projects = (data.projects ?? []).map((p) => ({
+          projectId: p.projectReference?.projectId ?? "",
+          friendlyName: p.friendlyName ?? p.projectReference?.projectId ?? ""
+        })).filter((p) => p.projectId !== "");
+        return { success: true, projects };
+      } finally {
+        clearTimeout(timeout);
+      }
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      return { success: false, error: msg };
+    }
+  }
+});
+
 // src/connectors/bigquery-oauth/setup.ts
+var listProjectsToolName2 = `bigquery-oauth_${listProjectsTool2.name}`;
 var bigquerySetup2 = new ConnectorSetup({
   ja: `## BigQuery \u30BB\u30C3\u30C8\u30A2\u30C3\u30D7\u624B\u9806
 
@@ -1226,32 +1455,27 @@ var bigquerySetup2 = new ConnectorSetup({
 
 ### \u624B\u9806
 
-#### \u30B9\u30C6\u30C3\u30D70: \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u9078\u629E
-1. \`listProjects\` \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:
+#### \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. \`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\u30D71\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\u30D71: \u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u9078\u629E
+#### \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
 2. \u7D50\u679C\u306B\u5FDC\u3058\u3066\u5206\u5C90:
    - **\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u304C2\u3064\u4EE5\u4E0A**: \`askUserQuestion\`\uFF08multiSelect: true\uFF09\u3067\u30E6\u30FC\u30B6\u30FC\u306B\u63D0\u793A\u3057\u3001\u4F7F\u7528\u3059\u308B\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u3092\u9078\u629E\u3055\u305B\u308B
    - **\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u304C1\u3064\u3060\u3051**: askUserQuestion \u306F\u4F7F\u308F\u305A\u81EA\u52D5\u63A1\u7528\u3002\u300C\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8 X \u3092\u81EA\u52D5\u9078\u629E\u3057\u307E\u3057\u305F\u300D\u30681\u6587\u3060\u3051\u66F8\u304F
 
-#### \u30B9\u30C6\u30C3\u30D72: \u30C6\u30FC\u30D6\u30EB\u9078\u629E
+#### \u30B9\u30C6\u30C3\u30D73: \u30C6\u30FC\u30D6\u30EB\u9078\u629E
 3. \u9078\u629E\u3055\u308C\u305F\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u306B\u5BFE\u3057\u3066 \`SELECT table_name FROM \\\`{project}.{dataset}.INFORMATION_SCHEMA.TABLES\\\`\` \u3092\u5B9F\u884C\u3057\u3066\u30C6\u30FC\u30D6\u30EB\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B\uFF08\u8907\u6570\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u306E\u5834\u5408\u306F UNION ALL \u3067\u4E00\u62EC\u53D6\u5F97\uFF09
 4. \u7D50\u679C\u306B\u5FDC\u3058\u3066\u5206\u5C90:
    - **\u30C6\u30FC\u30D6\u30EB\u304C2\u3064\u4EE5\u4E0A**: \`askUserQuestion\`\uFF08multiSelect: true\uFF09\u3067\u30E6\u30FC\u30B6\u30FC\u306B\u63D0\u793A\u3057\u3001\u4F7F\u7528\u3059\u308B\u30C6\u30FC\u30D6\u30EB\u3092\u9078\u629E\u3055\u305B\u308B\u3002description \u306B\u306F\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u540D\u3092\u8A18\u8F09\u3059\u308B
    - **\u30C6\u30FC\u30D6\u30EB\u304C1\u3064\u3060\u3051**: askUserQuestion \u306F\u4F7F\u308F\u305A\u81EA\u52D5\u63A1\u7528
 
-#### \u30B9\u30C6\u30C3\u30D73: \u4FDD\u5B58
+#### \u30B9\u30C6\u30C3\u30D74: \u4FDD\u5B58
 5. \`updateConnectionContext\` \u3067\u4EE5\u4E0B\u3092\u4FDD\u5B58\u3059\u308B:
-   - \`project\`: BigQuery\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8ID
    - \`dataset\`: \u9078\u629E\u3055\u308C\u305F\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u540D\uFF08\u8907\u6570\u306E\u5834\u5408\u306F\u30AB\u30F3\u30DE\u533A\u5207\u308A\uFF09
    - \`tables\`: \u9078\u629E\u3055\u308C\u305F\u30C6\u30FC\u30D6\u30EB\u540D\uFF08\u8907\u6570\u306E\u5834\u5408\u306F\u30AB\u30F3\u30DE\u533A\u5207\u308A\uFF09
    - \`note\`: \u30BB\u30C3\u30C8\u30A2\u30C3\u30D7\u5185\u5BB9\u306E\u7C21\u5358\u306A\u8AAC\u660E
@@ -1269,32 +1493,27 @@ Follow these steps to set up the BigQuery connection.
 
 ### Steps
 
-#### Step 0: Project Selection
-1. Call \`listProjects\` 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:
+#### Step 1: Project Selection
+1. Call \`${listProjectsToolName2}\` to get the list of GCP projects accessible with the OAuth credentials
+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 1
+   - \`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 1: Dataset Selection
+#### Step 2: Dataset Selection
 1. Run \`SELECT schema_name FROM INFORMATION_SCHEMA.SCHEMATA\` to get the list of datasets
 2. Branch based on results:
    - **2 or more datasets**: Present them to the user via \`askUserQuestion\` (multiSelect: true) and let them select which datasets to use
    - **Exactly 1 dataset**: Do NOT call askUserQuestion. Auto-select it. Just write "Auto-selected dataset X" in one sentence
 
-#### Step 2: Table Selection
+#### Step 3: Table Selection
 3. For the selected dataset(s), run \`SELECT table_name FROM \\\`{project}.{dataset}.INFORMATION_SCHEMA.TABLES\\\`\` to get the table list (use UNION ALL for multiple datasets)
 4. Branch based on results:
    - **2 or more tables**: Present them to the user via \`askUserQuestion\` (multiSelect: true) and let them select which tables to use. Include the dataset name in the description
    - **Exactly 1 table**: Do NOT call askUserQuestion. Auto-select it
 
-#### Step 3: Save
+#### Step 4: Save
 5. Call \`updateConnectionContext\` to save:
-   - \`project\`: BigQuery project ID
    - \`dataset\`: Selected dataset name(s) (comma-separated if multiple)
    - \`tables\`: Selected table name(s) (comma-separated if multiple)
    - \`note\`: Brief description of the setup
@@ -1321,14 +1540,14 @@ var parameters6 = {
   })
 };
 
-// src/connectors/bigquery-oauth/tools/execute-query.ts
-import { z as z7 } from "zod";
-var MAX_ROWS6 = 500;
-var REQUEST_TIMEOUT_MS = 6e4;
-var cachedToken = null;
-async function getProxyToken(config) {
-  if (cachedToken && cachedToken.expiresAt > Date.now() + 6e4) {
-    return cachedToken.token;
+// src/connectors/bigquery-oauth/tools/execute-query.ts
+import { z as z8 } from "zod";
+var MAX_ROWS6 = 500;
+var REQUEST_TIMEOUT_MS2 = 6e4;
+var cachedToken2 = null;
+async function getProxyToken2(config) {
+  if (cachedToken2 && cachedToken2.expiresAt > Date.now() + 6e4) {
+    return cachedToken2.token;
   }
   const url = `${config.appApiBaseUrl}/v0/database/${config.projectId}/environment/${config.environmentId}/oauth-request-proxy-token`;
   const res = await fetch(url, {
@@ -1348,31 +1567,31 @@ async function getProxyToken(config) {
     throw new Error(`Failed to get proxy token: HTTP ${res.status} ${errorText}`);
   }
   const data = await res.json();
-  cachedToken = {
+  cachedToken2 = {
     token: data.token,
     expiresAt: new Date(data.expiresAt).getTime()
   };
   return data.token;
 }
-var inputSchema7 = z7.object({
-  toolUseIntent: z7.string().optional().describe(
+var inputSchema8 = z8.object({
+  toolUseIntent: z8.string().optional().describe(
     "Brief description of what you intend to accomplish with this tool call"
   ),
-  connectionId: z7.string().describe("ID of the BigQuery OAuth connection to use"),
-  sql: z7.string().describe(
+  connectionId: z8.string().describe("ID of the BigQuery OAuth connection to use"),
+  sql: z8.string().describe(
     "BigQuery SQL (GoogleSQL) query. Use backtick-quoted fully qualified names `project.dataset.table` for table references."
   )
 });
-var outputSchema7 = z7.discriminatedUnion("success", [
-  z7.object({
-    success: z7.literal(true),
-    rowCount: z7.number(),
-    truncated: z7.boolean(),
-    rows: z7.array(z7.record(z7.string(), z7.unknown()))
+var outputSchema8 = z8.discriminatedUnion("success", [
+  z8.object({
+    success: z8.literal(true),
+    rowCount: z8.number(),
+    truncated: z8.boolean(),
+    rows: z8.array(z8.record(z8.string(), z8.unknown()))
   }),
-  z7.object({
-    success: z7.literal(false),
-    error: z7.string()
+  z8.object({
+    success: z8.literal(false),
+    error: z8.string()
   })
 ]);
 function parseQueryResponse(data) {
@@ -1393,8 +1612,8 @@ var executeQueryTool6 = new ConnectorTool({
   description: `Execute SQL against BigQuery via OAuth. Returns up to ${MAX_ROWS6} rows.
 Use for: schema exploration (INFORMATION_SCHEMA), data sampling, analytical queries.
 Avoid loading large amounts of data; always include LIMIT in queries.`,
-  inputSchema: inputSchema7,
-  outputSchema: outputSchema7,
+  inputSchema: inputSchema8,
+  outputSchema: outputSchema8,
   async execute({ connectionId, sql }, connections, config) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -1408,11 +1627,11 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
       `[connector-query] bigquery-oauth/${connection.name}: ${sql}`
     );
     try {
-      const token = await getProxyToken(config.oauthProxy);
+      const token = await getProxyToken2(config.oauthProxy);
       const proxyUrl = `https://${config.oauthProxy.sandboxId}.${config.oauthProxy.previewBaseDomain}/_sqcore/connections/${connectionId}/request`;
       const queryUrl = `https://bigquery.googleapis.com/bigquery/v2/projects/${gcpProjectId}/queries`;
       const controller = new AbortController();
-      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS);
+      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS2);
       try {
         const response = await fetch(proxyUrl, {
           method: "POST",
@@ -1450,113 +1669,6 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
   }
 });
 
-// src/connectors/bigquery-oauth/tools/list-projects.ts
-import { z as z8 } from "zod";
-var REQUEST_TIMEOUT_MS2 = 6e4;
-var cachedToken2 = null;
-async function getProxyToken2(config) {
-  if (cachedToken2 && cachedToken2.expiresAt > Date.now() + 6e4) {
-    return cachedToken2.token;
-  }
-  const url = `${config.appApiBaseUrl}/v0/database/${config.projectId}/environment/${config.environmentId}/oauth-request-proxy-token`;
-  const res = await fetch(url, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      "x-api-key": config.appApiKey,
-      "project-id": config.projectId
-    },
-    body: JSON.stringify({
-      sandboxId: config.sandboxId,
-      issuedBy: "coding-agent"
-    })
-  });
-  if (!res.ok) {
-    const errorText = await res.text().catch(() => res.statusText);
-    throw new Error(`Failed to get proxy token: HTTP ${res.status} ${errorText}`);
-  }
-  const data = await res.json();
-  cachedToken2 = {
-    token: data.token,
-    expiresAt: new Date(data.expiresAt).getTime()
-  };
-  return data.token;
-}
-var inputSchema8 = z8.object({
-  toolUseIntent: z8.string().optional().describe(
-    "Brief description of what you intend to accomplish with this tool call"
-  ),
-  connectionId: z8.string().describe("ID of the BigQuery OAuth connection to use")
-});
-var outputSchema8 = z8.discriminatedUnion("success", [
-  z8.object({
-    success: z8.literal(true),
-    projects: z8.array(
-      z8.object({
-        projectId: z8.string(),
-        friendlyName: z8.string()
-      })
-    )
-  }),
-  z8.object({
-    success: z8.literal(false),
-    error: z8.string()
-  })
-]);
-var listProjectsTool2 = new ConnectorTool({
-  name: "listProjects",
-  description: `List GCP projects accessible with the current OAuth credentials. Returns project IDs and friendly names.`,
-  inputSchema: inputSchema8,
-  outputSchema: outputSchema8,
-  async execute({ connectionId }, connections, config) {
-    const connection = connections.find((c) => c.id === connectionId);
-    if (!connection) {
-      return {
-        success: false,
-        error: `Connection ${connectionId} not found`
-      };
-    }
-    console.log(
-      `[connector-query] bigquery-oauth/${connection.name}: listProjects`
-    );
-    try {
-      const token = await getProxyToken2(config.oauthProxy);
-      const proxyUrl = `https://${config.oauthProxy.sandboxId}.${config.oauthProxy.previewBaseDomain}/_sqcore/connections/${connectionId}/request`;
-      const controller = new AbortController();
-      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS2);
-      try {
-        const response = await fetch(proxyUrl, {
-          method: "POST",
-          headers: {
-            "Content-Type": "application/json",
-            Authorization: `Bearer ${token}`
-          },
-          body: JSON.stringify({
-            url: "https://bigquery.googleapis.com/bigquery/v2/projects",
-            method: "GET"
-          }),
-          signal: controller.signal
-        });
-        const data = await response.json();
-        if (!response.ok) {
-          const errorMessage = typeof data?.error === "string" ? data.error : typeof data?.message === "string" ? data.message : `HTTP ${response.status} ${response.statusText}`;
-          return { success: false, error: errorMessage };
-        }
-        const projects = (data.projects ?? []).map((p) => ({
-          projectId: p.projectReference?.projectId ?? "",
-          friendlyName: p.friendlyName ?? p.projectReference?.projectId ?? ""
-        })).filter((p) => p.projectId !== "");
-        return { success: true, projects };
-      } finally {
-        clearTimeout(timeout);
-      }
-    } catch (err) {
-      const msg = err instanceof Error ? err.message : String(err);
-      return { success: false, error: msg };
-    }
-  }
-});
-
 // src/connectors/bigquery-oauth/index.ts
 var tools6 = { executeQuery: executeQueryTool6, listProjects: listProjectsTool2 };
 function parseQueryResponse2(data) {
@@ -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.3",
+  "version": "0.0.5",
   "description": "Squadbase Connectors",
   "type": "module",
   "main": "dist/index.js",