@retrivora-ai/rag-engine 1.8.1 → 1.8.2

package/dist/{ILLMProvider-BfRgI1Xh.d.mts → ILLMProvider-BOJFz3Na.d.mts}

@@ -7,6 +7,8 @@ interface RagMessage extends ChatMessage {
     id: string;
     sources?: VectorMatch[];
     uiTransformation?: unknown;
+    /** Full observability trace emitted by the backend. Only present on assistant messages. */
+    trace?: ObservabilityTrace;
     createdAt: string;
 }
 interface ChatOptions {
@@ -233,6 +235,48 @@ interface RagConfig {
     graphDb?: GraphDBConfig;
 }
 
+/** Per-stage latency breakdown for a single RAG request. */
+interface LatencyBreakdown {
+    embedMs: number;
+    retrieveMs: number;
+    rerankMs?: number;
+    generateMs: number;
+    totalMs: number;
+}
+/** Token usage reported by the LLM (or estimated when the provider doesn't expose it). */
+interface TokenUsage {
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+    /** Rough cost in USD — estimated from token counts and model pricing. */
+    estimatedCostUsd?: number;
+}
+/** A single retrieved chunk annotated with its retrieval context. */
+interface RetrievedChunk {
+    id: string | number;
+    score: number;
+    content: string;
+    metadata: Record<string, unknown>;
+    namespace: string;
+}
+/**
+ * Full observability trace for one RAG request.
+ * Emitted by the backend as an SSE frame and stored on each RagMessage.
+ */
+interface ObservabilityTrace {
+    requestId: string;
+    query: string;
+    rewrittenQuery?: string;
+    systemPrompt: string;
+    userPrompt: string;
+    chunks: RetrievedChunk[];
+    latency: LatencyBreakdown;
+    tokens?: TokenUsage;
+    /** 0 = fully grounded, 1 = likely hallucinated. */
+    hallucinationScore?: number;
+    hallucinationReason?: string;
+    timestamp: string;
+}
 interface VectorMatch {
     id: string | number;
     score: number;
@@ -255,6 +299,8 @@ interface ChatResponse {
     sources: VectorMatch[];
     graphData?: GraphSearchResult;
     ui_transformation?: unknown;
+    /** Observability trace — populated by the instrumented Pipeline. */
+    trace?: ObservabilityTrace;
 }
 interface GraphNode {
     id: string;
@@ -331,4 +377,4 @@ interface ILLMProvider {
     ping(): Promise<boolean>;
 }
 
-export type { ChatMessage as C, EmbedOptions as E, GraphDBConfig as G, ILLMProvider as I, LLMConfig as L, Product as P, RagMessage as R, UIConfig as U, VectorMatch as V, UseRagChatOptions as a, UseRagChatReturn as b, ChatOptions as c, ChatResponse as d, EmbeddingConfig as e, EmbeddingProvider as f, IngestDocument as g, LLMProvider as h, RAGConfig as i, RagConfig as j, UpsertDocument as k, VectorDBConfig as l, VectorDBProvider as m, RetrievalResult as n, GraphNode as o, Edge as p, GraphSearchResult as q };
+export type { ChatMessage as C, EmbedOptions as E, GraphDBConfig as G, ILLMProvider as I, LLMConfig as L, ObservabilityTrace as O, Product as P, RagMessage as R, TokenUsage as T, UIConfig as U, VectorMatch as V, UseRagChatOptions as a, UseRagChatReturn as b, ChatOptions as c, ChatResponse as d, EmbeddingConfig as e, EmbeddingProvider as f, IngestDocument as g, LLMProvider as h, LatencyBreakdown as i, RAGConfig as j, RagConfig as k, RetrievedChunk as l, UpsertDocument as m, VectorDBConfig as n, VectorDBProvider as o, RetrievalResult as p, GraphNode as q, Edge as r, GraphSearchResult as s };

package/dist/{ILLMProvider-BfRgI1Xh.d.ts → ILLMProvider-BOJFz3Na.d.ts}

@@ -7,6 +7,8 @@ interface RagMessage extends ChatMessage {
     id: string;
     sources?: VectorMatch[];
     uiTransformation?: unknown;
+    /** Full observability trace emitted by the backend. Only present on assistant messages. */
+    trace?: ObservabilityTrace;
     createdAt: string;
 }
 interface ChatOptions {
@@ -233,6 +235,48 @@ interface RagConfig {
     graphDb?: GraphDBConfig;
 }
 
+/** Per-stage latency breakdown for a single RAG request. */
+interface LatencyBreakdown {
+    embedMs: number;
+    retrieveMs: number;
+    rerankMs?: number;
+    generateMs: number;
+    totalMs: number;
+}
+/** Token usage reported by the LLM (or estimated when the provider doesn't expose it). */
+interface TokenUsage {
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+    /** Rough cost in USD — estimated from token counts and model pricing. */
+    estimatedCostUsd?: number;
+}
+/** A single retrieved chunk annotated with its retrieval context. */
+interface RetrievedChunk {
+    id: string | number;
+    score: number;
+    content: string;
+    metadata: Record<string, unknown>;
+    namespace: string;
+}
+/**
+ * Full observability trace for one RAG request.
+ * Emitted by the backend as an SSE frame and stored on each RagMessage.
+ */
+interface ObservabilityTrace {
+    requestId: string;
+    query: string;
+    rewrittenQuery?: string;
+    systemPrompt: string;
+    userPrompt: string;
+    chunks: RetrievedChunk[];
+    latency: LatencyBreakdown;
+    tokens?: TokenUsage;
+    /** 0 = fully grounded, 1 = likely hallucinated. */
+    hallucinationScore?: number;
+    hallucinationReason?: string;
+    timestamp: string;
+}
 interface VectorMatch {
     id: string | number;
     score: number;
@@ -255,6 +299,8 @@ interface ChatResponse {
     sources: VectorMatch[];
     graphData?: GraphSearchResult;
     ui_transformation?: unknown;
+    /** Observability trace — populated by the instrumented Pipeline. */
+    trace?: ObservabilityTrace;
 }
 interface GraphNode {
     id: string;
@@ -331,4 +377,4 @@ interface ILLMProvider {
     ping(): Promise<boolean>;
 }
 
-export type { ChatMessage as C, EmbedOptions as E, GraphDBConfig as G, ILLMProvider as I, LLMConfig as L, Product as P, RagMessage as R, UIConfig as U, VectorMatch as V, UseRagChatOptions as a, UseRagChatReturn as b, ChatOptions as c, ChatResponse as d, EmbeddingConfig as e, EmbeddingProvider as f, IngestDocument as g, LLMProvider as h, RAGConfig as i, RagConfig as j, UpsertDocument as k, VectorDBConfig as l, VectorDBProvider as m, RetrievalResult as n, GraphNode as o, Edge as p, GraphSearchResult as q };
+export type { ChatMessage as C, EmbedOptions as E, GraphDBConfig as G, ILLMProvider as I, LLMConfig as L, ObservabilityTrace as O, Product as P, RagMessage as R, TokenUsage as T, UIConfig as U, VectorMatch as V, UseRagChatOptions as a, UseRagChatReturn as b, ChatOptions as c, ChatResponse as d, EmbeddingConfig as e, EmbeddingProvider as f, IngestDocument as g, LLMProvider as h, LatencyBreakdown as i, RAGConfig as j, RagConfig as k, RetrievedChunk as l, UpsertDocument as m, VectorDBConfig as n, VectorDBProvider as o, RetrievalResult as p, GraphNode as q, Edge as r, GraphSearchResult as s };

package/dist/{MultiTablePostgresProvider-YY7LPNJK.mjs → MultiTablePostgresProvider-ZLGSKTJR.mjs}

@@ -1,6 +1,6 @@
 import {
   MultiTablePostgresProvider
-} from "./chunk-R3RGUMHE.mjs";
+} from "./chunk-OZFBG4BA.mjs";
 import "./chunk-IMP6FUCY.mjs";
 import "./chunk-X4TOT24V.mjs";
 export {

package/dist/chunk-ICKRMZQK.mjs

@@ -0,0 +1,76 @@
+// src/utils/synonyms.ts
+var FIELD_SYNONYMS = {
+  name: ["product", "item", "title", "label", "heading", "subject", "product name", "item name"],
+  price: ["cost", "amount", "msrp", "price", "rate", "value", "price_usd"],
+  brand: ["manufacturer", "vendor", "make", "company", "brand_name", "supplier"],
+  image: [
+    "imageUrl",
+    "thumbnail",
+    "img",
+    "url",
+    "photo",
+    "picture",
+    "media",
+    "image_url",
+    "main_image",
+    "product_image",
+    "thumb"
+  ],
+  stock: [
+    "inventory",
+    "quantity",
+    "count",
+    "availability",
+    "stock_level",
+    "inStock",
+    "is_available",
+    "in stock",
+    "status"
+  ],
+  description: ["summary", "content", "body", "text", "info", "details"],
+  link: ["url", "href", "product_url", "page_url", "link"]
+};
+function addSynonyms(customSynonyms) {
+  for (const [key, aliases] of Object.entries(customSynonyms)) {
+    if (!FIELD_SYNONYMS[key]) {
+      FIELD_SYNONYMS[key] = [];
+    }
+    const existing = new Set(FIELD_SYNONYMS[key].map((s) => s.toLowerCase()));
+    for (const alias of aliases) {
+      if (!existing.has(alias.toLowerCase())) {
+        FIELD_SYNONYMS[key].push(alias);
+        existing.add(alias.toLowerCase());
+      }
+    }
+  }
+}
+function resolveMetadataValue(meta, uiKey) {
+  var _a;
+  const synonyms = (_a = FIELD_SYNONYMS[uiKey]) != null ? _a : [];
+  const keys = Object.keys(meta);
+  const systemKeys = ["namespace", "filename", "filesize", "filetype", "docid", "uploadedat", "dimension", "chunkindex"];
+  let match = keys.find((k) => k.toLowerCase() === uiKey.toLowerCase());
+  if (match !== void 0) return meta[match];
+  match = keys.find((k) => synonyms.map((s) => s.toLowerCase()).includes(k.toLowerCase()));
+  if (match !== void 0) return meta[match];
+  const isBlacklisted = (kl) => {
+    return systemKeys.includes(kl) || kl.startsWith("option1") || kl.startsWith("option2") || kl.startsWith("option3");
+  };
+  match = keys.find((k) => {
+    const kl = k.toLowerCase();
+    if (isBlacklisted(kl)) return false;
+    return kl.includes(uiKey.toLowerCase());
+  });
+  if (match !== void 0) return meta[match];
+  match = keys.find((k) => {
+    const kl = k.toLowerCase();
+    if (isBlacklisted(kl)) return false;
+    return synonyms.some((sk) => kl.includes(sk.toLowerCase()) || sk.toLowerCase().includes(kl));
+  });
+  return match !== void 0 ? meta[match] : void 0;
+}
+
+export {
+  addSynonyms,
+  resolveMetadataValue
+};