@activepieces/piece-ai 0.0.1

package/src/lib/actions/image/generate-image.js

@@ -0,0 +1,216 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateImageAction = void 0;
+const tslib_1 = require("tslib");
+const pieces_framework_1 = require("@activepieces/pieces-framework");
+const ai_1 = require("ai");
+const ai_2 = require("ai");
+const mime_types_1 = tslib_1.__importDefault(require("mime-types"));
+const shared_1 = require("@activepieces/shared");
+const ai_sdk_1 = require("../../common/ai-sdk");
+const types_1 = require("../../common/types");
+const props_1 = require("../../common/props");
+exports.generateImageAction = (0, pieces_framework_1.createAction)({
+    name: 'generateImage',
+    displayName: 'Generate Image',
+    description: '',
+    props: {
+        provider: (0, props_1.aiProps)({ modelType: 'image' }).provider,
+        model: (0, props_1.aiProps)({ modelType: 'image' }).model,
+        prompt: pieces_framework_1.Property.LongText({
+            displayName: 'Prompt',
+            required: true,
+        }),
+        advancedOptions: pieces_framework_1.Property.DynamicProperties({
+            displayName: 'Advanced Options',
+            required: false,
+            auth: pieces_framework_1.PieceAuth.None(),
+            refreshers: ['provider', 'model'],
+            props: (propsValue) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+                const providerId = propsValue['provider'];
+                const modelId = propsValue['model'];
+                let options = {};
+                if (providerId === types_1.AIProviderName.OPENAI) {
+                    options = {
+                        quality: pieces_framework_1.Property.StaticDropdown({
+                            options: {
+                                options: modelId === 'dall-e-3'
+                                    ? [
+                                        { label: 'Standard', value: 'standard' },
+                                        { label: 'HD', value: 'hd' },
+                                    ]
+                                    : modelId === 'gpt-image-1'
+                                        ? [
+                                            { label: 'High', value: 'high' },
+                                            { label: 'Medium', value: 'medium' },
+                                            { label: 'Low', value: 'low' },
+                                        ]
+                                        : [],
+                                disabled: modelId === 'dall-e-2',
+                            },
+                            defaultValue: modelId === 'dall-e-3' ? 'standard' : 'high',
+                            displayName: 'Image Quality',
+                            required: false,
+                        }),
+                        size: pieces_framework_1.Property.StaticDropdown({
+                            options: {
+                                options: modelId === 'dall-e-3'
+                                    ? [
+                                        { label: '1024x1024', value: '1024x1024' },
+                                        { label: '1792x1024', value: '1792x1024' },
+                                        { label: '1024x1792', value: '1024x1792' },
+                                    ]
+                                    : modelId === 'gpt-image-1'
+                                        ? [
+                                            { label: '1024x1024', value: '1024x1024' },
+                                            { label: '1536x1024', value: '1536x1024' },
+                                            { label: '1024x1536', value: '1024x1536' },
+                                        ]
+                                        : [
+                                            { label: '256x256', value: '256x256' },
+                                            { label: '512x512', value: '512x512' },
+                                            { label: '1024x1024', value: '1024x1024' },
+                                        ],
+                            },
+                            displayName: 'Image Size',
+                            required: false,
+                        }),
+                    };
+                    if (modelId === 'gpt-image-1') {
+                        options = Object.assign(Object.assign({}, options), { background: pieces_framework_1.Property.StaticDropdown({
+                                options: {
+                                    options: [
+                                        { label: 'Auto', value: 'auto' },
+                                        { label: 'Transparent', value: 'transparent' },
+                                        { label: 'Opaque', value: 'opaque' },
+                                    ],
+                                },
+                                defaultValue: 'auto',
+                                description: 'The background of the image.',
+                                displayName: 'Background',
+                                required: true,
+                            }) });
+                    }
+                    return options;
+                }
+                if (providerId === types_1.AIProviderName.GOOGLE &&
+                    modelId === 'gemini-2.5-flash-image-preview') {
+                    options = {
+                        image: pieces_framework_1.Property.Array({
+                            displayName: 'Images',
+                            required: false,
+                            properties: {
+                                file: pieces_framework_1.Property.File({
+                                    displayName: 'Image File',
+                                    required: true,
+                                }),
+                            },
+                            description: 'The image(s) you want to edit/merge',
+                        }),
+                    };
+                }
+                return options;
+            }),
+        }),
+    },
+    run(context) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            const providerId = context.propsValue.provider;
+            const modelId = context.propsValue.model;
+            const image = yield getGeneratedImage({
+                providerId,
+                modelId,
+                engineToken: context.server.token,
+                apiUrl: context.server.apiUrl,
+                prompt: context.propsValue.prompt,
+                advancedOptions: context.propsValue.advancedOptions,
+            });
+            const imageData = image.base64 && image.base64.length > 0
+                ? Buffer.from(image.base64, 'base64')
+                : Buffer.from(image.uint8Array);
+            return context.files.write({
+                data: imageData,
+                fileName: 'image.png',
+            });
+        });
+    },
+});
+const getGeneratedImage = (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ providerId, modelId, engineToken, apiUrl, prompt, advancedOptions, }) {
+    const model = yield (0, ai_sdk_1.createAIModel)({
+        providerId,
+        modelId,
+        engineToken,
+        apiUrl,
+        isImage: true,
+    });
+    switch (providerId) {
+        case types_1.AIProviderName.GOOGLE:
+        case types_1.AIProviderName.ACTIVEPIECES:
+        case types_1.AIProviderName.OPENROUTER:
+            return generateImageUsingGenerateText({
+                model: model,
+                prompt,
+                advancedOptions,
+            });
+        default:
+            {
+                const { image } = yield (0, ai_2.experimental_generateImage)({
+                    model,
+                    prompt,
+                    providerOptions: {
+                        [providerId]: Object.assign({}, advancedOptions),
+                    },
+                });
+                return image;
+            }
+            ;
+    }
+});
+const generateImageUsingGenerateText = (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ model, prompt, advancedOptions, }) {
+    var _b;
+    const images = (_b = advancedOptions === null || advancedOptions === void 0 ? void 0 : advancedOptions['image']) !== null && _b !== void 0 ? _b : [];
+    const imageFiles = images.map((image) => {
+        const fileType = image.file.extension
+            ? mime_types_1.default.lookup(image.file.extension)
+            : 'image/jpeg';
+        return {
+            type: 'image',
+            image: `data:${fileType};base64,${image.file.base64}`,
+        };
+    });
+    const result = yield (0, ai_1.generateText)({
+        model,
+        providerOptions: {
+            google: { responseModalities: ['TEXT', 'IMAGE'] },
+            openrouter: { modalities: ['image', 'text'] },
+        },
+        messages: [
+            {
+                role: 'user',
+                content: [{ type: 'text', text: prompt }, ...imageFiles],
+            },
+        ],
+    });
+    assertImageGenerationSuccess(result);
+    return result.files[0];
+});
+const assertImageGenerationSuccess = (result) => {
+    const responseBody = result.response.body &&
+        typeof result.response.body === 'object' &&
+        'candidates' in result.response.body
+        ? result.response.body
+        : { candidates: [] };
+    const responseCandidates = Array.isArray(responseBody === null || responseBody === void 0 ? void 0 : responseBody.candidates)
+        ? responseBody === null || responseBody === void 0 ? void 0 : responseBody.candidates
+        : [];
+    responseCandidates.forEach((candidate) => {
+        if (candidate.finishReason !== 'STOP') {
+            throw new Error('Image generation failed Reason:\n ' +
+                JSON.stringify(responseCandidates, null, 2));
+        }
+    });
+    if ((0, shared_1.isNil)(result.files) || result.files.length === 0) {
+        throw new Error('No image generated');
+    }
+};
+//# sourceMappingURL=generate-image.js.map

package/src/lib/actions/image/generate-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-image.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/community/ai/src/lib/actions/image/generate-image.ts"],"names":[],"mappings":";;;;AAAA,qEAOwC;AACxC,2BAMY;AACZ,2BAAiE;AAEjE,oEAA8B;AAC9B,iDAA6C;AAC7C,gDAAoD;AACpD,8CAAoD;AACpD,8CAA6C;AAEhC,QAAA,mBAAmB,GAAG,IAAA,+BAAY,EAAC;IAC9C,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,gBAAgB;IAC7B,WAAW,EAAE,EAAE;IACf,KAAK,EAAE;QACL,QAAQ,EAAE,IAAA,eAAO,EAAC,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,QAAQ;QAClD,KAAK,EAAE,IAAA,eAAO,EAAC,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK;QAC5C,MAAM,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACxB,WAAW,EAAE,QAAQ;YACrB,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,eAAe,EAAE,2BAAQ,CAAC,iBAAiB,CAAC;YAC1C,WAAW,EAAE,kBAAkB;YAC/B,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,4BAAS,CAAC,IAAI,EAAE;YACtB,UAAU,EAAE,CAAC,UAAU,EAAE,OAAO,CAAC;YACjC,KAAK,EAAE,CAAO,UAAU,EAA6B,EAAE;gBACrD,MAAM,UAAU,GAAG,UAAU,CAAC,UAAU,CAAsB,CAAC;gBAC/D,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAsB,CAAC;gBAEzD,IAAI,OAAO,GAAqB,EAAE,CAAC;gBAEnC,IAAI,UAAU,KAAK,sBAAc,CAAC,MAAM,EAAE,CAAC;oBACzC,OAAO,GAAG;wBACR,OAAO,EAAE,2BAAQ,CAAC,cAAc,CAAC;4BAC/B,OAAO,EAAE;gCACP,OAAO,EACL,OAAO,KAAK,UAAU;oCACpB,CAAC,CAAC;wCACA,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;wCACxC,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE;qCAC7B;oCACD,CAAC,CAAC,OAAO,KAAK,aAAa;wCACzB,CAAC,CAAC;4CACA,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE;4CAChC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;4CACpC,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;yCAC/B;wCACD,CAAC,CAAC,EAAE;gCACV,QAAQ,EAAE,OAAO,KAAK,UAAU;6BACjC;4BACD,YAAY,EAAE,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM;4BAC1D,WAAW,EAAE,eAAe;4BAC5B,QAAQ,EAAE,KAAK;yBAChB,CAAC;wBACF,IAAI,EAAE,2BAAQ,CAAC,cAAc,CAAC;4BAC5B,OAAO,EAAE;gCACP,OAAO,EACL,OAAO,KAAK,UAAU;oCACpB,CAAC,CAAC;wCACA,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE;wCAC1C,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE;wCAC1C,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE;qCAC3C;oCACD,CAAC,CAAC,OAAO,KAAK,aAAa;wCACzB,CAAC,CAAC;4CACA,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE;4CAC1C,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE;4CAC1C,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE;yCAC3C;wCACD,CAAC,CAAC;4CACA,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;4CACtC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;4CACtC,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE;yCAC3C;6BACR;4BACD,WAAW,EAAE,YAAY;4BACzB,QAAQ,EAAE,KAAK;yBAChB,CAAC;qBACH,CAAC;oBAEF,IAAI,OAAO,KAAK,aAAa,EAAE,CAAC;wBAC9B,OAAO,mCACF,OAAO,KACV,UAAU,EAAE,2BAAQ,CAAC,cAAc,CAAC;gCAClC,OAAO,EAAE;oCACP,OAAO,EAAE;wCACP,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE;wCAChC,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,aAAa,EAAE;wCAC9C,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;qCACrC;iCACF;gCACD,YAAY,EAAE,MAAM;gCACpB,WAAW,EAAE,8BAA8B;gCAC3C,WAAW,EAAE,YAAY;gCACzB,QAAQ,EAAE,IAAI;6BACf,CAAC,GACH,CAAC;oBACJ,CAAC;oBAED,OAAO,OAAO,CAAC;gBACjB,CAAC;gBAED,IACE,UAAU,KAAK,sBAAc,CAAC,MAAM;oBACpC,OAAO,KAAK,gCAAgC,EAC5C,CAAC;oBACD,OAAO,GAAG;wBACR,KAAK,EAAE,2BAAQ,CAAC,KAAK,CAAC;4BACpB,WAAW,EAAE,QAAQ;4BACrB,QAAQ,EAAE,KAAK;4BACf,UAAU,EAAE;gCACV,IAAI,EAAE,2BAAQ,CAAC,IAAI,CAAC;oCAClB,WAAW,EAAE,YAAY;oCACzB,QAAQ,EAAE,IAAI;iCACf,CAAC;6BACH;4BACD,WAAW,EAAE,qCAAqC;yBACnD,CAAC;qBACH,CAAC;gBACJ,CAAC;gBAED,OAAO,OAAO,CAAC;YACjB,CAAC,CAAA;SACF,CAAC;KACH;IACK,GAAG,CAAC,OAAO;;YACf,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;YAC/C,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;YAEzC,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC;gBACpC,UAAU;gBACV,OAAO;gBACP,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;gBACjC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM;gBAC7B,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,MAAM;gBACjC,eAAe,EAAE,OAAO,CAAC,UAAU,CAAC,eAAe;aACpD,CAAC,CAAC;YAEH,MAAM,SAAS,GACb,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;gBACrC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC;gBACrC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAEpC,OAAO,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC;gBACzB,IAAI,EAAE,SAAS;gBACf,QAAQ,EAAE,WAAW;aACtB,CAAC,CAAC;QACL,CAAC;KAAA;CACF,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG,KAcC,EAAE,oDAdI,EAC/B,UAAU,EACV,OAAO,EACP,WAAW,EACX,MAAM,EACN,MAAM,EACN,eAAe,GAQhB;IACC,MAAM,KAAK,GAAG,MAAM,IAAA,sBAAa,EAAC;QAChC,UAAU;QACV,OAAO;QACP,WAAW;QACX,MAAM;QACN,OAAO,EAAE,IAAI;KACd,CAAC,CAAC;IAEH,QAAQ,UAAU,EAAE,CAAC;QACnB,KAAK,sBAAc,CAAC,MAAM,CAAC;QAC3B,KAAK,sBAAc,CAAC,YAAY,CAAC;QACjC,KAAK,sBAAc,CAAC,UAAU;YAC5B,OAAO,8BAA8B,CAAC;gBACpC,KAAK,EAAE,KAAmC;gBAC1C,MAAM;gBACN,eAAe;aAChB,CAAC,CAAC;QACL;YAAS,CAAC;gBACR,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAA,+BAAa,EAAC;oBACpC,KAAK;oBACL,MAAM;oBACN,eAAe,EAAE;wBACf,CAAC,UAAU,CAAC,oBAAO,eAAe,CAAE;qBACrC;iBACF,CAAC,CAAC;gBACH,OAAO,KAAK,CAAA;YACd,CAAC;YAAA,CAAC;IACJ,CAAC;AACH,CAAC,CAAA,CAAC;AAEF,MAAM,8BAA8B,GAAG,KAQZ,EAAE,oDARiB,EAC5C,KAAK,EACL,MAAM,EACN,eAAe,GAKhB;;IACC,MAAM,MAAM,GACV,MAAC,eAAe,aAAf,eAAe,uBAAf,eAAe,CAAG,OAAO,CAAyC,mCAAI,EAAE,CAAC;IAE5E,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAY,CAAC,KAAK,EAAE,EAAE;QACjD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS;YACnC,CAAC,CAAC,oBAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;YACnC,CAAC,CAAC,YAAY,CAAC;QAEjB,OAAO;YACL,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,QAAQ,QAAQ,WAAW,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE;SACtD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAY,EAAC;QAChC,KAAK;QACL,eAAe,EAAE;YACf,MAAM,EAAE,EAAE,kBAAkB,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE;YACjD,UAAU,EAAE,EAAE,UAAU,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE;SAC9C;QACD,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,GAAG,UAAU,CAAC;aACzD;SACF;KACF,CAAC,CAAC;IAEH,4BAA4B,CAAC,MAAM,CAAC,CAAC;IAErC,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AACzB,CAAC,CAAA,CAAC;AAEF,MAAM,4BAA4B,GAAG,CACnC,MAA0C,EACpC,EAAE;IACR,MAAM,YAAY,GAChB,MAAM,CAAC,QAAQ,CAAC,IAAI;QAClB,OAAO,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ;QACxC,YAAY,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI;QACpC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI;QACtB,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAEzB,MAAM,kBAAkB,GAAG,KAAK,CAAC,OAAO,CAAC,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,UAAU,CAAC;QAChE,CAAC,CAAC,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,UAAU;QAC1B,CAAC,CAAC,EAAE,CAAC;IAEP,kBAAkB,CAAC,OAAO,CAAC,CAAC,SAAmC,EAAE,EAAE;QACjE,IAAI,SAAS,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;YACtC,MAAM,IAAI,KAAK,CACb,oCAAoC;gBACpC,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,CAC5C,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,IAAA,cAAK,EAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrD,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACxC,CAAC;AACH,CAAC,CAAC"}

package/src/lib/actions/text/ask-ai.d.ts

@@ -0,0 +1,35 @@
+import { ToolSet } from 'ai';
+export declare const askAI: import("@activepieces/pieces-framework").IAction<import("@activepieces/pieces-framework").PieceAuthProperty, {
+    provider: import("@activepieces/pieces-framework").DropdownProperty<string, true, undefined>;
+    model: import("@activepieces/pieces-framework").DropdownProperty<string, true, undefined>;
+    prompt: import("@activepieces/pieces-framework").LongTextProperty<true>;
+    conversationKey: import("@activepieces/pieces-framework").ShortTextProperty<false>;
+    creativity: import("@activepieces/pieces-framework").NumberProperty<false>;
+    maxOutputTokens: import("@activepieces/pieces-framework").NumberProperty<false>;
+    webSearch: import("@activepieces/pieces-framework").CheckboxProperty<false>;
+    webSearchOptions: import("@activepieces/pieces-framework").DynamicProperties<false, undefined>;
+}>;
+export declare function createWebSearchTool(provider: string, options?: WebSearchOptions): ToolSet;
+type BaseWebSearchOptions = {
+    maxUses?: number;
+    includeSources?: boolean;
+};
+type UserLocationOptions = {
+    userLocationCity?: string;
+    userLocationRegion?: string;
+    userLocationCountry?: string;
+    userLocationTimezone?: string;
+};
+type AnthropicWebSearchOptions = BaseWebSearchOptions & UserLocationOptions & {
+    allowedDomains?: {
+        domain: string;
+    }[];
+    blockedDomains?: {
+        domain: string;
+    }[];
+};
+type OpenAIWebSearchOptions = BaseWebSearchOptions & UserLocationOptions & {
+    searchContextSize?: 'low' | 'medium' | 'high';
+};
+export type WebSearchOptions = AnthropicWebSearchOptions | OpenAIWebSearchOptions;
+export {};

package/src/lib/actions/text/ask-ai.js

@@ -0,0 +1,238 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.askAI = void 0;
+exports.createWebSearchTool = createWebSearchTool;
+const tslib_1 = require("tslib");
+const pieces_framework_1 = require("@activepieces/pieces-framework");
+const ai_1 = require("ai");
+const shared_1 = require("@activepieces/shared");
+const props_1 = require("../../common/props");
+const types_1 = require("../../common/types");
+const ai_sdk_1 = require("../../common/ai-sdk");
+exports.askAI = (0, pieces_framework_1.createAction)({
+    name: 'askAi',
+    displayName: 'Ask AI',
+    description: '',
+    props: {
+        provider: (0, props_1.aiProps)({ modelType: 'text' }).provider,
+        model: (0, props_1.aiProps)({ modelType: 'text' }).model,
+        prompt: pieces_framework_1.Property.LongText({
+            displayName: 'Prompt',
+            required: true,
+        }),
+        conversationKey: pieces_framework_1.Property.ShortText({
+            displayName: 'Conversation Key',
+            required: false,
+        }),
+        creativity: pieces_framework_1.Property.Number({
+            displayName: 'Creativity',
+            required: false,
+            defaultValue: 100,
+            description: 'Controls the creativity of the AI response. A higher value will make the AI more creative and a lower value will make it more deterministic.',
+        }),
+        maxOutputTokens: pieces_framework_1.Property.Number({
+            displayName: 'Max Tokens',
+            required: false,
+            defaultValue: 2000,
+        }),
+        webSearch: pieces_framework_1.Property.Checkbox({
+            displayName: 'Web Search',
+            required: false,
+            defaultValue: false,
+            description: 'Whether to use web search to find information for the AI to use in its response.',
+        }),
+        webSearchOptions: pieces_framework_1.Property.DynamicProperties({
+            displayName: 'Web Search Options',
+            required: false,
+            auth: pieces_framework_1.PieceAuth.None(),
+            refreshers: ['webSearch', 'provider', 'model'],
+            props: (propsValue) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+                const webSearchEnabled = propsValue['webSearch'];
+                const provider = propsValue['provider'];
+                if (!webSearchEnabled) {
+                    return {};
+                }
+                let options = {
+                    maxUses: pieces_framework_1.Property.Number({
+                        displayName: 'Max Web Search Uses',
+                        required: false,
+                        defaultValue: 5,
+                        description: 'Maximum number of searches to use. Default is 5.',
+                    }),
+                    includeSources: pieces_framework_1.Property.Checkbox({
+                        displayName: 'Include Sources',
+                        description: 'Whether to include the sources in the response. Useful for getting web search details (e.g. search queries, searched URLs, etc).',
+                        required: false,
+                        defaultValue: false,
+                    }),
+                };
+                const userLocationOptions = {
+                    userLocationCity: pieces_framework_1.Property.ShortText({
+                        displayName: 'User Location - City',
+                        required: false,
+                        description: 'The city name for localizing search results (e.g., San Francisco).',
+                    }),
+                    userLocationRegion: pieces_framework_1.Property.ShortText({
+                        displayName: 'User Location - Region',
+                        required: false,
+                        description: 'The region or state for localizing search results (e.g., California).',
+                    }),
+                    userLocationCountry: pieces_framework_1.Property.ShortText({
+                        displayName: 'User Location - Country',
+                        required: false,
+                        description: 'The country code for localizing search results (e.g., US).',
+                    }),
+                    userLocationTimezone: pieces_framework_1.Property.ShortText({
+                        displayName: 'User Location - Timezone',
+                        required: false,
+                        description: 'The IANA timezone ID for localizing search results (e.g., America/Los_Angeles).',
+                    }),
+                };
+                if (provider === types_1.AIProviderName.ANTHROPIC) {
+                    options = Object.assign(Object.assign(Object.assign({}, options), { allowedDomains: pieces_framework_1.Property.Array({
+                            displayName: 'Allowed Domains',
+                            required: false,
+                            description: 'List of domains to search (e.g., example.com, docs.example.com/blog). Domains should not include HTTP/HTTPS scheme. Subdomains are automatically included unless more specific subpaths are provided. Overrides Blocked Domains if both are provided.',
+                            properties: {
+                                domain: pieces_framework_1.Property.ShortText({
+                                    displayName: 'Domain',
+                                    required: true,
+                                }),
+                            },
+                        }), blockedDomains: pieces_framework_1.Property.Array({
+                            displayName: 'Blocked Domains',
+                            required: false,
+                            description: 'List of domains to exclude from search (e.g., example.com, docs.example.com/blog). Domains should not include HTTP/HTTPS scheme. Subdomains are automatically included unless more specific subpaths are provided. Overrided by Allowed Domains if both are provided.',
+                            properties: {
+                                domain: pieces_framework_1.Property.ShortText({
+                                    displayName: 'Domain',
+                                    required: true,
+                                }),
+                            },
+                        }) }), userLocationOptions);
+                }
+                if (provider === types_1.AIProviderName.OPENAI) {
+                    options = Object.assign(Object.assign(Object.assign({}, options), { searchContextSize: pieces_framework_1.Property.StaticDropdown({
+                            displayName: 'Search Context Size',
+                            required: false,
+                            defaultValue: 'medium',
+                            options: {
+                                options: [
+                                    { label: 'Low', value: 'low' },
+                                    { label: 'Medium', value: 'medium' },
+                                    { label: 'High', value: 'high' },
+                                ],
+                            },
+                            description: 'High level guidance for the amount of context window space to use for the search.',
+                        }) }), userLocationOptions);
+                }
+                return options;
+            }),
+        }),
+    },
+    run(context) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d;
+            const providerId = context.propsValue.provider;
+            const modelId = context.propsValue.model;
+            const storage = context.store;
+            const webSearchOptions = context.propsValue.webSearchOptions;
+            const model = yield (0, ai_sdk_1.createAIModel)({
+                providerId,
+                modelId,
+                engineToken: context.server.token,
+                apiUrl: context.server.apiUrl,
+                openaiResponsesModel: true,
+            });
+            const conversationKey = context.propsValue.conversationKey
+                ? `ask-ai-conversation:${context.propsValue.conversationKey}`
+                : null;
+            let conversation = null;
+            if (conversationKey) {
+                conversation = (_a = (yield storage.get(conversationKey))) !== null && _a !== void 0 ? _a : [];
+                if (!conversation) {
+                    yield storage.put(conversationKey, { messages: [] });
+                }
+            }
+            const response = yield (0, ai_1.generateText)({
+                model,
+                messages: [
+                    ...(conversation !== null && conversation !== void 0 ? conversation : []),
+                    {
+                        role: 'user',
+                        content: context.propsValue.prompt,
+                    },
+                ],
+                maxOutputTokens: context.propsValue.maxOutputTokens,
+                temperature: ((_b = context.propsValue.creativity) !== null && _b !== void 0 ? _b : 100) / 100,
+                tools: context.propsValue.webSearch
+                    ? createWebSearchTool(providerId, webSearchOptions)
+                    : undefined,
+                stopWhen: (0, ai_1.stepCountIs)((_c = webSearchOptions === null || webSearchOptions === void 0 ? void 0 : webSearchOptions.maxUses) !== null && _c !== void 0 ? _c : 5),
+            });
+            conversation === null || conversation === void 0 ? void 0 : conversation.push({
+                role: 'user',
+                content: context.propsValue.prompt,
+            });
+            conversation === null || conversation === void 0 ? void 0 : conversation.push({
+                role: 'assistant',
+                content: (_d = response.text) !== null && _d !== void 0 ? _d : '',
+            });
+            if (conversationKey) {
+                yield storage.put(conversationKey, conversation);
+            }
+            const includeSources = webSearchOptions.includeSources;
+            if (includeSources) {
+                return { text: response.text, sources: response.sources };
+            }
+            return response.text;
+        });
+    },
+});
+function createWebSearchTool(provider, options = {}) {
+    var _a;
+    const defaultMaxUses = 5;
+    switch (provider) {
+        case types_1.AIProviderName.ANTHROPIC: {
+            const anthropicOptions = options;
+            let allowedDomains;
+            let blockedDomains;
+            if (anthropicOptions.allowedDomains &&
+                anthropicOptions.allowedDomains.length > 0) {
+                allowedDomains = anthropicOptions.allowedDomains.map(({ domain }) => domain);
+            }
+            if (anthropicOptions.blockedDomains &&
+                anthropicOptions.blockedDomains.length > 0 &&
+                (!anthropicOptions.allowedDomains ||
+                    anthropicOptions.allowedDomains.length === 0)) {
+                blockedDomains = anthropicOptions.blockedDomains.map(({ domain }) => domain);
+            }
+            return {
+                web_search: (0, ai_sdk_1.anthropicSearchTool)(Object.assign(Object.assign(Object.assign({ maxUses: (_a = anthropicOptions.maxUses) !== null && _a !== void 0 ? _a : defaultMaxUses }, (0, shared_1.spreadIfDefined)('userLocation', buildUserLocation(anthropicOptions))), (0, shared_1.spreadIfDefined)('allowedDomains', allowedDomains)), (0, shared_1.spreadIfDefined)('blockedDomains', blockedDomains))),
+            };
+        }
+        case types_1.AIProviderName.OPENAI: {
+            const openaiOptions = options;
+            return {
+                web_search_preview: (0, ai_sdk_1.openaiSearchTool)(Object.assign(Object.assign({}, (0, shared_1.spreadIfDefined)('searchContextSize', openaiOptions.searchContextSize)), (0, shared_1.spreadIfDefined)('userLocation', buildUserLocation(openaiOptions)))),
+            };
+        }
+        case types_1.AIProviderName.GOOGLE: {
+            return {
+                google_search: (0, ai_sdk_1.googleSearchTool)({}),
+            };
+        }
+        default:
+            throw new Error(`Provider ${provider} is not supported for web search`);
+    }
+}
+function buildUserLocation(options) {
+    if (!options.userLocationCity &&
+        !options.userLocationRegion &&
+        !options.userLocationCountry &&
+        !options.userLocationTimezone) {
+        return undefined;
+    }
+    return Object.assign(Object.assign(Object.assign(Object.assign({ type: 'approximate' }, (0, shared_1.spreadIfDefined)('city', options.userLocationCity)), (0, shared_1.spreadIfDefined)('region', options.userLocationRegion)), (0, shared_1.spreadIfDefined)('country', options.userLocationCountry)), (0, shared_1.spreadIfDefined)('timezone', options.userLocationTimezone));
+}
+//# sourceMappingURL=ask-ai.js.map

package/src/lib/actions/text/ask-ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ask-ai.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/community/ai/src/lib/actions/text/ask-ai.ts"],"names":[],"mappings":";;;AA+NA,kDAqEC;;AApSD,qEAKwC;AACxC,2BAAsE;AACtE,iDAAuD;AACvD,8CAA6C;AAC7C,8CAAoD;AACpD,gDAA6G;AAEhG,QAAA,KAAK,GAAG,IAAA,+BAAY,EAAC;IAChC,IAAI,EAAE,OAAO;IACb,WAAW,EAAE,QAAQ;IACrB,WAAW,EAAE,EAAE;IACf,KAAK,EAAE;QACL,QAAQ,EAAE,IAAA,eAAO,EAAC,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC,QAAQ;QACjD,KAAK,EAAE,IAAA,eAAO,EAAC,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC,KAAK;QAC3C,MAAM,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACxB,WAAW,EAAE,QAAQ;YACrB,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,eAAe,EAAE,2BAAQ,CAAC,SAAS,CAAC;YAClC,WAAW,EAAE,kBAAkB;YAC/B,QAAQ,EAAE,KAAK;SAChB,CAAC;QACF,UAAU,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC1B,WAAW,EAAE,YAAY;YACzB,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,GAAG;YACjB,WAAW,EACT,8IAA8I;SACjJ,CAAC;QACF,eAAe,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC/B,WAAW,EAAE,YAAY;YACzB,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,IAAI;SACnB,CAAC;QACF,SAAS,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC3B,WAAW,EAAE,YAAY;YACzB,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;YACnB,WAAW,EACT,kFAAkF;SACrF,CAAC;QACF,gBAAgB,EAAE,2BAAQ,CAAC,iBAAiB,CAAC;YAC3C,WAAW,EAAE,oBAAoB;YACjC,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,4BAAS,CAAC,IAAI,EAAE;YACtB,UAAU,EAAE,CAAC,WAAW,EAAE,UAAU,EAAE,OAAO,CAAC;YAC9C,KAAK,EAAE,CAAO,UAAU,EAAE,EAAE;gBAC1B,MAAM,gBAAgB,GAAG,UAAU,CAAC,WAAW,CAAuB,CAAC;gBACvE,MAAM,QAAQ,GAAG,UAAU,CAAC,UAAU,CAAsB,CAAC;gBAE7D,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBACtB,OAAO,EAAE,CAAC;gBACZ,CAAC;gBAED,IAAI,OAAO,GAAqB;oBAC9B,OAAO,EAAE,2BAAQ,CAAC,MAAM,CAAC;wBACvB,WAAW,EAAE,qBAAqB;wBAClC,QAAQ,EAAE,KAAK;wBACf,YAAY,EAAE,CAAC;wBACf,WAAW,EAAE,kDAAkD;qBAChE,CAAC;oBACF,cAAc,EAAE,2BAAQ,CAAC,QAAQ,CAAC;wBAChC,WAAW,EAAE,iBAAiB;wBAC9B,WAAW,EACT,kIAAkI;wBACpI,QAAQ,EAAE,KAAK;wBACf,YAAY,EAAE,KAAK;qBACpB,CAAC;iBACH,CAAC;gBAEF,MAAM,mBAAmB,GAAG;oBAC1B,gBAAgB,EAAE,2BAAQ,CAAC,SAAS,CAAC;wBACnC,WAAW,EAAE,sBAAsB;wBACnC,QAAQ,EAAE,KAAK;wBACf,WAAW,EACT,oEAAoE;qBACvE,CAAC;oBACF,kBAAkB,EAAE,2BAAQ,CAAC,SAAS,CAAC;wBACrC,WAAW,EAAE,wBAAwB;wBACrC,QAAQ,EAAE,KAAK;wBACf,WAAW,EACT,uEAAuE;qBAC1E,CAAC;oBACF,mBAAmB,EAAE,2BAAQ,CAAC,SAAS,CAAC;wBACtC,WAAW,EAAE,yBAAyB;wBACtC,QAAQ,EAAE,KAAK;wBACf,WAAW,EACT,4DAA4D;qBAC/D,CAAC;oBACF,oBAAoB,EAAE,2BAAQ,CAAC,SAAS,CAAC;wBACvC,WAAW,EAAE,0BAA0B;wBACvC,QAAQ,EAAE,KAAK;wBACf,WAAW,EACT,iFAAiF;qBACpF,CAAC;iBACH,CAAC;gBAEF,IAAI,QAAQ,KAAK,sBAAc,CAAC,SAAS,EAAE,CAAC;oBAC1C,OAAO,iDACF,OAAO,KACV,cAAc,EAAE,2BAAQ,CAAC,KAAK,CAAC;4BAC7B,WAAW,EAAE,iBAAiB;4BAC9B,QAAQ,EAAE,KAAK;4BACf,WAAW,EACT,uPAAuP;4BACzP,UAAU,EAAE;gCACV,MAAM,EAAE,2BAAQ,CAAC,SAAS,CAAC;oCACzB,WAAW,EAAE,QAAQ;oCACrB,QAAQ,EAAE,IAAI;iCACf,CAAC;6BACH;yBACF,CAAC,EACF,cAAc,EAAE,2BAAQ,CAAC,KAAK,CAAC;4BAC7B,WAAW,EAAE,iBAAiB;4BAC9B,QAAQ,EAAE,KAAK;4BACf,WAAW,EACT,uQAAuQ;4BACzQ,UAAU,EAAE;gCACV,MAAM,EAAE,2BAAQ,CAAC,SAAS,CAAC;oCACzB,WAAW,EAAE,QAAQ;oCACrB,QAAQ,EAAE,IAAI;iCACf,CAAC;6BACH;yBACF,CAAC,KACC,mBAAmB,CACvB,CAAC;gBACJ,CAAC;gBAED,IAAI,QAAQ,KAAK,sBAAc,CAAC,MAAM,EAAE,CAAC;oBACvC,OAAO,iDACF,OAAO,KACV,iBAAiB,EAAE,2BAAQ,CAAC,cAAc,CAAC;4BACzC,WAAW,EAAE,qBAAqB;4BAClC,QAAQ,EAAE,KAAK;4BACf,YAAY,EAAE,QAAQ;4BACtB,OAAO,EAAE;gCACP,OAAO,EAAE;oCACP,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;oCAC9B,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;oCACpC,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE;iCACjC;6BACF;4BACD,WAAW,EACT,mFAAmF;yBACtF,CAAC,KACC,mBAAmB,CACvB,CAAC;gBACJ,CAAC;gBAED,OAAO,OAAO,CAAC;YACjB,CAAC,CAAA;SACF,CAAC;KACH;IACK,GAAG,CAAC,OAAO;;;YACf,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;YAC/C,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;YACzC,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC;YAC9B,MAAM,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,gBAAoC,CAAC;YAEjF,MAAM,KAAK,GAAG,MAAM,IAAA,sBAAa,EAAC;gBAChC,UAAU;gBACV,OAAO;gBACP,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;gBACjC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM;gBAC7B,oBAAoB,EAAE,IAAI;aAC3B,CAAC,CAAC;YAEH,MAAM,eAAe,GAAG,OAAO,CAAC,UAAU,CAAC,eAAe;gBACxD,CAAC,CAAC,uBAAuB,OAAO,CAAC,UAAU,CAAC,eAAe,EAAE;gBAC7D,CAAC,CAAC,IAAI,CAAC;YAET,IAAI,YAAY,GAAG,IAAI,CAAC;YACxB,IAAI,eAAe,EAAE,CAAC;gBACpB,YAAY,GAAG,MAAA,CAAC,MAAM,OAAO,CAAC,GAAG,CAAiB,eAAe,CAAC,CAAC,mCAAI,EAAE,CAAC;gBAC1E,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,MAAM,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAA,iBAAY,EAAC;gBAClC,KAAK;gBACL,QAAQ,EAAE;oBACR,GAAG,CAAC,YAAY,aAAZ,YAAY,cAAZ,YAAY,GAAI,EAAE,CAAC;oBACvB;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,MAAM;qBACnC;iBACF;gBACD,eAAe,EAAE,OAAO,CAAC,UAAU,CAAC,eAAe;gBACnD,WAAW,EAAE,CAAC,MAAA,OAAO,CAAC,UAAU,CAAC,UAAU,mCAAI,GAAG,CAAC,GAAG,GAAG;gBACzD,KAAK,EAAE,OAAO,CAAC,UAAU,CAAC,SAAS;oBACjC,CAAC,CAAC,mBAAmB,CAAC,UAAU,EAAE,gBAAgB,CAAC;oBACnD,CAAC,CAAC,SAAS;gBACb,QAAQ,EAAE,IAAA,gBAAW,EAAC,MAAA,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAE,OAAO,mCAAI,CAAC,CAAC;aACtD,CAAC,CAAC;YAEH,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,IAAI,CAAC;gBACjB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,MAAM;aACnC,CAAC,CAAC;YAEH,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,IAAI,CAAC;gBACjB,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,MAAA,QAAQ,CAAC,IAAI,mCAAI,EAAE;aAC7B,CAAC,CAAC;YAEH,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;YACnD,CAAC;YAED,MAAM,cAAc,GAAG,gBAAgB,CAAC,cAAc,CAAC;YACvD,IAAI,cAAc,EAAE,CAAC;gBACnB,OAAO,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC5D,CAAC;YACD,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;KAAA;CACF,CAAC,CAAC;AAEH,SAAgB,mBAAmB,CACjC,QAAgB,EAChB,UAA4B,EAAE;;IAE9B,MAAM,cAAc,GAAG,CAAC,CAAC;IAEzB,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,sBAAc,CAAC,SAAS,CAAC,CAAC,CAAC;YAC9B,MAAM,gBAAgB,GAAG,OAAoC,CAAC;YAE9D,IAAI,cAAoC,CAAC;YACzC,IAAI,cAAoC,CAAC;YAEzC,IACE,gBAAgB,CAAC,cAAc;gBAC/B,gBAAgB,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAC1C,CAAC;gBACD,cAAc,GAAG,gBAAgB,CAAC,cAAc,CAAC,GAAG,CAClD,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,MAAM,CACvB,CAAC;YACJ,CAAC;YAED,IACE,gBAAgB,CAAC,cAAc;gBAC/B,gBAAgB,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC;gBAC1C,CAAC,CAAC,gBAAgB,CAAC,cAAc;oBAC/B,gBAAgB,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC,CAAC,EAC/C,CAAC;gBACD,cAAc,GAAG,gBAAgB,CAAC,cAAc,CAAC,GAAG,CAClD,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,MAAM,CACvB,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,UAAU,EAAE,IAAA,4BAAmB,8CAC7B,OAAO,EAAE,MAAA,gBAAgB,CAAC,OAAO,mCAAI,cAAc,IAChD,IAAA,wBAAe,EAChB,cAAc,EACd,iBAAiB,CAAC,gBAAgB,CAAC,CACpC,GACE,IAAA,wBAAe,EAAC,gBAAgB,EAAE,cAAc,CAAC,GACjD,IAAA,wBAAe,EAAC,gBAAgB,EAAE,cAAc,CAAC,EACpD;aACI,CAAC;QACX,CAAC;QAED,KAAK,sBAAc,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3B,MAAM,aAAa,GAAG,OAAiC,CAAC;YAExD,OAAO;gBACL,kBAAkB,EAAE,IAAA,yBAAgB,kCAC/B,IAAA,wBAAe,EAChB,mBAAmB,EACnB,aAAa,CAAC,iBAAiB,CAChC,GACE,IAAA,wBAAe,EAAC,cAAc,EAAE,iBAAiB,CAAC,aAAa,CAAC,CAAC,EACpE;aACI,CAAC;QACX,CAAC;QAED,KAAK,sBAAc,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3B,OAAO;gBACL,aAAa,EAAE,IAAA,yBAAgB,EAAC,EAAE,CAAC;aAC7B,CAAC;QACX,CAAC;QAED;YACE,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,kCAAkC,CAAC,CAAC;IAC5E,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CACxB,OAA4B;IAE5B,IACE,CAAC,OAAO,CAAC,gBAAgB;QACzB,CAAC,OAAO,CAAC,kBAAkB;QAC3B,CAAC,OAAO,CAAC,mBAAmB;QAC5B,CAAC,OAAO,CAAC,oBAAoB,EAC7B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,iEACE,IAAI,EAAE,aAAsB,IACzB,IAAA,wBAAe,EAAC,MAAM,EAAE,OAAO,CAAC,gBAAgB,CAAC,GACjD,IAAA,wBAAe,EAAC,QAAQ,EAAE,OAAO,CAAC,kBAAkB,CAAC,GACrD,IAAA,wBAAe,EAAC,SAAS,EAAE,OAAO,CAAC,mBAAmB,CAAC,GACvD,IAAA,wBAAe,EAAC,UAAU,EAAE,OAAO,CAAC,oBAAoB,CAAC,EAC5D;AACJ,CAAC"}

package/src/lib/actions/text/summarize-text.d.ts

@@ -0,0 +1,7 @@
+export declare const summarizeText: import("@activepieces/pieces-framework").IAction<import("@activepieces/pieces-framework").PieceAuthProperty, {
+    provider: import("@activepieces/pieces-framework").DropdownProperty<string, true, undefined>;
+    model: import("@activepieces/pieces-framework").DropdownProperty<string, true, undefined>;
+    text: import("@activepieces/pieces-framework").LongTextProperty<true>;
+    prompt: import("@activepieces/pieces-framework").ShortTextProperty<true>;
+    maxOutputTokens: import("@activepieces/pieces-framework").NumberProperty<false>;
+}>;

package/src/lib/actions/text/summarize-text.js

@@ -0,0 +1,61 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.summarizeText = void 0;
+const tslib_1 = require("tslib");
+const types_1 = require("../../common/types");
+const ai_sdk_1 = require("../../common/ai-sdk");
+const pieces_framework_1 = require("@activepieces/pieces-framework");
+const ai_1 = require("ai");
+const props_1 = require("../../common/props");
+exports.summarizeText = (0, pieces_framework_1.createAction)({
+    name: 'summarizeText',
+    displayName: 'Summarize Text',
+    description: '',
+    props: {
+        provider: (0, props_1.aiProps)({ modelType: 'text' }).provider,
+        model: (0, props_1.aiProps)({ modelType: 'text' }).model,
+        text: pieces_framework_1.Property.LongText({
+            displayName: 'Text',
+            required: true,
+        }),
+        prompt: pieces_framework_1.Property.ShortText({
+            displayName: 'Prompt',
+            defaultValue: 'Summarize the following text in a clear and concise manner, capturing the key points and main ideas while keeping the summary brief and informative.',
+            required: true,
+        }),
+        maxOutputTokens: pieces_framework_1.Property.Number({
+            displayName: 'Max Tokens',
+            required: false,
+            defaultValue: 2000,
+        }),
+    },
+    run(context) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const providerId = context.propsValue.provider;
+            const modelId = context.propsValue.model;
+            const model = yield (0, ai_sdk_1.createAIModel)({
+                providerId,
+                modelId,
+                engineToken: context.server.token,
+                apiUrl: context.server.apiUrl,
+            });
+            const response = yield (0, ai_1.generateText)({
+                model,
+                messages: [
+                    {
+                        role: 'user',
+                        content: `${context.propsValue.prompt} Summarize the following text : ${context.propsValue.text}`
+                    },
+                ],
+                maxOutputTokens: context.propsValue.maxOutputTokens,
+                temperature: 1,
+                providerOptions: {
+                    [providerId]: Object.assign({}, (providerId === types_1.AIProviderName.OPENAI ? { reasoning_effort: 'minimal' } : {}))
+                }
+            });
+            return (_a = response.text) !== null && _a !== void 0 ? _a : '';
+        });
+    },
+});
+//# sourceMappingURL=summarize-text.js.map

package/src/lib/actions/text/summarize-text.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"summarize-text.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/community/ai/src/lib/actions/text/summarize-text.ts"],"names":[],"mappings":";;;;AAAA,8CAAoD;AACpD,gDAAoD;AACpD,qEAAwE;AACxE,2BAAkC;AAClC,8CAA6C;AAEhC,QAAA,aAAa,GAAG,IAAA,+BAAY,EAAC;IACxC,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,gBAAgB;IAC7B,WAAW,EAAE,EAAE;IACf,KAAK,EAAE;QACL,QAAQ,EAAE,IAAA,eAAO,EAAC,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC,QAAQ;QACjD,KAAK,EAAE,IAAA,eAAO,EAAC,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC,KAAK;QAC3C,IAAI,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACtB,WAAW,EAAE,MAAM;YACnB,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,MAAM,EAAE,2BAAQ,CAAC,SAAS,CAAC;YACzB,WAAW,EAAE,QAAQ;YACrB,YAAY,EACV,sJAAsJ;YACxJ,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,eAAe,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC/B,WAAW,EAAE,YAAY;YACzB,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,IAAI;SACnB,CAAC;KACH;IACK,GAAG,CAAC,OAAO;;;YACf,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;YAC/C,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;YAEzC,MAAM,KAAK,GAAG,MAAM,IAAA,sBAAa,EAAC;gBAChC,UAAU;gBACV,OAAO;gBACP,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;gBACjC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM;aAC9B,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,IAAA,iBAAY,EAAC;gBAClC,KAAK;gBACL,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,mCAAmC,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE;qBAClG;iBACF;gBACD,eAAe,EAAE,OAAO,CAAC,UAAU,CAAC,eAAe;gBACnD,WAAW,EAAE,CAAC;gBACd,eAAe,EAAE;oBACf,CAAC,UAAU,CAAC,oBACP,CAAC,UAAU,KAAK,sBAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CACjF;iBACF;aACF,CAAC,CAAC;YAEH,OAAO,MAAA,QAAQ,CAAC,IAAI,mCAAI,EAAE,CAAC;QAC7B,CAAC;KAAA;CACF,CAAC,CAAC"}

package/src/lib/actions/utility/classify-text.d.ts

@@ -0,0 +1,6 @@
+export declare const classifyText: import("@activepieces/pieces-framework").IAction<import("@activepieces/pieces-framework").PieceAuthProperty, {
+    provider: import("@activepieces/pieces-framework").DropdownProperty<string, true, undefined>;
+    model: import("@activepieces/pieces-framework").DropdownProperty<string, true, undefined>;
+    text: import("@activepieces/pieces-framework").LongTextProperty<true>;
+    categories: import("@activepieces/pieces-framework").ArrayProperty<true>;
+}>;

package/src/lib/actions/utility/classify-text.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.classifyText = void 0;
+const tslib_1 = require("tslib");
+const pieces_framework_1 = require("@activepieces/pieces-framework");
+const ai_1 = require("ai");
+const ai_sdk_1 = require("../../common/ai-sdk");
+const props_1 = require("../../common/props");
+exports.classifyText = (0, pieces_framework_1.createAction)({
+    name: 'classifyText',
+    displayName: 'Classify Text',
+    description: 'Classify your text into one of your provided categories.',
+    props: {
+        provider: (0, props_1.aiProps)({ modelType: 'text' }).provider,
+        model: (0, props_1.aiProps)({ modelType: 'text' }).model,
+        text: pieces_framework_1.Property.LongText({
+            displayName: 'Text to Classify',
+            required: true,
+        }),
+        categories: pieces_framework_1.Property.Array({
+            displayName: 'Categories',
+            description: 'Categories to classify text into.',
+            required: true,
+        }),
+    },
+    run(context) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const categories = (_a = context.propsValue.categories) !== null && _a !== void 0 ? _a : [];
+            const providerId = context.propsValue.provider;
+            const modelId = context.propsValue.model;
+            const model = yield (0, ai_sdk_1.createAIModel)({
+                providerId,
+                modelId,
+                engineToken: context.server.token,
+                apiUrl: context.server.apiUrl,
+            });
+            const response = yield (0, ai_1.generateText)({
+                model,
+                prompt: `As a text classifier, your task is to assign one of the following categories to the provided text: ${categories.join(', ')}. Please respond with only the selected category as a single word, and nothing else.
+      Text to classify: "${context.propsValue.text}"`,
+            });
+            const result = response.text.trim();
+            if (!categories.includes(result)) {
+                throw new Error('Unable to classify the text into the provided categories.');
+            }
+            return result;
+        });
+    },
+});
+//# sourceMappingURL=classify-text.js.map