@nest-yalc-2/sentinel 2.0.0

package/src/sentinel.interceptor.ts

@@ -0,0 +1,38 @@
+/**
+ * # NestJS Ferrox-Node RAG Groundedness & Response Interceptor (`sentinel.interceptor.ts`)
+ * Evaluates LLM generated response groundedness against retrieved context
+ */
+
+import { Injectable, NestInterceptor, ExecutionContext, CallHandler, BadGatewayException } from '@nestjs/common';
+import { Observable } from 'rxjs';
+import { map } from 'rxjs/operators';
+import { RagHallucinationGroundednessEngine } from './algorithms/rag-groundedness.js';
+
+@Injectable()
+export class RagGroundednessInterceptor implements NestInterceptor {
+  constructor(private readonly minGroundednessScore: number = 0.70) {}
+
+  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
+    const request = context.switchToHttp().getRequest();
+    const retrievedContexts: string[] = request.ragContexts || [];
+
+    return next.handle().pipe(
+      map((response) => {
+        if (typeof response === 'string' && retrievedContexts.length > 0) {
+          const assessment = RagHallucinationGroundednessEngine.evaluateGroundedness(
+            response,
+            retrievedContexts,
+            this.minGroundednessScore
+          );
+
+          if (!assessment.isGrounded) {
+            throw new BadGatewayException(
+              `[FerroxSentinel] LLM response failed groundedness check (Score: ${assessment.groundednessScore.toFixed(2)} < ${this.minGroundednessScore})`
+            );
+          }
+        }
+        return response;
+      })
+    );
+  }
+}

package/src/sentinel.module.d.ts

@@ -0,0 +1,9 @@
+import { DynamicModule } from '@nestjs/common';
+export interface SentinelModuleOptions {
+    enablePromptSanitizer?: boolean;
+    enableEntropyCheck?: boolean;
+    minGroundednessScore?: number;
+}
+export declare class FerroxSentinelModule {
+    static register(options?: SentinelModuleOptions): DynamicModule;
+}

package/src/sentinel.module.js

@@ -0,0 +1,28 @@
+"use strict";
+var FerroxSentinelModule_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FerroxSentinelModule = void 0;
+const tslib_1 = require("tslib");
+const common_1 = require("@nestjs/common");
+const sentinel_guard_js_1 = require("./sentinel.guard.js");
+const sentinel_interceptor_js_1 = require("./sentinel.interceptor.js");
+let FerroxSentinelModule = FerroxSentinelModule_1 = class FerroxSentinelModule {
+    static register(options = {}) {
+        return {
+            module: FerroxSentinelModule_1,
+            providers: [
+                sentinel_guard_js_1.FerroxSentinelGuard,
+                {
+                    provide: sentinel_interceptor_js_1.RagGroundednessInterceptor,
+                    useValue: new sentinel_interceptor_js_1.RagGroundednessInterceptor(options.minGroundednessScore ?? 0.70),
+                },
+            ],
+            exports: [sentinel_guard_js_1.FerroxSentinelGuard, sentinel_interceptor_js_1.RagGroundednessInterceptor],
+        };
+    }
+};
+exports.FerroxSentinelModule = FerroxSentinelModule;
+exports.FerroxSentinelModule = FerroxSentinelModule = FerroxSentinelModule_1 = tslib_1.__decorate([
+    (0, common_1.Module)({})
+], FerroxSentinelModule);
+//# sourceMappingURL=sentinel.module.js.map

package/src/sentinel.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sentinel.module.js","sourceRoot":"","sources":["sentinel.module.ts"],"names":[],"mappings":";;;;;AAIA,2CAAuD;AACvD,2DAA0D;AAC1D,uEAAuE;AAShE,IAAM,oBAAoB,4BAA1B,MAAM,oBAAoB;IAC/B,MAAM,CAAC,QAAQ,CAAC,UAAiC,EAAE;QACjD,OAAO;YACL,MAAM,EAAE,sBAAoB;YAC5B,SAAS,EAAE;gBACT,uCAAmB;gBACnB;oBACE,OAAO,EAAE,oDAA0B;oBACnC,QAAQ,EAAE,IAAI,oDAA0B,CAAC,OAAO,CAAC,oBAAoB,IAAI,IAAI,CAAC;iBAC/E;aACF;YACD,OAAO,EAAE,CAAC,uCAAmB,EAAE,oDAA0B,CAAC;SAC3D,CAAC;IACJ,CAAC;CACF,CAAA;AAdY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,eAAM,EAAC,EAAE,CAAC;GACE,oBAAoB,CAchC"}

package/src/sentinel.module.ts

@@ -0,0 +1,30 @@
+/**
+ * # NestJS Ferrox-Node Sentinel Module (`sentinel.module.ts`)
+ */
+
+import { Module, DynamicModule } from '@nestjs/common';
+import { FerroxSentinelGuard } from './sentinel.guard.js';
+import { RagGroundednessInterceptor } from './sentinel.interceptor.js';
+
+export interface SentinelModuleOptions {
+  enablePromptSanitizer?: boolean;
+  enableEntropyCheck?: boolean;
+  minGroundednessScore?: number;
+}
+
+@Module({})
+export class FerroxSentinelModule {
+  static register(options: SentinelModuleOptions = {}): DynamicModule {
+    return {
+      module: FerroxSentinelModule,
+      providers: [
+        FerroxSentinelGuard,
+        {
+          provide: RagGroundednessInterceptor,
+          useValue: new RagGroundednessInterceptor(options.minGroundednessScore ?? 0.70),
+        },
+      ],
+      exports: [FerroxSentinelGuard, RagGroundednessInterceptor],
+    };
+  }
+}