@retrivora-ai/rag-engine 0.4.3 → 0.4.4

package/dist/index.js

@@ -1,7 +1,9 @@
 "use strict";
 var __create = Object.create;
 var __defProp = Object.defineProperty;
+var __defProps = Object.defineProperties;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getOwnPropSymbols = Object.getOwnPropertySymbols;
 var __getProtoOf = Object.getPrototypeOf;
@@ -19,6 +21,7 @@ var __spreadValues = (a, b) => {
     }
   return a;
 };
+var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -47,6 +50,7 @@ __export(index_exports, {
   ChatWidget: () => ChatWidget,
   ChatWindow: () => ChatWindow,
   ConfigProvider: () => ConfigProvider,
+  DocumentUpload: () => DocumentUpload,
   MessageBubble: () => MessageBubble,
   SourceCard: () => SourceCard,
   useConfig: () => useConfig,
@@ -196,7 +200,8 @@ var defaultConfig = {
     showWidget: true,
     poweredBy: "RAG",
     visualStyle: "glass",
-    borderRadius: "xl"
+    borderRadius: "xl",
+    allowUpload: true
   }
 };
 var ConfigContext = (0, import_react3.createContext)(defaultConfig);
@@ -610,11 +615,166 @@ function ChatWidget({ position = "bottom-right" }) {
     )
   ] });
 }
+
+// src/components/DocumentUpload.tsx
+var import_react7 = require("react");
+var import_lucide_react6 = require("lucide-react");
+var import_jsx_runtime7 = require("react/jsx-runtime");
+function DocumentUpload({ namespace, onUploadComplete, className = "" }) {
+  const { ui } = useConfig();
+  const [fileStates, setFileStates] = (0, import_react7.useState)([]);
+  const [isDragging, setIsDragging] = (0, import_react7.useState)(false);
+  const fileInputRef = (0, import_react7.useRef)(null);
+  const addFiles = (files) => {
+    const newStates = files.map((file) => ({ file, status: "idle" }));
+    setFileStates((prev) => [...prev, ...newStates]);
+  };
+  const removeFile = (index) => {
+    setFileStates((prev) => prev.filter((_, i) => i !== index));
+  };
+  const onDragOver = (e) => {
+    e.preventDefault();
+    setIsDragging(true);
+  };
+  const onDragLeave = () => {
+    setIsDragging(false);
+  };
+  const onDrop = (e) => {
+    e.preventDefault();
+    setIsDragging(false);
+    if (e.dataTransfer.files) {
+      addFiles(Array.from(e.dataTransfer.files));
+    }
+  };
+  const onFileSelect = (e) => {
+    if (e.target.files) {
+      addFiles(Array.from(e.target.files));
+    }
+  };
+  const uploadFiles = async () => {
+    const idleFiles = fileStates.filter((s) => s.status === "idle");
+    if (idleFiles.length === 0) return;
+    setFileStates((prev) => prev.map((s) => s.status === "idle" ? __spreadProps(__spreadValues({}, s), { status: "uploading" }) : s));
+    const formData = new FormData();
+    idleFiles.forEach((s) => formData.append("files", s.file));
+    if (namespace) formData.append("namespace", namespace);
+    try {
+      const response = await fetch("/api/upload", {
+        method: "POST",
+        body: formData
+      });
+      const result = await response.json();
+      if (!response.ok) throw new Error(result.error || "Upload failed");
+      setFileStates((prev) => prev.map((s) => s.status === "uploading" ? __spreadProps(__spreadValues({}, s), { status: "success" }) : s));
+      if (onUploadComplete) onUploadComplete(result);
+    } catch (err) {
+      const message = err instanceof Error ? err.message : "Upload failed";
+      setFileStates((prev) => prev.map((s) => s.status === "uploading" ? __spreadProps(__spreadValues({}, s), { status: "error", error: message }) : s));
+    }
+  };
+  const isUploading = fileStates.some((s) => s.status === "uploading");
+  const hasIdle = fileStates.some((s) => s.status === "idle");
+  return /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)(
+    "div",
+    {
+      className: `p-6 border-2 border-dashed transition-all duration-300 rounded-2xl ${isDragging ? "border-emerald-500 bg-emerald-50/50 dark:bg-emerald-500/5" : "border-slate-200 dark:border-white/10 bg-white dark:bg-white/5"} ${className}`,
+      onDragOver,
+      onDragLeave,
+      onDrop,
+      children: [
+        /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)("div", { className: "flex flex-col items-center justify-center text-center", children: [
+          /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
+            "div",
+            {
+              className: "w-12 h-12 rounded-full flex items-center justify-center mb-4 shadow-sm",
+              style: { background: `linear-gradient(135deg, ${ui.primaryColor}20, ${ui.accentColor}20)` },
+              children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_lucide_react6.Upload, { className: "w-6 h-6", style: { color: ui.primaryColor } })
+            }
+          ),
+          /* @__PURE__ */ (0, import_jsx_runtime7.jsx)("h3", { className: "text-slate-900 dark:text-white font-semibold mb-1", children: "Upload Documents" }),
+          /* @__PURE__ */ (0, import_jsx_runtime7.jsx)("p", { className: "text-slate-500 dark:text-white/40 text-sm mb-6 max-w-xs", children: "Drag and drop PDF, DOCX, TXT, or JSON files to train your AI on your own data." }),
+          /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
+            "button",
+            {
+              onClick: () => {
+                var _a;
+                return (_a = fileInputRef.current) == null ? void 0 : _a.click();
+              },
+              disabled: isUploading,
+              className: "px-4 py-2 rounded-lg text-sm font-medium transition-all hover:scale-105 active:scale-95 disabled:opacity-50",
+              style: {
+                backgroundColor: `${ui.primaryColor}15`,
+                color: ui.primaryColor,
+                border: `1px solid ${ui.primaryColor}30`
+              },
+              children: "Select Files"
+            }
+          ),
+          /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
+            "input",
+            {
+              ref: fileInputRef,
+              type: "file",
+              multiple: true,
+              onChange: onFileSelect,
+              className: "hidden",
+              accept: ".pdf,.docx,.txt,.md,.json,.csv"
+            }
+          )
+        ] }),
+        fileStates.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)("div", { className: "mt-8 space-y-3", children: [
+          fileStates.map((state, i) => /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)(
+            "div",
+            {
+              className: "flex items-center justify-between p-3 rounded-xl bg-slate-50 dark:bg-white/5 border border-slate-100 dark:border-white/10",
+              children: [
+                /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)("div", { className: "flex items-center gap-3 overflow-hidden", children: [
+                  /* @__PURE__ */ (0, import_jsx_runtime7.jsx)("div", { className: "w-8 h-8 rounded-lg bg-white dark:bg-white/10 flex items-center justify-center shadow-sm", children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_lucide_react6.File, { className: "w-4 h-4 text-slate-400" }) }),
+                  /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)("div", { className: "truncate", children: [
+                    /* @__PURE__ */ (0, import_jsx_runtime7.jsx)("p", { className: "text-xs font-medium text-slate-700 dark:text-white/80 truncate", children: state.file.name }),
+                    /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)("p", { className: "text-[10px] text-slate-400", children: [
+                      (state.file.size / 1024).toFixed(1),
+                      " KB"
+                    ] })
+                  ] })
+                ] }),
+                /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)("div", { className: "flex items-center gap-2", children: [
+                  state.status === "uploading" && /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_lucide_react6.Loader2, { className: "w-4 h-4 text-slate-400 animate-spin" }),
+                  state.status === "success" && /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_lucide_react6.CheckCircle, { className: "w-4 h-4 text-emerald-500" }),
+                  state.status === "error" && /* @__PURE__ */ (0, import_jsx_runtime7.jsx)("div", { title: state.error, children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_lucide_react6.AlertCircle, { className: "w-4 h-4 text-rose-500" }) }),
+                  state.status !== "uploading" && /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
+                    "button",
+                    {
+                      onClick: () => removeFile(i),
+                      className: "p-1 rounded-md hover:bg-slate-200 dark:hover:bg-white/10 transition-colors",
+                      children: /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(import_lucide_react6.X, { className: "w-3.5 h-3.5 text-slate-400" })
+                    }
+                  )
+                ] })
+              ]
+            },
+            i
+          )),
+          hasIdle && !isUploading && /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
+            "button",
+            {
+              onClick: uploadFiles,
+              className: "w-full mt-4 py-2.5 rounded-xl text-sm font-semibold text-white shadow-lg transition-all hover:scale-[1.02] active:scale-[0.98]",
+              style: { background: `linear-gradient(135deg, ${ui.primaryColor}, ${ui.accentColor})` },
+              children: "Start Upload"
+            }
+          )
+        ] })
+      ]
+    }
+  );
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ChatWidget,
   ChatWindow,
   ConfigProvider,
+  DocumentUpload,
   MessageBubble,
   SourceCard,
   useConfig,

package/dist/index.mjs

@@ -2,6 +2,7 @@ import {
   mergeDefined
 } from "./chunk-EDLTMSNY.mjs";
 import {
+  __spreadProps,
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";
 
@@ -138,7 +139,8 @@ var defaultConfig = {
     showWidget: true,
     poweredBy: "RAG",
     visualStyle: "glass",
-    borderRadius: "xl"
+    borderRadius: "xl",
+    allowUpload: true
   }
 };
 var ConfigContext = createContext(defaultConfig);
@@ -570,10 +572,165 @@ function ChatWidget({ position = "bottom-right" }) {
     )
   ] });
 }
+
+// src/components/DocumentUpload.tsx
+import { useState as useState5, useRef as useRef3 } from "react";
+import { Upload, File, X as X3, CheckCircle, AlertCircle, Loader2 } from "lucide-react";
+import { jsx as jsx7, jsxs as jsxs5 } from "react/jsx-runtime";
+function DocumentUpload({ namespace, onUploadComplete, className = "" }) {
+  const { ui } = useConfig();
+  const [fileStates, setFileStates] = useState5([]);
+  const [isDragging, setIsDragging] = useState5(false);
+  const fileInputRef = useRef3(null);
+  const addFiles = (files) => {
+    const newStates = files.map((file) => ({ file, status: "idle" }));
+    setFileStates((prev) => [...prev, ...newStates]);
+  };
+  const removeFile = (index) => {
+    setFileStates((prev) => prev.filter((_, i) => i !== index));
+  };
+  const onDragOver = (e) => {
+    e.preventDefault();
+    setIsDragging(true);
+  };
+  const onDragLeave = () => {
+    setIsDragging(false);
+  };
+  const onDrop = (e) => {
+    e.preventDefault();
+    setIsDragging(false);
+    if (e.dataTransfer.files) {
+      addFiles(Array.from(e.dataTransfer.files));
+    }
+  };
+  const onFileSelect = (e) => {
+    if (e.target.files) {
+      addFiles(Array.from(e.target.files));
+    }
+  };
+  const uploadFiles = async () => {
+    const idleFiles = fileStates.filter((s) => s.status === "idle");
+    if (idleFiles.length === 0) return;
+    setFileStates((prev) => prev.map((s) => s.status === "idle" ? __spreadProps(__spreadValues({}, s), { status: "uploading" }) : s));
+    const formData = new FormData();
+    idleFiles.forEach((s) => formData.append("files", s.file));
+    if (namespace) formData.append("namespace", namespace);
+    try {
+      const response = await fetch("/api/upload", {
+        method: "POST",
+        body: formData
+      });
+      const result = await response.json();
+      if (!response.ok) throw new Error(result.error || "Upload failed");
+      setFileStates((prev) => prev.map((s) => s.status === "uploading" ? __spreadProps(__spreadValues({}, s), { status: "success" }) : s));
+      if (onUploadComplete) onUploadComplete(result);
+    } catch (err) {
+      const message = err instanceof Error ? err.message : "Upload failed";
+      setFileStates((prev) => prev.map((s) => s.status === "uploading" ? __spreadProps(__spreadValues({}, s), { status: "error", error: message }) : s));
+    }
+  };
+  const isUploading = fileStates.some((s) => s.status === "uploading");
+  const hasIdle = fileStates.some((s) => s.status === "idle");
+  return /* @__PURE__ */ jsxs5(
+    "div",
+    {
+      className: `p-6 border-2 border-dashed transition-all duration-300 rounded-2xl ${isDragging ? "border-emerald-500 bg-emerald-50/50 dark:bg-emerald-500/5" : "border-slate-200 dark:border-white/10 bg-white dark:bg-white/5"} ${className}`,
+      onDragOver,
+      onDragLeave,
+      onDrop,
+      children: [
+        /* @__PURE__ */ jsxs5("div", { className: "flex flex-col items-center justify-center text-center", children: [
+          /* @__PURE__ */ jsx7(
+            "div",
+            {
+              className: "w-12 h-12 rounded-full flex items-center justify-center mb-4 shadow-sm",
+              style: { background: `linear-gradient(135deg, ${ui.primaryColor}20, ${ui.accentColor}20)` },
+              children: /* @__PURE__ */ jsx7(Upload, { className: "w-6 h-6", style: { color: ui.primaryColor } })
+            }
+          ),
+          /* @__PURE__ */ jsx7("h3", { className: "text-slate-900 dark:text-white font-semibold mb-1", children: "Upload Documents" }),
+          /* @__PURE__ */ jsx7("p", { className: "text-slate-500 dark:text-white/40 text-sm mb-6 max-w-xs", children: "Drag and drop PDF, DOCX, TXT, or JSON files to train your AI on your own data." }),
+          /* @__PURE__ */ jsx7(
+            "button",
+            {
+              onClick: () => {
+                var _a;
+                return (_a = fileInputRef.current) == null ? void 0 : _a.click();
+              },
+              disabled: isUploading,
+              className: "px-4 py-2 rounded-lg text-sm font-medium transition-all hover:scale-105 active:scale-95 disabled:opacity-50",
+              style: {
+                backgroundColor: `${ui.primaryColor}15`,
+                color: ui.primaryColor,
+                border: `1px solid ${ui.primaryColor}30`
+              },
+              children: "Select Files"
+            }
+          ),
+          /* @__PURE__ */ jsx7(
+            "input",
+            {
+              ref: fileInputRef,
+              type: "file",
+              multiple: true,
+              onChange: onFileSelect,
+              className: "hidden",
+              accept: ".pdf,.docx,.txt,.md,.json,.csv"
+            }
+          )
+        ] }),
+        fileStates.length > 0 && /* @__PURE__ */ jsxs5("div", { className: "mt-8 space-y-3", children: [
+          fileStates.map((state, i) => /* @__PURE__ */ jsxs5(
+            "div",
+            {
+              className: "flex items-center justify-between p-3 rounded-xl bg-slate-50 dark:bg-white/5 border border-slate-100 dark:border-white/10",
+              children: [
+                /* @__PURE__ */ jsxs5("div", { className: "flex items-center gap-3 overflow-hidden", children: [
+                  /* @__PURE__ */ jsx7("div", { className: "w-8 h-8 rounded-lg bg-white dark:bg-white/10 flex items-center justify-center shadow-sm", children: /* @__PURE__ */ jsx7(File, { className: "w-4 h-4 text-slate-400" }) }),
+                  /* @__PURE__ */ jsxs5("div", { className: "truncate", children: [
+                    /* @__PURE__ */ jsx7("p", { className: "text-xs font-medium text-slate-700 dark:text-white/80 truncate", children: state.file.name }),
+                    /* @__PURE__ */ jsxs5("p", { className: "text-[10px] text-slate-400", children: [
+                      (state.file.size / 1024).toFixed(1),
+                      " KB"
+                    ] })
+                  ] })
+                ] }),
+                /* @__PURE__ */ jsxs5("div", { className: "flex items-center gap-2", children: [
+                  state.status === "uploading" && /* @__PURE__ */ jsx7(Loader2, { className: "w-4 h-4 text-slate-400 animate-spin" }),
+                  state.status === "success" && /* @__PURE__ */ jsx7(CheckCircle, { className: "w-4 h-4 text-emerald-500" }),
+                  state.status === "error" && /* @__PURE__ */ jsx7("div", { title: state.error, children: /* @__PURE__ */ jsx7(AlertCircle, { className: "w-4 h-4 text-rose-500" }) }),
+                  state.status !== "uploading" && /* @__PURE__ */ jsx7(
+                    "button",
+                    {
+                      onClick: () => removeFile(i),
+                      className: "p-1 rounded-md hover:bg-slate-200 dark:hover:bg-white/10 transition-colors",
+                      children: /* @__PURE__ */ jsx7(X3, { className: "w-3.5 h-3.5 text-slate-400" })
+                    }
+                  )
+                ] })
+              ]
+            },
+            i
+          )),
+          hasIdle && !isUploading && /* @__PURE__ */ jsx7(
+            "button",
+            {
+              onClick: uploadFiles,
+              className: "w-full mt-4 py-2.5 rounded-xl text-sm font-semibold text-white shadow-lg transition-all hover:scale-[1.02] active:scale-[0.98]",
+              style: { background: `linear-gradient(135deg, ${ui.primaryColor}, ${ui.accentColor})` },
+              children: "Start Upload"
+            }
+          )
+        ] })
+      ]
+    }
+  );
+}
 export {
   ChatWidget,
   ChatWindow,
   ConfigProvider,
+  DocumentUpload,
   MessageBubble,
   SourceCard,
   useConfig,

package/dist/server.d.mts

@@ -1,9 +1,9 @@
-import { c as RagConfig, C as ChatResponse, I as IngestDocument, e as VectorDBConfig, d as UpsertDocument, V as VectorMatch, L as LLMConfig, E as EmbeddingConfig, f as VectorDBProvider, b as LLMProvider, a as EmbeddingProvider } from './RagConfig-CVt24lbC.mjs';
-export { R as RAGConfig, U as UIConfig } from './RagConfig-CVt24lbC.mjs';
-import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-BICIjSuG.mjs';
-export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-BICIjSuG.mjs';
-import { H as HealthCheckResult } from './index-OI2--lvT.mjs';
-export { P as ProviderHealthCheck, c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from './index-OI2--lvT.mjs';
+import { c as RagConfig, C as ChatResponse, I as IngestDocument, e as VectorDBConfig, d as UpsertDocument, V as VectorMatch, G as GraphDBConfig, g as GraphNode, h as Edge, i as GraphSearchResult, L as LLMConfig, E as EmbeddingConfig, f as VectorDBProvider, b as LLMProvider, a as EmbeddingProvider } from './RagConfig-BgRDL9Vy.mjs';
+export { R as RAGConfig, U as UIConfig } from './RagConfig-BgRDL9Vy.mjs';
+import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-3yElxTO3.mjs';
+export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-3yElxTO3.mjs';
+import { H as HealthCheckResult } from './index-7qeLTPBL.mjs';
+export { P as ProviderHealthCheck, c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from './index-7qeLTPBL.mjs';
 import 'next/server';
 
 /**
@@ -233,9 +233,11 @@ declare class VectorPlugin {
  */
 declare class Pipeline {
     private vectorDB;
+    private graphDB?;
     private llmProvider;
     private embeddingProvider;
     private chunker;
+    private entityExtractor?;
     private config;
     private initialised;
     constructor(config: RagConfig);
@@ -249,6 +251,10 @@ declare class Pipeline {
         chunksIngested: number;
     }>>;
     ask(question: string, history?: ChatMessage[], namespace?: string): Promise<ChatResponse>;
+    /**
+     * Rewrite the user query for better retrieval performance.
+     */
+    private rewriteQuery;
 }
 
 /**
@@ -314,6 +320,38 @@ declare abstract class BaseVectorProvider {
     protected sanitizeFilter(filter?: Record<string, unknown>): Record<string, unknown>;
 }
 
+/**
+ * BaseGraphProvider — Abstract base class for Graph DB providers.
+ */
+declare abstract class BaseGraphProvider {
+    protected config: GraphDBConfig;
+    constructor(config: GraphDBConfig);
+    /**
+     * Initialise connection to the graph database.
+     */
+    abstract initialize(): Promise<void>;
+    /**
+     * Add nodes to the graph.
+     */
+    abstract addNodes(nodes: GraphNode[]): Promise<void>;
+    /**
+     * Add edges (relationships) to the graph.
+     */
+    abstract addEdges(edges: Edge[]): Promise<void>;
+    /**
+     * Query the graph for relevant nodes and edges.
+     */
+    abstract query(queryText: string, limit?: number): Promise<GraphSearchResult>;
+    /**
+     * Check if the database is reachable.
+     */
+    abstract ping(): Promise<boolean>;
+    /**
+     * Close connection.
+     */
+    abstract disconnect(): Promise<void>;
+}
+
 /**
  * ProviderRegistry — dynamic provider loader for Vector DBs and LLMs.
  *
@@ -323,19 +361,21 @@ declare abstract class BaseVectorProvider {
  */
 declare class ProviderRegistry {
     private static vectorProviders;
+    private static graphProviders;
+    static registerVectorProvider(name: string, providerClass: new (config: VectorDBConfig) => BaseVectorProvider): void;
     /**
-     * Register a custom vector provider class by name.
-     * The name must match the provider value used in VectorDBConfig.provider.
-     *
-     * @example
-     * ProviderRegistry.registerVectorProvider('my-db', MyCustomProvider);
+     * Register a custom graph provider class by name.
      */
-    static registerVectorProvider(name: string, providerClass: new (config: VectorDBConfig) => BaseVectorProvider): void;
+    static registerGraphProvider(name: string, providerClass: new (config: GraphDBConfig) => BaseGraphProvider): void;
     /**
      * Creates a vector database provider based on the configuration.
      * Built-in providers are dynamically imported to avoid bundling all SDKs.
      */
     static createVectorProvider(config: VectorDBConfig): Promise<BaseVectorProvider>;
+    /**
+     * Creates a graph database provider based on the configuration.
+     */
+    static createGraphProvider(config: GraphDBConfig): Promise<BaseGraphProvider>;
     /**
      * Creates an LLM provider based on the configuration.
      */

package/dist/server.d.ts

@@ -1,9 +1,9 @@
-import { c as RagConfig, C as ChatResponse, I as IngestDocument, e as VectorDBConfig, d as UpsertDocument, V as VectorMatch, L as LLMConfig, E as EmbeddingConfig, f as VectorDBProvider, b as LLMProvider, a as EmbeddingProvider } from './RagConfig-CVt24lbC.js';
-export { R as RAGConfig, U as UIConfig } from './RagConfig-CVt24lbC.js';
-import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-BICIjSuG.js';
-export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-BICIjSuG.js';
-import { H as HealthCheckResult } from './index-D1hoNXMT.js';
-export { P as ProviderHealthCheck, c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from './index-D1hoNXMT.js';
+import { c as RagConfig, C as ChatResponse, I as IngestDocument, e as VectorDBConfig, d as UpsertDocument, V as VectorMatch, G as GraphDBConfig, g as GraphNode, h as Edge, i as GraphSearchResult, L as LLMConfig, E as EmbeddingConfig, f as VectorDBProvider, b as LLMProvider, a as EmbeddingProvider } from './RagConfig-BgRDL9Vy.js';
+export { R as RAGConfig, U as UIConfig } from './RagConfig-BgRDL9Vy.js';
+import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-3yElxTO3.js';
+export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-3yElxTO3.js';
+import { H as HealthCheckResult } from './index-DowY4_K0.js';
+export { P as ProviderHealthCheck, c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from './index-DowY4_K0.js';
 import 'next/server';
 
 /**
@@ -233,9 +233,11 @@ declare class VectorPlugin {
  */
 declare class Pipeline {
     private vectorDB;
+    private graphDB?;
     private llmProvider;
     private embeddingProvider;
     private chunker;
+    private entityExtractor?;
     private config;
     private initialised;
     constructor(config: RagConfig);
@@ -249,6 +251,10 @@ declare class Pipeline {
         chunksIngested: number;
     }>>;
     ask(question: string, history?: ChatMessage[], namespace?: string): Promise<ChatResponse>;
+    /**
+     * Rewrite the user query for better retrieval performance.
+     */
+    private rewriteQuery;
 }
 
 /**
@@ -314,6 +320,38 @@ declare abstract class BaseVectorProvider {
     protected sanitizeFilter(filter?: Record<string, unknown>): Record<string, unknown>;
 }
 
+/**
+ * BaseGraphProvider — Abstract base class for Graph DB providers.
+ */
+declare abstract class BaseGraphProvider {
+    protected config: GraphDBConfig;
+    constructor(config: GraphDBConfig);
+    /**
+     * Initialise connection to the graph database.
+     */
+    abstract initialize(): Promise<void>;
+    /**
+     * Add nodes to the graph.
+     */
+    abstract addNodes(nodes: GraphNode[]): Promise<void>;
+    /**
+     * Add edges (relationships) to the graph.
+     */
+    abstract addEdges(edges: Edge[]): Promise<void>;
+    /**
+     * Query the graph for relevant nodes and edges.
+     */
+    abstract query(queryText: string, limit?: number): Promise<GraphSearchResult>;
+    /**
+     * Check if the database is reachable.
+     */
+    abstract ping(): Promise<boolean>;
+    /**
+     * Close connection.
+     */
+    abstract disconnect(): Promise<void>;
+}
+
 /**
  * ProviderRegistry — dynamic provider loader for Vector DBs and LLMs.
  *
@@ -323,19 +361,21 @@ declare abstract class BaseVectorProvider {
  */
 declare class ProviderRegistry {
     private static vectorProviders;
+    private static graphProviders;
+    static registerVectorProvider(name: string, providerClass: new (config: VectorDBConfig) => BaseVectorProvider): void;
     /**
-     * Register a custom vector provider class by name.
-     * The name must match the provider value used in VectorDBConfig.provider.
-     *
-     * @example
-     * ProviderRegistry.registerVectorProvider('my-db', MyCustomProvider);
+     * Register a custom graph provider class by name.
      */
-    static registerVectorProvider(name: string, providerClass: new (config: VectorDBConfig) => BaseVectorProvider): void;
+    static registerGraphProvider(name: string, providerClass: new (config: GraphDBConfig) => BaseGraphProvider): void;
     /**
      * Creates a vector database provider based on the configuration.
      * Built-in providers are dynamically imported to avoid bundling all SDKs.
      */
     static createVectorProvider(config: VectorDBConfig): Promise<BaseVectorProvider>;
+    /**
+     * Creates a graph database provider based on the configuration.
+     */
+    static createGraphProvider(config: GraphDBConfig): Promise<BaseGraphProvider>;
     /**
      * Creates an LLM provider based on the configuration.
      */