@squadbase/connectors 0.0.3 → 0.0.4

package/dist/index.js

@@ -942,7 +942,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,8 +1037,8 @@ 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
+#### \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
@@ -959,23 +1046,22 @@ var bigquerySetup = new ConnectorSetup({
    - \`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
+4. \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,8 +1079,8 @@ 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
+#### 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
@@ -1002,23 +1088,22 @@ Follow these steps to set up the BigQuery connection.
    - \`parameterSlug\`: \`"project-id"\`
    - \`value\`: the project ID
    - \`displayValue\`: \`"Project Name (id: project-id)"\`
-4. After receiving the continuation message, proceed to Step 1
+4. 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 +1117,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 +1146,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 +1184,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({
@@ -1218,7 +1219,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,8 +1335,8 @@ 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
+#### \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
@@ -1235,23 +1344,22 @@ var bigquerySetup2 = new ConnectorSetup({
    - \`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
+4. \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,8 +1377,8 @@ 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
+#### 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
@@ -1278,23 +1386,22 @@ Follow these steps to set up the BigQuery connection.
    - \`parameterSlug\`: \`"project-id"\`
    - \`value\`: the project ID
    - \`displayValue\`: \`"Project Name (id: project-id)"\`
-4. After receiving the continuation message, proceed to Step 1
+4. 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
@@ -1322,13 +1429,13 @@ var parameters6 = {
 };
 
 // src/connectors/bigquery-oauth/tools/execute-query.ts
-import { z as z7 } from "zod";
+import { z as z8 } 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;
+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 +1455,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 +1500,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 +1515,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 +1557,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) {

package/package.json

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