@squadbase/connectors 0.1.0 → 0.1.1-dev.1

package/dist/index.d.ts

@@ -228,6 +228,20 @@ declare const connectors: {
             success: false;
             error: string;
         }>;
+        listDatasets: ConnectorTool<{
+            connectionId: string;
+            toolUseIntent?: string | undefined;
+        }, {
+            success: true;
+            datasets: {
+                datasetId: string;
+                location: string;
+                friendlyName: string;
+            }[];
+        } | {
+            success: false;
+            error: string;
+        }>;
     }>;
     bigqueryOauth: ConnectorPlugin<{
         projectId: ParameterDefinition;
@@ -258,6 +272,20 @@ declare const connectors: {
             success: false;
             error: string;
         }>;
+        listDatasets: ConnectorTool<{
+            connectionId: string;
+            toolUseIntent?: string | undefined;
+        }, {
+            success: true;
+            datasets: {
+                datasetId: string;
+                location: string;
+                friendlyName: string;
+            }[];
+        } | {
+            success: false;
+            error: string;
+        }>;
     }>;
     databricks: ConnectorPlugin<{
         host: ParameterDefinition;
@@ -587,6 +615,20 @@ declare const bigqueryConnector: ConnectorPlugin<{
         success: false;
         error: string;
     }>;
+    listDatasets: ConnectorTool<{
+        connectionId: string;
+        toolUseIntent?: string | undefined;
+    }, {
+        success: true;
+        datasets: {
+            datasetId: string;
+            location: string;
+            friendlyName: string;
+        }[];
+    } | {
+        success: false;
+        error: string;
+    }>;
 }>;
 
 declare const bigqueryOauthConnector: ConnectorPlugin<{
@@ -618,6 +660,20 @@ declare const bigqueryOauthConnector: ConnectorPlugin<{
         success: false;
         error: string;
     }>;
+    listDatasets: ConnectorTool<{
+        connectionId: string;
+        toolUseIntent?: string | undefined;
+    }, {
+        success: true;
+        datasets: {
+            datasetId: string;
+            location: string;
+            friendlyName: string;
+        }[];
+    } | {
+        success: false;
+        error: string;
+    }>;
 }>;
 
 declare const awsAthenaConnector: ConnectorPlugin<{

package/dist/index.js

@@ -679,7 +679,7 @@ var snowflakePatConnector = new ConnectorPlugin({
   description: "Connect to Snowflake using a Personal Access Token (PAT).",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/6oyVtAcP3pMlXaOrts9unk/b7a9dc25d15c388b66e983041b855447/snowflake.svg",
   parameters: parameters8,
-  releaseFlag: { dev1: true, dev2: false, prod: false },
+  releaseFlag: { dev1: true, dev2: true, prod: true },
   setup: snowflakeSetup,
   systemPrompt: `## Snowflake SQL Notes
 - Use fully qualified names DB.SCHEMA.TABLE for table references
@@ -1135,8 +1135,68 @@ var listProjectsTool = new ConnectorTool({
   }
 });
 
+// src/connectors/bigquery/tools/list-datasets.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),
+    datasets: z6.array(
+      z6.object({
+        datasetId: z6.string(),
+        location: z6.string(),
+        friendlyName: z6.string()
+      })
+    )
+  }),
+  z6.object({
+    success: z6.literal(false),
+    error: z6.string()
+  })
+]);
+var listDatasetsTool = new ConnectorTool({
+  name: "listDatasets",
+  description: `List datasets in the selected BigQuery project. Returns dataset IDs, locations, 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 { BigQuery } = await import("@google-cloud/bigquery");
+      const projectId = parameters11.projectId.getValue(connection);
+      const serviceAccountJsonBase64 = parameters11.serviceAccountKeyJsonBase64.getValue(connection);
+      const credentials = JSON.parse(
+        Buffer.from(serviceAccountJsonBase64, "base64").toString("utf-8")
+      );
+      const bq = new BigQuery({ projectId, credentials });
+      const [datasetsResult] = await bq.getDatasets();
+      const datasets = datasetsResult.map((ds) => ({
+        datasetId: ds.id ?? "",
+        location: ds.metadata?.location ?? "",
+        friendlyName: ds.metadata?.friendlyName ?? ds.id ?? ""
+      })).filter((ds) => ds.datasetId !== "");
+      return { success: true, datasets };
+    } 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 listDatasetsToolName = `bigquery_${listDatasetsTool.name}`;
 var bigquerySetup = new ConnectorSetup({
   ja: `## BigQuery \u30BB\u30C3\u30C8\u30A2\u30C3\u30D7\u624B\u9806
 
@@ -1152,11 +1212,11 @@ var bigquerySetup = new ConnectorSetup({
 3. \u30E6\u30FC\u30B6\u30FC\u304C\u9078\u629E\u3057\u305F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u306E \`label\` \u304C\u30E1\u30C3\u30BB\u30FC\u30B8\u3068\u3057\u3066\u5C4A\u304F\u306E\u3067\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\u3001\u9078\u629E\u3055\u308C\u305F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u5185\u306E\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B
+1. \`${listDatasetsToolName}\` \u3092\u547C\u3073\u51FA\u3057\u3066\u3001\u9078\u629E\u3055\u308C\u305F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u5185\u306E\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B
 2. \`askUserQuestion\` \u3092\u547C\u3073\u51FA\u3059\uFF08multiSelect: true\uFF09:
    - \`question\`: \u4F7F\u7528\u3059\u308B\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u3092\u9078\u629E\u3059\u308B\u3088\u3046\u4FC3\u3059\u3002\u300C\u4E0A\u8A18\u4EE5\u5916\u306E\u30D1\u30D6\u30EA\u30C3\u30AF\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u306F \`bigquery-public-data.{\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u540D}\` \u306E\u5F62\u5F0F\u3067\u81EA\u7531\u5165\u529B\u3067\u304D\u307E\u3059\u300D\u3068\u660E\u8A18\u3059\u308B
    - \`options\`: \u4EE5\u4E0B\u3092\u9806\u756A\u306B\u7D50\u5408\u3059\u308B
-     1. \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u5185\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8: \u5404 \`{ label: schema_name, description: schema_name }\`
+     1. \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u5185\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8: \u5404 \`{ label: datasetId, description: location }\`
      2. \u30D1\u30D6\u30EA\u30C3\u30AF\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\uFF08\u56FA\u5B9A\u3067\u672B\u5C3E\u306B\u8FFD\u52A0\uFF09:
         - \`{ label: "bigquery-public-data.thelook_ecommerce (public dataset)", description: "EC\u30B5\u30A4\u30C8\u306E\u30C7\u30E2\u30C7\u30FC\u30BF" }\`
         - \`{ label: "bigquery-public-data.google_analytics_sample (public dataset)", description: "Google Analytics \u30B5\u30F3\u30D7\u30EB" }\`
@@ -1198,11 +1258,11 @@ Follow these steps to set up the BigQuery connection.
 3. The \`label\` of the user's selected project will arrive as a message. Proceed to Step 2
 
 #### Step 2: Dataset Selection
-1. Run \`SELECT schema_name FROM INFORMATION_SCHEMA.SCHEMATA\` to get the list of datasets in the selected project
+1. Call \`${listDatasetsToolName}\` to get the list of datasets in the selected project
 2. Call \`askUserQuestion\` (multiSelect: true):
    - \`question\`: Ask the user to select datasets. State that other public datasets can be entered freely in the format \`bigquery-public-data.{dataset_name}\`
    - \`options\`: Concatenate in order:
-     1. Project datasets: each \`{ label: schema_name, description: schema_name }\`
+     1. Project datasets: each \`{ label: datasetId, description: location }\`
      2. Public datasets (always append at the end):
         - \`{ label: "bigquery-public-data.thelook_ecommerce (public dataset)", description: "E-commerce demo data" }\`
         - \`{ label: "bigquery-public-data.google_analytics_sample (public dataset)", description: "Google Analytics sample" }\`
@@ -1233,27 +1293,27 @@ Follow these steps to set up the BigQuery connection.
 });
 
 // src/connectors/bigquery/tools/execute-query.ts
-import { z as z6 } from "zod";
+import { z as z7 } from "zod";
 var MAX_ROWS5 = 500;
-var inputSchema6 = z6.object({
-  toolUseIntent: z6.string().optional().describe(
+var inputSchema7 = z7.object({
+  toolUseIntent: z7.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"),
-  sql: z6.string().describe(
+  connectionId: z7.string().describe("ID of the BigQuery connection to use"),
+  sql: z7.string().describe(
     "BigQuery SQL (GoogleSQL) query. Use backtick-quoted fully qualified names `project.dataset.table` for table references."
   )
 });
-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()))
+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()))
   }),
-  z6.object({
-    success: z6.literal(false),
-    error: z6.string()
+  z7.object({
+    success: z7.literal(false),
+    error: z7.string()
   })
 ]);
 var executeQueryTool5 = new ConnectorTool({
@@ -1261,8 +1321,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: inputSchema6,
-  outputSchema: outputSchema6,
+  inputSchema: inputSchema7,
+  outputSchema: outputSchema7,
   async execute({ connectionId, sql }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -1300,7 +1360,7 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
 });
 
 // src/connectors/bigquery/index.ts
-var tools5 = { executeQuery: executeQueryTool5, listProjects: listProjectsTool };
+var tools5 = { executeQuery: executeQueryTool5, listProjects: listProjectsTool, listDatasets: listDatasetsTool };
 var bigqueryConnector = new ConnectorPlugin({
   slug: "bigquery",
   authType: null,
@@ -1313,9 +1373,9 @@ var bigqueryConnector = new ConnectorPlugin({
   systemPrompt: `## BigQuery SQL Notes
 - Use backtick-quoted fully qualified names \`project.dataset.table\` for table references
 - Use INFORMATION_SCHEMA for schema exploration
-  - List datasets: \`SELECT schema_name FROM \\\`project_id\\\`.INFORMATION_SCHEMA.SCHEMATA\`
   - List tables: \`SELECT table_name FROM \\\`project_id.dataset\\\`.INFORMATION_SCHEMA.TABLES\`
   - List columns: \`SELECT column_name, data_type FROM \\\`project_id.dataset\\\`.INFORMATION_SCHEMA.COLUMNS WHERE table_name = 'xxx'\`
+- To list datasets, use the listDatasets tool instead of INFORMATION_SCHEMA.SCHEMATA
 - Always specify project_id explicitly in queries`,
   tools: tools5,
   async checkConnection(params, _config) {
@@ -1352,7 +1412,7 @@ var bigqueryConnector = new ConnectorPlugin({
 });
 
 // src/connectors/bigquery-oauth/tools/list-projects.ts
-import { z as z7 } from "zod";
+import { z as z8 } from "zod";
 var REQUEST_TIMEOUT_MS = 6e4;
 var cachedToken = null;
 async function getProxyToken(config) {
@@ -1383,32 +1443,32 @@ async function getProxyToken(config) {
   };
   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")
+  connectionId: z8.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()
+var outputSchema8 = z8.discriminatedUnion("success", [
+  z8.object({
+    success: z8.literal(true),
+    projects: z8.array(
+      z8.object({
+        projectId: z8.string(),
+        friendlyName: z8.string()
       })
     )
   }),
-  z7.object({
-    success: z7.literal(false),
-    error: z7.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: inputSchema7,
-  outputSchema: outputSchema7,
+  inputSchema: inputSchema8,
+  outputSchema: outputSchema8,
   async execute({ connectionId }, connections, config) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -1458,8 +1518,134 @@ var listProjectsTool2 = new ConnectorTool({
   }
 });
 
+// src/connectors/bigquery-oauth/tools/list-datasets.ts
+import { z as z9 } from "zod";
+
+// src/connectors/bigquery-oauth/parameters.ts
+var parameters12 = {
+  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_OAUTH_PROJECT_ID",
+    type: "text",
+    secret: false,
+    required: false
+  })
+};
+
+// src/connectors/bigquery-oauth/tools/list-datasets.ts
+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 inputSchema9 = z9.object({
+  toolUseIntent: z9.string().optional().describe(
+    "Brief description of what you intend to accomplish with this tool call"
+  ),
+  connectionId: z9.string().describe("ID of the BigQuery OAuth connection to use")
+});
+var outputSchema9 = z9.discriminatedUnion("success", [
+  z9.object({
+    success: z9.literal(true),
+    datasets: z9.array(
+      z9.object({
+        datasetId: z9.string(),
+        location: z9.string(),
+        friendlyName: z9.string()
+      })
+    )
+  }),
+  z9.object({
+    success: z9.literal(false),
+    error: z9.string()
+  })
+]);
+var listDatasetsTool2 = new ConnectorTool({
+  name: "listDatasets",
+  description: `List datasets in the selected BigQuery project. Returns dataset IDs, locations, and friendly names.`,
+  inputSchema: inputSchema9,
+  outputSchema: outputSchema9,
+  async execute({ connectionId }, connections, config) {
+    const connection = connections.find((c) => c.id === connectionId);
+    if (!connection) {
+      return {
+        success: false,
+        error: `Connection ${connectionId} not found`
+      };
+    }
+    const gcpProjectId = parameters12.projectId.getValue(connection);
+    console.log(
+      `[connector-query] bigquery-oauth/${connection.name}: listDatasets`
+    );
+    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/${gcpProjectId}/datasets`,
+            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 datasets = (data.datasets ?? []).map((ds) => ({
+          datasetId: ds.datasetReference?.datasetId ?? "",
+          location: ds.location ?? "",
+          friendlyName: ds.friendlyName ?? ds.datasetReference?.datasetId ?? ""
+        })).filter((ds) => ds.datasetId !== "");
+        return { success: true, datasets };
+      } 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 listDatasetsToolName2 = `bigquery-oauth_${listDatasetsTool2.name}`;
 var bigquerySetup2 = new ConnectorSetup({
   ja: `## BigQuery \u30BB\u30C3\u30C8\u30A2\u30C3\u30D7\u624B\u9806
 
@@ -1475,11 +1661,11 @@ var bigquerySetup2 = new ConnectorSetup({
 3. \u30E6\u30FC\u30B6\u30FC\u304C\u9078\u629E\u3057\u305F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u306E \`label\` \u304C\u30E1\u30C3\u30BB\u30FC\u30B8\u3068\u3057\u3066\u5C4A\u304F\u306E\u3067\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\u3001\u9078\u629E\u3055\u308C\u305F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u5185\u306E\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B
+1. \`${listDatasetsToolName2}\` \u3092\u547C\u3073\u51FA\u3057\u3066\u3001\u9078\u629E\u3055\u308C\u305F\u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u5185\u306E\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u4E00\u89A7\u3092\u53D6\u5F97\u3059\u308B
 2. \`askUserQuestion\` \u3092\u547C\u3073\u51FA\u3059\uFF08multiSelect: true\uFF09:
    - \`question\`: \u4F7F\u7528\u3059\u308B\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u3092\u9078\u629E\u3059\u308B\u3088\u3046\u4FC3\u3059\u3002\u300C\u4E0A\u8A18\u4EE5\u5916\u306E\u30D1\u30D6\u30EA\u30C3\u30AF\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u306F \`bigquery-public-data.{\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\u540D}\` \u306E\u5F62\u5F0F\u3067\u81EA\u7531\u5165\u529B\u3067\u304D\u307E\u3059\u300D\u3068\u660E\u8A18\u3059\u308B
    - \`options\`: \u4EE5\u4E0B\u3092\u9806\u756A\u306B\u7D50\u5408\u3059\u308B
-     1. \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u5185\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8: \u5404 \`{ label: schema_name, description: schema_name }\`
+     1. \u30D7\u30ED\u30B8\u30A7\u30AF\u30C8\u5185\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8: \u5404 \`{ label: datasetId, description: location }\`
      2. \u30D1\u30D6\u30EA\u30C3\u30AF\u30C7\u30FC\u30BF\u30BB\u30C3\u30C8\uFF08\u56FA\u5B9A\u3067\u672B\u5C3E\u306B\u8FFD\u52A0\uFF09:
         - \`{ label: "bigquery-public-data.thelook_ecommerce (public dataset)", description: "EC\u30B5\u30A4\u30C8\u306E\u30C7\u30E2\u30C7\u30FC\u30BF" }\`
         - \`{ label: "bigquery-public-data.google_analytics_sample (public dataset)", description: "Google Analytics \u30B5\u30F3\u30D7\u30EB" }\`
@@ -1521,11 +1707,11 @@ Follow these steps to set up the BigQuery connection.
 3. The \`label\` of the user's selected project will arrive as a message. Proceed to Step 2
 
 #### Step 2: Dataset Selection
-1. Run \`SELECT schema_name FROM INFORMATION_SCHEMA.SCHEMATA\` to get the list of datasets in the selected project
+1. Call \`${listDatasetsToolName2}\` to get the list of datasets in the selected project
 2. Call \`askUserQuestion\` (multiSelect: true):
    - \`question\`: Ask the user to select datasets. State that other public datasets can be entered freely in the format \`bigquery-public-data.{dataset_name}\`
    - \`options\`: Concatenate in order:
-     1. Project datasets: each \`{ label: schema_name, description: schema_name }\`
+     1. Project datasets: each \`{ label: datasetId, description: location }\`
      2. Public datasets (always append at the end):
         - \`{ label: "bigquery-public-data.thelook_ecommerce (public dataset)", description: "E-commerce demo data" }\`
         - \`{ label: "bigquery-public-data.google_analytics_sample (public dataset)", description: "Google Analytics sample" }\`
@@ -1555,27 +1741,14 @@ Follow these steps to set up the BigQuery connection.
 - Skip unnecessary explanations and proceed efficiently`
 });
 
-// src/connectors/bigquery-oauth/parameters.ts
-var parameters12 = {
-  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_OAUTH_PROJECT_ID",
-    type: "text",
-    secret: false,
-    required: false
-  })
-};
-
 // src/connectors/bigquery-oauth/tools/execute-query.ts
-import { z as z8 } from "zod";
+import { z as z10 } 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;
+var REQUEST_TIMEOUT_MS3 = 6e4;
+var cachedToken3 = null;
+async function getProxyToken3(config) {
+  if (cachedToken3 && cachedToken3.expiresAt > Date.now() + 6e4) {
+    return cachedToken3.token;
   }
   const url = `${config.appApiBaseUrl}/v0/database/${config.projectId}/environment/${config.environmentId}/oauth-request-proxy-token`;
   const res = await fetch(url, {
@@ -1595,31 +1768,31 @@ async function getProxyToken2(config) {
     throw new Error(`Failed to get proxy token: HTTP ${res.status} ${errorText}`);
   }
   const data = await res.json();
-  cachedToken2 = {
+  cachedToken3 = {
     token: data.token,
     expiresAt: new Date(data.expiresAt).getTime()
   };
   return data.token;
 }
-var inputSchema8 = z8.object({
-  toolUseIntent: z8.string().optional().describe(
+var inputSchema10 = z10.object({
+  toolUseIntent: z10.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"),
-  sql: z8.string().describe(
+  connectionId: z10.string().describe("ID of the BigQuery OAuth connection to use"),
+  sql: z10.string().describe(
     "BigQuery SQL (GoogleSQL) query. Use backtick-quoted fully qualified names `project.dataset.table` for table references."
   )
 });
-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()))
+var outputSchema10 = z10.discriminatedUnion("success", [
+  z10.object({
+    success: z10.literal(true),
+    rowCount: z10.number(),
+    truncated: z10.boolean(),
+    rows: z10.array(z10.record(z10.string(), z10.unknown()))
   }),
-  z8.object({
-    success: z8.literal(false),
-    error: z8.string()
+  z10.object({
+    success: z10.literal(false),
+    error: z10.string()
   })
 ]);
 function parseQueryResponse(data) {
@@ -1640,8 +1813,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: inputSchema8,
-  outputSchema: outputSchema8,
+  inputSchema: inputSchema10,
+  outputSchema: outputSchema10,
   async execute({ connectionId, sql }, connections, config) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -1655,11 +1828,11 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
       `[connector-query] bigquery-oauth/${connection.name}: ${sql}`
     );
     try {
-      const token = await getProxyToken2(config.oauthProxy);
+      const token = await getProxyToken3(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_MS2);
+      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS3);
       try {
         const response = await fetch(proxyUrl, {
           method: "POST",
@@ -1698,7 +1871,7 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
 });
 
 // src/connectors/bigquery-oauth/index.ts
-var tools6 = { executeQuery: executeQueryTool6, listProjects: listProjectsTool2 };
+var tools6 = { executeQuery: executeQueryTool6, listProjects: listProjectsTool2, listDatasets: listDatasetsTool2 };
 function parseQueryResponse2(data) {
   const schema = data.schema;
   const rawRows = data.rows;
@@ -1718,7 +1891,7 @@ var bigqueryOauthConnector = new ConnectorPlugin({
   description: "Connect to Google BigQuery for data warehouse and analytics using OAuth.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/6nlehQyOmdbktG5hOYkYMr/6ca559140d5ddc7dadc5eac88858a563/bigquery.svg",
   parameters: parameters12,
-  releaseFlag: { dev1: true, dev2: false, prod: false },
+  releaseFlag: { dev1: true, dev2: true, prod: true },
   setup: bigquerySetup2,
   proxyPolicy: {
     allowlist: [
@@ -1729,9 +1902,9 @@ var bigqueryOauthConnector = new ConnectorPlugin({
   systemPrompt: `## BigQuery SQL Notes (OAuth)
 - Use backtick-quoted fully qualified names \`project.dataset.table\` for table references
 - Use INFORMATION_SCHEMA for schema exploration
-  - List datasets: \`SELECT schema_name FROM \\\`project_id\\\`.INFORMATION_SCHEMA.SCHEMATA\`
   - List tables: \`SELECT table_name FROM \\\`project_id.dataset\\\`.INFORMATION_SCHEMA.TABLES\`
   - List columns: \`SELECT column_name, data_type FROM \\\`project_id.dataset\\\`.INFORMATION_SCHEMA.COLUMNS WHERE table_name = 'xxx'\`
+- To list datasets, use the listDatasets tool instead of INFORMATION_SCHEMA.SCHEMATA
 - Always specify project_id explicitly in queries`,
   tools: tools6,
   async checkConnection(params, config) {
@@ -1822,27 +1995,27 @@ var parameters13 = {
 };
 
 // src/connectors/aws-athena/tools/execute-query.ts
-import { z as z9 } from "zod";
+import { z as z11 } from "zod";
 var MAX_ROWS7 = 500;
 var POLL_INTERVAL_MS = 1e3;
 var POLL_TIMEOUT_MS = 12e4;
-var inputSchema9 = z9.object({
-  toolUseIntent: z9.string().optional().describe(
+var inputSchema11 = z11.object({
+  toolUseIntent: z11.string().optional().describe(
     "Brief description of what you intend to accomplish with this tool call"
   ),
-  connectionId: z9.string().describe("ID of the AWS Athena connection to use"),
-  sql: z9.string().describe("Athena SQL query (Presto/Trino based)")
+  connectionId: z11.string().describe("ID of the AWS Athena connection to use"),
+  sql: z11.string().describe("Athena SQL query (Presto/Trino based)")
 });
-var outputSchema9 = z9.discriminatedUnion("success", [
-  z9.object({
-    success: z9.literal(true),
-    rowCount: z9.number(),
-    truncated: z9.boolean(),
-    rows: z9.array(z9.record(z9.string(), z9.unknown()))
+var outputSchema11 = z11.discriminatedUnion("success", [
+  z11.object({
+    success: z11.literal(true),
+    rowCount: z11.number(),
+    truncated: z11.boolean(),
+    rows: z11.array(z11.record(z11.string(), z11.unknown()))
   }),
-  z9.object({
-    success: z9.literal(false),
-    error: z9.string()
+  z11.object({
+    success: z11.literal(false),
+    error: z11.string()
   })
 ]);
 var executeQueryTool7 = new ConnectorTool({
@@ -1850,8 +2023,8 @@ var executeQueryTool7 = new ConnectorTool({
   description: `Execute SQL against AWS Athena. Returns up to ${MAX_ROWS7} rows.
 Use for: schema exploration (SHOW DATABASES/TABLES, DESCRIBE TABLE), data sampling, analytical queries on S3 data.
 Avoid loading large amounts of data; always include LIMIT in queries.`,
-  inputSchema: inputSchema9,
-  outputSchema: outputSchema9,
+  inputSchema: inputSchema11,
+  outputSchema: outputSchema11,
   async execute({ connectionId, sql }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -2151,29 +2324,29 @@ var parameters14 = {
 };
 
 // src/connectors/redshift/tools/execute-query.ts
-import { z as z10 } from "zod";
+import { z as z12 } from "zod";
 var MAX_ROWS8 = 500;
 var POLL_INTERVAL_MS2 = 1e3;
 var POLL_TIMEOUT_MS2 = 12e4;
-var inputSchema10 = z10.object({
-  toolUseIntent: z10.string().optional().describe(
+var inputSchema12 = z12.object({
+  toolUseIntent: z12.string().optional().describe(
     "Brief description of what you intend to accomplish with this tool call"
   ),
-  connectionId: z10.string().describe("ID of the Redshift connection to use"),
-  sql: z10.string().describe(
+  connectionId: z12.string().describe("ID of the Redshift connection to use"),
+  sql: z12.string().describe(
     "SQL query to execute against Amazon Redshift."
   )
 });
-var outputSchema10 = z10.discriminatedUnion("success", [
-  z10.object({
-    success: z10.literal(true),
-    rowCount: z10.number(),
-    truncated: z10.boolean(),
-    rows: z10.array(z10.record(z10.string(), z10.unknown()))
+var outputSchema12 = z12.discriminatedUnion("success", [
+  z12.object({
+    success: z12.literal(true),
+    rowCount: z12.number(),
+    truncated: z12.boolean(),
+    rows: z12.array(z12.record(z12.string(), z12.unknown()))
   }),
-  z10.object({
-    success: z10.literal(false),
-    error: z10.string()
+  z12.object({
+    success: z12.literal(false),
+    error: z12.string()
   })
 ]);
 var executeQueryTool8 = new ConnectorTool({
@@ -2181,8 +2354,8 @@ var executeQueryTool8 = new ConnectorTool({
   description: `Execute SQL against Amazon Redshift. Returns up to ${MAX_ROWS8} rows.
 Use for: schema exploration, data sampling, analytical queries.
 Avoid loading large amounts of data; always include LIMIT in queries.`,
-  inputSchema: inputSchema10,
-  outputSchema: outputSchema10,
+  inputSchema: inputSchema12,
+  outputSchema: outputSchema12,
   async execute({ connectionId, sql }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -2442,25 +2615,25 @@ var parameters15 = {
 };
 
 // src/connectors/databricks/tools/execute-query.ts
-import { z as z11 } from "zod";
+import { z as z13 } from "zod";
 var MAX_ROWS9 = 500;
-var inputSchema11 = z11.object({
-  toolUseIntent: z11.string().optional().describe(
+var inputSchema13 = z13.object({
+  toolUseIntent: z13.string().optional().describe(
     "Brief description of what you intend to accomplish with this tool call"
   ),
-  connectionId: z11.string().describe("ID of the Databricks connection to use"),
-  sql: z11.string().describe("Databricks SQL query (Spark SQL based)")
+  connectionId: z13.string().describe("ID of the Databricks connection to use"),
+  sql: z13.string().describe("Databricks SQL query (Spark SQL based)")
 });
-var outputSchema11 = z11.discriminatedUnion("success", [
-  z11.object({
-    success: z11.literal(true),
-    rowCount: z11.number(),
-    truncated: z11.boolean(),
-    rows: z11.array(z11.record(z11.string(), z11.unknown()))
+var outputSchema13 = z13.discriminatedUnion("success", [
+  z13.object({
+    success: z13.literal(true),
+    rowCount: z13.number(),
+    truncated: z13.boolean(),
+    rows: z13.array(z13.record(z13.string(), z13.unknown()))
   }),
-  z11.object({
-    success: z11.literal(false),
-    error: z11.string()
+  z13.object({
+    success: z13.literal(false),
+    error: z13.string()
   })
 ]);
 var executeQueryTool9 = new ConnectorTool({
@@ -2468,8 +2641,8 @@ var executeQueryTool9 = new ConnectorTool({
   description: `Execute SQL against Databricks. Returns up to ${MAX_ROWS9} rows.
 Use for: schema exploration (SHOW CATALOGS/DATABASES/TABLES, DESCRIBE TABLE), data sampling, analytical queries.
 Avoid loading large amounts of data; always include LIMIT in queries.`,
-  inputSchema: inputSchema11,
-  outputSchema: outputSchema11,
+  inputSchema: inputSchema13,
+  outputSchema: outputSchema13,
   async execute({ connectionId, sql }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -2595,25 +2768,25 @@ var databricksConnector = new ConnectorPlugin({
 });
 
 // src/connectors/airtable/tools/request.ts
-import { z as z12 } from "zod";
+import { z as z14 } from "zod";
 var BASE_URL = "https://api.airtable.com/v0/";
-var REQUEST_TIMEOUT_MS3 = 6e4;
-var inputSchema12 = z12.object({
-  toolUseIntent: z12.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
-  connectionId: z12.string().describe("ID of the Airtable connection to use"),
-  method: z12.enum(["GET", "POST", "PATCH", "DELETE"]).describe("HTTP method"),
-  path: z12.string().describe("API path (e.g., '{baseId}/{tableIdOrName}', 'meta/bases/{baseId}/tables'). {baseId} is automatically replaced."),
-  body: z12.record(z12.string(), z12.unknown()).optional().describe("Request body (JSON)")
+var REQUEST_TIMEOUT_MS4 = 6e4;
+var inputSchema14 = z14.object({
+  toolUseIntent: z14.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
+  connectionId: z14.string().describe("ID of the Airtable connection to use"),
+  method: z14.enum(["GET", "POST", "PATCH", "DELETE"]).describe("HTTP method"),
+  path: z14.string().describe("API path (e.g., '{baseId}/{tableIdOrName}', 'meta/bases/{baseId}/tables'). {baseId} is automatically replaced."),
+  body: z14.record(z14.string(), z14.unknown()).optional().describe("Request body (JSON)")
 });
-var outputSchema12 = z12.discriminatedUnion("success", [
-  z12.object({
-    success: z12.literal(true),
-    status: z12.number(),
-    data: z12.record(z12.string(), z12.unknown())
+var outputSchema14 = z14.discriminatedUnion("success", [
+  z14.object({
+    success: z14.literal(true),
+    status: z14.number(),
+    data: z14.record(z14.string(), z14.unknown())
   }),
-  z12.object({
-    success: z12.literal(false),
-    error: z12.string()
+  z14.object({
+    success: z14.literal(false),
+    error: z14.string()
   })
 ]);
 var requestTool = new ConnectorTool({
@@ -2621,8 +2794,8 @@ var requestTool = new ConnectorTool({
   description: `Send authenticated requests to the Airtable API.
 Authentication is handled automatically using the API Key.
 {baseId} in the path is automatically replaced with the connection's base-id.`,
-  inputSchema: inputSchema12,
-  outputSchema: outputSchema12,
+  inputSchema: inputSchema14,
+  outputSchema: outputSchema14,
   async execute({ connectionId, method, path, body }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -2635,7 +2808,7 @@ Authentication is handled automatically using the API Key.
       const resolvedPath = path.replace(/\{baseId\}/g, baseId);
       const url = `${BASE_URL}${resolvedPath}`;
       const controller = new AbortController();
-      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS3);
+      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS4);
       try {
         const response = await fetch(url, {
           method,
@@ -2726,25 +2899,25 @@ await airtable.updateRecords("Tasks", [
 });
 
 // src/connectors/google-analytics/tools/request.ts
-import { z as z13 } from "zod";
+import { z as z15 } from "zod";
 var BASE_URL2 = "https://analyticsdata.googleapis.com/v1beta/";
-var REQUEST_TIMEOUT_MS4 = 6e4;
-var inputSchema13 = z13.object({
-  toolUseIntent: z13.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
-  connectionId: z13.string().describe("ID of the Google Analytics connection to use"),
-  method: z13.enum(["GET", "POST"]).describe("HTTP method"),
-  path: z13.string().describe("API path (e.g., 'properties/{propertyId}:runReport'). {propertyId} is automatically replaced."),
-  body: z13.record(z13.string(), z13.unknown()).optional().describe("POST request body (JSON)")
+var REQUEST_TIMEOUT_MS5 = 6e4;
+var inputSchema15 = z15.object({
+  toolUseIntent: z15.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
+  connectionId: z15.string().describe("ID of the Google Analytics connection to use"),
+  method: z15.enum(["GET", "POST"]).describe("HTTP method"),
+  path: z15.string().describe("API path (e.g., 'properties/{propertyId}:runReport'). {propertyId} is automatically replaced."),
+  body: z15.record(z15.string(), z15.unknown()).optional().describe("POST request body (JSON)")
 });
-var outputSchema13 = z13.discriminatedUnion("success", [
-  z13.object({
-    success: z13.literal(true),
-    status: z13.number(),
-    data: z13.record(z13.string(), z13.unknown())
+var outputSchema15 = z15.discriminatedUnion("success", [
+  z15.object({
+    success: z15.literal(true),
+    status: z15.number(),
+    data: z15.record(z15.string(), z15.unknown())
   }),
-  z13.object({
-    success: z13.literal(false),
-    error: z13.string()
+  z15.object({
+    success: z15.literal(false),
+    error: z15.string()
   })
 ]);
 var requestTool2 = new ConnectorTool({
@@ -2752,8 +2925,8 @@ var requestTool2 = new ConnectorTool({
   description: `Send authenticated requests to the Google Analytics Data API.
 Authentication is handled automatically using a service account.
 {propertyId} in the path is automatically replaced with the connection's property-id.`,
-  inputSchema: inputSchema13,
-  outputSchema: outputSchema13,
+  inputSchema: inputSchema15,
+  outputSchema: outputSchema15,
   async execute({ connectionId, method, path, body }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -2778,7 +2951,7 @@ Authentication is handled automatically using a service account.
       const resolvedPath = path.replace(/\{propertyId\}/g, propertyId);
       const url = `${BASE_URL2}${resolvedPath}`;
       const controller = new AbortController();
-      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS4);
+      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS5);
       try {
         const response = await fetch(url, {
           method,
@@ -2879,32 +3052,32 @@ const realtime = await ga.runRealtimeReport({
 });
 
 // src/connectors/kintone/tools/request.ts
-import { z as z14 } from "zod";
-var REQUEST_TIMEOUT_MS5 = 6e4;
-var inputSchema14 = z14.object({
-  toolUseIntent: z14.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
-  connectionId: z14.string().describe("ID of the kintone connection to use"),
-  method: z14.enum(["GET", "POST", "PUT", "DELETE"]).describe("HTTP method"),
-  path: z14.string().describe("API path (e.g., 'apps.json', 'records.json?app=1&query=...')"),
-  body: z14.record(z14.string(), z14.unknown()).optional().describe("Request body (JSON)")
+import { z as z16 } from "zod";
+var REQUEST_TIMEOUT_MS6 = 6e4;
+var inputSchema16 = z16.object({
+  toolUseIntent: z16.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
+  connectionId: z16.string().describe("ID of the kintone connection to use"),
+  method: z16.enum(["GET", "POST", "PUT", "DELETE"]).describe("HTTP method"),
+  path: z16.string().describe("API path (e.g., 'apps.json', 'records.json?app=1&query=...')"),
+  body: z16.record(z16.string(), z16.unknown()).optional().describe("Request body (JSON)")
 });
-var outputSchema14 = z14.discriminatedUnion("success", [
-  z14.object({
-    success: z14.literal(true),
-    status: z14.number(),
-    data: z14.record(z14.string(), z14.unknown())
+var outputSchema16 = z16.discriminatedUnion("success", [
+  z16.object({
+    success: z16.literal(true),
+    status: z16.number(),
+    data: z16.record(z16.string(), z16.unknown())
   }),
-  z14.object({
-    success: z14.literal(false),
-    error: z14.string()
+  z16.object({
+    success: z16.literal(false),
+    error: z16.string()
   })
 ]);
 var requestTool3 = new ConnectorTool({
   name: "request",
   description: `Send authenticated requests to the kintone REST API.
 Authentication is handled automatically using username and password.`,
-  inputSchema: inputSchema14,
-  outputSchema: outputSchema14,
+  inputSchema: inputSchema16,
+  outputSchema: outputSchema16,
   async execute({ connectionId, method, path, body }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -2918,7 +3091,7 @@ Authentication is handled automatically using username and password.`,
       const authToken = Buffer.from(`${username}:${password}`).toString("base64");
       const url = `${baseUrl.replace(/\/+$/, "")}/k/v1/${path}`;
       const controller = new AbortController();
-      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS5);
+      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS6);
       try {
         const headers = {
           "X-Cybozu-Authorization": authToken
@@ -3015,33 +3188,33 @@ const { apps } = await kintone.listApps();
 });
 
 // src/connectors/wix-store/tools/request.ts
-import { z as z15 } from "zod";
+import { z as z17 } from "zod";
 var BASE_URL3 = "https://www.wixapis.com/";
-var REQUEST_TIMEOUT_MS6 = 6e4;
-var inputSchema15 = z15.object({
-  toolUseIntent: z15.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
-  connectionId: z15.string().describe("ID of the Wix Store connection to use"),
-  method: z15.enum(["GET", "POST"]).describe("HTTP method"),
-  path: z15.string().describe("API path (e.g., 'stores/v1/products/query', 'stores/v2/orders/query')"),
-  body: z15.record(z15.string(), z15.unknown()).optional().describe("Request body (JSON)")
+var REQUEST_TIMEOUT_MS7 = 6e4;
+var inputSchema17 = z17.object({
+  toolUseIntent: z17.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
+  connectionId: z17.string().describe("ID of the Wix Store connection to use"),
+  method: z17.enum(["GET", "POST"]).describe("HTTP method"),
+  path: z17.string().describe("API path (e.g., 'stores/v1/products/query', 'stores/v2/orders/query')"),
+  body: z17.record(z17.string(), z17.unknown()).optional().describe("Request body (JSON)")
 });
-var outputSchema15 = z15.discriminatedUnion("success", [
-  z15.object({
-    success: z15.literal(true),
-    status: z15.number(),
-    data: z15.record(z15.string(), z15.unknown())
+var outputSchema17 = z17.discriminatedUnion("success", [
+  z17.object({
+    success: z17.literal(true),
+    status: z17.number(),
+    data: z17.record(z17.string(), z17.unknown())
   }),
-  z15.object({
-    success: z15.literal(false),
-    error: z15.string()
+  z17.object({
+    success: z17.literal(false),
+    error: z17.string()
   })
 ]);
 var requestTool4 = new ConnectorTool({
   name: "request",
   description: `Send authenticated requests to the Wix Store API.
 Authentication is handled automatically using the API Key and Site ID.`,
-  inputSchema: inputSchema15,
-  outputSchema: outputSchema15,
+  inputSchema: inputSchema17,
+  outputSchema: outputSchema17,
   async execute({ connectionId, method, path, body }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -3053,7 +3226,7 @@ Authentication is handled automatically using the API Key and Site ID.`,
       const siteId = parameters4.siteId.getValue(connection);
       const url = `${BASE_URL3}${path}`;
       const controller = new AbortController();
-      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS6);
+      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS7);
       try {
         const response = await fetch(url, {
           method,
@@ -3151,27 +3324,27 @@ const { collections } = await wix.queryCollections({ paging: { limit: 50 } });
 });
 
 // src/connectors/dbt/tools/request.ts
-import { z as z16 } from "zod";
-var REQUEST_TIMEOUT_MS7 = 6e4;
+import { z as z18 } from "zod";
+var REQUEST_TIMEOUT_MS8 = 6e4;
 function resolveGraphqlEndpoint(host) {
   if (host.includes("emea")) return "https://metadata.emea.dbt.com/graphql";
   if (host.includes(".au.")) return "https://metadata.au.dbt.com/graphql";
   return "https://metadata.cloud.getdbt.com/graphql";
 }
-var inputSchema16 = z16.object({
-  toolUseIntent: z16.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
-  connectionId: z16.string().describe("ID of the dbt Cloud connection to use"),
-  query: z16.string().describe("GraphQL query"),
-  variables: z16.record(z16.string(), z16.unknown()).optional().describe("GraphQL variables (JSON)")
+var inputSchema18 = z18.object({
+  toolUseIntent: z18.string().optional().describe("Brief description of what you intend to accomplish with this tool call"),
+  connectionId: z18.string().describe("ID of the dbt Cloud connection to use"),
+  query: z18.string().describe("GraphQL query"),
+  variables: z18.record(z18.string(), z18.unknown()).optional().describe("GraphQL variables (JSON)")
 });
-var outputSchema16 = z16.discriminatedUnion("success", [
-  z16.object({
-    success: z16.literal(true),
-    data: z16.record(z16.string(), z16.unknown())
+var outputSchema18 = z18.discriminatedUnion("success", [
+  z18.object({
+    success: z18.literal(true),
+    data: z18.record(z18.string(), z18.unknown())
   }),
-  z16.object({
-    success: z16.literal(false),
-    error: z16.string()
+  z18.object({
+    success: z18.literal(false),
+    error: z18.string()
   })
 ]);
 var requestTool5 = new ConnectorTool({
@@ -3179,8 +3352,8 @@ var requestTool5 = new ConnectorTool({
   description: `Send authenticated requests to the dbt Cloud Discovery API (GraphQL).
 Authentication is handled automatically using the API token.
 {environmentId} in GraphQL variables is automatically replaced with the prod-env-id.`,
-  inputSchema: inputSchema16,
-  outputSchema: outputSchema16,
+  inputSchema: inputSchema18,
+  outputSchema: outputSchema18,
   async execute({ connectionId, query, variables }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -3196,7 +3369,7 @@ Authentication is handled automatically using the API token.
       ) : void 0;
       const endpoint = resolveGraphqlEndpoint(host);
       const controller = new AbortController();
-      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS7);
+      const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS8);
       try {
         const response = await fetch(endpoint, {
           method: "POST",
@@ -3339,27 +3512,27 @@ var parameters16 = {
 };
 
 // src/connectors/squadbase-db/tools/execute-query.ts
-import { z as z17 } from "zod";
+import { z as z19 } from "zod";
 var MAX_ROWS10 = 500;
 var CONNECT_TIMEOUT_MS3 = 1e4;
 var STATEMENT_TIMEOUT_MS2 = 6e4;
-var inputSchema17 = z17.object({
-  toolUseIntent: z17.string().optional().describe(
+var inputSchema19 = z19.object({
+  toolUseIntent: z19.string().optional().describe(
     "Brief description of what you intend to accomplish with this tool call"
   ),
-  connectionId: z17.string().describe("ID of the Squadbase DB connection to use"),
-  sql: z17.string().describe("PostgreSQL SQL query. Always include LIMIT in queries.")
+  connectionId: z19.string().describe("ID of the Squadbase DB connection to use"),
+  sql: z19.string().describe("PostgreSQL SQL query. Always include LIMIT in queries.")
 });
-var outputSchema17 = z17.discriminatedUnion("success", [
-  z17.object({
-    success: z17.literal(true),
-    rowCount: z17.number(),
-    truncated: z17.boolean(),
-    rows: z17.array(z17.record(z17.string(), z17.unknown()))
+var outputSchema19 = z19.discriminatedUnion("success", [
+  z19.object({
+    success: z19.literal(true),
+    rowCount: z19.number(),
+    truncated: z19.boolean(),
+    rows: z19.array(z19.record(z19.string(), z19.unknown()))
   }),
-  z17.object({
-    success: z17.literal(false),
-    error: z17.string()
+  z19.object({
+    success: z19.literal(false),
+    error: z19.string()
   })
 ]);
 var executeQueryTool10 = new ConnectorTool({
@@ -3367,8 +3540,8 @@ var executeQueryTool10 = new ConnectorTool({
   description: `Execute SQL against Squadbase DB (PostgreSQL). Returns up to ${MAX_ROWS10} rows.
 Use for: schema exploration (information_schema), data sampling, analytical queries.
 Avoid loading large amounts of data; always include LIMIT in queries.`,
-  inputSchema: inputSchema17,
-  outputSchema: outputSchema17,
+  inputSchema: inputSchema19,
+  outputSchema: outputSchema19,
   async execute({ connectionId, sql }, connections) {
     const connection = connections.find((c) => c.id === connectionId);
     if (!connection) {
@@ -3473,7 +3646,7 @@ var openaiConnector = new ConnectorPlugin({
   description: "Connect to OpenAI for AI model inference, embeddings, and image generation.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/53XJtCgUlW10x6i1X8xpxM/0bfd634069f1d74241296543cb20427a/openai.svg",
   parameters: parameters6,
-  releaseFlag: { dev1: true, dev2: false, prod: false },
+  releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## OpenAI SDK (TypeScript handler)
 Use the OpenAI connector via the SDK in TypeScript handlers:
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@squadbase/connectors",
-  "version": "0.1.0",
+  "version": "0.1.1-dev.1",
   "description": "Squadbase Connectors",
   "type": "module",
   "main": "dist/index.js",
@@ -30,6 +30,8 @@
     "sync:dev1": "API_BASE_URL=https://dev1-api.squadbase.dev/v0 dotenv tsx scripts/sync-connectors.ts",
     "sync:dev2": "API_BASE_URL=https://dev2-api.squadbase.dev/v0 dotenv tsx scripts/sync-connectors.ts",
     "sync:prod": "API_BASE_URL=https://api.squadbase.dev/v0 dotenv tsx scripts/sync-connectors.ts",
+    "release": "npm run build && npm version patch && npm publish",
+    "release:dev": "npm run build && npm version prerelease --preid=dev && npm publish --tag dev",
     "test": "vitest",
     "test:e2e": "dotenv -e e2e/fixtures/secrets/.env -- vitest run --config vitest.config.e2e.ts"
   },