entity-predictor 1.2.1 → 1.2.3

package/README.md

@@ -63,7 +63,8 @@ Output:
 {
   entity: "STATE BANK OF INDIA",
   confidence: 1,
-  confidenceLevel: "Trustable"
+  confidenceLevel: "Trustable",
+  input: "sbi"
 }
 */
 ```
@@ -78,7 +79,7 @@ console.log(result);
 Output:
 {
   entity: "ICICI BANK",
-  confidence: 0.714...,
+  confidence: 0.71,
   confidenceLevel: "Moderate Confidence"
 }
 */
@@ -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 or `{ entity: "UNKNOWN", ... }`.
+- **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.1",
+  "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,
     };
   }
 
@@ -146,11 +147,12 @@ export class EntityPredictor {
     const matches = findBestMatch(normalizedInput, this.searchCandidates);
 
     // Map all ratings to our format and sort
-    const sortedMatches = matches.ratings
+    const sortedMatches = (matches?.ratings || [])
       .map((rating, index) => ({
         entity: this.candidateToEntity[index],
-        confidence: rating.rating,
-        confidenceLevel: this._getConfidenceLevel(rating.rating),
+        confidence: Number(rating?.rating?.toFixed(2)),
+        confidenceLevel: this._getConfidenceLevel(rating?.rating),
+        input,
       }))
       .filter((m) => m.confidence >= threshold)
       .sort((a, b) => b.confidence - a.confidence);