n8n-nodes-fasttext-language-detector 0.1.4 → 0.1.5

package/dist/nodes/FastTextLanguageDetector/FastTextLanguageDetector.node.js

@@ -201,25 +201,34 @@ class FastTextLanguageDetector {
                     try {
                         // Используем require для загрузки CommonJS модуля
                         // eslint-disable-next-line @typescript-eslint/no-var-requires
-                        const fastText = require('fasttext.js');
-                        resolve(fastText);
+                        const FastText = require('fasttext.js');
+                        resolve(FastText);
                     }
                     catch (error) {
                         reject(error);
                     }
                 });
-                // Обрабатываем разные варианты экспорта (CommonJS/ESM)
-                // fasttext.js может экспортировать через default или напрямую
-                const FastTextClassifier = fastTextModule.Classifier ||
-                    (fastTextModule.default && fastTextModule.default.Classifier) ||
-                    fastTextModule.default;
-                if (!FastTextClassifier || typeof FastTextClassifier !== 'function') {
+                // fasttext.js экспортирует класс FastText напрямую
+                const FastTextClass = fastTextModule ||
+                    fastTextModule.default ||
+                    (fastTextModule.default && fastTextModule.default.default);
+                if (!FastTextClass || typeof FastTextClass !== 'function') {
                     const availableExports = Object.keys(fastTextModule).join(', ');
-                    throw new n8n_workflow_1.ApplicationError(`Failed to load fastText module: Classifier not found. Available exports: ${availableExports}`, { level: 'error' });
+                    throw new n8n_workflow_1.ApplicationError(`Failed to load fastText module: FastText class not found. Available exports: ${availableExports}`, { level: 'error' });
                 }
-                const loadedModel = new FastTextClassifier(modelPath);
-                model = loadedModel;
-                FastTextLanguageDetector.modelCache.set(modelPath, loadedModel);
+                // Создаем экземпляр FastText с путем к модели
+                const fastTextInstance = new FastTextClass({
+                    loadModel: modelPath,
+                    predict: {
+                        mostlikely: topK,
+                        verbosity: 0,
+                        normalize: false
+                    }
+                });
+                // Загружаем модель
+                await fastTextInstance.load();
+                model = fastTextInstance;
+                FastTextLanguageDetector.modelCache.set(modelPath, fastTextInstance);
             }
         }
         catch (error) {
@@ -265,19 +274,48 @@ class FastTextLanguageDetector {
                     throw new n8n_workflow_1.ApplicationError('Model not initialized', { level: 'error' });
                 }
                 // Определяем язык
-                const predictions = await model.predict(processedText, topK);
+                // fasttext.js возвращает массив массивов [label, probability] или массив объектов
+                const predictionsRaw = await model.predict(processedText);
+                if (!Array.isArray(predictionsRaw) || predictionsRaw.length === 0) {
+                    throw new n8n_workflow_1.ApplicationError('No predictions returned from fastText model', { level: 'error' });
+                }
+                // Преобразуем формат ответа fasttext.js в наш формат
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                const predictions = predictionsRaw.map((pred) => {
+                    // fasttext.js возвращает либо [label, value], либо объект
+                    let label;
+                    let probability;
+                    if (Array.isArray(pred)) {
+                        label = String(pred[0] || '');
+                        probability = Number(pred[1] || 0);
+                    }
+                    else if (pred && typeof pred === 'object') {
+                        label = String(pred.label || pred[0] || '');
+                        probability = Number(pred.value || pred.probability || pred[1] || 0);
+                    }
+                    else {
+                        label = String(pred || '');
+                        probability = 0;
+                    }
+                    // Убираем префикс __label__ если есть
+                    const cleanLabel = label.replace(/^__label__/, '');
+                    return {
+                        label: cleanLabel,
+                        probability: probability
+                    };
+                });
                 // Фильтруем по порогу уверенности
                 const filteredPredictions = predictions.filter((pred) => pred.probability >= confidenceThreshold);
                 if (filteredPredictions.length === 0) {
                     // Если нет предсказаний выше порога, возвращаем лучшее
                     const result = {
                         ...json,
-                        [outputField]: predictions[0]?.label?.replace('__label__', '') || 'unknown',
+                        [outputField]: predictions[0]?.label || 'unknown',
                     };
                     if (includeConfidence) {
                         result[`${outputField}Confidence`] = predictions[0]?.probability || 0;
                         result[`${outputField}All`] = predictions.map((p) => ({
-                            language: p.label?.replace('__label__', '') || 'unknown',
+                            language: p.label || 'unknown',
                             confidence: p.probability || 0,
                         }));
                     }
@@ -288,7 +326,7 @@ class FastTextLanguageDetector {
                 const topPrediction = filteredPredictions[0];
                 const result = {
                     ...json,
-                    [outputField]: topPrediction.label?.replace('__label__', '') || 'unknown',
+                    [outputField]: topPrediction.label || 'unknown',
                 };
                 if (includeConfidence) {
                     result[`${outputField}Confidence`] = topPrediction.probability || 0;

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "n8n-nodes-fasttext-language-detector",
-    "version": "0.1.4",
+    "version": "0.1.5",
     "description": "n8n node for language detection using fastText",
     "author": {
         "name": "Your Name",