@credal/actions 0.1.23 → 0.1.25

package/dist/actions/actionMapper.js

@@ -27,6 +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"));
 exports.ActionMapper = {
     math: {
         add: {
@@ -98,6 +99,11 @@ exports.ActionMapper = {
             paramsSchema: types_1.snowflakeGetRowByFieldValueParamsSchema,
             outputSchema: types_1.snowflakeGetRowByFieldValueOutputSchema,
         },
+        runSnowflakeQueryWriteResultsToS3: {
+            fn: runSnowflakeQueryWriteResultsToS3_1.default,
+            paramsSchema: types_1.snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema,
+            outputSchema: types_1.snowflakeRunSnowflakeQueryWriteResultsToS3OutputSchema,
+        },
     },
     linkedin: {
         createShareLinkedinPostUrl: {

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

@@ -13,6 +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 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.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.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.slackSendMessageDefinition = {
     description: "Sends a message to a Slack channel",
     scopes: ["chat:write"],
@@ -538,6 +538,29 @@ exports.credalCallCopilotDefinition = {
                     },
                 },
             },
+            webSearchResults: {
+                type: "array",
+                description: "The web search results in the response",
+                items: {
+                    type: "object",
+                    description: "The web search result in the response",
+                    required: ["title", "url"],
+                    properties: {
+                        title: {
+                            type: "string",
+                            description: "The title of the web search result",
+                        },
+                        url: {
+                            type: "string",
+                            description: "The url of the web search result",
+                        },
+                        contents: {
+                            type: "string",
+                            description: "The contents of the web search result",
+                        },
+                    },
+                },
+            },
         },
     },
     name: "callCopilot",
@@ -750,6 +773,69 @@ exports.snowflakeGetRowByFieldValueDefinition = {
     name: "getRowByFieldValue",
     provider: "snowflake",
 };
+exports.snowflakeRunSnowflakeQueryWriteResultsToS3Definition = {
+    description: "Execute a Snowflake query and write results to an S3 bucket",
+    scopes: [],
+    parameters: {
+        type: "object",
+        required: ["databaseName", "warehouse", "query", "user", "accountName", "s3BucketName", "s3Region"],
+        properties: {
+            databaseName: {
+                type: "string",
+                description: "The name of the database to query",
+            },
+            warehouse: {
+                type: "string",
+                description: "The warehouse to use for executing the query",
+            },
+            query: {
+                type: "string",
+                description: "The SQL query to execute",
+            },
+            user: {
+                type: "string",
+                description: "The username to authenticate with",
+            },
+            accountName: {
+                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)",
+                enum: ["json", "csv"],
+            },
+        },
+    },
+    output: {
+        type: "object",
+        required: ["bucketUrl", "message", "rowCount"],
+        properties: {
+            bucketUrl: {
+                type: "string",
+                description: "The URL of the S3 bucket where the results are stored",
+            },
+            message: {
+                type: "string",
+                description: "A message describing the result or error",
+            },
+            rowCount: {
+                type: "number",
+                description: "The number of rows returned by the query",
+            },
+        },
+    },
+    name: "runSnowflakeQueryWriteResultsToS3",
+    provider: "snowflake",
+};
 exports.openstreetmapGetLatitudeLongitudeFromLocationDefinition = {
     description: "Get the latitude and longitude of a location",
     scopes: [],

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

@@ -13,6 +13,8 @@ export declare const AuthParamsSchema: z.ZodObject<{
     emailReplyTo: z.ZodOptional<z.ZodString>;
     emailBcc: z.ZodOptional<z.ZodString>;
     cloudId: z.ZodOptional<z.ZodString>;
+    awsAccessKeyId: z.ZodOptional<z.ZodString>;
+    awsSecretAccessKey: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
     authToken?: string | undefined;
     baseUrl?: string | undefined;
@@ -23,6 +25,8 @@ export declare const AuthParamsSchema: z.ZodObject<{
     emailReplyTo?: string | undefined;
     emailBcc?: string | undefined;
     cloudId?: string | undefined;
+    awsAccessKeyId?: string | undefined;
+    awsSecretAccessKey?: string | undefined;
 }, {
     authToken?: string | undefined;
     baseUrl?: string | undefined;
@@ -33,6 +37,8 @@ export declare const AuthParamsSchema: z.ZodObject<{
     emailReplyTo?: string | undefined;
     emailBcc?: string | undefined;
     cloudId?: string | undefined;
+    awsAccessKeyId?: string | undefined;
+    awsSecretAccessKey?: string | undefined;
 }>;
 export type AuthParamsType = z.infer<typeof AuthParamsSchema>;
 export declare const slackSendMessageParamsSchema: z.ZodObject<{
@@ -545,6 +551,19 @@ export declare const credalCallCopilotOutputSchema: z.ZodObject<{
         };
         url?: string | undefined;
     }>, "many">>;
+    webSearchResults: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        title: z.ZodString;
+        url: z.ZodString;
+        contents: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        title: string;
+        url: string;
+        contents?: string | undefined;
+    }, {
+        title: string;
+        url: string;
+        contents?: string | undefined;
+    }>, "many">>;
 }, "strip", z.ZodTypeAny, {
     response: string;
     referencedSources?: {
@@ -565,6 +584,11 @@ export declare const credalCallCopilotOutputSchema: z.ZodObject<{
         };
         url?: string | undefined;
     }[] | undefined;
+    webSearchResults?: {
+        title: string;
+        url: string;
+        contents?: string | undefined;
+    }[] | undefined;
 }, {
     response: string;
     referencedSources?: {
@@ -585,6 +609,11 @@ export declare const credalCallCopilotOutputSchema: z.ZodObject<{
         };
         url?: string | undefined;
     }[] | undefined;
+    webSearchResults?: {
+        title: string;
+        url: string;
+        contents?: string | undefined;
+    }[] | undefined;
 }>;
 export type credalCallCopilotOutputType = z.infer<typeof credalCallCopilotOutputSchema>;
 export type credalCallCopilotFunction = ActionFunction<credalCallCopilotParamsType, AuthParamsType, credalCallCopilotOutputType>;
@@ -618,11 +647,11 @@ export declare const linkedinCreateShareLinkedinPostUrlParamsSchema: z.ZodObject
     text: z.ZodOptional<z.ZodString>;
     url: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
-    text?: string | undefined;
     url?: string | undefined;
-}, {
     text?: string | undefined;
+}, {
     url?: string | undefined;
+    text?: string | undefined;
 }>;
 export type linkedinCreateShareLinkedinPostUrlParamsType = z.infer<typeof linkedinCreateShareLinkedinPostUrlParamsSchema>;
 export declare const linkedinCreateShareLinkedinPostUrlOutputSchema: z.ZodObject<{
@@ -740,6 +769,50 @@ 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<{
+    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;
+    databaseName: string;
+    accountName: string;
+    user: string;
+    warehouse: string;
+    s3BucketName: string;
+    s3Region: string;
+    outputFormat?: "json" | "csv" | undefined;
+}, {
+    query: string;
+    databaseName: string;
+    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;
+    rowCount: z.ZodNumber;
+}, "strip", z.ZodTypeAny, {
+    message: string;
+    bucketUrl: string;
+    rowCount: number;
+}, {
+    message: string;
+    bucketUrl: string;
+    rowCount: number;
+}>;
+export type snowflakeRunSnowflakeQueryWriteResultsToS3OutputType = z.infer<typeof snowflakeRunSnowflakeQueryWriteResultsToS3OutputSchema>;
+export type snowflakeRunSnowflakeQueryWriteResultsToS3Function = ActionFunction<snowflakeRunSnowflakeQueryWriteResultsToS3ParamsType, AuthParamsType, snowflakeRunSnowflakeQueryWriteResultsToS3OutputType>;
 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.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.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;
 const zod_1 = require("zod");
 exports.AuthParamsSchema = zod_1.z.object({
     authToken: zod_1.z.string().optional(),
@@ -12,6 +12,8 @@ exports.AuthParamsSchema = zod_1.z.object({
     emailReplyTo: zod_1.z.string().optional(),
     emailBcc: zod_1.z.string().optional(),
     cloudId: zod_1.z.string().optional(),
+    awsAccessKeyId: zod_1.z.string().optional(),
+    awsSecretAccessKey: zod_1.z.string().optional(),
 });
 exports.slackSendMessageParamsSchema = zod_1.z.object({
     channelName: zod_1.z.string().describe("The name of the Slack channel to send the message to (e.g. general, alerts)"),
@@ -185,6 +187,16 @@ exports.credalCallCopilotOutputSchema = zod_1.z.object({
         .describe("The source in the data context of the response"))
         .describe("The sources in the data context of the response")
         .optional(),
+    webSearchResults: zod_1.z
+        .array(zod_1.z
+        .object({
+        title: zod_1.z.string().describe("The title of the web search result"),
+        url: zod_1.z.string().describe("The url of the web search result"),
+        contents: zod_1.z.string().describe("The contents of the web search result").optional(),
+    })
+        .describe("The web search result in the response"))
+        .describe("The web search results in the response")
+        .optional(),
 });
 exports.zendeskCreateZendeskTicketParamsSchema = zod_1.z.object({
     subject: zod_1.z.string().describe("The subject of the ticket"),
@@ -237,6 +249,24 @@ exports.snowflakeGetRowByFieldValueOutputSchema = zod_1.z.object({
     })
         .describe("The row from the Snowflake table"),
 });
+exports.snowflakeRunSnowflakeQueryWriteResultsToS3ParamsSchema = 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(),
+});
+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"),
+    rowCount: zod_1.z.number().describe("The number of rows returned by the query"),
+});
 exports.openstreetmapGetLatitudeLongitudeFromLocationParamsSchema = zod_1.z.object({
     location: zod_1.z.string().describe("The location to get the latitude and longitude of"),
 });

package/dist/actions/groups.js

@@ -39,9 +39,9 @@ exports.ACTION_GROUPS = {
         description: "Action for inserting a document into a MongoDB collection",
         actions: [templates_1.mongoInsertMongoDocDefinition],
     },
-    SNOWFLAKE_GET_ROW_BY_FIELD_VALUE: {
-        description: "Action for getting a row from a Snowflake table by field value",
-        actions: [templates_1.snowflakeGetRowByFieldValueDefinition],
+    SNOWFLAKE_ACTIONS: {
+        description: "Action for getting content from a Snowflake table",
+        actions: [templates_1.snowflakeGetRowByFieldValueDefinition, templates_1.snowflakeRunSnowflakeQueryWriteResultsToS3Definition],
     },
     JIRA_CREATE_TICKET: {
         description: "Action for creating a Jira ticket",

package/dist/actions/parse.js

@@ -49,6 +49,8 @@ z.object({
     emailReplyTo: z.string().optional(),
     emailBcc: z.string().optional(),
     cloudId: z.string().optional(),
+    awsAccessKeyId: z.string().optional(),
+    awsSecretAccessKey: z.string().optional(),
 })
 `;
 function validateObject(object) {

package/dist/actions/providers/credal/callCopilot.js

@@ -30,6 +30,7 @@ const callCopilot = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params,
             : "Error getting response",
         referencedSources: response.sendChatResult.type === "ai_response_result" ? response.sendChatResult.referencedSources : [],
         sourcesInDataContext: response.sendChatResult.type === "ai_response_result" ? response.sendChatResult.sourcesInDataContext : [],
+        webSearchResults: response.sendChatResult.type === "ai_response_result" ? response.sendChatResult.webSearchResults : [],
     };
 });
 exports.default = callCopilot;

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

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

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

@@ -0,0 +1,154 @@
+"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"));
+const client_s3_1 = require("@aws-sdk/client-s3");
+const s3_request_presigner_1 = require("@aws-sdk/s3-request-presigner");
+const uuid_1 = require("uuid");
+// Only log errors.
+snowflake_sdk_1.default.configure({ logLevel: "ERROR" });
+const runSnowflakeQueryWriteResultsToS3 = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
+    const { databaseName, warehouse, query, user, accountName, s3BucketName, s3Region, outputFormat = "json" } = params;
+    const { apiKey: privateKey, awsAccessKeyId, awsSecretAccessKey } = authParams;
+    if (!privateKey) {
+        throw new Error("Snowflake private key is required");
+    }
+    if (!awsAccessKeyId || !awsSecretAccessKey) {
+        throw new Error("AWS credentials are required");
+    }
+    if (!accountName || !user || !databaseName || !warehouse || !query || !s3BucketName) {
+        throw new Error("Missing required parameters for Snowflake query or S3 destination");
+    }
+    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, null, 2);
+        }
+        return { formattedData, resultsLength: queryResults.length };
+    });
+    const uploadToS3AndGetURL = (formattedData) => __awaiter(void 0, void 0, void 0, function* () {
+        // Create S3 client
+        const s3Client = new client_s3_1.S3Client({
+            region: s3Region,
+            credentials: {
+                accessKeyId: awsAccessKeyId,
+                secretAccessKey: awsSecretAccessKey,
+            },
+        });
+        const contentType = outputFormat.toLowerCase() === "csv" ? "text/csv" : "application/json";
+        const fileExtension = outputFormat.toLowerCase() === "csv" ? "csv" : "json";
+        const finalKey = `${databaseName}/${(0, uuid_1.v4)()}.${fileExtension}`;
+        // Upload to S3 without ACL
+        const uploadCommand = new client_s3_1.PutObjectCommand({
+            Bucket: s3BucketName,
+            Key: finalKey,
+            Body: formattedData,
+            ContentType: contentType,
+        });
+        yield s3Client.send(uploadCommand);
+        // Generate a presigned URL (valid for an hour)
+        const getObjectCommand = new client_s3_1.GetObjectCommand({
+            Bucket: s3BucketName,
+            Key: finalKey,
+        });
+        const presignedUrl = yield (0, s3_request_presigner_1.getSignedUrl)(s3Client, getObjectCommand, { expiresIn: 3600 });
+        return presignedUrl;
+    });
+    // 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();
+        const presignedUrl = yield uploadToS3AndGetURL(formattedData);
+        // Return fields to match schema definition
+        connection.destroy(err => {
+            if (err) {
+                console.log("Failed to disconnect from Snowflake:", err);
+            }
+        });
+        return {
+            bucketUrl: presignedUrl,
+            message: `Query results successfully written to S3. URL valid for 1 hour.`,
+            rowCount: resultsLength,
+        };
+    }
+    catch (error) {
+        connection.destroy(err => {
+            if (err) {
+                console.log("Failed to disconnect from Snowflake:", err);
+            }
+        });
+        throw Error(`An error occurred: ${error}`);
+    }
+});
+exports.default = runSnowflakeQueryWriteResultsToS3;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@credal/actions",
-  "version": "0.1.23",
+  "version": "0.1.25",
   "description": "AI Actions by Credal AI",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -38,6 +38,7 @@
     "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",
@@ -49,6 +50,7 @@
     "resend": "^4.1.2",
     "snowflake-sdk": "^2.0.2",
     "ts-node": "^10.9.2",
+    "uuid": "^11.1.0",
     "zod": "^3.24.1"
   }
 }