@taskmagic/apps-hugging-face 0.0.1

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

@@ -0,0 +1,344 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textClassification = void 0;
+const tslib_1 = require("tslib");
+const pieces_framework_1 = require("@taskmagic/pieces-framework");
+const pieces_common_1 = require("@taskmagic/pieces-common");
+const auth_1 = require("../auth");
+const common_1 = require("../common");
+exports.textClassification = (0, pieces_framework_1.createAction)({
+    name: 'text_classification',
+    auth: auth_1.huggingFaceAuth,
+    displayName: 'Text Classification',
+    description: 'Classify text into categories using Hugging Face models - includes zero-shot classification for custom categories',
+    props: {
+        classificationMode: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Classification Type',
+            description: 'Choose your classification approach',
+            required: true,
+            options: {
+                disabled: false,
+                options: [
+                    {
+                        label: '🎯 Zero-Shot (Custom Categories)',
+                        value: 'zero-shot',
+                    },
+                    {
+                        label: '📊 Pre-trained Models',
+                        value: 'pretrained',
+                    },
+                    {
+                        label: '🔍 Search All Models',
+                        value: 'search',
+                    },
+                ],
+            },
+            defaultValue: 'zero-shot',
+        }),
+        zeroShotModel: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Zero-Shot Model',
+            description: 'Model for classifying into your custom categories',
+            required: false,
+            options: {
+                disabled: false,
+                options: [
+                    {
+                        label: '🔥 Facebook BART-Large (5.4M downloads)',
+                        value: 'facebook/bart-large-mnli',
+                    },
+                    {
+                        label: '🔥 DeBERTa TaskSource (904K downloads)',
+                        value: 'sileod/deberta-v3-base-tasksource-nli',
+                    },
+                    {
+                        label: '🔥 DeBERTa MNLI-FEVER (668K downloads)',
+                        value: 'MoritzLaurer/DeBERTa-v3-base-mnli-fever-anli',
+                    },
+                    {
+                        label: '🌍 DistilBERT Multilingual (520K downloads)',
+                        value: 'lxyuan/distilbert-base-multilingual-cased-sentiments-student',
+                    },
+                ],
+            },
+            defaultValue: 'facebook/bart-large-mnli',
+        }),
+        customLabels: pieces_framework_1.Property.LongText({
+            displayName: 'Custom Categories',
+            description: 'Enter categories separated by commas (e.g., "customer support, sales inquiry, spam, billing question")',
+            required: false,
+        }),
+        pretrainedModel: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Pre-trained Model',
+            description: 'Select a specialized pre-trained classification model',
+            required: false,
+            options: {
+                disabled: false,
+                options: [
+                    // === SENTIMENT ANALYSIS ===
+                    {
+                        label: '😊 Sentiment: Twitter RoBERTa (3M downloads)',
+                        value: 'cardiffnlp/twitter-roberta-base-sentiment-latest',
+                    },
+                    {
+                        label: '😊 Sentiment: DistilBERT SST-2 (2.9M downloads)',
+                        value: 'distilbert/distilbert-base-uncased-finetuned-sst-2-english',
+                    },
+                    {
+                        label: '🌍 Sentiment: Multilingual BERT (1.7M downloads)',
+                        value: 'nlptown/bert-base-multilingual-uncased-sentiment',
+                    },
+                    // === BUSINESS & FINANCE ===
+                    {
+                        label: '💰 Finance: FinBERT Sentiment (1.5M downloads)',
+                        value: 'ProsusAI/finbert',
+                    },
+                    {
+                        label: '💰 Finance: FinBERT Tone (1.4M downloads)',
+                        value: 'yiyanghkust/finbert-tone',
+                    },
+                    // === CONTENT MODERATION ===
+                    {
+                        label: '🛡️ Hate Speech: RoBERTa (2M downloads)',
+                        value: 'facebook/roberta-hate-speech-dynabench-r4-target',
+                    },
+                    {
+                        label: '🛡️ Toxicity: HateBERT (1.1M downloads)',
+                        value: 'tomh/toxigen_hatebert',
+                    },
+                    {
+                        label: '🛡️ Toxicity: ToxDect RoBERTa (1.1M downloads)',
+                        value: 'Xuhui/ToxDect-roberta-large',
+                    },
+                    // === CUSTOMER SERVICE ===
+                    {
+                        label: '🎧 Customer Service: Banking Intent',
+                        value: 'atulgupta002/banking_customer_service_query_intent_classifier',
+                    },
+                    {
+                        label: '📧 Spam: SMS Classification',
+                        value: 'wesleyacheng/sms-spam-classification-with-bert',
+                    },
+                ],
+            },
+            defaultValue: 'cardiffnlp/twitter-roberta-base-sentiment-latest',
+        }),
+        searchModel: pieces_framework_1.Property.Dropdown({
+            displayName: 'Search Models',
+            description: 'Search from all available text classification models',
+            required: false,
+            refreshers: [],
+            options: (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ auth }) {
+                const popularModels = [
+                    {
+                        label: '🔥 Facebook BART Zero-Shot (5.4M downloads)',
+                        value: 'facebook/bart-large-mnli',
+                    },
+                    {
+                        label: '🔥 Twitter Sentiment RoBERTa (3M downloads)',
+                        value: 'cardiffnlp/twitter-roberta-base-sentiment-latest',
+                    },
+                    {
+                        label: '🔥 DistilBERT Sentiment (2.9M downloads)',
+                        value: 'distilbert/distilbert-base-uncased-finetuned-sst-2-english',
+                    },
+                ];
+                if (!auth) {
+                    return {
+                        disabled: false,
+                        options: popularModels,
+                    };
+                }
+                try {
+                    const response = yield pieces_common_1.httpClient.sendRequest({
+                        method: pieces_common_1.HttpMethod.GET,
+                        url: 'https://huggingface.co/api/models',
+                        queryParams: {
+                            filter: 'text-classification',
+                            sort: 'downloads',
+                            direction: '-1',
+                            limit: '100',
+                        },
+                    });
+                    const models = response.body;
+                    const classificationModels = models
+                        .filter((model) => model.pipeline_tag === 'text-classification' ||
+                        model.pipeline_tag === 'zero-shot-classification')
+                        .map((model) => {
+                        var _a;
+                        return ({
+                            label: `${model.id} (${((_a = model.downloads) === null || _a === void 0 ? void 0 : _a.toLocaleString()) || 0} downloads)`,
+                            value: model.id,
+                        });
+                    })
+                        .slice(0, 50);
+                    const allOptions = [
+                        ...popularModels,
+                        ...classificationModels.filter((model) => !popularModels.some((popular) => popular.value === model.value)),
+                    ];
+                    return {
+                        disabled: false,
+                        options: allOptions,
+                    };
+                }
+                catch (error) {
+                    return {
+                        disabled: false,
+                        options: popularModels,
+                    };
+                }
+            }),
+        }),
+        text: pieces_framework_1.Property.LongText({
+            displayName: 'Text to Classify',
+            description: 'The text content you want to classify',
+            required: true,
+        }),
+        topK: pieces_framework_1.Property.Number({
+            displayName: 'Number of Results',
+            description: 'Number of top predictions to return',
+            required: false,
+            defaultValue: 3,
+        }),
+        functionToApply: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Output Function',
+            description: 'How to calculate confidence scores',
+            required: false,
+            options: {
+                disabled: false,
+                options: [
+                    { label: 'Softmax (Recommended)', value: 'softmax' },
+                    { label: 'Sigmoid', value: 'sigmoid' },
+                    { label: 'None (Raw Scores)', value: 'none' },
+                ],
+            },
+            defaultValue: 'softmax',
+        }),
+        useCache: pieces_framework_1.Property.Checkbox({
+            displayName: 'Use Cache',
+            description: 'Use cached results for faster responses',
+            required: false,
+            defaultValue: true,
+        }),
+        waitForModel: pieces_framework_1.Property.Checkbox({
+            displayName: 'Wait for Model',
+            description: 'Wait for model to load if not immediately available',
+            required: false,
+            defaultValue: false,
+        }),
+    },
+    run(context) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            const { classificationMode, zeroShotModel, customLabels, pretrainedModel, searchModel, text, topK, functionToApply, useCache, waitForModel, } = context.propsValue;
+            // Determine which model to use
+            let modelToUse;
+            let isZeroShot = false;
+            switch (classificationMode) {
+                case 'zero-shot':
+                    if (!(customLabels === null || customLabels === void 0 ? void 0 : customLabels.trim())) {
+                        throw new Error('Please provide custom categories for zero-shot classification (e.g., "positive, negative, neutral")');
+                    }
+                    modelToUse = zeroShotModel || 'facebook/bart-large-mnli';
+                    isZeroShot = true;
+                    break;
+                case 'pretrained':
+                    modelToUse =
+                        pretrainedModel || 'cardiffnlp/twitter-roberta-base-sentiment-latest';
+                    break;
+                case 'search':
+                    if (!searchModel) {
+                        throw new Error('Please select a model from the search dropdown');
+                    }
+                    modelToUse = searchModel;
+                    break;
+                default:
+                    modelToUse = zeroShotModel || 'facebook/bart-large-mnli';
+                    isZeroShot = true;
+            }
+            const options = {
+                use_cache: useCache !== null && useCache !== void 0 ? useCache : true,
+                wait_for_model: waitForModel !== null && waitForModel !== void 0 ? waitForModel : false,
+            };
+            if (isZeroShot) {
+                // Handle zero-shot classification
+                const labels = (customLabels || '')
+                    .split(',')
+                    .map((label) => label.trim())
+                    .filter((label) => label.length > 0);
+                if (labels.length === 0) {
+                    throw new Error('Please provide at least one category for zero-shot classification');
+                }
+                const zeroShotRaw = yield (0, common_1.callPipelineTask)({
+                    auth: context.auth,
+                    model: modelToUse,
+                    body: {
+                        inputs: text,
+                        parameters: {
+                            candidate_labels: labels,
+                        },
+                        options,
+                    },
+                });
+                // Normalize into an array of {label, score} objects
+                const classifications = zeroShotRaw.labels.map((label, index) => ({
+                    label,
+                    score: zeroShotRaw.scores[index],
+                }));
+                // Sort by score and limit to topK
+                const sortedResults = classifications
+                    .sort((a, b) => b.score - a.score)
+                    .slice(0, topK || 3);
+                return {
+                    predictions: sortedResults,
+                    topPrediction: sortedResults[0],
+                    text: text,
+                    model: modelToUse,
+                    classificationMode: 'zero-shot',
+                    customCategories: labels,
+                    confidenceThreshold: 0.5,
+                    highConfidencePredictions: sortedResults.filter((r) => r.score > 0.5),
+                    rawResult: zeroShotRaw,
+                };
+            }
+            else {
+                // Handle regular text classification
+                const parameters = {};
+                if (topK !== undefined && topK > 0) {
+                    parameters.top_k = topK;
+                }
+                if (functionToApply &&
+                    (functionToApply === 'softmax' ||
+                        functionToApply === 'sigmoid' ||
+                        functionToApply === 'none')) {
+                    parameters.function_to_apply = functionToApply;
+                }
+                const body = {
+                    inputs: text,
+                    options,
+                };
+                if (Object.keys(parameters).length > 0) {
+                    body['parameters'] = parameters;
+                }
+                const classificationResult = yield (0, common_1.callPipelineTask)({
+                    auth: context.auth,
+                    model: modelToUse,
+                    body,
+                });
+                // The API may return either an array of {label, score} or a nested array
+                const results = Array.isArray(classificationResult[0])
+                    ? classificationResult[0]
+                    : classificationResult;
+                return {
+                    predictions: results,
+                    topPrediction: results[0],
+                    text: text,
+                    model: modelToUse,
+                    classificationMode: 'pretrained',
+                    confidenceThreshold: 0.5,
+                    highConfidencePredictions: results.filter((r) => r.score > 0.5),
+                    rawResult: classificationResult,
+                };
+            }
+        });
+    },
+});
+//# sourceMappingURL=text-classification.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"text-classification.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/hugging-face/src/lib/actions/text-classification.ts"],"names":[],"mappings":";;;;AAAA,kEAAqE;AACrE,4DAAkE;AAClE,kCAA0C;AAC1C,sCAA6C;AAShC,QAAA,kBAAkB,GAAG,IAAA,+BAAY,EAAC;IAC7C,IAAI,EAAE,qBAAqB;IAC3B,IAAI,EAAE,sBAAe;IACrB,WAAW,EAAE,qBAAqB;IAClC,WAAW,EACT,mHAAmH;IACrH,KAAK,EAAE;QACL,kBAAkB,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAC1C,WAAW,EAAE,qBAAqB;YAClC,WAAW,EAAE,qCAAqC;YAClD,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,kCAAkC;wBACzC,KAAK,EAAE,WAAW;qBACnB;oBACD;wBACE,KAAK,EAAE,uBAAuB;wBAC9B,KAAK,EAAE,YAAY;qBACpB;oBACD;wBACE,KAAK,EAAE,sBAAsB;wBAC7B,KAAK,EAAE,QAAQ;qBAChB;iBACF;aACF;YACD,YAAY,EAAE,WAAW;SAC1B,CAAC;QACF,aAAa,EAAE,2BAAQ,CAAC,cAAc,CAAC;YACrC,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EAAE,mDAAmD;YAChE,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,yCAAyC;wBAChD,KAAK,EAAE,0BAA0B;qBAClC;oBACD;wBACE,KAAK,EAAE,wCAAwC;wBAC/C,KAAK,EAAE,uCAAuC;qBAC/C;oBACD;wBACE,KAAK,EAAE,wCAAwC;wBAC/C,KAAK,EAAE,8CAA8C;qBACtD;oBACD;wBACE,KAAK,EAAE,6CAA6C;wBACpD,KAAK,EACH,8DAA8D;qBACjE;iBACF;aACF;YACD,YAAY,EAAE,0BAA0B;SACzC,CAAC;QACF,YAAY,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC9B,WAAW,EAAE,mBAAmB;YAChC,WAAW,EACT,wGAAwG;YAC1G,QAAQ,EAAE,KAAK;SAChB,CAAC;QACF,eAAe,EAAE,2BAAQ,CAAC,cAAc,CAAC;YACvC,WAAW,EAAE,mBAAmB;YAChC,WAAW,EAAE,uDAAuD;YACpE,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP,6BAA6B;oBAC7B;wBACE,KAAK,EAAE,8CAA8C;wBACrD,KAAK,EAAE,kDAAkD;qBAC1D;oBACD;wBACE,KAAK,EAAE,iDAAiD;wBACxD,KAAK,EAAE,4DAA4D;qBACpE;oBACD;wBACE,KAAK,EAAE,kDAAkD;wBACzD,KAAK,EAAE,kDAAkD;qBAC1D;oBAED,6BAA6B;oBAC7B;wBACE,KAAK,EAAE,gDAAgD;wBACvD,KAAK,EAAE,kBAAkB;qBAC1B;oBACD;wBACE,KAAK,EAAE,2CAA2C;wBAClD,KAAK,EAAE,0BAA0B;qBAClC;oBAED,6BAA6B;oBAC7B;wBACE,KAAK,EAAE,yCAAyC;wBAChD,KAAK,EAAE,kDAAkD;qBAC1D;oBACD;wBACE,KAAK,EAAE,yCAAyC;wBAChD,KAAK,EAAE,uBAAuB;qBAC/B;oBACD;wBACE,KAAK,EAAE,gDAAgD;wBACvD,KAAK,EAAE,6BAA6B;qBACrC;oBAED,2BAA2B;oBAC3B;wBACE,KAAK,EAAE,qCAAqC;wBAC5C,KAAK,EACH,+DAA+D;qBAClE;oBACD;wBACE,KAAK,EAAE,6BAA6B;wBACpC,KAAK,EAAE,gDAAgD;qBACxD;iBACF;aACF;YACD,YAAY,EAAE,kDAAkD;SACjE,CAAC;QACF,WAAW,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC7B,WAAW,EAAE,eAAe;YAC5B,WAAW,EAAE,sDAAsD;YACnE,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,EAAE;YACd,OAAO,EAAE,KAAiB,EAAE,oDAAZ,EAAE,IAAI,EAAE;gBACtB,MAAM,aAAa,GAAG;oBACpB;wBACE,KAAK,EAAE,6CAA6C;wBACpD,KAAK,EAAE,0BAA0B;qBAClC;oBACD;wBACE,KAAK,EAAE,6CAA6C;wBACpD,KAAK,EAAE,kDAAkD;qBAC1D;oBACD;wBACE,KAAK,EAAE,0CAA0C;wBACjD,KAAK,EAAE,4DAA4D;qBACpE;iBACF,CAAC;gBAEF,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,OAAO;wBACL,QAAQ,EAAE,KAAK;wBACf,OAAO,EAAE,aAAa;qBACvB,CAAC;gBACJ,CAAC;gBAED,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,0BAAU,CAAC,WAAW,CAAC;wBAC5C,MAAM,EAAE,0BAAU,CAAC,GAAG;wBACtB,GAAG,EAAE,mCAAmC;wBACxC,WAAW,EAAE;4BACX,MAAM,EAAE,qBAAqB;4BAC7B,IAAI,EAAE,WAAW;4BACjB,SAAS,EAAE,IAAI;4BACf,KAAK,EAAE,KAAK;yBACb;qBACF,CAAC,CAAC;oBAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAItB,CAAC;oBAEH,MAAM,oBAAoB,GAAG,MAAM;yBAChC,MAAM,CACL,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,CAAC,YAAY,KAAK,qBAAqB;wBAC5C,KAAK,CAAC,YAAY,KAAK,0BAA0B,CACpD;yBACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;;wBAAC,OAAA,CAAC;4BACf,KAAK,EAAE,GAAG,KAAK,CAAC,EAAE,KAChB,CAAA,MAAA,KAAK,CAAC,SAAS,0CAAE,cAAc,EAAE,KAAI,CACvC,aAAa;4BACb,KAAK,EAAE,KAAK,CAAC,EAAE;yBAChB,CAAC,CAAA;qBAAA,CAAC;yBACF,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAEhB,MAAM,UAAU,GAAG;wBACjB,GAAG,aAAa;wBAChB,GAAG,oBAAoB,CAAC,MAAM,CAC5B,CAAC,KAAK,EAAE,EAAE,CACR,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK,CAAC,CAClE;qBACF,CAAC;oBAEF,OAAO;wBACL,QAAQ,EAAE,KAAK;wBACf,OAAO,EAAE,UAAU;qBACpB,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO;wBACL,QAAQ,EAAE,KAAK;wBACf,OAAO,EAAE,aAAa;qBACvB,CAAC;gBACJ,CAAC;YACH,CAAC,CAAA;SACF,CAAC;QACF,IAAI,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACtB,WAAW,EAAE,kBAAkB;YAC/B,WAAW,EAAE,uCAAuC;YACpD,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,IAAI,EAAE,2BAAQ,CAAC,MAAM,CAAC;YACpB,WAAW,EAAE,mBAAmB;YAChC,WAAW,EAAE,qCAAqC;YAClD,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,CAAC;SAChB,CAAC;QACF,eAAe,EAAE,2BAAQ,CAAC,cAAc,CAAC;YACvC,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EAAE,oCAAoC;YACjD,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP,EAAE,KAAK,EAAE,uBAAuB,EAAE,KAAK,EAAE,SAAS,EAAE;oBACpD,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;oBACtC,EAAE,KAAK,EAAE,mBAAmB,EAAE,KAAK,EAAE,MAAM,EAAE;iBAC9C;aACF;YACD,YAAY,EAAE,SAAS;SACxB,CAAC;QACF,QAAQ,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC1B,WAAW,EAAE,WAAW;YACxB,WAAW,EAAE,yCAAyC;YACtD,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,IAAI;SACnB,CAAC;QACF,YAAY,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC9B,WAAW,EAAE,gBAAgB;YAC7B,WAAW,EAAE,qDAAqD;YAClE,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;SACpB,CAAC;KACH;IACK,GAAG,CAAC,OAAO;;YACf,MAAM,EACJ,kBAAkB,EAClB,aAAa,EACb,YAAY,EACZ,eAAe,EACf,WAAW,EACX,IAAI,EACJ,IAAI,EACJ,eAAe,EACf,QAAQ,EACR,YAAY,GACb,GAAG,OAAO,CAAC,UAAU,CAAC;YAEvB,+BAA+B;YAC/B,IAAI,UAAkB,CAAC;YACvB,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,QAAQ,kBAAkB,EAAE,CAAC;gBAC3B,KAAK,WAAW;oBACd,IAAI,CAAC,CAAA,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,IAAI,EAAE,CAAA,EAAE,CAAC;wBAC1B,MAAM,IAAI,KAAK,CACb,qGAAqG,CACtG,CAAC;oBACJ,CAAC;oBACD,UAAU,GAAG,aAAa,IAAI,0BAA0B,CAAC;oBACzD,UAAU,GAAG,IAAI,CAAC;oBAClB,MAAM;gBACR,KAAK,YAAY;oBACf,UAAU;wBACR,eAAe,IAAI,kDAAkD,CAAC;oBACxE,MAAM;gBACR,KAAK,QAAQ;oBACX,IAAI,CAAC,WAAW,EAAE,CAAC;wBACjB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;oBACpE,CAAC;oBACD,UAAU,GAAG,WAAW,CAAC;oBACzB,MAAM;gBACR;oBACE,UAAU,GAAG,aAAa,IAAI,0BAA0B,CAAC;oBACzD,UAAU,GAAG,IAAI,CAAC;YACtB,CAAC;YAED,MAAM,OAAO,GAAG;gBACd,SAAS,EAAE,QAAQ,aAAR,QAAQ,cAAR,QAAQ,GAAI,IAAI;gBAC3B,cAAc,EAAE,YAAY,aAAZ,YAAY,cAAZ,YAAY,GAAI,KAAK;aACtC,CAAC;YAEF,IAAI,UAAU,EAAE,CAAC;gBACf,kCAAkC;gBAClC,MAAM,MAAM,GAAG,CAAC,YAAY,IAAI,EAAE,CAAC;qBAChC,KAAK,CAAC,GAAG,CAAC;qBACV,GAAG,CAAC,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;qBACpC,MAAM,CAAC,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAE/C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACxB,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;gBACJ,CAAC;gBAED,MAAM,WAAW,GAAG,MAAM,IAAA,yBAAgB,EAAoB;oBAC5D,IAAI,EAAE,OAAO,CAAC,IAAc;oBAC5B,KAAK,EAAE,UAAU;oBACjB,IAAI,EAAE;wBACJ,MAAM,EAAE,IAAI;wBACZ,UAAU,EAAE;4BACV,gBAAgB,EAAE,MAAM;yBACzB;wBACD,OAAO;qBACR;iBACF,CAAC,CAAC;gBAEH,oDAAoD;gBACpD,MAAM,eAAe,GAAiB,WAAW,CAAC,MAAM,CAAC,GAAG,CAC1D,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;oBACjB,KAAK;oBACL,KAAK,EAAE,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC;iBACjC,CAAC,CACH,CAAC;gBAEF,kCAAkC;gBAClC,MAAM,aAAa,GAAG,eAAe;qBAClC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;qBACjC,KAAK,CAAC,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC;gBAEvB,OAAO;oBACL,WAAW,EAAE,aAAa;oBAC1B,aAAa,EAAE,aAAa,CAAC,CAAC,CAAC;oBAC/B,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,UAAU;oBACjB,kBAAkB,EAAE,WAAW;oBAC/B,gBAAgB,EAAE,MAAM;oBACxB,mBAAmB,EAAE,GAAG;oBACxB,yBAAyB,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC;oBACrE,SAAS,EAAE,WAAW;iBACvB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,qCAAqC;gBACrC,MAAM,UAAU,GAGZ,EAAE,CAAC;gBAEP,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;oBACnC,UAAU,CAAC,KAAK,GAAG,IAAI,CAAC;gBAC1B,CAAC;gBAED,IACE,eAAe;oBACf,CAAC,eAAe,KAAK,SAAS;wBAC5B,eAAe,KAAK,SAAS;wBAC7B,eAAe,KAAK,MAAM,CAAC,EAC7B,CAAC;oBACD,UAAU,CAAC,iBAAiB,GAAG,eAAe,CAAC;gBACjD,CAAC;gBAED,MAAM,IAAI,GAA4B;oBACpC,MAAM,EAAE,IAAI;oBACZ,OAAO;iBACR,CAAC;gBAEF,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvC,IAAI,CAAC,YAAY,CAAC,GAAG,UAAU,CAAC;gBAClC,CAAC;gBAED,MAAM,oBAAoB,GAAG,MAAM,IAAA,yBAAgB,EAEjD;oBACA,IAAI,EAAE,OAAO,CAAC,IAAc;oBAC5B,KAAK,EAAE,UAAU;oBACjB,IAAI;iBACL,CAAC,CAAC;gBAEH,yEAAyE;gBACzE,MAAM,OAAO,GAAiB,KAAK,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;oBAClE,CAAC,CAAE,oBAAoB,CAAC,CAAC,CAAkB;oBAC3C,CAAC,CAAE,oBAAqC,CAAC;gBAE3C,OAAO;oBACL,WAAW,EAAE,OAAO;oBACpB,aAAa,EAAE,OAAO,CAAC,CAAC,CAAC;oBACzB,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,UAAU;oBACjB,kBAAkB,EAAE,YAAY;oBAChC,mBAAmB,EAAE,GAAG;oBACxB,yBAAyB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC;oBAC/D,SAAS,EAAE,oBAAoB;iBAChC,CAAC;YACJ,CAAC;QACH,CAAC;KAAA;CACF,CAAC,CAAC"}

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

@@ -0,0 +1,12 @@
+export declare const textSummarization: import("@taskmagic/pieces-framework").IAction<import("@taskmagic/pieces-framework").SecretTextProperty<true>, {
+    contentType: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, true>;
+    model: import("@taskmagic/pieces-framework").DropdownProperty<string, true>;
+    text: import("@taskmagic/pieces-framework").LongTextProperty<true>;
+    summaryLength: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, false>;
+    customMinLength: import("@taskmagic/pieces-framework").NumberProperty<false>;
+    customMaxLength: import("@taskmagic/pieces-framework").NumberProperty<false>;
+    cleanUpSpaces: import("@taskmagic/pieces-framework").CheckboxProperty<false>;
+    truncationStrategy: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, false>;
+    useCache: import("@taskmagic/pieces-framework").CheckboxProperty<false>;
+    waitForModel: import("@taskmagic/pieces-framework").CheckboxProperty<false>;
+}>;