@credal/actions 0.1.25 → 0.1.26

package/dist/actions/actionMapper.js

@@ -27,7 +27,7 @@ const symbolLookup_1 = __importDefault(require("./providers/finnhub/symbolLookup
 const getBasicFinancials_1 = __importDefault(require("./providers/finnhub/getBasicFinancials"));
 const overwritePage_1 = __importDefault(require("./providers/confluence/overwritePage"));
 const fetchPageContent_1 = __importDefault(require("./providers/confluence/fetchPageContent"));
-const runSnowflakeQueryWriteResultsToS3_1 = __importDefault(require("./providers/snowflake/runSnowflakeQueryWriteResultsToS3"));
+const runSnowflakeQuery_1 = __importDefault(require("./providers/snowflake/runSnowflakeQuery"));
 exports.ActionMapper = {
     math: {
         add: {
@@ -99,10 +99,10 @@ exports.ActionMapper = {
             paramsSchema: types_1.snowflakeGetRowByFieldValueParamsSchema,
             outputSchema: types_1.snowflakeGetRowByFieldValueOutputSchema,
         },
-        runSnowflakeQueryWriteResultsToS3: {
-            fn: runSnowflakeQueryWriteResultsToS3_1.default,
-            paramsSchema: types_1.snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema,
-            outputSchema: types_1.snowflakeRunSnowflakeQueryWriteResultsToS3OutputSchema,
+        runSnowflakeQuery: {
+            fn: runSnowflakeQuery_1.default,
+            paramsSchema: types_1.snowflakeRunSnowflakeQueryParamsSchema,
+            outputSchema: types_1.snowflakeRunSnowflakeQueryOutputSchema,
         },
     },
     linkedin: {

package/dist/actions/autogen/templates.d.ts

@@ -13,7 +13,7 @@ export declare const linkedinCreateShareLinkedinPostUrlDefinition: ActionTemplat
 export declare const xCreateShareXPostUrlDefinition: ActionTemplate;
 export declare const mongoInsertMongoDocDefinition: ActionTemplate;
 export declare const snowflakeGetRowByFieldValueDefinition: ActionTemplate;
-export declare const snowflakeRunSnowflakeQueryWriteResultsToS3Definition: ActionTemplate;
+export declare const snowflakeRunSnowflakeQueryDefinition: ActionTemplate;
 export declare const openstreetmapGetLatitudeLongitudeFromLocationDefinition: ActionTemplate;
 export declare const nwsGetForecastForLocationDefinition: ActionTemplate;
 export declare const firecrawlScrapeUrlDefinition: ActionTemplate;

package/dist/actions/autogen/templates.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.finnhubGetBasicFinancialsDefinition = exports.finnhubSymbolLookupDefinition = exports.googleOauthCreateNewGoogleDocDefinition = exports.resendSendEmailDefinition = exports.firecrawlScrapeTweetDataWithNitterDefinition = exports.firecrawlScrapeUrlDefinition = exports.nwsGetForecastForLocationDefinition = exports.openstreetmapGetLatitudeLongitudeFromLocationDefinition = exports.snowflakeRunSnowflakeQueryWriteResultsToS3Definition = exports.snowflakeGetRowByFieldValueDefinition = exports.mongoInsertMongoDocDefinition = exports.xCreateShareXPostUrlDefinition = exports.linkedinCreateShareLinkedinPostUrlDefinition = exports.zendeskCreateZendeskTicketDefinition = exports.credalCallCopilotDefinition = exports.googlemapsNearbysearchRestaurantsDefinition = exports.googlemapsValidateAddressDefinition = exports.jiraCreateJiraTicketDefinition = exports.confluenceFetchPageContentDefinition = exports.confluenceOverwritePageDefinition = exports.mathAddDefinition = exports.slackListConversationsDefinition = exports.slackSendMessageDefinition = void 0;
+exports.finnhubGetBasicFinancialsDefinition = exports.finnhubSymbolLookupDefinition = exports.googleOauthCreateNewGoogleDocDefinition = exports.resendSendEmailDefinition = exports.firecrawlScrapeTweetDataWithNitterDefinition = exports.firecrawlScrapeUrlDefinition = exports.nwsGetForecastForLocationDefinition = exports.openstreetmapGetLatitudeLongitudeFromLocationDefinition = exports.snowflakeRunSnowflakeQueryDefinition = exports.snowflakeGetRowByFieldValueDefinition = exports.mongoInsertMongoDocDefinition = exports.xCreateShareXPostUrlDefinition = exports.linkedinCreateShareLinkedinPostUrlDefinition = exports.zendeskCreateZendeskTicketDefinition = exports.credalCallCopilotDefinition = exports.googlemapsNearbysearchRestaurantsDefinition = exports.googlemapsValidateAddressDefinition = exports.jiraCreateJiraTicketDefinition = exports.confluenceFetchPageContentDefinition = exports.confluenceOverwritePageDefinition = exports.mathAddDefinition = exports.slackListConversationsDefinition = exports.slackSendMessageDefinition = void 0;
 exports.slackSendMessageDefinition = {
     description: "Sends a message to a Slack channel",
     scopes: ["chat:write"],
@@ -773,12 +773,12 @@ exports.snowflakeGetRowByFieldValueDefinition = {
     name: "getRowByFieldValue",
     provider: "snowflake",
 };
-exports.snowflakeRunSnowflakeQueryWriteResultsToS3Definition = {
-    description: "Execute a Snowflake query and write results to an S3 bucket",
+exports.snowflakeRunSnowflakeQueryDefinition = {
+    description: "Execute a Snowflake query and return output.",
     scopes: [],
     parameters: {
         type: "object",
-        required: ["databaseName", "warehouse", "query", "user", "accountName", "s3BucketName", "s3Region"],
+        required: ["databaseName", "warehouse", "query", "user", "accountName"],
         properties: {
             databaseName: {
                 type: "string",
@@ -800,32 +800,20 @@ exports.snowflakeRunSnowflakeQueryWriteResultsToS3Definition = {
                 type: "string",
                 description: "The name of the Snowflake account",
             },
-            s3BucketName: {
-                type: "string",
-                description: "The name of the S3 bucket to write results to",
-            },
-            s3Region: {
-                type: "string",
-                description: "The AWS region where the S3 bucket is located",
-            },
             outputFormat: {
                 type: "string",
-                description: "Format for the output file (json or csv, defaults to json)",
+                description: "The format of the output",
                 enum: ["json", "csv"],
             },
         },
     },
     output: {
         type: "object",
-        required: ["bucketUrl", "message", "rowCount"],
+        required: ["content", "rowCount"],
         properties: {
-            bucketUrl: {
-                type: "string",
-                description: "The URL of the S3 bucket where the results are stored",
-            },
-            message: {
+            content: {
                 type: "string",
-                description: "A message describing the result or error",
+                description: "The content of the query result (json)",
             },
             rowCount: {
                 type: "number",
@@ -833,7 +821,7 @@ exports.snowflakeRunSnowflakeQueryWriteResultsToS3Definition = {
             },
         },
     },
-    name: "runSnowflakeQueryWriteResultsToS3",
+    name: "runSnowflakeQuery",
     provider: "snowflake",
 };
 exports.openstreetmapGetLatitudeLongitudeFromLocationDefinition = {

package/dist/actions/autogen/types.d.ts

@@ -769,14 +769,12 @@ export declare const snowflakeGetRowByFieldValueOutputSchema: z.ZodObject<{
 }>;
 export type snowflakeGetRowByFieldValueOutputType = z.infer<typeof snowflakeGetRowByFieldValueOutputSchema>;
 export type snowflakeGetRowByFieldValueFunction = ActionFunction<snowflakeGetRowByFieldValueParamsType, AuthParamsType, snowflakeGetRowByFieldValueOutputType>;
-export declare const snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema: z.ZodObject<{
+export declare const snowflakeRunSnowflakeQueryParamsSchema: z.ZodObject<{
     databaseName: z.ZodString;
     warehouse: z.ZodString;
     query: z.ZodString;
     user: z.ZodString;
     accountName: z.ZodString;
-    s3BucketName: z.ZodString;
-    s3Region: z.ZodString;
     outputFormat: z.ZodOptional<z.ZodEnum<["json", "csv"]>>;
 }, "strip", z.ZodTypeAny, {
     query: string;
@@ -784,8 +782,6 @@ export declare const snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema: z.Z
     accountName: string;
     user: string;
     warehouse: string;
-    s3BucketName: string;
-    s3Region: string;
     outputFormat?: "json" | "csv" | undefined;
 }, {
     query: string;
@@ -793,26 +789,21 @@ export declare const snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema: z.Z
     accountName: string;
     user: string;
     warehouse: string;
-    s3BucketName: string;
-    s3Region: string;
     outputFormat?: "json" | "csv" | undefined;
 }>;
-export type snowflakeRunSnowflakeQueryWriteResultsToS3ParamsType = z.infer<typeof snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema>;
-export declare const snowflakeRunSnowflakeQueryWriteResultsToS3OutputSchema: z.ZodObject<{
-    bucketUrl: z.ZodString;
-    message: z.ZodString;
+export type snowflakeRunSnowflakeQueryParamsType = z.infer<typeof snowflakeRunSnowflakeQueryParamsSchema>;
+export declare const snowflakeRunSnowflakeQueryOutputSchema: z.ZodObject<{
+    content: z.ZodString;
     rowCount: z.ZodNumber;
 }, "strip", z.ZodTypeAny, {
-    message: string;
-    bucketUrl: string;
+    content: string;
     rowCount: number;
 }, {
-    message: string;
-    bucketUrl: string;
+    content: string;
     rowCount: number;
 }>;
-export type snowflakeRunSnowflakeQueryWriteResultsToS3OutputType = z.infer<typeof snowflakeRunSnowflakeQueryWriteResultsToS3OutputSchema>;
-export type snowflakeRunSnowflakeQueryWriteResultsToS3Function = ActionFunction<snowflakeRunSnowflakeQueryWriteResultsToS3ParamsType, AuthParamsType, snowflakeRunSnowflakeQueryWriteResultsToS3OutputType>;
+export type snowflakeRunSnowflakeQueryOutputType = z.infer<typeof snowflakeRunSnowflakeQueryOutputSchema>;
+export type snowflakeRunSnowflakeQueryFunction = ActionFunction<snowflakeRunSnowflakeQueryParamsType, AuthParamsType, snowflakeRunSnowflakeQueryOutputType>;
 export declare const openstreetmapGetLatitudeLongitudeFromLocationParamsSchema: z.ZodObject<{
     location: z.ZodString;
 }, "strip", z.ZodTypeAny, {

package/dist/actions/autogen/types.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.finnhubGetBasicFinancialsOutputSchema = exports.finnhubGetBasicFinancialsParamsSchema = exports.finnhubSymbolLookupOutputSchema = exports.finnhubSymbolLookupParamsSchema = exports.googleOauthCreateNewGoogleDocOutputSchema = exports.googleOauthCreateNewGoogleDocParamsSchema = exports.resendSendEmailOutputSchema = exports.resendSendEmailParamsSchema = exports.firecrawlScrapeTweetDataWithNitterOutputSchema = exports.firecrawlScrapeTweetDataWithNitterParamsSchema = exports.firecrawlScrapeUrlOutputSchema = exports.firecrawlScrapeUrlParamsSchema = exports.nwsGetForecastForLocationOutputSchema = exports.nwsGetForecastForLocationParamsSchema = exports.openstreetmapGetLatitudeLongitudeFromLocationOutputSchema = exports.openstreetmapGetLatitudeLongitudeFromLocationParamsSchema = exports.snowflakeRunSnowflakeQueryWriteResultsToS3OutputSchema = exports.snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema = exports.snowflakeGetRowByFieldValueOutputSchema = exports.snowflakeGetRowByFieldValueParamsSchema = exports.mongoInsertMongoDocOutputSchema = exports.mongoInsertMongoDocParamsSchema = exports.xCreateShareXPostUrlOutputSchema = exports.xCreateShareXPostUrlParamsSchema = exports.linkedinCreateShareLinkedinPostUrlOutputSchema = exports.linkedinCreateShareLinkedinPostUrlParamsSchema = exports.zendeskCreateZendeskTicketOutputSchema = exports.zendeskCreateZendeskTicketParamsSchema = exports.credalCallCopilotOutputSchema = exports.credalCallCopilotParamsSchema = exports.googlemapsNearbysearchRestaurantsOutputSchema = exports.googlemapsNearbysearchRestaurantsParamsSchema = exports.googlemapsValidateAddressOutputSchema = exports.googlemapsValidateAddressParamsSchema = exports.jiraCreateJiraTicketOutputSchema = exports.jiraCreateJiraTicketParamsSchema = exports.confluenceFetchPageContentOutputSchema = exports.confluenceFetchPageContentParamsSchema = exports.confluenceOverwritePageOutputSchema = exports.confluenceOverwritePageParamsSchema = exports.mathAddOutputSchema = exports.mathAddParamsSchema = exports.slackListConversationsOutputSchema = exports.slackListConversationsParamsSchema = exports.slackSendMessageOutputSchema = exports.slackSendMessageParamsSchema = exports.AuthParamsSchema = void 0;
+exports.finnhubGetBasicFinancialsOutputSchema = exports.finnhubGetBasicFinancialsParamsSchema = exports.finnhubSymbolLookupOutputSchema = exports.finnhubSymbolLookupParamsSchema = exports.googleOauthCreateNewGoogleDocOutputSchema = exports.googleOauthCreateNewGoogleDocParamsSchema = exports.resendSendEmailOutputSchema = exports.resendSendEmailParamsSchema = exports.firecrawlScrapeTweetDataWithNitterOutputSchema = exports.firecrawlScrapeTweetDataWithNitterParamsSchema = exports.firecrawlScrapeUrlOutputSchema = exports.firecrawlScrapeUrlParamsSchema = exports.nwsGetForecastForLocationOutputSchema = exports.nwsGetForecastForLocationParamsSchema = exports.openstreetmapGetLatitudeLongitudeFromLocationOutputSchema = exports.openstreetmapGetLatitudeLongitudeFromLocationParamsSchema = exports.snowflakeRunSnowflakeQueryOutputSchema = exports.snowflakeRunSnowflakeQueryParamsSchema = exports.snowflakeGetRowByFieldValueOutputSchema = exports.snowflakeGetRowByFieldValueParamsSchema = exports.mongoInsertMongoDocOutputSchema = exports.mongoInsertMongoDocParamsSchema = exports.xCreateShareXPostUrlOutputSchema = exports.xCreateShareXPostUrlParamsSchema = exports.linkedinCreateShareLinkedinPostUrlOutputSchema = exports.linkedinCreateShareLinkedinPostUrlParamsSchema = exports.zendeskCreateZendeskTicketOutputSchema = exports.zendeskCreateZendeskTicketParamsSchema = exports.credalCallCopilotOutputSchema = exports.credalCallCopilotParamsSchema = exports.googlemapsNearbysearchRestaurantsOutputSchema = exports.googlemapsNearbysearchRestaurantsParamsSchema = exports.googlemapsValidateAddressOutputSchema = exports.googlemapsValidateAddressParamsSchema = exports.jiraCreateJiraTicketOutputSchema = exports.jiraCreateJiraTicketParamsSchema = exports.confluenceFetchPageContentOutputSchema = exports.confluenceFetchPageContentParamsSchema = exports.confluenceOverwritePageOutputSchema = exports.confluenceOverwritePageParamsSchema = exports.mathAddOutputSchema = exports.mathAddParamsSchema = exports.slackListConversationsOutputSchema = exports.slackListConversationsParamsSchema = exports.slackSendMessageOutputSchema = exports.slackSendMessageParamsSchema = exports.AuthParamsSchema = void 0;
 const zod_1 = require("zod");
 exports.AuthParamsSchema = zod_1.z.object({
     authToken: zod_1.z.string().optional(),
@@ -249,22 +249,16 @@ exports.snowflakeGetRowByFieldValueOutputSchema = zod_1.z.object({
     })
         .describe("The row from the Snowflake table"),
 });
-exports.snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema = zod_1.z.object({
+exports.snowflakeRunSnowflakeQueryParamsSchema = zod_1.z.object({
     databaseName: zod_1.z.string().describe("The name of the database to query"),
     warehouse: zod_1.z.string().describe("The warehouse to use for executing the query"),
     query: zod_1.z.string().describe("The SQL query to execute"),
     user: zod_1.z.string().describe("The username to authenticate with"),
     accountName: zod_1.z.string().describe("The name of the Snowflake account"),
-    s3BucketName: zod_1.z.string().describe("The name of the S3 bucket to write results to"),
-    s3Region: zod_1.z.string().describe("The AWS region where the S3 bucket is located"),
-    outputFormat: zod_1.z
-        .enum(["json", "csv"])
-        .describe("Format for the output file (json or csv, defaults to json)")
-        .optional(),
+    outputFormat: zod_1.z.enum(["json", "csv"]).describe("The format of the output").optional(),
 });
-exports.snowflakeRunSnowflakeQueryWriteResultsToS3OutputSchema = zod_1.z.object({
-    bucketUrl: zod_1.z.string().describe("The URL of the S3 bucket where the results are stored"),
-    message: zod_1.z.string().describe("A message describing the result or error"),
+exports.snowflakeRunSnowflakeQueryOutputSchema = zod_1.z.object({
+    content: zod_1.z.string().describe("The content of the query result (json)"),
     rowCount: zod_1.z.number().describe("The number of rows returned by the query"),
 });
 exports.openstreetmapGetLatitudeLongitudeFromLocationParamsSchema = zod_1.z.object({

package/dist/actions/groups.js

@@ -41,7 +41,7 @@ exports.ACTION_GROUPS = {
     },
     SNOWFLAKE_ACTIONS: {
         description: "Action for getting content from a Snowflake table",
-        actions: [templates_1.snowflakeGetRowByFieldValueDefinition, templates_1.snowflakeRunSnowflakeQueryWriteResultsToS3Definition],
+        actions: [templates_1.snowflakeGetRowByFieldValueDefinition, templates_1.snowflakeRunSnowflakeQueryDefinition],
     },
     JIRA_CREATE_TICKET: {
         description: "Action for creating a Jira ticket",

package/dist/actions/providers/snowflake/runSnowflakeQuery.d.ts

@@ -0,0 +1,3 @@
+import { snowflakeRunSnowflakeQueryFunction } from "../../autogen/types";
+declare const runSnowflakeQuery: snowflakeRunSnowflakeQueryFunction;
+export default runSnowflakeQuery;

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

@@ -0,0 +1,118 @@
+"use strict";
+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 });
+const snowflake_sdk_1 = __importDefault(require("snowflake-sdk"));
+const crypto_1 = __importDefault(require("crypto"));
+// Only log errors.
+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;
+    const { apiKey: privateKey } = authParams;
+    if (!privateKey) {
+        throw new Error("Snowflake private key is required");
+    }
+    if (!accountName || !user || !databaseName || !warehouse || !query) {
+        throw new Error("Missing required parameters for Snowflake query");
+    }
+    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 executeQueryAndFormatData = () => __awaiter(void 0, void 0, void 0, function* () {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const queryResults = yield new Promise((resolve, reject) => {
+            connection.execute({
+                sqlText: query,
+                complete: (err, stmt, rows) => {
+                    if (err) {
+                        return reject(err);
+                    }
+                    return resolve(rows || []);
+                },
+            });
+        });
+        // 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 };
+    });
+    // Process the private key
+    const privateKeyCorrectFormatString = getPrivateKeyCorrectFormat(privateKey);
+    // Set up a connection using snowflake-sdk
+    const connection = snowflake_sdk_1.default.createConnection({
+        account: accountName,
+        username: user,
+        privateKey: privateKeyCorrectFormatString,
+        authenticator: "SNOWFLAKE_JWT",
+        role: "ACCOUNTADMIN",
+        warehouse: warehouse,
+        database: databaseName,
+    });
+    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);
+            });
+        });
+        const { formattedData, resultsLength } = yield executeQueryAndFormatData();
+        // Return fields to match schema definition
+        connection.destroy(err => {
+            if (err) {
+                console.log("Failed to disconnect from Snowflake:", err);
+            }
+        });
+        return {
+            rowCount: resultsLength,
+            content: formattedData,
+        };
+    }
+    catch (error) {
+        connection.destroy(err => {
+            if (err) {
+                console.log("Failed to disconnect from Snowflake:", err);
+            }
+        });
+        throw Error(`An error occurred: ${error}`);
+    }
+});
+exports.default = runSnowflakeQuery;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@credal/actions",
-  "version": "0.1.25",
+  "version": "0.1.26",
   "description": "AI Actions by Credal AI",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -38,7 +38,6 @@
     "typescript-eslint": "^8.18.0"
   },
   "dependencies": {
-    "@aws-sdk/s3-request-presigner": "^3.758.0",
     "@credal/sdk": "^0.0.21",
     "@mendable/firecrawl-js": "^1.19.0",
     "@slack/web-api": "^7.8.0",
@@ -50,7 +49,6 @@
     "resend": "^4.1.2",
     "snowflake-sdk": "^2.0.2",
     "ts-node": "^10.9.2",
-    "uuid": "^11.1.0",
     "zod": "^3.24.1"
   }
 }