@activeboxes/pieces-common 0.4.0

package/src/lib/ai/providers/openai/index.js

@@ -0,0 +1,224 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.openaiModels = exports.openai = void 0;
+const tslib_1 = require("tslib");
+const __1 = require("../..");
+const shared_1 = require("@activeboxes/shared");
+const openai_1 = tslib_1.__importDefault(require("openai"));
+const utils_1 = require("../utils");
+const pieces_framework_1 = require("@activeboxes/pieces-framework");
+const mime_types_1 = tslib_1.__importDefault(require("mime-types"));
+const openai = ({ proxyUrl, engineToken }) => {
+    const openaiApiVersion = 'v1';
+    const sdk = new openai_1.default({
+        apiKey: engineToken,
+        baseURL: `${proxyUrl}/${openaiApiVersion}`,
+    });
+    return {
+        provider: 'OPENAI',
+        function: {
+            call: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+                var _a;
+                const messages = params.messages.map((message) => ({
+                    role: __1.AIChatRole.USER,
+                    content: [{ type: 'text', text: message.content }],
+                }));
+                if (params.files.length) {
+                    const contents = [];
+                    for (const file of params.files) {
+                        const fileType = file.extension ? mime_types_1.default.lookup(file.extension) : 'image/jpeg';
+                        if (fileType && fileType.startsWith('image')) {
+                            contents.push({
+                                type: 'image_url',
+                                image_url: {
+                                    url: `data:image/${file.extension};base64,${file.base64}`,
+                                },
+                            });
+                        }
+                    }
+                    if (contents.length) {
+                        const lastMessage = messages[messages.length - 1];
+                        if (lastMessage && lastMessage.role === __1.AIChatRole.USER) {
+                            const exitingContent = lastMessage.content;
+                            lastMessage.content = [...exitingContent, ...contents];
+                        }
+                        else {
+                            messages.push({
+                                role: __1.AIChatRole.USER,
+                                content: contents,
+                            });
+                        }
+                    }
+                }
+                const completion = yield sdk.chat.completions.create({
+                    model: params.model,
+                    messages: messages,
+                    max_tokens: params.maxTokens,
+                    tools: params.functions.map((functionDefinition) => ({
+                        type: 'function',
+                        function: {
+                            name: functionDefinition.name,
+                            description: functionDefinition.description,
+                            parameters: functionDefinition.arguments,
+                        },
+                    })),
+                });
+                const toolCall = (_a = completion.choices[0].message.tool_calls) === null || _a === void 0 ? void 0 : _a[0];
+                return {
+                    choices: completion.choices.map((choice) => {
+                        var _a;
+                        return ({
+                            role: __1.AIChatRole.ASSISTANT,
+                            content: (_a = choice.message.content) !== null && _a !== void 0 ? _a : '',
+                        });
+                    }),
+                    call: toolCall
+                        ? {
+                            id: toolCall.id,
+                            function: {
+                                name: toolCall.function.name,
+                                arguments: JSON.parse(toolCall.function.arguments),
+                            },
+                        }
+                        : null,
+                    created: completion.created,
+                    model: completion.model,
+                    usage: completion.usage && {
+                        completionTokens: completion.usage.completion_tokens,
+                        promptTokens: completion.usage.prompt_tokens,
+                        totalTokens: completion.usage.total_tokens,
+                    },
+                };
+            }),
+        },
+        image: {
+            generate: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+                const mapper = findImageMapper(params.model);
+                const input = yield mapper.encodeInput(params);
+                const response = yield sdk.images.generate(input);
+                return mapper.decodeOutput(response);
+            }),
+        },
+        chat: {
+            text: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+                var _a;
+                const completion = yield sdk.chat.completions.create({
+                    model: params.model,
+                    messages: params.messages.map((message) => ({
+                        role: message.role === 'user' ? 'user' : 'assistant',
+                        content: message.content,
+                    })),
+                    temperature: Math.tanh((_a = params.creativity) !== null && _a !== void 0 ? _a : 100),
+                    max_tokens: params.maxTokens,
+                    stop: params.stop,
+                });
+                return {
+                    choices: completion.choices.map((choice) => {
+                        var _a;
+                        return ({
+                            role: __1.AIChatRole.ASSISTANT,
+                            content: (_a = choice.message.content) !== null && _a !== void 0 ? _a : '',
+                        });
+                    }),
+                    created: completion.created,
+                    model: completion.model,
+                    usage: completion.usage && {
+                        completionTokens: completion.usage.completion_tokens,
+                        promptTokens: completion.usage.prompt_tokens,
+                        totalTokens: completion.usage.total_tokens,
+                    },
+                };
+            }),
+        },
+        moderation: {
+            create: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+                var _a;
+                const inputs = [];
+                if (params.text) {
+                    inputs.push({ type: 'text', text: params.text });
+                }
+                for (const image of (_a = params.images) !== null && _a !== void 0 ? _a : []) {
+                    inputs.push({
+                        type: 'image_url',
+                        image_url: {
+                            url: `data:image/${image.extension};base64,${image.base64}`,
+                        },
+                    });
+                }
+                const response = yield sdk.moderations.create({
+                    input: inputs,
+                    model: params.model,
+                });
+                return response.results[0];
+            }),
+        },
+    };
+};
+exports.openai = openai;
+const findImageMapper = (model) => {
+    var _a;
+    const mapper = (_a = exports.openaiModels.find((m) => m.value === model)) === null || _a === void 0 ? void 0 : _a.mapper;
+    if ((0, shared_1.isNil)(mapper) || !('__tag' in mapper) || mapper.__tag !== utils_1.ModelType.IMAGE) {
+        throw new Error(`OpenAI image model ${model} not found`);
+    }
+    return mapper;
+};
+const openaiImageMapper = (0, utils_1.imageMapper)({
+    encodeInput: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+        var _a;
+        return {
+            model: params.model,
+            prompt: params.prompt,
+            quality: (_a = params.advancedOptions) === null || _a === void 0 ? void 0 : _a['quality'],
+            size: params.size,
+            response_format: 'b64_json',
+        };
+    }),
+    decodeOutput: (result) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+        const response = result;
+        const imageBase64 = response.data[0].b64_json;
+        return imageBase64 ? { image: imageBase64 } : null;
+    }),
+    advancedOptions: {
+        quality: pieces_framework_1.Property.StaticDropdown({
+            options: {
+                options: [
+                    { label: 'Standard', value: 'standard' },
+                    { label: 'HD', value: 'hd' },
+                ],
+                disabled: false,
+                placeholder: 'Select Image Quality',
+            },
+            defaultValue: 'standard',
+            description: 'Standard images are less detailed and faster to generate, while HD images are more detailed but slower to generate.',
+            displayName: 'Image Quality',
+            required: true,
+        }),
+    },
+});
+exports.openaiModels = [
+    (0, utils_1.model)({ label: 'gpt-4o', value: 'gpt-4o', supported: ['text', 'function'] }),
+    (0, utils_1.model)({
+        label: 'gpt-4o-mini',
+        value: 'gpt-4o-mini',
+        supported: ['text', 'function'],
+    }),
+    (0, utils_1.model)({
+        label: 'gpt-4-turbo',
+        value: 'gpt-4-turbo',
+        supported: ['text', 'function'],
+    }),
+    (0, utils_1.model)({
+        label: 'gpt-3.5-turbo',
+        value: 'gpt-3.5-turbo',
+        supported: ['text'],
+    }),
+    (0, utils_1.model)({ label: 'dall-e-3', value: 'dall-e-3', supported: ['image'] }).mapper(openaiImageMapper),
+    (0, utils_1.model)({ label: 'dall-e-2', value: 'dall-e-2', supported: ['image'] }).mapper(openaiImageMapper),
+    (0, utils_1.model)({
+        label: 'omni-moderation-latest',
+        value: 'omni-moderation-latest',
+        supported: ['moderation'],
+    }),
+];
+//# sourceMappingURL=index.js.map

package/src/lib/ai/providers/openai/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../../../../../packages/pieces/community/common/src/lib/ai/providers/openai/index.ts"],"names":[],"mappings":";;;;AAAA,6BAAkD;AAClD,gDAA4C;AAC5C,4DAA4B;AAC5B,oCAAyD;AACzD,oEAAyD;AAQzD,oEAA8B;AAEvB,MAAM,MAAM,GAAc,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,EAAM,EAAE;IAClE,MAAM,gBAAgB,GAAG,IAAI,CAAC;IAC9B,MAAM,GAAG,GAAG,IAAI,gBAAM,CAAC;QACtB,MAAM,EAAE,WAAW;QACnB,OAAO,EAAE,GAAG,QAAQ,IAAI,gBAAgB,EAAE;KAC1C,CAAC,CAAC;IACH,OAAO;QACN,QAAQ,EAAE,QAAQ;QAClB,QAAQ,EAAE;YACT,IAAI,EAAE,CAAO,MAAM,EAAE,EAAE;;gBACtB,MAAM,QAAQ,GAAiC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBAChF,IAAI,EAAE,cAAU,CAAC,IAAI;oBACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;iBAClD,CAAC,CAAC,CAAC;gBAEJ,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;oBACzB,MAAM,QAAQ,GAA0C,EAAE,CAAC;oBAC3D,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;wBACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,oBAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;wBAE7E,IAAI,QAAQ,IAAI,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;4BAC9C,QAAQ,CAAC,IAAI,CAAC;gCACb,IAAI,EAAE,WAAW;gCACjB,SAAS,EAAE;oCACV,GAAG,EAAE,cAAc,IAAI,CAAC,SAAS,WAAW,IAAI,CAAC,MAAM,EAAE;iCACzD;6BACD,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;oBACD,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;wBACrB,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;wBAElD,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,cAAU,CAAC,IAAI,EAAE,CAAC;4BACzD,MAAM,cAAc,GAAG,WAAW,CAAC,OAA2C,CAAC;4BAC/E,WAAW,CAAC,OAAO,GAAG,CAAC,GAAG,cAAc,EAAE,GAAG,QAAQ,CAAC,CAAC;wBACxD,CAAC;6BAAM,CAAC;4BACP,QAAQ,CAAC,IAAI,CAAC;gCACb,IAAI,EAAE,cAAU,CAAC,IAAI;gCACrB,OAAO,EAAE,QAAQ;6BACjB,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;gBACF,CAAC;gBAED,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;oBACpD,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,QAAQ,EAAE,QAAQ;oBAClB,UAAU,EAAE,MAAM,CAAC,SAAS;oBAC5B,KAAK,EAAE,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;wBACpD,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE;4BACT,IAAI,EAAE,kBAAkB,CAAC,IAAI;4BAC7B,WAAW,EAAE,kBAAkB,CAAC,WAAW;4BAC3C,UAAU,EAAE,kBAAkB,CAAC,SAA0C;yBACzE;qBACD,CAAC,CAAC;iBACH,CAAC,CAAC;gBAEH,MAAM,QAAQ,GAAG,MAAA,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,0CAAG,CAAC,CAAC,CAAC;gBAE/D,OAAO;oBACN,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;;wBAAC,OAAA,CAAC;4BAC5C,IAAI,EAAE,cAAU,CAAC,SAAS;4BAC1B,OAAO,EAAE,MAAA,MAAM,CAAC,OAAO,CAAC,OAAO,mCAAI,EAAE;yBACrC,CAAC,CAAA;qBAAA,CAAC;oBACH,IAAI,EAAE,QAAQ;wBACb,CAAC,CAAC;4BACA,EAAE,EAAE,QAAQ,CAAC,EAAE;4BACf,QAAQ,EAAE;gCACT,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;gCAC5B,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAmB,CAAC;6BAC5D;yBACA;wBACH,CAAC,CAAC,IAAI;oBACP,OAAO,EAAE,UAAU,CAAC,OAAO;oBAC3B,KAAK,EAAE,UAAU,CAAC,KAAK;oBACvB,KAAK,EAAE,UAAU,CAAC,KAAK,IAAI;wBAC1B,gBAAgB,EAAE,UAAU,CAAC,KAAK,CAAC,iBAAiB;wBACpD,YAAY,EAAE,UAAU,CAAC,KAAK,CAAC,aAAa;wBAC5C,WAAW,EAAE,UAAU,CAAC,KAAK,CAAC,YAAY;qBAC1C;iBACD,CAAC;YACH,CAAC,CAAA;SACD;QACD,KAAK,EAAE;YACN,QAAQ,EAAE,CAAO,MAAM,EAAE,EAAE;gBAC1B,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC7C,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAY,CAAC,CAAC;gBACzD,OAAO,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YACtC,CAAC,CAAA;SACD;QACD,IAAI,EAAE;YACL,IAAI,EAAE,CAAO,MAAM,EAAE,EAAE;;gBACtB,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;oBACpD,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;wBAC3C,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW;wBACpD,OAAO,EAAE,OAAO,CAAC,OAAO;qBACxB,CAAC,CAAC;oBACH,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,MAAA,MAAM,CAAC,UAAU,mCAAI,GAAG,CAAC;oBAChD,UAAU,EAAE,MAAM,CAAC,SAAS;oBAC5B,IAAI,EAAE,MAAM,CAAC,IAAI;iBACjB,CAAC,CAAC;gBAEH,OAAO;oBACN,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;;wBAAC,OAAA,CAAC;4BAC5C,IAAI,EAAE,cAAU,CAAC,SAAS;4BAC1B,OAAO,EAAE,MAAA,MAAM,CAAC,OAAO,CAAC,OAAO,mCAAI,EAAE;yBACrC,CAAC,CAAA;qBAAA,CAAC;oBACH,OAAO,EAAE,UAAU,CAAC,OAAO;oBAC3B,KAAK,EAAE,UAAU,CAAC,KAAK;oBACvB,KAAK,EAAE,UAAU,CAAC,KAAK,IAAI;wBAC1B,gBAAgB,EAAE,UAAU,CAAC,KAAK,CAAC,iBAAiB;wBACpD,YAAY,EAAE,UAAU,CAAC,KAAK,CAAC,aAAa;wBAC5C,WAAW,EAAE,UAAU,CAAC,KAAK,CAAC,YAAY;qBAC1C;iBACD,CAAC;YACH,CAAC,CAAA;SACD;QACD,UAAU,EAAE;YACX,MAAM,EAAE,CAAO,MAAM,EAAE,EAAE;;gBACxB,MAAM,MAAM,GAAgC,EAAE,CAAC;gBAE/C,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;oBACjB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;gBAClD,CAAC;gBACD,KAAK,MAAM,KAAK,IAAI,MAAA,MAAM,CAAC,MAAM,mCAAI,EAAE,EAAE,CAAC;oBACzC,MAAM,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,WAAW;wBACjB,SAAS,EAAE;4BACV,GAAG,EAAE,cAAc,KAAK,CAAC,SAAS,WAAW,KAAK,CAAC,MAAM,EAAE;yBAC3D;qBACD,CAAC,CAAC;gBACJ,CAAC;gBAED,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC;oBAC7C,KAAK,EAAE,MAAM;oBACb,KAAK,EAAE,MAAM,CAAC,KAAK;iBACnB,CAAC,CAAC;gBAEH,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC5B,CAAC,CAAA;SACD;KACD,CAAC;AACH,CAAC,CAAC;AAjJW,QAAA,MAAM,UAiJjB;AAEF,MAAM,eAAe,GAAG,CAAC,KAAa,EAAE,EAAE;;IACzC,MAAM,MAAM,GAAG,MAAA,oBAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,0CAAE,MAAM,CAAC;IACnE,IAAI,IAAA,cAAK,EAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,KAAK,iBAAS,CAAC,KAAK,EAAE,CAAC;QAC/E,MAAM,IAAI,KAAK,CAAC,sBAAsB,KAAK,YAAY,CAAC,CAAC;IAC1D,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,iBAAiB,GAAG,IAAA,mBAAW,EAAC;IACrC,WAAW,EAAE,CAAO,MAAM,EAAE,EAAE;;QAC7B,OAAO;YACN,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,OAAO,EAAE,MAAA,MAAM,CAAC,eAAe,0CAAG,SAAS,CAAQ;YACnD,IAAI,EAAE,MAAM,CAAC,IAAW;YACxB,eAAe,EAAE,UAAU;SAC3B,CAAC;IACH,CAAC,CAAA;IACD,YAAY,EAAE,CAAO,MAAM,EAAE,EAAE;QAC9B,MAAM,QAAQ,GAAG,MAAsC,CAAC;QACxD,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC9C,OAAO,WAAW,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IACpD,CAAC,CAAA;IACD,eAAe,EAAE;QAChB,OAAO,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAChC,OAAO,EAAE;gBACR,OAAO,EAAE;oBACR,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;oBACxC,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE;iBAC5B;gBACD,QAAQ,EAAE,KAAK;gBACf,WAAW,EAAE,sBAAsB;aACnC;YACD,YAAY,EAAE,UAAU;YACxB,WAAW,EACV,qHAAqH;YACtH,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,IAAI;SACd,CAAC;KACF;CACD,CAAC,CAAC;AAEU,QAAA,YAAY,GAAG;IAC3B,IAAA,aAAK,EAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,MAAM,EAAE,UAAU,CAAC,EAAE,CAAC;IAC5E,IAAA,aAAK,EAAC;QACL,KAAK,EAAE,aAAa;QACpB,KAAK,EAAE,aAAa;QACpB,SAAS,EAAE,CAAC,MAAM,EAAE,UAAU,CAAC;KAC/B,CAAC;IACF,IAAA,aAAK,EAAC;QACL,KAAK,EAAE,aAAa;QACpB,KAAK,EAAE,aAAa;QACpB,SAAS,EAAE,CAAC,MAAM,EAAE,UAAU,CAAC;KAC/B,CAAC;IACF,IAAA,aAAK,EAAC;QACL,KAAK,EAAE,eAAe;QACtB,KAAK,EAAE,eAAe;QACtB,SAAS,EAAE,CAAC,MAAM,CAAC;KACnB,CAAC;IACF,IAAA,aAAK,EAAC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC;IAC/F,IAAA,aAAK,EAAC,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC;IAC/F,IAAA,aAAK,EAAC;QACL,KAAK,EAAE,wBAAwB;QAC/B,KAAK,EAAE,wBAAwB;QAC/B,SAAS,EAAE,CAAC,YAAY,CAAC;KACzB,CAAC;CACF,CAAC"}

package/src/lib/ai/providers/replicate/index.d.ts

@@ -0,0 +1,10 @@
+import { AIFactory } from '../..';
+import { ChatModelMapper, ImageModelMapper, ModelType } from '../utils';
+export declare const replicate: AIFactory;
+export declare const replicateModels: {
+    mapper: ImageModelMapper | ChatModelMapper;
+    __tag: ModelType.IMAGE | ModelType.CHAT;
+    label: string;
+    value: string;
+    supported: ("function" | "text" | "image" | "moderation")[];
+}[];

package/src/lib/ai/providers/replicate/index.js

@@ -0,0 +1,185 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.replicateModels = exports.replicate = void 0;
+const tslib_1 = require("tslib");
+const pieces_framework_1 = require("@activeboxes/pieces-framework");
+const shared_1 = require("@activeboxes/shared");
+const replicate_1 = tslib_1.__importDefault(require("replicate"));
+const __1 = require("../..");
+const utils_1 = require("../utils");
+const replicate = ({ proxyUrl, engineToken }) => {
+    const sdk = new replicate_1.default({
+        auth: engineToken,
+        baseUrl: `${proxyUrl}/v1`,
+    });
+    return {
+        provider: 'replicate',
+        image: {
+            generate: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+                const mapper = findMapper(params.model, utils_1.ModelType.IMAGE);
+                const modelOwner = params.model.split('/')[0];
+                const modelName = params.model.split('/')[1];
+                const versionId = params.model.split(':')[1];
+                const prediction = yield sdk.predictions.create({
+                    model: `${modelOwner}/${modelName}`,
+                    version: versionId,
+                    input: yield mapper.encodeInput(params),
+                });
+                const result = yield sdk.wait(prediction, { interval: 500, mode: "poll" });
+                return mapper.decodeOutput(result.output);
+            }),
+        },
+        chat: {
+            text: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+                const mapper = findMapper(params.model, utils_1.ModelType.CHAT);
+                const modelOwner = params.model.split('/')[0];
+                const modelName = params.model.split('/')[1];
+                const versionId = params.model.split(':')[1];
+                const prediction = yield sdk.predictions.create({
+                    model: `${modelOwner}/${modelName}`,
+                    version: versionId,
+                    input: yield mapper.encodeInput(params),
+                });
+                const result = yield sdk.wait(prediction, { interval: 500, mode: "poll" });
+                return mapper.decodeOutput(result.output);
+            }),
+        },
+    };
+};
+exports.replicate = replicate;
+const commonImageMapper = (0, utils_1.imageMapper)({
+    encodeInput: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+        var _a, _b, _c, _d;
+        const [width, height] = (_b = (_a = params.size) === null || _a === void 0 ? void 0 : _a.split('x').map(Number)) !== null && _b !== void 0 ? _b : [512, 512];
+        const negativePrompt = (_d = (_c = params.advancedOptions) === null || _c === void 0 ? void 0 : _c['negativePrompt']) !== null && _d !== void 0 ? _d : null;
+        return {
+            prompt: params.prompt,
+            negative_prompt: negativePrompt,
+            width,
+            height,
+            num_outputs: 1,
+        };
+    }),
+    decodeOutput: (output) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+        const imageUrls = output;
+        const imageUrl = imageUrls[0];
+        const image = yield fetch(imageUrl);
+        const imageBuffer = yield image.arrayBuffer();
+        const imageBase64 = Buffer.from(imageBuffer).toString('base64');
+        return {
+            image: imageBase64,
+        };
+    }),
+    advancedOptions: {
+        negativePrompt: pieces_framework_1.Property.ShortText({
+            displayName: 'Negative Prompt',
+            required: true,
+            description: 'A prompt to avoid in the generated image.',
+        }),
+    }
+});
+const llamaMapper = (0, utils_1.chatMapper)({
+    encodeInput: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+        var _a;
+        const concatenatedSystemMessage = params.messages
+            .filter((message) => message.role === __1.AIChatRole.SYSTEM)
+            .map((message) => message.content)
+            .join('\n');
+        return {
+            system_prompt: concatenatedSystemMessage,
+            temperature: Math.tanh((_a = params.creativity) !== null && _a !== void 0 ? _a : 100),
+            prompt: params.messages
+                .filter((message) => message.role !== __1.AIChatRole.SYSTEM)
+                .map((message) => `${message.role}: ${message.content}`)
+                .join('\n'),
+            stop_sequences: params.stop,
+            max_tokens: params.maxTokens,
+        };
+    }),
+    decodeOutput: (output) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+        const choices = output;
+        return {
+            choices: [{
+                    content: choices.join(""),
+                    role: __1.AIChatRole.ASSISTANT,
+                }],
+        };
+    }),
+});
+const mistralMapper = (0, utils_1.chatMapper)({
+    encodeInput: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+        var _a;
+        const concatenatedSystemMessage = params.messages
+            .filter((message) => message.role === __1.AIChatRole.SYSTEM)
+            .map((message) => message.content)
+            .join('\n');
+        return {
+            system_prompt: concatenatedSystemMessage,
+            temperature: Math.tanh((_a = params.creativity) !== null && _a !== void 0 ? _a : 100),
+            prompt: params.messages
+                .filter((message) => message.role !== __1.AIChatRole.SYSTEM)
+                .map((message) => `${message.role}: ${message.content}`)
+                .join('\n'),
+            stop_sequences: params.stop,
+            max_new_tokens: params.maxTokens,
+        };
+    }),
+    decodeOutput: (output) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+        const choices = output;
+        return {
+            choices: [{
+                    content: choices.join(""),
+                    role: __1.AIChatRole.ASSISTANT,
+                }],
+        };
+    }),
+});
+exports.replicateModels = [
+    (0, utils_1.model)({
+        label: "meta/meta-llama-3-70b-instruct",
+        value: "meta/meta-llama-3-70b-instruct",
+        supported: ['text']
+    }).mapper(llamaMapper),
+    (0, utils_1.model)({
+        label: "meta/meta-llama-3-8b-instruct",
+        value: "meta/meta-llama-3-8b-instruct",
+        supported: ['text']
+    }).mapper(llamaMapper),
+    (0, utils_1.model)({
+        label: "mistralai/mixtral-8x7b-instruct-v0.1",
+        value: "mistralai/mixtral-8x7b-instruct-v0.1",
+        supported: ['text']
+    }).mapper(mistralMapper),
+    (0, utils_1.model)({
+        label: "mistralai/mistral-7b-instruct-v0.2",
+        value: "mistralai/mistral-7b-instruct-v0.2",
+        supported: ['text']
+    }).mapper(mistralMapper),
+    (0, utils_1.model)({ label: 'bytedance/sdxl-lightning-4step', value: 'bytedance/sdxl-lightning-4step:5599ed30703defd1d160a25a63321b4dec97101d98b4674bcc56e41f62f35637', supported: ['image'] })
+        .mapper(commonImageMapper),
+    (0, utils_1.model)({ label: 'stability-ai/stable-diffusion', value: 'stability-ai/stable-diffusion:ac732df83cea7fff18b8472768c88ad041fa750ff7682a21affe81863cbe77e4', supported: ['image'] })
+        .mapper(commonImageMapper),
+    (0, utils_1.model)({ label: 'black-forest-labs/flux-schnell', value: 'black-forest-labs/flux-schnell', supported: ['image'] })
+        .mapper((0, utils_1.imageMapper)({
+        advancedOptions: {},
+        decodeOutput: commonImageMapper.decodeOutput,
+        encodeInput: (params) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+            const commonParams = yield commonImageMapper.encodeInput(params);
+            const width = commonParams.width;
+            const height = commonParams.height;
+            const gcd = (a, b) => b === 0 ? a : gcd(b, a % b);
+            const divisor = gcd(width, height);
+            const aspectRatio = `${width / divisor}:${height / divisor}`;
+            return Object.assign(Object.assign({}, commonParams), { megapixels: "1", aspect_ratio: aspectRatio, output_format: "png" });
+        }),
+    })),
+];
+const findMapper = (model, type) => {
+    var _a;
+    const mapper = (_a = exports.replicateModels.find(m => m.value === model)) === null || _a === void 0 ? void 0 : _a.mapper;
+    if ((0, shared_1.isNil)(mapper) || mapper.__tag !== type) {
+        throw new Error(`${type} model ${model} not found`);
+    }
+    return mapper;
+};
+//# sourceMappingURL=index.js.map

package/src/lib/ai/providers/replicate/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../../../../../packages/pieces/community/common/src/lib/ai/providers/replicate/index.ts"],"names":[],"mappings":";;;;AAAA,oEAAyD;AACzD,gDAA4C;AAC5C,kEAAkC;AAClC,6BAAkD;AAClD,oCAAwG;AAEjG,MAAM,SAAS,GAAc,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,EAAM,EAAE;IACpE,MAAM,GAAG,GAAG,IAAI,mBAAS,CAAC;QACxB,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,GAAG,QAAQ,KAAK;KAC1B,CAAC,CAAC;IAEH,OAAO;QACL,QAAQ,EAAE,WAAW;QACrB,KAAK,EAAE;YACL,QAAQ,EAAE,CAAO,MAAM,EAAE,EAAE;gBACzB,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,iBAAS,CAAC,KAAK,CAAqB,CAAC;gBAC7E,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7C,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7C,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC;oBAC9C,KAAK,EAAE,GAAG,UAAU,IAAI,SAAS,EAAE;oBACnC,OAAO,EAAE,SAAS;oBAClB,KAAK,EAAE,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC;iBACxC,CAAC,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC3E,OAAO,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC5C,CAAC,CAAA;SACF;QACD,IAAI,EAAE;YACJ,IAAI,EAAE,CAAO,MAAM,EAAE,EAAE;gBACrB,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,iBAAS,CAAC,IAAI,CAAoB,CAAC;gBAC3E,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9C,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7C,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7C,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC;oBAC9C,KAAK,EAAE,GAAG,UAAU,IAAI,SAAS,EAAE;oBACnC,OAAO,EAAE,SAAS;oBAClB,KAAK,EAAE,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC;iBACxC,CAAC,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC3E,OAAO,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC5C,CAAC,CAAA;SACF;KACF,CAAC;AACJ,CAAC,CAAC;AAvCW,QAAA,SAAS,aAuCpB;AAEF,MAAM,iBAAiB,GAAG,IAAA,mBAAW,EAAC;IACpC,WAAW,EAAE,CAAO,MAAM,EAAE,EAAE;;QAC5B,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,MAAA,MAAA,MAAM,CAAC,IAAI,0CAAE,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,MAAM,CAAC,mCAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC1E,MAAM,cAAc,GAAG,MAAA,MAAA,MAAM,CAAC,eAAe,0CAAG,gBAAgB,CAAC,mCAAI,IAAI,CAAC;QAC1E,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,eAAe,EAAE,cAAc;YAC/B,KAAK;YACL,MAAM;YACN,WAAW,EAAE,CAAC;SACf,CAAC;IACJ,CAAC,CAAA;IACD,YAAY,EAAE,CAAO,MAAM,EAAE,EAAE;QAC7B,MAAM,SAAS,GAAG,MAAuB,CAAC;QAC1C,MAAM,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;QAC9C,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAChE,OAAO;YACL,KAAK,EAAE,WAAW;SACnB,CAAC;IACJ,CAAC,CAAA;IACD,eAAe,EAAE;QACf,cAAc,EAAE,2BAAQ,CAAC,SAAS,CAAC;YACjC,WAAW,EAAE,iBAAiB;YAC9B,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,2CAA2C;SACzD,CAAC;KACH;CACF,CAAC,CAAA;AAEF,MAAM,WAAW,GAAG,IAAA,kBAAU,EAAC;IAC7B,WAAW,EAAE,CAAO,MAAM,EAAE,EAAE;;QAC5B,MAAM,yBAAyB,GAAG,MAAM,CAAC,QAAQ;aAC9C,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,cAAU,CAAC,MAAM,CAAC;aACvD,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,OAAO;YACL,aAAa,EAAE,yBAAyB;YACxC,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,MAAA,MAAM,CAAC,UAAU,mCAAI,GAAG,CAAC;YAChD,MAAM,EAAE,MAAM,CAAC,QAAQ;iBACpB,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,cAAU,CAAC,MAAM,CAAC;iBACvD,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,OAAO,EAAE,CAAC;iBACvD,IAAI,CAAC,IAAI,CAAC;YACb,cAAc,EAAE,MAAM,CAAC,IAAI;YAC3B,UAAU,EAAE,MAAM,CAAC,SAAS;SAC7B,CAAC;IACJ,CAAC,CAAA;IACD,YAAY,EAAE,CAAO,MAAM,EAAE,EAAE;QAC7B,MAAM,OAAO,GAAG,MAAuB,CAAC;QACxC,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;oBACzB,IAAI,EAAE,cAAU,CAAC,SAAS;iBAC3B,CAAC;SACH,CAAA;IACH,CAAC,CAAA;CACF,CAAC,CAAC;AAEH,MAAM,aAAa,GAAG,IAAA,kBAAU,EAAC;IAC/B,WAAW,EAAE,CAAO,MAAM,EAAE,EAAE;;QAC5B,MAAM,yBAAyB,GAAG,MAAM,CAAC,QAAQ;aAC9C,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,cAAU,CAAC,MAAM,CAAC;aACvD,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,OAAO;YACL,aAAa,EAAE,yBAAyB;YACxC,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,MAAA,MAAM,CAAC,UAAU,mCAAI,GAAG,CAAC;YAChD,MAAM,EAAE,MAAM,CAAC,QAAQ;iBACpB,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,cAAU,CAAC,MAAM,CAAC;iBACvD,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,OAAO,EAAE,CAAC;iBACvD,IAAI,CAAC,IAAI,CAAC;YACb,cAAc,EAAE,MAAM,CAAC,IAAI;YAC3B,cAAc,EAAE,MAAM,CAAC,SAAS;SACjC,CAAC;IACJ,CAAC,CAAA;IACD,YAAY,EAAE,CAAO,MAAM,EAAE,EAAE;QAC7B,MAAM,OAAO,GAAG,MAAuB,CAAC;QACxC,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;oBACzB,IAAI,EAAE,cAAU,CAAC,SAAS;iBAC3B,CAAC;SACH,CAAA;IACH,CAAC,CAAA;CACF,CAAC,CAAC;AAEU,QAAA,eAAe,GAAG;IAC7B,IAAA,aAAK,EAAC;QACJ,KAAK,EAAE,gCAAgC;QACvC,KAAK,EAAE,gCAAgC;QACvC,SAAS,EAAE,CAAC,MAAM,CAAC;KACpB,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC;IACtB,IAAA,aAAK,EAAC;QACJ,KAAK,EAAE,+BAA+B;QACtC,KAAK,EAAE,+BAA+B;QACtC,SAAS,EAAE,CAAC,MAAM,CAAC;KACpB,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC;IACtB,IAAA,aAAK,EAAC;QACJ,KAAK,EAAE,sCAAsC;QAC7C,KAAK,EAAE,sCAAsC;QAC7C,SAAS,EAAE,CAAC,MAAM,CAAC;KACpB,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;IACxB,IAAA,aAAK,EAAC;QACJ,KAAK,EAAE,oCAAoC;QAC3C,KAAK,EAAE,oCAAoC;QAC3C,SAAS,EAAE,CAAC,MAAM,CAAC;KACpB,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;IACxB,IAAA,aAAK,EAAC,EAAE,KAAK,EAAE,gCAAgC,EAAE,KAAK,EAAE,iGAAiG,EAAE,SAAS,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;SAC/K,MAAM,CAAC,iBAAiB,CAAC;IAC5B,IAAA,aAAK,EAAC,EAAE,KAAK,EAAE,+BAA+B,EAAE,KAAK,EAAE,gGAAgG,EAAE,SAAS,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;SAC7K,MAAM,CAAC,iBAAiB,CAAC;IAC5B,IAAA,aAAK,EAAC,EAAE,KAAK,EAAE,gCAAgC,EAAE,KAAK,EAAE,gCAAgC,EAAE,SAAS,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;SAC9G,MAAM,CAAC,IAAA,mBAAW,EAAC;QAClB,eAAe,EAAE,EAAE;QACnB,YAAY,EAAE,iBAAiB,CAAC,YAAY;QAC5C,WAAW,EAAE,CAAO,MAAM,EAAE,EAAE;YAC5B,MAAM,YAAY,GAAG,MAAM,iBAAiB,CAAC,WAAW,CAAC,MAAM,CAAQ,CAAC;YACxE,MAAM,KAAK,GAAG,YAAY,CAAC,KAAe,CAAC;YAC3C,MAAM,MAAM,GAAG,YAAY,CAAC,MAAgB,CAAC;YAC7C,MAAM,GAAG,GAAG,CAAC,CAAS,EAAE,CAAS,EAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1E,MAAM,OAAO,GAAG,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACnC,MAAM,WAAW,GAAG,GAAG,KAAK,GAAG,OAAO,IAAI,MAAM,GAAG,OAAO,EAAE,CAAC;YAC7D,uCACK,YAAY,KACf,UAAU,EAAE,GAAG,EACf,YAAY,EAAE,WAAW,EACzB,aAAa,EAAE,KAAK,IACpB;QACJ,CAAC,CAAA;KACF,CAAC,CAAC;CACN,CAAA;AAED,MAAM,UAAU,GAAG,CAAC,KAAa,EAAE,IAAe,EAAE,EAAE;;IACpD,MAAM,MAAM,GAAG,MAAA,uBAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,0CAAE,MAAM,CAAC;IACpE,IAAI,IAAA,cAAK,EAAC,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,KAAK,IAAI,EAAE,CAAC;QAC3C,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,UAAU,KAAK,YAAY,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC"}

package/src/lib/ai/providers/utils.d.ts

@@ -0,0 +1,58 @@
+import { DynamicProp } from '@activeboxes/pieces-framework';
+import { AIChatCompletion, AIChatCompletionsCreateParams, AIImageCompletion, AIImageGenerateParams } from '../index';
+export declare function authHeader(options: AuthHeaderOptions): AuthHeader;
+export type HeaderValueMapper = (value: string) => string;
+export type AuthHeader = {
+    name: string;
+    mapper: HeaderValueMapper;
+};
+type AuthHeaderOptions = {
+    bearer: true;
+} | {
+    bearer: false;
+    name: string;
+    mapper?: HeaderValueMapper;
+};
+export type ModelParameterValue = string | number | boolean;
+export type Model = ReturnType<typeof model> | ReturnType<ReturnType<typeof model>['mapper']>;
+export declare enum ModelType {
+    IMAGE = "image",
+    CHAT = "chat",
+    NO_MAPPER = "no-mapper"
+}
+export declare function model(options: {
+    label: string;
+    value: string;
+    supported: Array<'text' | 'image' | 'function' | 'moderation'>;
+}): {
+    __tag: ModelType;
+    label: string;
+    value: string;
+    supported: ("function" | "text" | "image" | "moderation")[];
+    mapper: (mapper: ImageModelMapper | ChatModelMapper) => {
+        mapper: ImageModelMapper | ChatModelMapper;
+        __tag: ModelType.IMAGE | ModelType.CHAT;
+        label: string;
+        value: string;
+        supported: ("function" | "text" | "image" | "moderation")[];
+    };
+};
+export type ModelMapper<T extends ModelType> = T extends ModelType.IMAGE ? ImageModelMapper : T extends ModelType.CHAT ? ChatModelMapper : never;
+export interface ImageModelMapper {
+    __tag: ModelType.IMAGE;
+    encodeInput: (input: AIImageGenerateParams) => Promise<object>;
+    decodeOutput: (output: object) => Promise<AIImageCompletion | null>;
+    advancedOptions?: Record<string, DynamicProp>;
+}
+export interface ChatModelMapper {
+    __tag: ModelType.CHAT;
+    encodeInput: (input: AIChatCompletionsCreateParams) => Promise<object>;
+    decodeOutput: (output: object) => Promise<AIChatCompletion>;
+    advancedOptions?: Record<string, DynamicProp>;
+}
+export declare function imageMapper(mapper: Omit<ImageModelMapper, '__tag'>): ImageModelMapper;
+export declare function chatMapper(mapper: Omit<ChatModelMapper, '__tag'>): ChatModelMapper;
+export declare const hasMapper: (m: Model) => m is Model & {
+    mapper: ModelMapper<ModelType>;
+};
+export {};

package/src/lib/ai/providers/utils.js

@@ -0,0 +1,48 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.hasMapper = exports.ModelType = void 0;
+exports.authHeader = authHeader;
+exports.model = model;
+exports.imageMapper = imageMapper;
+exports.chatMapper = chatMapper;
+function authHeader(options) {
+    var _a;
+    return {
+        name: options.bearer ? 'Authorization' : options.name,
+        mapper: options.bearer
+            ? headerValueMappers.bearer
+            : (_a = options.mapper) !== null && _a !== void 0 ? _a : headerValueMappers.default,
+    };
+}
+const headerValueMappers = {
+    bearer: (value) => `Bearer ${value}`,
+    default: (value) => value,
+};
+var ModelType;
+(function (ModelType) {
+    ModelType["IMAGE"] = "image";
+    ModelType["CHAT"] = "chat";
+    ModelType["NO_MAPPER"] = "no-mapper";
+})(ModelType || (exports.ModelType = ModelType = {}));
+function model(options) {
+    return {
+        __tag: ModelType.NO_MAPPER,
+        label: options.label,
+        value: options.value,
+        supported: options.supported,
+        mapper: function (mapper) {
+            return Object.assign(Object.assign({}, this), { mapper: mapper, __tag: mapper.__tag });
+        },
+    };
+}
+function imageMapper(mapper) {
+    return Object.assign(Object.assign({}, mapper), { __tag: ModelType.IMAGE });
+}
+function chatMapper(mapper) {
+    return Object.assign(Object.assign({}, mapper), { __tag: ModelType.CHAT });
+}
+const hasMapper = (m) => {
+    return m.__tag === ModelType.IMAGE || m.__tag === ModelType.CHAT;
+};
+exports.hasMapper = hasMapper;
+//# sourceMappingURL=utils.js.map

package/src/lib/ai/providers/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/community/common/src/lib/ai/providers/utils.ts"],"names":[],"mappings":";;;AAUA,gCAOC;AA8BD,sBAcC;AAqBD,kCAIC;AAED,gCAIC;AAlFD,SAAgB,UAAU,CAAC,OAA0B;;IACnD,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAE,eAAyB,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI;QAChE,MAAM,EAAE,OAAO,CAAC,MAAM;YACpB,CAAC,CAAC,kBAAkB,CAAC,MAAM;YAC3B,CAAC,CAAC,MAAA,OAAO,CAAC,MAAM,mCAAI,kBAAkB,CAAC,OAAO;KACjD,CAAC;AACJ,CAAC;AAaD,MAAM,kBAAkB,GAAG;IACzB,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,UAAU,KAAK,EAAE;IAC5C,OAAO,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK;CAClC,CAAC;AAQF,IAAY,SAIX;AAJD,WAAY,SAAS;IACnB,4BAAe,CAAA;IACf,0BAAa,CAAA;IACb,oCAAuB,CAAA;AACzB,CAAC,EAJW,SAAS,yBAAT,SAAS,QAIpB;AAED,SAAgB,KAAK,CAAC,OAIrB;IACC,OAAO;QACL,KAAK,EAAE,SAAS,CAAC,SAAS;QAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,MAAM,EAAE,UAAU,MAA0C;YAC1D,uCAAY,IAAI,KAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAG;QAC1D,CAAC;KACF,CAAC;AACJ,CAAC;AAqBD,SAAgB,WAAW,CACzB,MAAuC;IAEvC,uCAAY,MAAM,KAAE,KAAK,EAAE,SAAS,CAAC,KAAK,IAAG;AAC/C,CAAC;AAED,SAAgB,UAAU,CACxB,MAAsC;IAEtC,uCAAY,MAAM,KAAE,KAAK,EAAE,SAAS,CAAC,IAAI,IAAG;AAC9C,CAAC;AAEM,MAAM,SAAS,GAAG,CACvB,CAAQ,EACyC,EAAE;IACnD,OAAO,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,IAAI,CAAC;AACnE,CAAC,CAAC;AAJW,QAAA,SAAS,aAIpB"}

package/src/lib/authentication/index.d.ts

@@ -0,0 +1,15 @@
+export type Authentication = BearerTokenAuthentication | BasicAuthentication;
+export declare enum AuthenticationType {
+    BEARER_TOKEN = "BEARER_TOKEN",
+    BASIC = "BASIC"
+}
+export type BaseAuthentication<T extends AuthenticationType> = {
+    type: T;
+};
+export type BearerTokenAuthentication = BaseAuthentication<AuthenticationType.BEARER_TOKEN> & {
+    token: string;
+};
+export type BasicAuthentication = BaseAuthentication<AuthenticationType.BASIC> & {
+    username: string;
+    password: string;
+};

package/src/lib/authentication/index.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AuthenticationType = void 0;
+var AuthenticationType;
+(function (AuthenticationType) {
+    AuthenticationType["BEARER_TOKEN"] = "BEARER_TOKEN";
+    AuthenticationType["BASIC"] = "BASIC";
+})(AuthenticationType || (exports.AuthenticationType = AuthenticationType = {}));
+//# sourceMappingURL=index.js.map

package/src/lib/authentication/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/common/src/lib/authentication/index.ts"],"names":[],"mappings":";;;AAEA,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,mDAA6B,CAAA;IAC7B,qCAAe,CAAA;AACjB,CAAC,EAHW,kBAAkB,kCAAlB,kBAAkB,QAG7B"}

package/src/lib/helpers/index.d.ts

@@ -0,0 +1,29 @@
+import { OAuth2PropertyValue, PieceAuthProperty, Property, StaticDropdownProperty, StaticPropsValue, InputPropertyMap } from '@activeboxes/pieces-framework';
+import { HttpHeaders, HttpMethod } from '../http';
+export declare const getAccessTokenOrThrow: (auth: OAuth2PropertyValue | undefined) => string;
+export declare function createCustomApiCallAction({ auth, baseUrl, authMapping, description, displayName, name, props, extraProps, }: {
+    auth?: PieceAuthProperty;
+    baseUrl: (auth?: unknown) => string;
+    authMapping?: (auth: unknown, propsValue: StaticPropsValue<any>) => Promise<HttpHeaders>;
+    description?: string | null;
+    displayName?: string | null;
+    name?: string | null;
+    props?: {
+        url?: Partial<ReturnType<typeof Property.ShortText>>;
+        method?: Partial<StaticDropdownProperty<HttpMethod, boolean>>;
+        headers?: Partial<ReturnType<typeof Property.Object>>;
+        queryParams?: Partial<ReturnType<typeof Property.Object>>;
+        body?: Partial<ReturnType<typeof Property.Json>>;
+        failsafe?: Partial<ReturnType<typeof Property.Checkbox>>;
+        timeout?: Partial<ReturnType<typeof Property.Number>>;
+    };
+    extraProps?: InputPropertyMap;
+}): import("@activeboxes/pieces-framework").IAction<import("@activeboxes/pieces-framework").BasicAuthProperty | import("@activeboxes/pieces-framework").CustomAuthProperty<any> | import("@activeboxes/pieces-framework").OAuth2Property<any> | import("@activeboxes/pieces-framework").SecretTextProperty<boolean>, {
+    url: import("@activeboxes/pieces-framework").DynamicProperties<true>;
+    method: StaticDropdownProperty<HttpMethod, false> | StaticDropdownProperty<HttpMethod, true>;
+    headers: import("@activeboxes/pieces-framework").ObjectProperty<false> | import("@activeboxes/pieces-framework").ObjectProperty<true>;
+    queryParams: import("@activeboxes/pieces-framework").ObjectProperty<false> | import("@activeboxes/pieces-framework").ObjectProperty<true>;
+    body: import("@activeboxes/pieces-framework").JsonProperty<false> | import("@activeboxes/pieces-framework").JsonProperty<true>;
+    failsafe: import("@activeboxes/pieces-framework").CheckboxProperty<false> | import("@activeboxes/pieces-framework").CheckboxProperty<true>;
+    timeout: import("@activeboxes/pieces-framework").NumberProperty<false> | import("@activeboxes/pieces-framework").NumberProperty<true>;
+}>;