@retrivora-ai/rag-engine 2.1.3 → 2.1.4

package/src/core/Pipeline.ts

@@ -1,4 +1,5 @@
 import { BaseVectorProvider } from '../providers/vectordb/BaseVectorProvider';
+import { SDK_VERSION } from '../version';
 import { BaseGraphProvider } from '../providers/graphdb/BaseGraphProvider';
 import { ILLMProvider } from '../llm/ILLMProvider';
 import { ChatMessage } from '../types';
@@ -933,6 +934,14 @@ You are a helpful assistant. Use the provided context to answer questions accura
               }
             }
 
+            const modelName = finalTrace?.model || this.config.llm?.model || 'llama-3.1-8b-instant';
+            const providerName = finalTrace?.provider || this.config.llm?.provider || 'groq';
+            const tokenCount = Number(finalTrace?.tokens?.totalTokens || 0);
+            const costEst = Number(finalTrace?.tokens?.estimatedCostUsd || 0);
+            const latencyDuration = Number(finalTrace?.latency?.totalMs || 0);
+
+            console.log(`[Retrivora Pipeline Telemetry] 📊 Dispatching RAG telemetry -> Project: "${ns}", Model: "${providerName}/${modelName}", Tokens: ${tokenCount}, Latency: ${latencyDuration}ms`);
+
             await fetch(absoluteUrl, {
               method: 'POST',
               headers: {
@@ -941,7 +950,20 @@ You are a helpful assistant. Use the provided context to answer questions accura
               body: JSON.stringify({
                 trace: finalTrace,
                 licenseKey: this.config.licenseKey,
-                projectId: ns
+                projectId: ns,
+                organization: process.env.RETRIVORA_ORGANIZATION || 'default-org',
+                sdkVersion: SDK_VERSION,
+                model: modelName,
+                provider: providerName,
+                tokens: tokenCount,
+                cost: costEst,
+                costUsd: costEst,
+                latencyMs: latencyDuration,
+                status: 'success',
+                action: 'RAG_QUERY',
+                feature: 'RAG',
+                timestamp: new Date().toISOString(),
+                details: `Org: default-org | Model: ${providerName}/${modelName} | Tokens: ${tokenCount} | Latency: ${latencyDuration}ms | Cost: $${costEst} | SDK: v${SDK_VERSION} | Feature: RAG`
               })
             });
           } catch (err: any) {

package/src/handlers/index.ts

@@ -147,6 +147,8 @@ function getOrCreatePlugin(configOrPlugin: Partial<RagConfig> | VectorPlugin | u
 
 // ─── Telemetry Reporter Helper ───────────────────────────────────────────────
 
+import { SDK_VERSION } from '../version';
+
 /**
  * Asynchronously sends telemetry payload to telemetryUrl when telemetry is enabled
  * or when a licenseKey is set. Non-blocking.
@@ -157,7 +159,7 @@ function reportTelemetry(
   action: string,
   status: 'success' | 'error',
   details: string,
-  trace?: unknown
+  trace?: any
 ) {
   try {
     const config = plugin.getConfig();
@@ -171,6 +173,7 @@ function reportTelemetry(
 
     const telemetryUrl = telemetryConfig?.url || process.env.TELEMETRY_URL || process.env.NEXT_PUBLIC_TELEMETRY_URL || defaultUrl;
     const host = req.headers.get('host') || 'localhost';
+    const userAgent = req.headers.get('user-agent') || `Retrivora-SDK/${SDK_VERSION}`;
 
     let absoluteUrl = telemetryUrl;
     if (!telemetryUrl.startsWith('http')) {
@@ -179,6 +182,33 @@ function reportTelemetry(
     }
 
     const projectId = config.projectId || 'default';
+    const model = trace?.model || config.llm?.model || config.embedding?.model || 'llama-3.1-8b-instant';
+    const provider = trace?.provider || config.llm?.provider || config.embedding?.provider || 'groq';
+    const tokens = Number(trace?.tokens?.totalTokens || trace?.totalTokens || 0);
+    const costUsd = Number(trace?.tokens?.estimatedCostUsd || trace?.costUsd || 0);
+    const latencyMs = Number(trace?.latency?.totalMs || trace?.latencyMs || 0);
+
+    const payload = {
+      trace,
+      licenseKey,
+      projectId,
+      organization: process.env.RETRIVORA_ORGANIZATION || 'default-org',
+      sdkVersion: SDK_VERSION,
+      model,
+      provider,
+      tokens,
+      cost: costUsd,
+      costUsd,
+      latencyMs,
+      status,
+      action,
+      feature: action,
+      userAgent,
+      timestamp: new Date().toISOString(),
+      details,
+    };
+
+    console.log(`[Retrivora SDK Telemetry] 📊 Dispatching telemetry -> Feature: "${action}", Status: "${status}", Project: "${projectId}", Model: "${provider}/${model}", Latency: ${latencyMs}ms`);
 
     fetch(absoluteUrl, {
       method: 'POST',
@@ -186,17 +216,11 @@ function reportTelemetry(
         'Content-Type': 'application/json',
         'x-forwarded-for': req.headers.get('x-forwarded-for') || '',
         'x-real-ip': req.headers.get('x-real-ip') || '',
+        'user-agent': userAgent,
       },
-      body: JSON.stringify({
-        trace,
-        licenseKey,
-        projectId,
-        action,
-        status,
-        details,
-      }),
+      body: JSON.stringify(payload),
     }).catch(err => {
-      console.warn('[Retrivora Telemetry] Async report warning:', err.message);
+      console.warn('[Retrivora SDK Telemetry] Async report warning:', err.message);
     });
   } catch {
     /* non-blocking */

package/src/index.css

@@ -2259,6 +2259,11 @@
       cursor: not-allowed;
     }
   }
+  .disabled\:text-slate-400 {
+    &:disabled {
+      color: var(--color-slate-400);
+    }
+  }
   .disabled\:opacity-30 {
     &:disabled {
       opacity: 30%;
@@ -2912,6 +2917,14 @@
       color: var(--color-amber-400);
     }
   }
+  .dark\:text-amber-400\/90 {
+    @media (prefers-color-scheme: dark) {
+      color: color-mix(in srgb, oklch(82.8% 0.189 84.429) 90%, transparent);
+      @supports (color: color-mix(in lab, red, red)) {
+        color: color-mix(in oklab, var(--color-amber-400) 90%, transparent);
+      }
+    }
+  }
   .dark\:text-blue-100 {
     @media (prefers-color-scheme: dark) {
       color: var(--color-blue-100);
@@ -3371,6 +3384,16 @@
       }
     }
   }
+  .dark\:disabled\:text-white\/30 {
+    @media (prefers-color-scheme: dark) {
+      &:disabled {
+        color: color-mix(in srgb, #fff 30%, transparent);
+        @supports (color: color-mix(in lab, red, red)) {
+          color: color-mix(in oklab, var(--color-white) 30%, transparent);
+        }
+      }
+    }
+  }
 }
 @layer base {
   :root {

package/src/index.ts

@@ -15,6 +15,7 @@ export { MessageBubble } from './components/MessageBubble';
 export { SourceCard } from './components/SourceCard';
 export { ObservabilityPanel } from './components/ObservabilityPanel';
 export { ConfigProvider, useConfig } from './components/ConfigProvider';
+export { SDK_VERSION } from './version';
 export { CodeViewer } from './components/CodeViewer';
 export { ProductCard } from './components/ProductCard';
 export { ProductCarousel } from './components/ProductCarousel';

package/src/server.ts

@@ -36,6 +36,7 @@ export { ConfigResolver } from './core/ConfigResolver';
 export { ProviderRegistry } from './core/ProviderRegistry';
 export { ConfigValidator } from './core/ConfigValidator';
 export { LicenseVerifier } from './core/LicenseVerifier';
+export { SDK_VERSION } from './version';
 export { ProviderHealthCheck } from './core/ProviderHealthCheck';
 export { BatchProcessor } from './core/BatchProcessor';
 

package/src/types/index.ts

@@ -40,6 +40,8 @@ export interface ObservabilityTrace {
   chunks: RetrievedChunk[];
   latency: LatencyBreakdown;
   tokens?: TokenUsage;
+  model?: string;
+  provider?: string;
   /** 0 = fully grounded, 1 = likely hallucinated. */
   hallucinationScore?: number;
   hallucinationReason?: string;

package/src/types/props.ts

@@ -109,6 +109,10 @@ export interface ProductCarouselProps {
 export interface DocumentUploadProps {
   /** Optional namespace for the upload */
   namespace?: string;
+  /** Custom upload API endpoint URL (defaults to /api/retrivora/upload or /api/upload) */
+  uploadUrl?: string;
+  /** Base URL for Retrivora SDK endpoints (defaults to /api/retrivora) */
+  retrivoraApiBase?: string;
   /** Callback when upload completes */
   onUploadComplete?: (results: unknown) => void;
   /** Additional className */

package/src/version.ts

@@ -0,0 +1,6 @@
+import pkg from '../package.json';
+
+/**
+ * Dynamically resolved SDK Version from package.json
+ */
+export const SDK_VERSION: string = pkg.version || '2.1.3';