@credal/actions 0.1.43 → 0.1.45

package/dist/actions/providers/snowflake/auth/getSnowflakeConnection.d.ts

@@ -6,3 +6,4 @@ export declare function getSnowflakeConnection(snowflakeData: {
     warehouse: string;
     database: string;
 }, authParams: AuthParamsType): Connection;
+export declare function connectToSnowflakeAndWarehouse(connection: Connection, warehouse?: string): Promise<void>;

package/dist/actions/providers/snowflake/auth/getSnowflakeConnection.js

@@ -1,16 +1,71 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getSnowflakeConnection = getSnowflakeConnection;
-const crypto_1 = __importDefault(require("crypto"));
+exports.connectToSnowflakeAndWarehouse = connectToSnowflakeAndWarehouse;
 const snowflake_sdk_1 = __importDefault(require("snowflake-sdk"));
+const forge = __importStar(require("node-forge"));
+const getPrivateKeyCorrectFormat = (privateKey) => {
+    try {
+        // Parse the private key
+        const pemKey = forge.pem.decode(privateKey)[0];
+        // Re-encode it properly with correct formatting
+        return forge.pem.encode(pemKey);
+    }
+    catch (error) {
+        console.error("Error processing private key:", error);
+        throw new Error("Invalid private key format. Please check the key format and try again.");
+    }
+};
 function getSnowflakeConnection(snowflakeData, authParams) {
     const { authToken, apiKey } = authParams;
     const { account, username, warehouse, database } = snowflakeData;
     if (authToken) {
-        // Always try to use Nango-Snowflake OAuth
+        // Always try to use Nango-Snowflake OAuth (unused for now)
         return snowflake_sdk_1.default.createConnection({
             account: account,
             username: username,
@@ -22,20 +77,6 @@ function getSnowflakeConnection(snowflakeData, authParams) {
         });
     }
     else if (apiKey) {
-        // Use the apiKey for authentication (one off)
-        const getPrivateKeyCorrectFormat = (privateKey) => {
-            const buffer = Buffer.from(privateKey);
-            const privateKeyObject = crypto_1.default.createPrivateKey({
-                key: buffer,
-                format: "pem",
-                passphrase: "password",
-            });
-            const privateKeyCorrectFormat = privateKeyObject.export({
-                format: "pem",
-                type: "pkcs8",
-            });
-            return privateKeyCorrectFormat.toString();
-        };
         const privateKeyCorrectFormatString = getPrivateKeyCorrectFormat(apiKey);
         return snowflake_sdk_1.default.createConnection({
             account: account,
@@ -51,3 +92,30 @@ function getSnowflakeConnection(snowflakeData, authParams) {
         throw new Error("Snowflake authToken or apiKey is required");
     }
 }
+function connectToSnowflakeAndWarehouse(connection, warehouse) {
+    return __awaiter(this, void 0, void 0, function* () {
+        yield new Promise((resolve, reject) => {
+            connection.connect((err, conn) => {
+                if (err) {
+                    console.error("Unable to connect to Snowflake:", err.message);
+                    return reject(err);
+                }
+                resolve(conn);
+            });
+        });
+        if (warehouse) {
+            yield new Promise((resolve, reject) => {
+                connection.execute({
+                    sqlText: `USE WAREHOUSE ${warehouse}`,
+                    complete: (err, stmt, rows) => {
+                        if (err) {
+                            console.error("Unable to use warehouse:", err.message);
+                            return reject(err);
+                        }
+                        resolve(rows);
+                    },
+                });
+            });
+        }
+    });
+}

package/dist/actions/providers/snowflake/runSnowflakeQuery.js

@@ -14,6 +14,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const snowflake_sdk_1 = __importDefault(require("snowflake-sdk"));
 const getSnowflakeConnection_1 = require("./auth/getSnowflakeConnection");
+const formatDataForCodeInterpreter_1 = require("../../util/formatDataForCodeInterpreter");
 snowflake_sdk_1.default.configure({ logLevel: "ERROR" });
 const runSnowflakeQuery = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
     const { databaseName, warehouse, query, user, accountName, outputFormat = "json" } = params;
@@ -34,24 +35,8 @@ const runSnowflakeQuery = (_a) => __awaiter(void 0, [_a], void 0, function* ({ p
             });
         });
         // Format the results based on the output format
-        let formattedData;
-        if (outputFormat.toLowerCase() === "csv") {
-            if (queryResults.length === 0) {
-                formattedData = "";
-            }
-            else {
-                const headers = Object.keys(queryResults[0]).join(",");
-                const rows = queryResults.map(row => Object.values(row)
-                    .map(value => (typeof value === "object" && value !== null ? JSON.stringify(value) : value))
-                    .join(","));
-                formattedData = [headers, ...rows].join("\n");
-            }
-        }
-        else {
-            // Default to JSON
-            formattedData = JSON.stringify(queryResults).replace(/\s+/g, "");
-        }
-        return { formattedData, resultsLength: queryResults.length };
+        const { formattedData, resultsLength } = (0, formatDataForCodeInterpreter_1.formatDataForCodeInterpreter)(queryResults, outputFormat);
+        return { formattedData, resultsLength };
     });
     // Set up a connection using snowflake-sdk
     const connection = (0, getSnowflakeConnection_1.getSnowflakeConnection)({
@@ -62,15 +47,7 @@ const runSnowflakeQuery = (_a) => __awaiter(void 0, [_a], void 0, function* ({ p
     }, { authToken: authParams.authToken, apiKey: authParams.apiKey });
     try {
         // Connect to Snowflake
-        yield new Promise((resolve, reject) => {
-            connection.connect((err, conn) => {
-                if (err) {
-                    console.error("Unable to connect to Snowflake:", err.message);
-                    return reject(err);
-                }
-                resolve(conn);
-            });
-        });
+        yield (0, getSnowflakeConnection_1.connectToSnowflakeAndWarehouse)(connection, warehouse);
         const { formattedData, resultsLength } = yield executeQueryAndFormatData();
         // Return fields to match schema definition
         connection.destroy(err => {

package/dist/actions/util/formatDataForCodeInterpreter.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Formats query results into either CSV or JSON format for code interpreter
+ * @param queryResults The results from a database query
+ * @param outputFormat The desired output format ("csv" or "json")
+ * @returns Object containing formatted data and result count
+ */
+export declare function formatDataForCodeInterpreter(queryResults: any[], outputFormat?: string): {
+    formattedData: string;
+    resultsLength: number;
+};

package/dist/actions/util/formatDataForCodeInterpreter.js

@@ -0,0 +1,43 @@
+"use strict";
+/* eslint-disable @typescript-eslint/no-explicit-any */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatDataForCodeInterpreter = formatDataForCodeInterpreter;
+/**
+ * Formats query results into either CSV or JSON format for code interpreter
+ * @param queryResults The results from a database query
+ * @param outputFormat The desired output format ("csv" or "json")
+ * @returns Object containing formatted data and result count
+ */
+function formatDataForCodeInterpreter(queryResults, outputFormat = "json") {
+    let formattedData;
+    if (outputFormat.toLowerCase() === "csv") {
+        if (queryResults.length === 0) {
+            formattedData = "";
+        }
+        else {
+            const headers = Object.keys(queryResults[0]).join(",");
+            // Helper function to properly escape CSV fields
+            const escapeCsvField = (field) => {
+                if (field === null || field === undefined)
+                    return "";
+                const stringValue = typeof field === "object" ? JSON.stringify(field) : String(field);
+                // If the field contains commas, quotes, or newlines, it needs to be quoted
+                if (stringValue.includes(",") ||
+                    stringValue.includes('"') ||
+                    stringValue.includes("\n") ||
+                    stringValue.includes("\r")) {
+                    // Escape quotes by doubling them and wrap in quotes
+                    return '"' + stringValue.replace(/"/g, '""') + '"';
+                }
+                return stringValue;
+            };
+            const rows = queryResults.map(row => Object.values(row).map(escapeCsvField).join(","));
+            formattedData = [headers, ...rows].join("\n");
+        }
+    }
+    else {
+        // Default to JSON - keep it formatted for readability
+        formattedData = JSON.stringify(queryResults, null, 2);
+    }
+    return { formattedData, resultsLength: queryResults.length };
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@credal/actions",
-  "version": "0.1.43",
+  "version": "0.1.45",
   "description": "AI Actions by Credal AI",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -27,6 +27,7 @@
     "@types/js-yaml": "^4.0.9",
     "@types/jsonwebtoken": "^9.0.9",
     "@types/node": "^22.10.1",
+    "@types/node-forge": "^1.3.11",
     "@typescript-eslint/eslint-plugin": "^8.18.0",
     "@typescript-eslint/parser": "^8.18.0",
     "eslint": "^9.16.0",
@@ -52,6 +53,7 @@
     "json-schema-to-zod": "^2.5.0",
     "jsonwebtoken": "^9.0.2",
     "mongodb": "^6.13.1",
+    "node-forge": "^1.3.1",
     "octokit": "^4.1.2",
     "resend": "^4.1.2",
     "snowflake-sdk": "^2.0.2",