@retrivora-ai/rag-engine 2.2.7 → 2.2.8

package/src/hooks/useRagChat.ts

@@ -129,8 +129,27 @@ export function useRagChat(
           userMessageId: userMsgId,
         });
 
-        const fetchHeaders = {
+        const getHeader = (headersObj: Record<string, string> | undefined, name: string) => {
+          if (!headersObj) return undefined;
+          const target = name.toLowerCase();
+          for (const [k, v] of Object.entries(headersObj)) {
+            if (k.toLowerCase() === target) return v;
+          }
+          return undefined;
+        };
+
+        const resolvedLicenseKey =
+          options.licenseKey ||
+          getHeader(options.headers, 'x-license-key') ||
+          getHeader(options.headers, 'authorization')?.replace(/^Bearer\s+/i, '') ||
+          (typeof process !== 'undefined'
+            ? (process.env.NEXT_PUBLIC_RETRIVORA_LICENSE_KEY || process.env.RETRIVORA_LICENSE_KEY)
+            : '') ||
+          '';
+
+        const fetchHeaders: Record<string, string> = {
           'Content-Type': 'application/json',
+          ...(resolvedLicenseKey ? { 'x-license-key': resolvedLicenseKey } : {}),
           ...(options.headers || {})
         };
 

package/src/types/chat.ts

@@ -57,6 +57,8 @@ export interface UseRagChatOptions {
   onError?: (error: string) => void;
   /** Optional custom headers to send with the chat request */
   headers?: Record<string, string>;
+  /** Optional Retrivora license key override */
+  licenseKey?: string;
   /** Session ID for history and feedback tracking (default: 'default') */
   sessionId?: string;
 }

package/src/types/props.ts

@@ -57,11 +57,12 @@ export interface ChatWindowProps {
   onAddToCart?: (product: Product) => void;
   /** Optional custom API URL to send chat messages to */
   apiUrl?: string;
-  /**
-   * Base URL for Retrivora SDK history/feedback/sessions endpoints.
+  /** Base URL for Retrivora SDK history/feedback/sessions endpoints.
    * Defaults to '/api/retrivora'. Set this to match your catch-all route location.
    */
   retrivoraApiBase?: string;
+  /** Optional Retrivora license key override */
+  licenseKey?: string;
   /** Optional custom headers to send with the chat request */
   headers?: Record<string, string>;
 }
@@ -78,6 +79,8 @@ export interface ChatWidgetProps {
    * Defaults to '/api/retrivora'.
    */
   retrivoraApiBase?: string;
+  /** Optional Retrivora license key override */
+  licenseKey?: string;
   /** Optional custom headers to send with the chat request */
   headers?: Record<string, string>;
 }