@retrivora-ai/rag-engine 2.2.5 → 2.2.7

package/src/core/LicenseValidator.ts

@@ -0,0 +1,146 @@
+import { SDK_VERSION } from '../version';
+import { SDKVersionUnsupportedError, LicenseValidationError } from '../exceptions';
+
+export interface LicenseValidationRequest {
+  licenseKey: string;
+  projectId?: string;
+  sdkVersion?: string;
+  sdk?: string;
+  platform?: string;
+  retrivoraApiBase?: string;
+  headers?: Record<string, string>;
+}
+
+export interface LicenseValidationResponse {
+  valid: boolean;
+  licenseStatus: 'ACTIVE' | 'SUSPENDED' | 'TERMINATED' | 'EXPIRED' | 'REVOKED';
+  minimumSupportedVersion: string;
+  latestVersion: string;
+  forceUpgrade: boolean;
+  expiresAt: string | null;
+  message: string;
+}
+
+interface CachedSuccessValidation {
+  response: LicenseValidationResponse;
+  timestamp: number;
+}
+
+export class LicenseValidator {
+  private static instance: LicenseValidator;
+  private cache: Map<string, CachedSuccessValidation> = new Map();
+  private static readonly SUCCESS_CACHE_TTL_MS = 5 * 60 * 1000; // 5-minute TTL for successful validation only
+
+  private constructor() {}
+
+  public static getInstance(): LicenseValidator {
+    if (!LicenseValidator.instance) {
+      LicenseValidator.instance = new LicenseValidator();
+    }
+    return LicenseValidator.instance;
+  }
+
+  /**
+   * Invalidate local validation cache for a license key
+   */
+  public purgeCache(licenseKey?: string): void {
+    if (licenseKey) {
+      this.cache.delete(licenseKey);
+    } else {
+      this.cache.clear();
+    }
+  }
+
+  /**
+   * Validate license status and SDK version against Retrivora License Service
+   */
+  public async validate(request: LicenseValidationRequest): Promise<LicenseValidationResponse> {
+    const {
+      licenseKey,
+      projectId,
+      sdkVersion = SDK_VERSION,
+      sdk = '@retrivora-ai/rag-engine',
+      platform = typeof window !== 'undefined' ? 'browser' : 'node',
+      retrivoraApiBase,
+      headers: customHeaders,
+    } = request;
+
+    if (!licenseKey) {
+      this.purgeCache(licenseKey);
+      throw new LicenseValidationError('Missing RETRIVORA_LICENSE_KEY in request.');
+    }
+
+    // 1. Check successful 5-minute cache
+    const cached = this.cache.get(licenseKey);
+    if (cached) {
+      const isFresh = Date.now() - cached.timestamp < LicenseValidator.SUCCESS_CACHE_TTL_MS;
+      if (isFresh && cached.response.valid && cached.response.licenseStatus === 'ACTIVE' && !cached.response.forceUpgrade) {
+        return cached.response;
+      } else {
+        this.cache.delete(licenseKey);
+      }
+    }
+
+    // 2. Fetch live validation from Retrivora backend API
+    const base = retrivoraApiBase || (typeof process !== 'undefined' ? process.env.NEXT_PUBLIC_RETRIVORA_API_BASE || process.env.RETRIVORA_API_BASE : '') || '/api/retrivora';
+    const validateUrl = `${base.replace(/\/$/, '')}/v1/license/validate`;
+
+    let res: Response;
+    try {
+      res = await fetch(validateUrl, {
+        method: 'POST',
+        cache: 'no-store',
+        headers: {
+          'Content-Type': 'application/json',
+          'x-license-key': licenseKey,
+          'x-sdk-version': sdkVersion,
+          ...(customHeaders || {}),
+        },
+        body: JSON.stringify({
+          licenseKey,
+          projectId,
+          sdkVersion,
+          sdk,
+          platform,
+        }),
+      });
+    } catch (err: any) {
+      // Fallback: If network error occurs, do not cache and throw error
+      this.purgeCache(licenseKey);
+      throw new LicenseValidationError(`License validation request failed: ${err?.message || 'Network error'}`);
+    }
+
+    const data: LicenseValidationResponse = await res.json().catch(() => ({
+      valid: false,
+      licenseStatus: 'REVOKED' as const,
+      minimumSupportedVersion: '2.1.0',
+      latestVersion: SDK_VERSION,
+      forceUpgrade: false,
+      expiresAt: null,
+      message: 'Failed to parse license validation response.',
+    }));
+
+    // 3. Negative validation responses immediately purge any existing cache
+    if (!res.ok || !data.valid || data.licenseStatus !== 'ACTIVE' || data.forceUpgrade) {
+      this.purgeCache(licenseKey);
+
+      if (data.forceUpgrade || res.status === 426) {
+        throw new SDKVersionUnsupportedError(
+          data.message || `This SDK version (${sdkVersion}) is no longer supported. Upgrade to version ${data.latestVersion || '2.2.6'} or later.`,
+          data
+        );
+      }
+
+      const errMsg = data.message || `License validation failed with status: ${data.licenseStatus}. Access denied.`;
+      throw new LicenseValidationError(errMsg, data);
+    }
+
+    // 4. Cache SUCCESSFUL validation only (TTL = 5 minutes)
+    this.cache.set(licenseKey, {
+      response: data,
+      timestamp: Date.now(),
+    });
+
+    return data;
+  }
+}

package/src/exceptions/index.ts

@@ -8,7 +8,9 @@ export type RetrivoraErrorCode =
   | 'RETRIEVAL_FAILED'
   | 'RATE_LIMITED'
   | 'CONFIGURATION_ERROR'
-  | 'AUTHENTICATION_ERROR';
+  | 'AUTHENTICATION_ERROR'
+  | 'SDK_VERSION_UNSUPPORTED'
+  | 'LICENSE_VALIDATION_ERROR';
 
 export class RetrivoraError extends Error {
   readonly code: RetrivoraErrorCode;
@@ -58,6 +60,18 @@ export class AuthenticationException extends RetrivoraError {
   }
 }
 
+export class SDKVersionUnsupportedError extends RetrivoraError {
+  constructor(message = 'This SDK version is no longer supported.', details?: unknown) {
+    super(message, 'SDK_VERSION_UNSUPPORTED', details);
+  }
+}
+
+export class LicenseValidationError extends RetrivoraError {
+  constructor(message = 'License validation failed.', details?: unknown) {
+    super(message, 'LICENSE_VALIDATION_ERROR', details);
+  }
+}
+
 /**
  * Wraps any unknown error into an appropriate RetrivoraError subclass.
  */
@@ -80,14 +94,20 @@ export function wrapError(
     return new RateLimitException(message, err);
   }
 
-  // 2. Authentication Recognition
+  // 2. Authentication / License Recognition
   if (
     status === 401 ||
     status === 403 ||
-    /unauthorized|auth|api[- ]?key/i.test(message) ||
-    code === 'INVALID_API_KEY'
+    /unauthorized|auth|license|revoked|terminated|api[- ]?key/i.test(message) ||
+    code === 'INVALID_API_KEY' ||
+    code === 'LICENSE_DELETED' ||
+    code === 'LICENSE_INACTIVE'
   ) {
-    return new AuthenticationException(message, err);
+    const isLicenseError = /403|license|revoked|terminated/i.test(message) || status === 403;
+    const cleanMsg = isLicenseError
+      ? 'Your Retrivora license key has been terminated, suspended, or revoked. Access denied.'
+      : message;
+    return new AuthenticationException(cleanMsg, err);
   }
 
   // Fallback Mapping based on defaultCode

package/src/handlers/index.ts

@@ -492,8 +492,18 @@ export function createStreamHandler(
           }
         } catch (streamError) {
           if (isActive) {
-            const errorMessage =
+            let errorMessage =
               streamError instanceof Error ? streamError.message : String(streamError);
+
+            if (
+              errorMessage.includes('403') ||
+              errorMessage.toLowerCase().includes('license') ||
+              errorMessage.toLowerCase().includes('revoked') ||
+              errorMessage.toLowerCase().includes('terminated')
+            ) {
+              errorMessage = 'Your Retrivora license key has been terminated, suspended, or revoked. Access denied.';
+            }
+
             console.error('[createStreamHandler] Stream error:', streamError);
             reportTelemetry(req, plugin, 'QUERY_GENERATION', 'error', errorMessage);
             try {

package/src/hooks/useRagChat.ts

@@ -154,7 +154,15 @@ export function useRagChat(
 
         if (!response.ok) {
           const errorData = await response.json().catch(() => ({}));
-          throw new Error(errorData.error || `Server returned ${response.status}`);
+          if (
+            response.status === 403 ||
+            response.status === 401 ||
+            (typeof errorData.error === 'string' && errorData.error.toLowerCase().includes('license')) ||
+            errorData.error?.type === 'license_revoked'
+          ) {
+            throw new Error('Your Retrivora license is no longer valid. Please contact your administrator or obtain a valid license key.');
+          }
+          throw new Error(errorData.error?.message || errorData.error || `Server returned ${response.status}`);
         }
 
         if (!response.body) {
@@ -284,7 +292,15 @@ export function useRagChat(
           console.log('[useRagChat] Streaming stopped by user.');
           return;
         }
-        const msg = err instanceof Error ? err.message : 'Something went wrong. Please try again.';
+        let msg = err instanceof Error ? err.message : 'Something went wrong. Please try again.';
+        if (
+          msg.includes('403') ||
+          msg.toLowerCase().includes('license') ||
+          msg.toLowerCase().includes('revoked') ||
+          msg.toLowerCase().includes('terminated')
+        ) {
+          msg = 'Your Retrivora license key has been terminated, suspended, or revoked. Access denied.';
+        }
         setError(msg);
         onError?.(msg);
       } finally {

package/src/index.css

@@ -96,6 +96,9 @@
     --color-rose-400: oklch(71.2% 0.194 13.428);
     --color-rose-500: oklch(64.5% 0.246 16.439);
     --color-rose-600: oklch(58.6% 0.253 17.585);
+    --color-rose-800: oklch(45.5% 0.188 13.697);
+    --color-rose-900: oklch(41% 0.159 10.272);
+    --color-rose-950: oklch(27.1% 0.105 12.094);
     --color-slate-50: oklch(98.4% 0.003 247.858);
     --color-slate-100: oklch(96.8% 0.007 247.896);
     --color-slate-200: oklch(92.9% 0.013 255.508);
@@ -370,6 +373,9 @@
   .bottom-20 {
     bottom: calc(var(--spacing) * 20);
   }
+  .bottom-24 {
+    bottom: calc(var(--spacing) * 24);
+  }
   .-left-4 {
     left: calc(var(--spacing) * -4);
   }
@@ -451,6 +457,9 @@
   .my-6 {
     margin-block: calc(var(--spacing) * 6);
   }
+  .my-auto {
+    margin-block: auto;
+  }
   .mt-0\.5 {
     margin-top: calc(var(--spacing) * 0.5);
   }
@@ -1121,6 +1130,12 @@
   .border-amber-200 {
     border-color: var(--color-amber-200);
   }
+  .border-amber-200\/80 {
+    border-color: color-mix(in srgb, oklch(92.4% 0.12 95.746) 80%, transparent);
+    @supports (color: color-mix(in lab, red, red)) {
+      border-color: color-mix(in oklab, var(--color-amber-200) 80%, transparent);
+    }
+  }
   .border-emerald-100 {
     border-color: var(--color-emerald-100);
   }
@@ -1157,6 +1172,12 @@
   .border-rose-200 {
     border-color: var(--color-rose-200);
   }
+  .border-rose-200\/60 {
+    border-color: color-mix(in srgb, oklch(89.2% 0.058 10.001) 60%, transparent);
+    @supports (color: color-mix(in lab, red, red)) {
+      border-color: color-mix(in oklab, var(--color-rose-200) 60%, transparent);
+    }
+  }
   .border-slate-100 {
     border-color: var(--color-slate-100);
   }
@@ -1217,6 +1238,12 @@
   .bg-amber-50 {
     background-color: var(--color-amber-50);
   }
+  .bg-amber-50\/70 {
+    background-color: color-mix(in srgb, oklch(98.7% 0.022 95.277) 70%, transparent);
+    @supports (color: color-mix(in lab, red, red)) {
+      background-color: color-mix(in oklab, var(--color-amber-50) 70%, transparent);
+    }
+  }
   .bg-amber-100 {
     background-color: var(--color-amber-100);
   }
@@ -1322,6 +1349,15 @@
   .bg-rose-50 {
     background-color: var(--color-rose-50);
   }
+  .bg-rose-50\/50 {
+    background-color: color-mix(in srgb, oklch(96.9% 0.015 12.422) 50%, transparent);
+    @supports (color: color-mix(in lab, red, red)) {
+      background-color: color-mix(in oklab, var(--color-rose-50) 50%, transparent);
+    }
+  }
+  .bg-rose-100 {
+    background-color: var(--color-rose-100);
+  }
   .bg-rose-500 {
     background-color: var(--color-rose-500);
   }
@@ -1409,6 +1445,12 @@
       background-color: color-mix(in oklab, var(--color-white) 90%, transparent);
     }
   }
+  .bg-white\/95 {
+    background-color: color-mix(in srgb, #fff 95%, transparent);
+    @supports (color: color-mix(in lab, red, red)) {
+      background-color: color-mix(in oklab, var(--color-white) 95%, transparent);
+    }
+  }
   .bg-gradient-to-br {
     --tw-gradient-position: to bottom right in oklab;
     background-image: linear-gradient(var(--tw-gradient-stops));
@@ -1619,6 +1661,9 @@
   .whitespace-pre-wrap {
     white-space: pre-wrap;
   }
+  .text-amber-300 {
+    color: var(--color-amber-300);
+  }
   .text-amber-400 {
     color: var(--color-amber-400);
   }
@@ -1950,6 +1995,10 @@
     --tw-outline-style: none;
     outline-style: none;
   }
+  .select-all {
+    -webkit-user-select: all;
+    user-select: all;
+  }
   .select-none {
     -webkit-user-select: none;
     user-select: none;
@@ -2391,6 +2440,14 @@
       }
     }
   }
+  .dark\:border-amber-800\/40 {
+    @media (prefers-color-scheme: dark) {
+      border-color: color-mix(in srgb, oklch(47.3% 0.137 46.201) 40%, transparent);
+      @supports (color: color-mix(in lab, red, red)) {
+        border-color: color-mix(in oklab, var(--color-amber-800) 40%, transparent);
+      }
+    }
+  }
   .dark\:border-emerald-400\/20 {
     @media (prefers-color-scheme: dark) {
       border-color: color-mix(in srgb, oklch(76.5% 0.177 163.223) 20%, transparent);
@@ -2463,6 +2520,22 @@
       }
     }
   }
+  .dark\:border-rose-800\/40 {
+    @media (prefers-color-scheme: dark) {
+      border-color: color-mix(in srgb, oklch(45.5% 0.188 13.697) 40%, transparent);
+      @supports (color: color-mix(in lab, red, red)) {
+        border-color: color-mix(in oklab, var(--color-rose-800) 40%, transparent);
+      }
+    }
+  }
+  .dark\:border-rose-900\/60 {
+    @media (prefers-color-scheme: dark) {
+      border-color: color-mix(in srgb, oklch(41% 0.159 10.272) 60%, transparent);
+      @supports (color: color-mix(in lab, red, red)) {
+        border-color: color-mix(in oklab, var(--color-rose-900) 60%, transparent);
+      }
+    }
+  }
   .dark\:border-slate-700 {
     @media (prefers-color-scheme: dark) {
       border-color: var(--color-slate-700);
@@ -2515,6 +2588,11 @@
       background-color: color-mix(in oklab, #0f0f1a 90%, transparent);
     }
   }
+  .dark\:bg-\[\#0f0f1a\]\/95 {
+    @media (prefers-color-scheme: dark) {
+      background-color: color-mix(in oklab, #0f0f1a 95%, transparent);
+    }
+  }
   .dark\:bg-\[\#080811\] {
     @media (prefers-color-scheme: dark) {
       background-color: #080811;
@@ -2560,6 +2638,22 @@
       }
     }
   }
+  .dark\:bg-amber-900\/40 {
+    @media (prefers-color-scheme: dark) {
+      background-color: color-mix(in srgb, oklch(41.4% 0.112 45.904) 40%, transparent);
+      @supports (color: color-mix(in lab, red, red)) {
+        background-color: color-mix(in oklab, var(--color-amber-900) 40%, transparent);
+      }
+    }
+  }
+  .dark\:bg-amber-950\/20 {
+    @media (prefers-color-scheme: dark) {
+      background-color: color-mix(in srgb, oklch(27.9% 0.077 45.635) 20%, transparent);
+      @supports (color: color-mix(in lab, red, red)) {
+        background-color: color-mix(in oklab, var(--color-amber-950) 20%, transparent);
+      }
+    }
+  }
   .dark\:bg-amber-950\/40 {
     @media (prefers-color-scheme: dark) {
       background-color: color-mix(in srgb, oklch(27.9% 0.077 45.635) 40%, transparent);
@@ -2826,6 +2920,22 @@
       }
     }
   }
+  .dark\:bg-rose-900\/40 {
+    @media (prefers-color-scheme: dark) {
+      background-color: color-mix(in srgb, oklch(41% 0.159 10.272) 40%, transparent);
+      @supports (color: color-mix(in lab, red, red)) {
+        background-color: color-mix(in oklab, var(--color-rose-900) 40%, transparent);
+      }
+    }
+  }
+  .dark\:bg-rose-950\/20 {
+    @media (prefers-color-scheme: dark) {
+      background-color: color-mix(in srgb, oklch(27.1% 0.105 12.094) 20%, transparent);
+      @supports (color: color-mix(in lab, red, red)) {
+        background-color: color-mix(in oklab, var(--color-rose-950) 20%, transparent);
+      }
+    }
+  }
   .dark\:bg-slate-700 {
     @media (prefers-color-scheme: dark) {
       background-color: var(--color-slate-700);
@@ -3372,6 +3482,15 @@
       }
     }
   }
+  .dark\:hover\:text-white {
+    @media (prefers-color-scheme: dark) {
+      &:hover {
+        @media (hover: hover) {
+          color: var(--color-white);
+        }
+      }
+    }
+  }
   .dark\:hover\:text-white\/60 {
     @media (prefers-color-scheme: dark) {
       &:hover {

package/src/index.ts

@@ -63,6 +63,8 @@ export type { ILLMProvider, EmbedOptions } from './llm/ILLMProvider';
 export type { Chunk, ChunkOptions } from './rag/DocumentChunker';
 export type { Retrivora } from './core/Retrivora';
 export type { RetrivoraErrorCode } from './exceptions';
+export { LicenseValidator } from './core/LicenseValidator';
+export type { LicenseValidationRequest, LicenseValidationResponse } from './core/LicenseValidator';
 export {
   RetrivoraError,
   ProviderNotFoundException,
@@ -71,6 +73,8 @@ export {
   RateLimitException,
   ConfigurationException,
   AuthenticationException,
+  SDKVersionUnsupportedError,
+  LicenseValidationError,
 } from './exceptions';
 export type {
   VectorMatch,

package/src/server.ts

@@ -90,6 +90,10 @@ export {
   sseErrorFrame,
 } from './handlers';
 
+// ── License Validator ─────────────────────────────────────────
+export { LicenseValidator } from './core/LicenseValidator';
+export type { LicenseValidationRequest, LicenseValidationResponse } from './core/LicenseValidator';
+
 // ── Exceptions ────────────────────────────────────────────────
 export type { RetrivoraErrorCode } from './exceptions';
 export {
@@ -100,6 +104,8 @@ export {
   RateLimitException,
   ConfigurationException,
   AuthenticationException,
+  SDKVersionUnsupportedError,
+  LicenseValidationError,
   wrapError,
 } from './exceptions';