@squadbase/connectors 0.0.10 → 0.0.12

package/dist/{chunk-5YEHPSNW.js → chunk-E5AVUXWJ.js}

@@ -73,7 +73,21 @@ var parameters = {
   })
 };
 
+// src/connectors/openai/parameters.ts
+var parameters2 = {
+  apiKey: new ParameterDefinition({
+    slug: "api-key",
+    name: "OpenAI API Key",
+    description: "The OpenAI API key for authentication (starts with sk-).",
+    envVarBaseKey: "OPENAI_API_KEY",
+    type: "text",
+    secret: true,
+    required: true
+  })
+};
+
 export {
   ParameterDefinition,
-  parameters
+  parameters,
+  parameters2
 };

package/dist/index.d.ts

@@ -161,6 +161,7 @@ declare const connectors: {
         role: ParameterDefinition;
         warehouse: ParameterDefinition;
         privateKeyBase64: ParameterDefinition;
+        privateKeyPassphrase: ParameterDefinition;
     }, {
         executeQuery: ConnectorTool<{
             connectionId: string;
@@ -470,6 +471,9 @@ declare const connectors: {
             error: string;
         }>;
     }>;
+    openai: ConnectorPlugin<{
+        apiKey: ParameterDefinition;
+    }, {}>;
 };
 
 declare const snowflakeConnector: ConnectorPlugin<{
@@ -478,6 +482,7 @@ declare const snowflakeConnector: ConnectorPlugin<{
     role: ParameterDefinition;
     warehouse: ParameterDefinition;
     privateKeyBase64: ParameterDefinition;
+    privateKeyPassphrase: ParameterDefinition;
 }, {
     executeQuery: ConnectorTool<{
         connectionId: string;
@@ -802,4 +807,8 @@ declare const squadbaseDbConnector: ConnectorPlugin<{
     }>;
 }>;
 
-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 };
+declare const openaiConnector: ConnectorPlugin<{
+    apiKey: ParameterDefinition;
+}, {}>;
+
+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, openaiConnector, postgresqlConnector, redshiftConnector, snowflakeConnector, snowflakePatConnector, squadbaseDbConnector, wixStoreConnector };

package/dist/index.js

@@ -1,7 +1,8 @@
 import {
   ParameterDefinition,
-  parameters
-} from "./chunk-5YEHPSNW.js";
+  parameters,
+  parameters2
+} from "./chunk-E5AVUXWJ.js";
 
 // src/connector-setup.ts
 var ConnectorSetup = class {
@@ -225,7 +226,7 @@ Follow these steps to set up the Snowflake connection.
 });
 
 // src/connectors/snowflake/parameters.ts
-var parameters2 = {
+var parameters3 = {
   account: new ParameterDefinition({
     slug: "account",
     name: "Snowflake Account",
@@ -270,6 +271,15 @@ var parameters2 = {
     type: "base64EncodedText",
     secret: true,
     required: true
+  }),
+  privateKeyPassphrase: new ParameterDefinition({
+    slug: "private-key-passphrase",
+    name: "Snowflake Private Key Passphrase",
+    description: "The passphrase for decrypting an encrypted private key. Leave empty if the key is not encrypted.",
+    envVarBaseKey: "SNOWFLAKE_PRIVATE_KEY_PASSPHRASE",
+    type: "text",
+    secret: true,
+    required: false
   })
 };
 
@@ -319,21 +329,23 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
     try {
       const snowflake = (await import("snowflake-sdk")).default;
       snowflake.configure({ logLevel: "ERROR" });
-      const account = parameters2.account.getValue(connection);
-      const user = parameters2.user.getValue(connection);
-      const role = parameters2.role.getValue(connection);
-      const warehouse = parameters2.warehouse.getValue(connection);
-      const privateKeyBase64 = parameters2.privateKeyBase64.getValue(connection);
+      const account = parameters3.account.getValue(connection);
+      const user = parameters3.user.getValue(connection);
+      const role = parameters3.role.getValue(connection);
+      const warehouse = parameters3.warehouse.getValue(connection);
+      const privateKeyBase64 = parameters3.privateKeyBase64.getValue(connection);
       const privateKey = Buffer.from(privateKeyBase64, "base64").toString(
         "utf-8"
       );
+      const privateKeyPass = parameters3.privateKeyPassphrase.tryGetValue(connection);
       const conn = snowflake.createConnection({
         account,
         username: user,
         role,
         warehouse,
         authenticator: "SNOWFLAKE_JWT",
-        privateKey
+        privateKey,
+        ...privateKeyPass ? { privateKeyPass } : {}
       });
       await new Promise((resolve, reject) => {
         conn.connect((err) => {
@@ -390,7 +402,7 @@ var snowflakeConnector = new ConnectorPlugin({
   name: "Snowflake",
   description: "Connect to Snowflake for cloud data warehousing and analytics.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/6oyVtAcP3pMlXaOrts9unk/b7a9dc25d15c388b66e983041b855447/snowflake.svg",
-  parameters: parameters2,
+  parameters: parameters3,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   setup: snowflakeSetup,
   systemPrompt: `## Snowflake SQL Notes
@@ -407,16 +419,18 @@ var snowflakeConnector = new ConnectorPlugin({
       const snowflake = (await import("snowflake-sdk")).default;
       snowflake.configure({ logLevel: "ERROR" });
       const privateKey = Buffer.from(
-        params[parameters2.privateKeyBase64.slug],
+        params[parameters3.privateKeyBase64.slug],
         "base64"
       ).toString("utf-8");
+      const privateKeyPass = params[parameters3.privateKeyPassphrase.slug] || void 0;
       const conn = snowflake.createConnection({
-        account: params[parameters2.account.slug],
-        username: params[parameters2.user.slug],
-        role: params[parameters2.role.slug],
-        warehouse: params[parameters2.warehouse.slug],
+        account: params[parameters3.account.slug],
+        username: params[parameters3.user.slug],
+        role: params[parameters3.role.slug],
+        warehouse: params[parameters3.warehouse.slug],
         authenticator: "SNOWFLAKE_JWT",
-        privateKey
+        privateKey,
+        ...privateKeyPass ? { privateKeyPass } : {}
       });
       await new Promise((resolve, reject) => {
         conn.connect((err) => {
@@ -446,16 +460,18 @@ var snowflakeConnector = new ConnectorPlugin({
     const snowflake = (await import("snowflake-sdk")).default;
     snowflake.configure({ logLevel: "ERROR" });
     const privateKey = Buffer.from(
-      params[parameters2.privateKeyBase64.slug],
+      params[parameters3.privateKeyBase64.slug],
       "base64"
     ).toString("utf-8");
+    const privateKeyPass = params[parameters3.privateKeyPassphrase.slug] || void 0;
     const conn = snowflake.createConnection({
-      account: params[parameters2.account.slug],
-      username: params[parameters2.user.slug],
-      role: params[parameters2.role.slug],
-      warehouse: params[parameters2.warehouse.slug],
+      account: params[parameters3.account.slug],
+      username: params[parameters3.user.slug],
+      role: params[parameters3.role.slug],
+      warehouse: params[parameters3.warehouse.slug],
       authenticator: "SNOWFLAKE_JWT",
-      privateKey
+      privateKey,
+      ...privateKeyPass ? { privateKeyPass } : {}
     });
     await new Promise((resolve, reject) => {
       conn.connect((err) => {
@@ -482,7 +498,7 @@ var snowflakeConnector = new ConnectorPlugin({
 });
 
 // src/connectors/snowflake-pat/parameters.ts
-var parameters3 = {
+var parameters4 = {
   account: new ParameterDefinition({
     slug: "account",
     name: "Snowflake Account",
@@ -576,11 +592,11 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
     try {
       const snowflake = (await import("snowflake-sdk")).default;
       snowflake.configure({ logLevel: "ERROR" });
-      const account = parameters3.account.getValue(connection);
-      const user = parameters3.user.getValue(connection);
-      const role = parameters3.role.getValue(connection);
-      const warehouse = parameters3.warehouse.getValue(connection);
-      const password = parameters3.pat.getValue(connection);
+      const account = parameters4.account.getValue(connection);
+      const user = parameters4.user.getValue(connection);
+      const role = parameters4.role.getValue(connection);
+      const warehouse = parameters4.warehouse.getValue(connection);
+      const password = parameters4.pat.getValue(connection);
       const conn = snowflake.createConnection({
         account,
         username: user,
@@ -643,7 +659,7 @@ var snowflakePatConnector = new ConnectorPlugin({
   name: "Snowflake (PAT)",
   description: "Connect to Snowflake using a Personal Access Token (PAT).",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/6oyVtAcP3pMlXaOrts9unk/b7a9dc25d15c388b66e983041b855447/snowflake.svg",
-  parameters: parameters3,
+  parameters: parameters4,
   releaseFlag: { dev1: true, dev2: false, prod: false },
   setup: snowflakeSetup,
   systemPrompt: `## Snowflake SQL Notes
@@ -660,11 +676,11 @@ var snowflakePatConnector = new ConnectorPlugin({
       const snowflake = (await import("snowflake-sdk")).default;
       snowflake.configure({ logLevel: "ERROR" });
       const conn = snowflake.createConnection({
-        account: params[parameters3.account.slug],
-        username: params[parameters3.user.slug],
-        role: params[parameters3.role.slug],
-        warehouse: params[parameters3.warehouse.slug],
-        password: params[parameters3.pat.slug]
+        account: params[parameters4.account.slug],
+        username: params[parameters4.user.slug],
+        role: params[parameters4.role.slug],
+        warehouse: params[parameters4.warehouse.slug],
+        password: params[parameters4.pat.slug]
       });
       await new Promise((resolve, reject) => {
         conn.connect((err) => {
@@ -699,11 +715,11 @@ var snowflakePatConnector = new ConnectorPlugin({
     const snowflake = (await import("snowflake-sdk")).default;
     snowflake.configure({ logLevel: "ERROR" });
     const conn = snowflake.createConnection({
-      account: params[parameters3.account.slug],
-      username: params[parameters3.user.slug],
-      role: params[parameters3.role.slug],
-      warehouse: params[parameters3.warehouse.slug],
-      password: params[parameters3.pat.slug]
+      account: params[parameters4.account.slug],
+      username: params[parameters4.user.slug],
+      role: params[parameters4.role.slug],
+      warehouse: params[parameters4.warehouse.slug],
+      password: params[parameters4.pat.slug]
     });
     await new Promise((resolve, reject) => {
       conn.connect((err) => {
@@ -735,7 +751,7 @@ var snowflakePatConnector = new ConnectorPlugin({
 });
 
 // src/connectors/postgresql/parameters.ts
-var parameters4 = {
+var parameters5 = {
   connectionUrl: new ParameterDefinition({
     slug: "connection-url",
     name: "PostgreSQL Connection URL",
@@ -792,7 +808,7 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
     let connectionUrl;
     try {
       const { Pool } = await import("pg");
-      connectionUrl = parameters4.connectionUrl.getValue(connection);
+      connectionUrl = parameters5.connectionUrl.getValue(connection);
       const pool = new Pool({
         connectionString: connectionUrl,
         ssl: { rejectUnauthorized: false },
@@ -830,7 +846,7 @@ var postgresqlConnector = new ConnectorPlugin({
   name: "PostgreSQL",
   description: "Connect to PostgreSQL databases for relational data storage and querying.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/42AHi2uqteUn65MyqdN6V7/a0f68f12af6aac96bbcda5980f43de07/elephant.png",
-  parameters: parameters4,
+  parameters: parameters5,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## PostgreSQL SQL Notes
 - Schema exploration:
@@ -841,7 +857,7 @@ var postgresqlConnector = new ConnectorPlugin({
   async checkConnection(params, _config) {
     const { Pool } = await import("pg");
     const pool = new Pool({
-      connectionString: params[parameters4.connectionUrl.slug],
+      connectionString: params[parameters5.connectionUrl.slug],
       ssl: { rejectUnauthorized: false },
       connectionTimeoutMillis: 1e4
     });
@@ -858,7 +874,7 @@ var postgresqlConnector = new ConnectorPlugin({
     const { Pool } = await import("pg");
     const { text, values } = buildPositionalParams(sql, namedParams);
     const pool = new Pool({
-      connectionString: params[parameters4.connectionUrl.slug],
+      connectionString: params[parameters5.connectionUrl.slug],
       ssl: { rejectUnauthorized: false },
       connectionTimeoutMillis: 1e4,
       statement_timeout: 6e4
@@ -873,7 +889,7 @@ var postgresqlConnector = new ConnectorPlugin({
 });
 
 // src/connectors/mysql/parameters.ts
-var parameters5 = {
+var parameters6 = {
   connectionUrl: new ParameterDefinition({
     slug: "connection-url",
     name: "MySQL Connection URL",
@@ -930,7 +946,7 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
     let connectionUrl;
     try {
       const mysql = await import("mysql2/promise");
-      connectionUrl = parameters5.connectionUrl.getValue(connection);
+      connectionUrl = parameters6.connectionUrl.getValue(connection);
       const pool = mysql.createPool({
         uri: connectionUrl,
         connectTimeout: CONNECT_TIMEOUT_MS2
@@ -970,7 +986,7 @@ var mysqlConnector = new ConnectorPlugin({
   name: "MySQL",
   description: "Connect to MySQL databases for relational data storage and querying.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/6ghPFeGgl7uBs5NHH1a4L/512c9433beec5b595caa41f04921c1f9/logo-mysql-170x115.png",
-  parameters: parameters5,
+  parameters: parameters6,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## MySQL SQL Notes
 - Schema exploration:
@@ -981,7 +997,7 @@ var mysqlConnector = new ConnectorPlugin({
   async checkConnection(params, _config) {
     const mysql = await import("mysql2/promise");
     const pool = mysql.createPool({
-      uri: params[parameters5.connectionUrl.slug],
+      uri: params[parameters6.connectionUrl.slug],
       connectTimeout: 1e4
     });
     try {
@@ -997,7 +1013,7 @@ var mysqlConnector = new ConnectorPlugin({
     const mysql = await import("mysql2/promise");
     const { text, values } = buildQuestionmarkParams(sql, namedParams);
     const pool = mysql.createPool({
-      uri: params[parameters5.connectionUrl.slug],
+      uri: params[parameters6.connectionUrl.slug],
       connectTimeout: 1e4
     });
     try {
@@ -1018,7 +1034,7 @@ var mysqlConnector = new ConnectorPlugin({
 import { z as z5 } from "zod";
 
 // src/connectors/bigquery/parameters.ts
-var parameters6 = {
+var parameters7 = {
   serviceAccountKeyJsonBase64: new ParameterDefinition({
     slug: "service-account-key-json-base64",
     name: "Google Cloud Service Account JSON",
@@ -1075,7 +1091,7 @@ var listProjectsTool = new ConnectorTool({
       };
     }
     try {
-      const serviceAccountJsonBase64 = parameters6.serviceAccountKeyJsonBase64.getValue(connection);
+      const serviceAccountJsonBase64 = parameters7.serviceAccountKeyJsonBase64.getValue(connection);
       const credentials = JSON.parse(
         Buffer.from(serviceAccountJsonBase64, "base64").toString("utf-8")
       );
@@ -1223,8 +1239,8 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
     );
     try {
       const { BigQuery } = await import("@google-cloud/bigquery");
-      const projectId = parameters6.projectId.getValue(connection);
-      const serviceAccountJsonBase64 = parameters6.serviceAccountKeyJsonBase64.getValue(connection);
+      const projectId = parameters7.projectId.getValue(connection);
+      const serviceAccountJsonBase64 = parameters7.serviceAccountKeyJsonBase64.getValue(connection);
       const credentials = JSON.parse(
         Buffer.from(serviceAccountJsonBase64, "base64").toString("utf-8")
       );
@@ -1254,7 +1270,7 @@ var bigqueryConnector = new ConnectorPlugin({
   name: "BigQuery",
   description: "Connect to Google BigQuery for data warehouse and analytics.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/6nlehQyOmdbktG5hOYkYMr/6ca559140d5ddc7dadc5eac88858a563/bigquery.svg",
-  parameters: parameters6,
+  parameters: parameters7,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   setup: bigquerySetup,
   systemPrompt: `## BigQuery SQL Notes
@@ -1268,10 +1284,10 @@ var bigqueryConnector = new ConnectorPlugin({
   async checkConnection(params, _config) {
     const { BigQuery } = await import("@google-cloud/bigquery");
     const credentials = JSON.parse(
-      Buffer.from(params[parameters6.serviceAccountKeyJsonBase64.slug], "base64").toString("utf-8")
+      Buffer.from(params[parameters7.serviceAccountKeyJsonBase64.slug], "base64").toString("utf-8")
     );
     const bq = new BigQuery({
-      projectId: params[parameters6.projectId.slug],
+      projectId: params[parameters7.projectId.slug],
       credentials
     });
     try {
@@ -1286,10 +1302,10 @@ var bigqueryConnector = new ConnectorPlugin({
     const { BigQuery } = await import("@google-cloud/bigquery");
     const resolvedSql = replaceLiteralParams(sql, namedParams);
     const credentials = JSON.parse(
-      Buffer.from(params[parameters6.serviceAccountKeyJsonBase64.slug], "base64").toString("utf-8")
+      Buffer.from(params[parameters7.serviceAccountKeyJsonBase64.slug], "base64").toString("utf-8")
     );
     const bq = new BigQuery({
-      projectId: params[parameters6.projectId.slug],
+      projectId: params[parameters7.projectId.slug],
       credentials
     });
     const [job] = await bq.createQueryJob({ query: resolvedSql });
@@ -1485,7 +1501,7 @@ Follow these steps to set up the BigQuery connection.
 });
 
 // src/connectors/bigquery-oauth/parameters.ts
-var parameters7 = {
+var parameters8 = {
   projectId: new ParameterDefinition({
     slug: "project-id",
     name: "Google Cloud Project ID",
@@ -1579,7 +1595,7 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
         error: `Connection ${connectionId} not found`
       };
     }
-    const gcpProjectId = parameters7.projectId.getValue(connection);
+    const gcpProjectId = parameters8.projectId.getValue(connection);
     console.log(
       `[connector-query] bigquery-oauth/${connection.name}: ${sql}`
     );
@@ -1646,7 +1662,7 @@ var bigqueryOauthConnector = new ConnectorPlugin({
   name: "BigQuery (OAuth)",
   description: "Connect to Google BigQuery for data warehouse and analytics using OAuth.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/6nlehQyOmdbktG5hOYkYMr/6ca559140d5ddc7dadc5eac88858a563/bigquery.svg",
-  parameters: parameters7,
+  parameters: parameters8,
   releaseFlag: { dev1: true, dev2: false, prod: false },
   setup: bigquerySetup2,
   proxyPolicy: {
@@ -1665,7 +1681,7 @@ var bigqueryOauthConnector = new ConnectorPlugin({
   tools: tools6,
   async checkConnection(params, config) {
     const { proxyFetch } = config;
-    const projectId = params[parameters7.projectId.slug];
+    const projectId = params[parameters8.projectId.slug];
     const url = `https://bigquery.googleapis.com/bigquery/v2/projects/${projectId}/queries`;
     try {
       const res = await proxyFetch(url, {
@@ -1684,7 +1700,7 @@ var bigqueryOauthConnector = new ConnectorPlugin({
   },
   async query(params, sql, namedParams, context) {
     const { proxyFetch } = context;
-    const projectId = params[parameters7.projectId.slug];
+    const projectId = params[parameters8.projectId.slug];
     const resolvedSql = replaceLiteralParams(sql, namedParams);
     const url = `https://bigquery.googleapis.com/bigquery/v2/projects/${projectId}/queries`;
     const res = await proxyFetch(url, {
@@ -1702,7 +1718,7 @@ var bigqueryOauthConnector = new ConnectorPlugin({
 });
 
 // src/connectors/aws-athena/parameters.ts
-var parameters8 = {
+var parameters9 = {
   awsAccessKeyId: new ParameterDefinition({
     slug: "aws-access-key-id",
     name: "AWS Access Key ID",
@@ -1799,8 +1815,8 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
         GetQueryExecutionCommand,
         GetQueryResultsCommand
       } = await import("@aws-sdk/client-athena");
-      const workgroup = parameters8.workgroup.tryGetValue(connection);
-      const outputLocation = parameters8.outputLocation.tryGetValue(connection);
+      const workgroup = parameters9.workgroup.tryGetValue(connection);
+      const outputLocation = parameters9.outputLocation.tryGetValue(connection);
       if (!workgroup && !outputLocation) {
         return {
           success: false,
@@ -1808,10 +1824,10 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
         };
       }
       const client = new AthenaClient({
-        region: parameters8.awsRegion.getValue(connection),
+        region: parameters9.awsRegion.getValue(connection),
         credentials: {
-          accessKeyId: parameters8.awsAccessKeyId.getValue(connection),
-          secretAccessKey: parameters8.awsSecretAccessKey.getValue(connection)
+          accessKeyId: parameters9.awsAccessKeyId.getValue(connection),
+          secretAccessKey: parameters9.awsSecretAccessKey.getValue(connection)
         }
       });
       const startParams = { QueryString: sql };
@@ -1875,7 +1891,7 @@ var awsAthenaConnector = new ConnectorPlugin({
   name: "AWS Athena",
   description: "Connect to AWS Athena for serverless SQL queries on S3 data.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/5x0vIHtUHfJJMZUv4RFOYZ/5059bac389f0169542f39cdb4b387d2c/Athena.svg",
-  parameters: parameters8,
+  parameters: parameters9,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## AWS Athena SQL Notes
 - Uses Presto/Trino based SQL syntax
@@ -1893,8 +1909,8 @@ var awsAthenaConnector = new ConnectorPlugin({
         StartQueryExecutionCommand,
         GetQueryExecutionCommand
       } = await import("@aws-sdk/client-athena");
-      const workgroup = params[parameters8.workgroup.slug];
-      const outputLocation = params[parameters8.outputLocation.slug];
+      const workgroup = params[parameters9.workgroup.slug];
+      const outputLocation = params[parameters9.outputLocation.slug];
       if (!workgroup && !outputLocation) {
         return {
           success: false,
@@ -1902,10 +1918,10 @@ var awsAthenaConnector = new ConnectorPlugin({
         };
       }
       const client = new AthenaClient({
-        region: params[parameters8.awsRegion.slug],
+        region: params[parameters9.awsRegion.slug],
         credentials: {
-          accessKeyId: params[parameters8.awsAccessKeyId.slug],
-          secretAccessKey: params[parameters8.awsSecretAccessKey.slug]
+          accessKeyId: params[parameters9.awsAccessKeyId.slug],
+          secretAccessKey: params[parameters9.awsSecretAccessKey.slug]
         }
       });
       const startParams = { QueryString: "SELECT 1" };
@@ -1950,16 +1966,16 @@ var awsAthenaConnector = new ConnectorPlugin({
       GetQueryResultsCommand
     } = await import("@aws-sdk/client-athena");
     const resolvedSql = replaceLiteralParams(sql, namedParams);
-    const workgroup = params[parameters8.workgroup.slug];
-    const outputLocation = params[parameters8.outputLocation.slug];
+    const workgroup = params[parameters9.workgroup.slug];
+    const outputLocation = params[parameters9.outputLocation.slug];
     if (!workgroup && !outputLocation) {
       throw new Error("Either workgroup or output-location is required");
     }
     const client = new AthenaClient({
-      region: params[parameters8.awsRegion.slug],
+      region: params[parameters9.awsRegion.slug],
       credentials: {
-        accessKeyId: params[parameters8.awsAccessKeyId.slug],
-        secretAccessKey: params[parameters8.awsSecretAccessKey.slug]
+        accessKeyId: params[parameters9.awsAccessKeyId.slug],
+        secretAccessKey: params[parameters9.awsSecretAccessKey.slug]
       }
     });
     const startParams = { QueryString: resolvedSql };
@@ -2004,7 +2020,7 @@ var awsAthenaConnector = new ConnectorPlugin({
 });
 
 // src/connectors/redshift/parameters.ts
-var parameters9 = {
+var parameters10 = {
   awsAccessKeyId: new ParameterDefinition({
     slug: "aws-access-key-id",
     name: "AWS Access Key ID",
@@ -2130,14 +2146,14 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
         DescribeStatementCommand,
         GetStatementResultCommand
       } = await import("@aws-sdk/client-redshift-data");
-      const awsAccessKeyId = parameters9.awsAccessKeyId.getValue(connection);
-      const awsSecretAccessKey = parameters9.awsSecretAccessKey.getValue(connection);
-      const awsRegion = parameters9.awsRegion.getValue(connection);
-      const database = parameters9.database.getValue(connection);
-      const clusterIdentifier = parameters9.clusterIdentifier.tryGetValue(connection);
-      const workgroupName = parameters9.workgroupName.tryGetValue(connection);
-      const secretArn = parameters9.secretArn.tryGetValue(connection);
-      const dbUser = parameters9.dbUser.tryGetValue(connection);
+      const awsAccessKeyId = parameters10.awsAccessKeyId.getValue(connection);
+      const awsSecretAccessKey = parameters10.awsSecretAccessKey.getValue(connection);
+      const awsRegion = parameters10.awsRegion.getValue(connection);
+      const database = parameters10.database.getValue(connection);
+      const clusterIdentifier = parameters10.clusterIdentifier.tryGetValue(connection);
+      const workgroupName = parameters10.workgroupName.tryGetValue(connection);
+      const secretArn = parameters10.secretArn.tryGetValue(connection);
+      const dbUser = parameters10.dbUser.tryGetValue(connection);
       if (!clusterIdentifier && !workgroupName) {
         return {
           success: false,
@@ -2208,7 +2224,7 @@ var redshiftConnector = new ConnectorPlugin({
   name: "Redshift",
   description: "Connect to Amazon Redshift for data warehouse analytics.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/AEwW2psmrnZ7htTVsgA9t/a637e31707c5d760be73ce1d8ec75580/aws-redshift-logo.svg",
-  parameters: parameters9,
+  parameters: parameters10,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## Redshift SQL Notes
 - Uses PostgreSQL based SQL syntax
@@ -2226,10 +2242,10 @@ var redshiftConnector = new ConnectorPlugin({
         ExecuteStatementCommand,
         DescribeStatementCommand
       } = await import("@aws-sdk/client-redshift-data");
-      const clusterIdentifier = params[parameters9.clusterIdentifier.slug];
-      const workgroupName = params[parameters9.workgroupName.slug];
-      const secretArn = params[parameters9.secretArn.slug];
-      const dbUser = params[parameters9.dbUser.slug];
+      const clusterIdentifier = params[parameters10.clusterIdentifier.slug];
+      const workgroupName = params[parameters10.workgroupName.slug];
+      const secretArn = params[parameters10.secretArn.slug];
+      const dbUser = params[parameters10.dbUser.slug];
       if (!clusterIdentifier && !workgroupName) {
         return {
           success: false,
@@ -2237,15 +2253,15 @@ var redshiftConnector = new ConnectorPlugin({
         };
       }
       const client = new RedshiftDataClient({
-        region: params[parameters9.awsRegion.slug],
+        region: params[parameters10.awsRegion.slug],
         credentials: {
-          accessKeyId: params[parameters9.awsAccessKeyId.slug],
-          secretAccessKey: params[parameters9.awsSecretAccessKey.slug]
+          accessKeyId: params[parameters10.awsAccessKeyId.slug],
+          secretAccessKey: params[parameters10.awsSecretAccessKey.slug]
         }
       });
       const { Id: statementId } = await client.send(
         new ExecuteStatementCommand({
-          Database: params[parameters9.database.slug],
+          Database: params[parameters10.database.slug],
           Sql: "SELECT 1",
           ...clusterIdentifier && { ClusterIdentifier: clusterIdentifier },
           ...workgroupName && { WorkgroupName: workgroupName },
@@ -2286,23 +2302,23 @@ var redshiftConnector = new ConnectorPlugin({
       GetStatementResultCommand
     } = await import("@aws-sdk/client-redshift-data");
     const resolvedSql = replaceLiteralParams(sql, namedParams);
-    const clusterIdentifier = params[parameters9.clusterIdentifier.slug];
-    const workgroupName = params[parameters9.workgroupName.slug];
-    const secretArn = params[parameters9.secretArn.slug];
-    const dbUser = params[parameters9.dbUser.slug];
+    const clusterIdentifier = params[parameters10.clusterIdentifier.slug];
+    const workgroupName = params[parameters10.workgroupName.slug];
+    const secretArn = params[parameters10.secretArn.slug];
+    const dbUser = params[parameters10.dbUser.slug];
     if (!clusterIdentifier && !workgroupName) {
       throw new Error("Either cluster-identifier or workgroup-name is required");
     }
     const client = new RedshiftDataClient({
-      region: params[parameters9.awsRegion.slug],
+      region: params[parameters10.awsRegion.slug],
       credentials: {
-        accessKeyId: params[parameters9.awsAccessKeyId.slug],
-        secretAccessKey: params[parameters9.awsSecretAccessKey.slug]
+        accessKeyId: params[parameters10.awsAccessKeyId.slug],
+        secretAccessKey: params[parameters10.awsSecretAccessKey.slug]
       }
     });
     const { Id: statementId } = await client.send(
       new ExecuteStatementCommand({
-        Database: params[parameters9.database.slug],
+        Database: params[parameters10.database.slug],
         Sql: resolvedSql,
         ...clusterIdentifier && { ClusterIdentifier: clusterIdentifier },
         ...workgroupName && { WorkgroupName: workgroupName },
@@ -2340,7 +2356,7 @@ var redshiftConnector = new ConnectorPlugin({
 });
 
 // src/connectors/databricks/parameters.ts
-var parameters10 = {
+var parameters11 = {
   host: new ParameterDefinition({
     slug: "host",
     name: "Databricks Workspace Host",
@@ -2412,9 +2428,9 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
     );
     try {
       const { DBSQLClient } = await import("@databricks/sql");
-      const host = parameters10.host.getValue(connection);
-      const token = parameters10.token.getValue(connection);
-      const httpPath = parameters10.httpPath.getValue(connection);
+      const host = parameters11.host.getValue(connection);
+      const token = parameters11.token.getValue(connection);
+      const httpPath = parameters11.httpPath.getValue(connection);
       const client = new DBSQLClient();
       await client.connect({ host, path: httpPath, token });
       let session;
@@ -2455,7 +2471,7 @@ var databricksConnector = new ConnectorPlugin({
   name: "Databricks",
   description: "Connect to Databricks for data lakehouse and SQL analytics.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/6QgcrfpQOKg18P7DdgKerd/af55bf0d871339049824dd167b97a29f/databricks-icon.svg",
-  parameters: parameters10,
+  parameters: parameters11,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## Databricks SQL Notes
 - Uses Spark SQL / Databricks SQL syntax
@@ -2470,9 +2486,9 @@ var databricksConnector = new ConnectorPlugin({
     const { DBSQLClient } = await import("@databricks/sql");
     const client = new DBSQLClient();
     await client.connect({
-      host: params[parameters10.host.slug],
-      path: params[parameters10.httpPath.slug],
-      token: params[parameters10.token.slug]
+      host: params[parameters11.host.slug],
+      path: params[parameters11.httpPath.slug],
+      token: params[parameters11.token.slug]
     });
     let session;
     let operation;
@@ -2499,9 +2515,9 @@ var databricksConnector = new ConnectorPlugin({
     const resolvedSql = replaceLiteralParams(sql, namedParams);
     const client = new DBSQLClient();
     await client.connect({
-      host: params[parameters10.host.slug],
-      path: params[parameters10.httpPath.slug],
-      token: params[parameters10.token.slug]
+      host: params[parameters11.host.slug],
+      path: params[parameters11.httpPath.slug],
+      token: params[parameters11.token.slug]
     });
     let session;
     let operation;
@@ -2524,7 +2540,7 @@ var databricksConnector = new ConnectorPlugin({
 });
 
 // src/connectors/airtable/parameters.ts
-var parameters11 = {
+var parameters12 = {
   baseId: new ParameterDefinition({
     slug: "base-id",
     name: "Airtable Base ID",
@@ -2581,8 +2597,8 @@ Authentication is handled automatically using the API Key.
     }
     console.log(`[connector-request] airtable/${connection.name}: ${method} ${path}`);
     try {
-      const apiKey = parameters11.apiKey.getValue(connection);
-      const baseId = parameters11.baseId.getValue(connection);
+      const apiKey = parameters12.apiKey.getValue(connection);
+      const baseId = parameters12.baseId.getValue(connection);
       const resolvedPath = path.replace(/\{baseId\}/g, baseId);
       const url = `${BASE_URL}${resolvedPath}`;
       const controller = new AbortController();
@@ -2624,7 +2640,7 @@ var airtableConnector = new ConnectorPlugin({
   name: "Airtable",
   description: "Connect to Airtable for spreadsheet-database hybrid data management.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/19JUphfOZjyjTK6Zg4NGCf/8c56227b088cada52d3a2d9385a3be97/airtable.svg",
-  parameters: parameters11,
+  parameters: parameters12,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## Airtable API
 - Call the Airtable REST API using the authenticated request tool
@@ -2648,7 +2664,7 @@ var airtableConnector = new ConnectorPlugin({
 });
 
 // src/connectors/google-analytics/parameters.ts
-var parameters12 = {
+var parameters13 = {
   serviceAccountKeyJsonBase64: new ParameterDefinition({
     slug: "service-account-key-json-base64",
     name: "Google Cloud Service Account JSON",
@@ -2706,8 +2722,8 @@ Authentication is handled automatically using a service account.
     console.log(`[connector-request] google-analytics/${connection.name}: ${method} ${path}`);
     try {
       const { GoogleAuth } = await import("google-auth-library");
-      const keyJsonBase64 = parameters12.serviceAccountKeyJsonBase64.getValue(connection);
-      const propertyId = parameters12.propertyId.getValue(connection);
+      const keyJsonBase64 = parameters13.serviceAccountKeyJsonBase64.getValue(connection);
+      const propertyId = parameters13.propertyId.getValue(connection);
       const credentials = JSON.parse(
         Buffer.from(keyJsonBase64, "base64").toString("utf-8")
       );
@@ -2760,7 +2776,7 @@ var googleAnalyticsConnector = new ConnectorPlugin({
   name: "Google Analytics",
   description: "Connect to Google Analytics for web analytics and reporting.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/7fs0ipzxuD9mACDzBATtxX/3c53ed90d15c96483e4f78cb29dab5e9/google-analytics.svg",
-  parameters: parameters12,
+  parameters: parameters13,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## Google Analytics Data API
 - Call the GA4 Data API using the authenticated request tool
@@ -2930,7 +2946,7 @@ const { apps } = await kintone.listApps();
 });
 
 // src/connectors/wix-store/parameters.ts
-var parameters13 = {
+var parameters14 = {
   accountId: new ParameterDefinition({
     slug: "account-id",
     name: "Account ID",
@@ -2995,8 +3011,8 @@ Authentication is handled automatically using the API Key and Site ID.`,
     }
     console.log(`[connector-request] wix-store/${connection.name}: ${method} ${path}`);
     try {
-      const apiKey = parameters13.apiKey.getValue(connection);
-      const siteId = parameters13.siteId.getValue(connection);
+      const apiKey = parameters14.apiKey.getValue(connection);
+      const siteId = parameters14.siteId.getValue(connection);
       const url = `${BASE_URL3}${path}`;
       const controller = new AbortController();
       const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS6);
@@ -3038,7 +3054,7 @@ var wixStoreConnector = new ConnectorPlugin({
   name: "Wix Store",
   description: "Connect to Wix Store.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/YyFxclQFzROIYpFam6vRK/e7e75d3feac49a1cc5e433c147216d23/Wix_logo_black.svg",
-  parameters: parameters13,
+  parameters: parameters14,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## Wix Store API
 - Call the Wix Store REST API using the authenticated request tool
@@ -3064,7 +3080,7 @@ var wixStoreConnector = new ConnectorPlugin({
 });
 
 // src/connectors/dbt/parameters.ts
-var parameters14 = {
+var parameters15 = {
   host: new ParameterDefinition({
     slug: "host",
     name: "dbt Cloud Host",
@@ -3141,9 +3157,9 @@ Authentication is handled automatically using the API token.
     }
     console.log(`[connector-request] dbt/${connection.name}: GraphQL query`);
     try {
-      const host = parameters14.host.getValue(connection);
-      const token = parameters14.token.getValue(connection);
-      const environmentId = parameters14.prodEnvId.getValue(connection);
+      const host = parameters15.host.getValue(connection);
+      const token = parameters15.token.getValue(connection);
+      const environmentId = parameters15.prodEnvId.getValue(connection);
       const resolvedVariables = variables ? JSON.parse(
         JSON.stringify(variables).replace(/\{environmentId\}/g, environmentId)
       ) : void 0;
@@ -3193,7 +3209,7 @@ var dbtConnector = new ConnectorPlugin({
   name: "dbt",
   description: "Connect to dbt Cloud for data transformation and analytics engineering.",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/4iT6ncXtdtHdkXexU0WgfZ/0367a38d245f2568eab5eb511f9ee692/dbt.png",
-  parameters: parameters14,
+  parameters: parameters15,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## dbt Cloud Discovery API (GraphQL)
 - Call the dbt Cloud Discovery API using the authenticated request tool
@@ -3250,7 +3266,7 @@ query:
 });
 
 // src/connectors/squadbase-db/parameters.ts
-var parameters15 = {
+var parameters16 = {
   connectionUrl: new ParameterDefinition({
     slug: "connection-url",
     name: "Connection URL",
@@ -3307,7 +3323,7 @@ Avoid loading large amounts of data; always include LIMIT in queries.`,
     let connectionUrl;
     try {
       const { Pool } = await import("pg");
-      connectionUrl = parameters15.connectionUrl.getValue(connection);
+      connectionUrl = parameters16.connectionUrl.getValue(connection);
       const pool = new Pool({
         connectionString: connectionUrl,
         ssl: { rejectUnauthorized: false },
@@ -3345,7 +3361,7 @@ var squadbaseDbConnector = new ConnectorPlugin({
   name: "Squadbase DB",
   description: "Connect to Squadbase DB (PostgreSQL).",
   iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/25y0XqMxIufeD3egWH3bEl/659b4ade405890654cfaf91c03a4b458/icon.svg",
-  parameters: parameters15,
+  parameters: parameters16,
   releaseFlag: { dev1: true, dev2: true, prod: true },
   systemPrompt: `## Squadbase DB SQL Notes
 - Uses PostgreSQL based SQL syntax
@@ -3357,7 +3373,7 @@ var squadbaseDbConnector = new ConnectorPlugin({
   async checkConnection(params, _config) {
     const { Pool } = await import("pg");
     const pool = new Pool({
-      connectionString: params[parameters15.connectionUrl.slug],
+      connectionString: params[parameters16.connectionUrl.slug],
       ssl: { rejectUnauthorized: false },
       connectionTimeoutMillis: 1e4
     });
@@ -3374,7 +3390,7 @@ var squadbaseDbConnector = new ConnectorPlugin({
     const { Pool } = await import("pg");
     const { text, values } = buildPositionalParams(sql, namedParams);
     const pool = new Pool({
-      connectionString: params[parameters15.connectionUrl.slug],
+      connectionString: params[parameters16.connectionUrl.slug],
       ssl: { rejectUnauthorized: false },
       connectionTimeoutMillis: 1e4,
       statement_timeout: 6e4
@@ -3388,6 +3404,34 @@ var squadbaseDbConnector = new ConnectorPlugin({
   }
 });
 
+// src/connectors/openai/index.ts
+var tools16 = {};
+var openaiConnector = new ConnectorPlugin({
+  slug: "openai",
+  authType: null,
+  name: "OpenAI",
+  description: "Connect to OpenAI for AI model inference, embeddings, and image generation.",
+  iconUrl: "https://images.ctfassets.net/9ncizv60xc5y/53XJtCgUlW10x6i1X8xpxM/0bfd634069f1d74241296543cb20427a/openai.svg",
+  parameters: parameters2,
+  releaseFlag: { dev1: true, dev2: false, prod: false },
+  systemPrompt: `## OpenAI SDK (TypeScript handler)
+Use the OpenAI connector via the SDK in TypeScript handlers:
+
+\`\`\`ts
+import { connection } from "@squadbase/vite-server/connectors/openai";
+import OpenAI from "openai";
+
+const { apiKey } = connection("<connectionId>");
+const client = new OpenAI({ apiKey });
+
+const response = await client.chat.completions.create({
+  model: "gpt-4o",
+  messages: [{ role: "user", content: "Hello" }],
+});
+\`\`\``,
+  tools: tools16
+});
+
 // src/connectors/registry.ts
 var plugins = {
   snowflake: snowflakeConnector,
@@ -3404,7 +3448,8 @@ var plugins = {
   airtable: airtableConnector,
   squadbaseDb: squadbaseDbConnector,
   kintone: kintoneConnector,
-  wixStore: wixStoreConnector
+  wixStore: wixStoreConnector,
+  openai: openaiConnector
 };
 var connectors = {
   ...plugins,
@@ -3443,6 +3488,7 @@ export {
   googleAnalyticsConnector,
   kintoneConnector,
   mysqlConnector,
+  openaiConnector,
   postgresqlConnector,
   redshiftConnector,
   snowflakeConnector,

package/dist/sdk.d.ts

@@ -34,6 +34,16 @@ interface KintoneConnectorSdk {
  * @param params - Resolved parameter values keyed by parameter slug
  *                 ("base-url", "username", "password")
  */
-declare function createClient(params: Record<string, string>): KintoneConnectorSdk;
+declare function createClient$1(params: Record<string, string>): KintoneConnectorSdk;
 
-export { type KintoneConnectorSdk, createClient as kintone };
+interface OpenAIConnectorSdk {
+    apiKey: string;
+}
+/**
+ * Create an OpenAI client from resolved connection parameters.
+ *
+ * @param params - Resolved parameter values keyed by parameter slug ("api-key")
+ */
+declare function createClient(params: Record<string, string>): OpenAIConnectorSdk;
+
+export { type KintoneConnectorSdk, type OpenAIConnectorSdk, createClient$1 as kintone, createClient as openai };

package/dist/sdk.js

@@ -1,6 +1,7 @@
 import {
-  parameters
-} from "./chunk-5YEHPSNW.js";
+  parameters,
+  parameters2
+} from "./chunk-E5AVUXWJ.js";
 
 // src/connectors/kintone/sdk/index.ts
 function createClient(params) {
@@ -70,6 +71,18 @@ function createClient(params) {
     }
   };
 }
+
+// src/connectors/openai/sdk/index.ts
+function createClient2(params) {
+  const apiKey = params[parameters2.apiKey.slug];
+  if (!apiKey) {
+    throw new Error(
+      `openai: missing required parameter: ${parameters2.apiKey.slug}`
+    );
+  }
+  return { apiKey };
+}
 export {
-  createClient as kintone
+  createClient as kintone,
+  createClient2 as openai
 };

package/package.json

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