entity-predictor 1.2.2 → 1.2.3

package/README.md

@@ -63,7 +63,8 @@ Output:
 {
   entity: "STATE BANK OF INDIA",
   confidence: 1,
-  confidenceLevel: "Trustable"
+  confidenceLevel: "Trustable",
+  input: "sbi"
 }
 */
 ```
@@ -141,7 +142,7 @@ predictor.addEntity("PUNJAB NATIONAL BANK", ["PNB"]);
 
 - `input`: String to search for.
 - `threshold`: (Optional) Minimum confidence score (default `0.6`).
-- **Returns**: Best match object, `{ entity: "UNKNOWN", ... }` if no match found, or `null` if input is invalid.
+- **Returns**: Best match object `{ entity: string, confidence: number, input: string, ... }`, `{ entity: "UNKNOWN", input: string, ... }` if no match found, or `null` if input is invalid.
 
 ### `predictTop(input, limit, threshold)`
 

package/index.d.ts

@@ -13,6 +13,7 @@ export interface PredictionResult {
     entity: string;
     confidence: number;
     confidenceLevel: "Trustable" | "High Confidence" | "Moderate Confidence" | "Low Confidence";
+    input: string;
 }
 
 export class EntityPredictor {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "entity-predictor",
-  "version": "1.2.2",
+  "version": "1.2.3",
   "description": "Lightweight, Zero Dependency Node.js library for entity name prediction and normalization.",
   "types": "index.d.ts",
   "type": "module",

package/src/predictor.js

@@ -127,6 +127,7 @@ export class EntityPredictor {
       entity: "UNKNOWN",
       confidence: 0,
       confidenceLevel: "Low Confidence",
+      input,
     };
   }
 
@@ -151,6 +152,7 @@ export class EntityPredictor {
         entity: this.candidateToEntity[index],
         confidence: Number(rating?.rating?.toFixed(2)),
         confidenceLevel: this._getConfidenceLevel(rating?.rating),
+        input,
       }))
       .filter((m) => m.confidence >= threshold)
       .sort((a, b) => b.confidence - a.confidence);