@joliegg/moderation 0.3.1 → 0.3.3

package/dist/index.d.ts

@@ -29,7 +29,7 @@ declare class ModerationClient {
      * Returns a list of moderation categories detected on an image
      *
      * @param {string} url
-     * @param {number} [minimumConfidence = 95]  The minimum confidence required for a category to be considered
+     * @param {number} [minimumConfidence = 50]  The minimum confidence required for a category to be considered
      *
      *
      * @returns {Promise<ModerationResult[]>} The list of results that were detected with the minimum confidence specified

package/dist/index.js

@@ -57,9 +57,10 @@ class ModerationClient {
             const normalizedText = text.toLowerCase();
             const matches = this.banList.filter(w => normalizedText.indexOf(w) > -1);
             if (matches.length > 0) {
+                const words = normalizedText.split(' ');
                 categories.push({
                     category: 'BAN_LIST',
-                    confidence: matches.length,
+                    confidence: (matches.length / words.length) * 100,
                 });
             }
         }
@@ -87,12 +88,12 @@ class ModerationClient {
      * Returns a list of moderation categories detected on an image
      *
      * @param {string} url
-     * @param {number} [minimumConfidence = 95]  The minimum confidence required for a category to be considered
+     * @param {number} [minimumConfidence = 50]  The minimum confidence required for a category to be considered
      *
      *
      * @returns {Promise<ModerationResult[]>} The list of results that were detected with the minimum confidence specified
      */
-    async moderateImage(url, minimumConfidence = 95) {
+    async moderateImage(url, minimumConfidence = 50) {
         if (typeof this.rekognitionClient === 'undefined') {
             return { source: url, moderation: [] };
         }
@@ -115,6 +116,7 @@ class ModerationClient {
             },
             MinConfidence: minimumConfidence
         });
+        console.log(ModerationLabels);
         if (Array.isArray(ModerationLabels)) {
             const moderation = ModerationLabels.map(l => ({
                 category: l.Name ?? 'Unknown',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@joliegg/moderation",
-  "version": "0.3.1",
+  "version": "0.3.3",
   "description": "A set of tools for chat moderation",
   "author": "Diana Islas Ocampo",
   "main": "dist/index.js",
@@ -12,7 +12,7 @@
     "test": "node test/index.js"
   },
   "engines": {
-    "node": "20.x"
+    "node": ">=20.x"
   },
   "devDependencies": {
     "@babel/eslint-parser": "^7.24.8",

package/src/index.ts

@@ -73,9 +73,11 @@ class ModerationClient {
       const matches = this.banList.filter(w => normalizedText.indexOf(w) > -1);
 
       if (matches.length > 0) {
+        const words = normalizedText.split(' ');
+
         categories.push({
           category: 'BAN_LIST',
-          confidence: matches.length,
+          confidence: (matches.length / words.length) * 100,
         });
       }
     }
@@ -109,12 +111,12 @@ class ModerationClient {
    * Returns a list of moderation categories detected on an image
    *
    * @param {string} url
-   * @param {number} [minimumConfidence = 95]  The minimum confidence required for a category to be considered
+   * @param {number} [minimumConfidence = 50]  The minimum confidence required for a category to be considered
    *
    *
    * @returns {Promise<ModerationResult[]>} The list of results that were detected with the minimum confidence specified
    */
-  async moderateImage (url: string, minimumConfidence: number = 95): Promise<ModerationResult> {
+  async moderateImage (url: string, minimumConfidence: number = 50): Promise<ModerationResult> {
     if (typeof this.rekognitionClient === 'undefined') {
       return { source: url, moderation: [] };
     }
@@ -140,6 +142,8 @@ class ModerationClient {
       MinConfidence: minimumConfidence
     });
 
+    console.log(ModerationLabels);
+
     if (Array.isArray(ModerationLabels)) {
       const moderation = ModerationLabels.map(l => ({
         category: l.Name ?? 'Unknown',