@memberjunction/ai 3.1.1 → 3.3.0

package/dist/generic/baseReranker.d.ts

@@ -0,0 +1,13 @@
+import { BaseModel } from './baseModel';
+import { RerankParams, RerankResponse, RerankResult } from './reranker.types';
+export declare abstract class BaseReranker extends BaseModel {
+    protected _modelName: string;
+    constructor(apiKey: string, modelName?: string);
+    get ModelName(): string;
+    Rerank(params: RerankParams): Promise<RerankResponse>;
+    protected abstract doRerank(params: RerankParams): Promise<RerankResult[]>;
+    protected createErrorResponse(message: string, startTime: number): RerankResponse;
+    protected sortByRelevance(results: RerankResult[]): RerankResult[];
+    protected filterByThreshold(results: RerankResult[], minScore: number): RerankResult[];
+}
+//# sourceMappingURL=baseReranker.d.ts.map

package/dist/generic/baseReranker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"baseReranker.d.ts","sourceRoot":"","sources":["../../src/generic/baseReranker.ts"],"names":[],"mappings":"AAeA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAqB9E,8BAAsB,YAAa,SAAQ,SAAS;IAChD,SAAS,CAAC,UAAU,EAAE,MAAM,CAAC;gBAOjB,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM;IAQ9C,IAAW,SAAS,IAAI,MAAM,CAE7B;IAeY,MAAM,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,cAAc,CAAC;IAoElE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAM1E,SAAS,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,cAAc;IAcjF,SAAS,CAAC,eAAe,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,YAAY,EAAE;IAQlE,SAAS,CAAC,iBAAiB,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,QAAQ,EAAE,MAAM,GAAG,YAAY,EAAE;CAGzF"}

package/dist/generic/baseReranker.js

@@ -0,0 +1,70 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseReranker = void 0;
+const baseModel_1 = require("./baseModel");
+class BaseReranker extends baseModel_1.BaseModel {
+    constructor(apiKey, modelName) {
+        super(apiKey);
+        this._modelName = modelName || '';
+    }
+    get ModelName() {
+        return this._modelName;
+    }
+    async Rerank(params) {
+        const startTime = Date.now();
+        try {
+            if (!params.query || params.query.trim().length === 0) {
+                return this.createErrorResponse('Query cannot be empty', startTime);
+            }
+            if (!params.documents || params.documents.length === 0) {
+                return {
+                    success: true,
+                    results: [],
+                    durationMs: Date.now() - startTime,
+                    modelName: this._modelName
+                };
+            }
+            for (let i = 0; i < params.documents.length; i++) {
+                const doc = params.documents[i];
+                if (!doc.id || !doc.text) {
+                    return this.createErrorResponse(`Document at index ${i} missing required field (id or text)`, startTime);
+                }
+            }
+            const results = await this.doRerank(params);
+            const topK = params.topK || results.length;
+            const limitedResults = results.slice(0, topK);
+            const rankedResults = limitedResults.map((r, idx) => ({
+                ...r,
+                rank: idx
+            }));
+            return {
+                success: true,
+                results: rankedResults,
+                durationMs: Date.now() - startTime,
+                modelName: this._modelName
+            };
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            console.error(`BaseReranker.Rerank error: ${message}`);
+            return this.createErrorResponse(message, startTime);
+        }
+    }
+    createErrorResponse(message, startTime) {
+        return {
+            success: false,
+            results: [],
+            errorMessage: message,
+            durationMs: Date.now() - startTime,
+            modelName: this._modelName
+        };
+    }
+    sortByRelevance(results) {
+        return [...results].sort((a, b) => b.relevanceScore - a.relevanceScore);
+    }
+    filterByThreshold(results, minScore) {
+        return results.filter(r => r.relevanceScore >= minScore);
+    }
+}
+exports.BaseReranker = BaseReranker;
+//# sourceMappingURL=baseReranker.js.map

package/dist/generic/baseReranker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"baseReranker.js","sourceRoot":"","sources":["../../src/generic/baseReranker.ts"],"names":[],"mappings":";;;AAeA,2CAAwC;AAsBxC,MAAsB,YAAa,SAAQ,qBAAS;IAQhD,YAAY,MAAc,EAAE,SAAkB;QAC1C,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,UAAU,GAAG,SAAS,IAAI,EAAE,CAAC;IACtC,CAAC;IAKD,IAAW,SAAS;QAChB,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;IAeM,KAAK,CAAC,MAAM,CAAC,MAAoB;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACpD,OAAO,IAAI,CAAC,mBAAmB,CAAC,uBAAuB,EAAE,SAAS,CAAC,CAAC;YACxE,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrD,OAAO;oBACH,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE,EAAE;oBACX,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;oBAClC,SAAS,EAAE,IAAI,CAAC,UAAU;iBAC7B,CAAC;YACN,CAAC;YAGD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/C,MAAM,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBAChC,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;oBACvB,OAAO,IAAI,CAAC,mBAAmB,CAC3B,qBAAqB,CAAC,sCAAsC,EAC5D,SAAS,CACZ,CAAC;gBACN,CAAC;YACL,CAAC;YAGD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAG5C,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC;YAC3C,MAAM,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YAG9C,MAAM,aAAa,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAClD,GAAG,CAAC;gBACJ,IAAI,EAAE,GAAG;aACZ,CAAC,CAAC,CAAC;YAEJ,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,aAAa;gBACtB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,SAAS,EAAE,IAAI,CAAC,UAAU;aAC7B,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,OAAO,CAAC,KAAK,CAAC,8BAA8B,OAAO,EAAE,CAAC,CAAC;YACvD,OAAO,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QACxD,CAAC;IACL,CAAC;IAqBS,mBAAmB,CAAC,OAAe,EAAE,SAAiB;QAC5D,OAAO;YACH,OAAO,EAAE,KAAK;YACd,OAAO,EAAE,EAAE;YACX,YAAY,EAAE,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,SAAS,EAAE,IAAI,CAAC,UAAU;SAC7B,CAAC;IACN,CAAC;IAMS,eAAe,CAAC,OAAuB;QAC7C,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC,cAAc,CAAC,CAAC;IAC5E,CAAC;IAMS,iBAAiB,CAAC,OAAuB,EAAE,QAAgB;QACjE,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc,IAAI,QAAQ,CAAC,CAAC;IAC7D,CAAC;CACJ;AApID,oCAoIC"}

package/dist/generic/reranker.types.d.ts

@@ -0,0 +1,26 @@
+export interface RerankDocument {
+    id: string;
+    text: string;
+    metadata?: Record<string, unknown>;
+    originalScore?: number;
+}
+export interface RerankResult {
+    id: string;
+    relevanceScore: number;
+    document: RerankDocument;
+    rank: number;
+}
+export interface RerankParams {
+    query: string;
+    documents: RerankDocument[];
+    topK?: number;
+    options?: Record<string, unknown>;
+}
+export interface RerankResponse {
+    success: boolean;
+    results: RerankResult[];
+    errorMessage?: string;
+    durationMs: number;
+    modelName?: string;
+}
+//# sourceMappingURL=reranker.types.d.ts.map

package/dist/generic/reranker.types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.types.d.ts","sourceRoot":"","sources":["../../src/generic/reranker.types.ts"],"names":[],"mappings":"AAaA,MAAM,WAAW,cAAc;IAI3B,EAAE,EAAE,MAAM,CAAC;IAKX,IAAI,EAAE,MAAM,CAAC;IAMb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAKnC,aAAa,CAAC,EAAE,MAAM,CAAC;CAC1B;AAKD,MAAM,WAAW,YAAY;IAIzB,EAAE,EAAE,MAAM,CAAC;IAKX,cAAc,EAAE,MAAM,CAAC;IAKvB,QAAQ,EAAE,cAAc,CAAC;IAKzB,IAAI,EAAE,MAAM,CAAC;CAChB;AAKD,MAAM,WAAW,YAAY;IAIzB,KAAK,EAAE,MAAM,CAAC;IAKd,SAAS,EAAE,cAAc,EAAE,CAAC;IAM5B,IAAI,CAAC,EAAE,MAAM,CAAC;IAKd,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAKD,MAAM,WAAW,cAAc;IAI3B,OAAO,EAAE,OAAO,CAAC;IAKjB,OAAO,EAAE,YAAY,EAAE,CAAC;IAKxB,YAAY,CAAC,EAAE,MAAM,CAAC;IAKtB,UAAU,EAAE,MAAM,CAAC;IAKnB,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB"}

package/dist/generic/reranker.types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=reranker.types.js.map

package/dist/generic/reranker.types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.types.js","sourceRoot":"","sources":["../../src/generic/reranker.types.ts"],"names":[],"mappings":""}

package/dist/index.d.ts

@@ -1,9 +1,11 @@
 export * from './generic/baseModel';
 export * from './generic/baseLLM';
 export * from './generic/baseImage';
+export * from './generic/baseReranker';
 export * from './generic/chat.types';
 export * from './generic/classify.types';
 export * from './generic/summarize.types';
+export * from './generic/reranker.types';
 export * from './generic/apiKeyDictionary';
 export * from './generic/embed.types';
 export * from './generic/baseEmbeddings';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAA;AACnC,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,0BAA0B,CAAC;AACzC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,4BAA4B,CAAC;AAC3C,cAAc,uBAAuB,CAAC;AACtC,cAAc,0BAA0B,CAAC;AACzC,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAA;AACnC,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,wBAAwB,CAAC;AACvC,cAAc,sBAAsB,CAAC;AACrC,cAAc,0BAA0B,CAAC;AACzC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,0BAA0B,CAAC;AACzC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,uBAAuB,CAAC;AACtC,cAAc,0BAA0B,CAAC;AACzC,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC"}

package/dist/index.js

@@ -17,9 +17,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./generic/baseModel"), exports);
 __exportStar(require("./generic/baseLLM"), exports);
 __exportStar(require("./generic/baseImage"), exports);
+__exportStar(require("./generic/baseReranker"), exports);
 __exportStar(require("./generic/chat.types"), exports);
 __exportStar(require("./generic/classify.types"), exports);
 __exportStar(require("./generic/summarize.types"), exports);
+__exportStar(require("./generic/reranker.types"), exports);
 __exportStar(require("./generic/apiKeyDictionary"), exports);
 __exportStar(require("./generic/embed.types"), exports);
 __exportStar(require("./generic/baseEmbeddings"), exports);

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,sDAAmC;AACnC,oDAAkC;AAClC,sDAAoC;AACpC,uDAAqC;AACrC,2DAAyC;AACzC,4DAA0C;AAC1C,6DAA2C;AAC3C,wDAAsC;AACtC,2DAAyC;AACzC,sDAAoC;AACpC,sDAAoC;AACpC,uDAAqC;AACrC,0DAAwC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,sDAAmC;AACnC,oDAAkC;AAClC,sDAAoC;AACpC,yDAAuC;AACvC,uDAAqC;AACrC,2DAAyC;AACzC,4DAA0C;AAC1C,2DAAyC;AACzC,6DAA2C;AAC3C,wDAAsC;AACtC,2DAAyC;AACzC,sDAAoC;AACpC,sDAAoC;AACpC,uDAAqC;AACrC,0DAAwC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memberjunction/ai",
-  "version": "3.1.1",
+  "version": "3.3.0",
   "description": "MemberJunction: AI - core components for abstracting LLMs and other AI model types that are usable anywhere without ANY other MJ dependencies past @memberjunction/global which itself has zero additional dependencies.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -15,7 +15,7 @@
   "author": "MemberJunction.com",
   "license": "ISC",
   "dependencies": {
-    "@memberjunction/global": "3.1.1",
+    "@memberjunction/global": "3.3.0",
     "dotenv": "^16.4.1",
     "rxjs": "^7.8.1"
   },