@retrivora-ai/rag-engine 0.1.3 → 0.1.5

package/dist/handlers/index.mjs

@@ -3,8 +3,10 @@ import {
   createHealthHandler,
   createIngestHandler,
   createUploadHandler
-} from "../chunk-NVOMLHXW.mjs";
-import "../chunk-ZPXLQR5Q.mjs";
+} from "../chunk-BP4U4TT5.mjs";
+import "../chunk-JI6VD5TJ.mjs";
+import "../chunk-UKDXCXW7.mjs";
+import "../chunk-I4E63NIC.mjs";
 export {
   createChatHandler,
   createHealthHandler,

package/dist/index.d.mts

@@ -1,8 +1,9 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import React$1, { ReactNode } from 'react';
-import { C as ChatMessage, V as VectorMatch, U as UIConfig } from './RAGPipeline-BmkIv1HD.mjs';
-export { a as ChatOptions, b as ChatResponse, E as EmbedOptions, c as EmbeddingConfig, I as ILLMProvider, d as IVectorDB, e as IngestDocument, L as LLMConfig, R as RAGConfig, f as RagConfig, g as UpsertDocument, h as VectorDBConfig } from './RAGPipeline-BmkIv1HD.mjs';
-export { C as Chunk, a as ChunkOptions } from './DocumentChunker-BUrIrcPk.mjs';
+import { C as ChatMessage } from './DocumentChunker-cfaMidtA.mjs';
+export { a as ChatOptions, b as Chunk, c as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-cfaMidtA.mjs';
+import { V as VectorMatch, U as UIConfig } from './RagConfig-DG_0f8ka.mjs';
+export { C as ChatResponse, E as EmbeddingConfig, I as IngestDocument, L as LLMConfig, R as RAGConfig, a as RagConfig, b as UpsertDocument, c as VectorDBConfig } from './RagConfig-DG_0f8ka.mjs';
 
 interface ChatWidgetProps {
     /** Position of the floating button. Defaults to bottom-right. */

package/dist/index.d.ts

@@ -1,8 +1,9 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import React$1, { ReactNode } from 'react';
-import { C as ChatMessage, V as VectorMatch, U as UIConfig } from './RAGPipeline-BmkIv1HD.js';
-export { a as ChatOptions, b as ChatResponse, E as EmbedOptions, c as EmbeddingConfig, I as ILLMProvider, d as IVectorDB, e as IngestDocument, L as LLMConfig, R as RAGConfig, f as RagConfig, g as UpsertDocument, h as VectorDBConfig } from './RAGPipeline-BmkIv1HD.js';
-export { C as Chunk, a as ChunkOptions } from './DocumentChunker-BUrIrcPk.js';
+import { C as ChatMessage } from './DocumentChunker-cfaMidtA.js';
+export { a as ChatOptions, b as Chunk, c as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-cfaMidtA.js';
+import { V as VectorMatch, U as UIConfig } from './RagConfig-DG_0f8ka.js';
+export { C as ChatResponse, E as EmbeddingConfig, I as IngestDocument, L as LLMConfig, R as RAGConfig, a as RagConfig, b as UpsertDocument, c as VectorDBConfig } from './RagConfig-DG_0f8ka.js';
 
 interface ChatWidgetProps {
     /** Position of the floating button. Defaults to bottom-right. */

package/dist/index.js

@@ -168,10 +168,10 @@ var import_react3 = require("react");
 
 // src/utils/templateUtils.ts
 function mergeDefined(base, override) {
+  const merged = __spreadValues({}, base || {});
   if (!override) {
-    return __spreadValues({}, base);
+    return merged;
   }
-  const merged = __spreadValues({}, base);
   for (const [key, value] of Object.entries(override)) {
     if (value !== void 0 && value !== null && value !== "") {
       merged[key] = value;

package/dist/index.mjs

@@ -1,7 +1,9 @@
 import {
-  __spreadValues,
   mergeDefined
-} from "./chunk-ZPXLQR5Q.mjs";
+} from "./chunk-UKDXCXW7.mjs";
+import {
+  __spreadValues
+} from "./chunk-I4E63NIC.mjs";
 
 // src/components/ChatWidget.tsx
 import { useState as useState4 } from "react";

package/dist/server.d.mts

@@ -1,42 +1,151 @@
-import { f as RagConfig, h as VectorDBConfig, d as IVectorDB, g as UpsertDocument, V as VectorMatch, L as LLMConfig, c as EmbeddingConfig, I as ILLMProvider, C as ChatMessage, a as ChatOptions, E as EmbedOptions } from './RAGPipeline-BmkIv1HD.mjs';
-export { b as ChatResponse, e as IngestDocument, R as RAGConfig, i as RAGPipeline, U as UIConfig } from './RAGPipeline-BmkIv1HD.mjs';
-export { C as Chunk, a as ChunkOptions, D as DocumentChunker } from './DocumentChunker-BUrIrcPk.mjs';
+import { a as RagConfig, C as ChatResponse, I as IngestDocument, c as VectorDBConfig, b as UpsertDocument, V as VectorMatch, L as LLMConfig, E as EmbeddingConfig } from './RagConfig-DG_0f8ka.mjs';
+export { R as RAGConfig, U as UIConfig } from './RagConfig-DG_0f8ka.mjs';
+import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-cfaMidtA.mjs';
+export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-cfaMidtA.mjs';
 export { createChatHandler, createHealthHandler, createIngestHandler, createUploadHandler } from './handlers/index.mjs';
 import 'next/server';
 
 /**
- * serverConfig.ts — reads RagConfig from environment variables at runtime.
- *
- * This keeps secrets server-side and never exposes them to the browser.
- * Consumer projects set these env vars in their .env.local file.
+ * VectorPlugin — main orchestrator class.
+ * This is the primary interface for host applications.
  */
+declare class VectorPlugin {
+    private config;
+    private pipeline;
+    /**
+     * Initializes the plugin with the host configuration.
+     * @param hostConfig - Configuration object passed from the host application.
+     */
+    constructor(hostConfig?: Partial<RagConfig>);
+    /**
+     * Get the current resolved configuration.
+     */
+    getConfig(): RagConfig;
+    /**
+     * Run a chat query.
+     */
+    chat(message: string, history?: ChatMessage[], namespace?: string): Promise<ChatResponse>;
+    /**
+     * Ingest documents into the vector database.
+     */
+    ingest(documents: IngestDocument[], namespace?: string): Promise<Array<{
+        docId: string;
+        chunksIngested: number;
+    }>>;
+    /**
+     * Check the health of the connected providers.
+     */
+    health(): Promise<Record<string, boolean>>;
+}
 
-declare function getRagConfig(env?: Record<string, string | undefined>): RagConfig;
+/**
+ * Pipeline — orchestrates the RAG flow: Embed → Search → Augment → Generate.
+ */
+declare class Pipeline {
+    private vectorDB;
+    private llmProvider;
+    private embeddingProvider;
+    private chunker;
+    private config;
+    private initialised;
+    constructor(config: RagConfig);
+    initialize(): Promise<void>;
+    ingest(documents: IngestDocument[], namespace?: string): Promise<Array<{
+        docId: string;
+        chunksIngested: number;
+    }>>;
+    ask(question: string, history?: ChatMessage[], namespace?: string): Promise<ChatResponse>;
+    health(): Promise<Record<string, boolean>>;
+}
 
 /**
- * VectorDBFactory — instantiates the correct IVectorDB adapter
- * based on the VectorDBConfig.provider value.
- *
- * Usage:
- *   const db = VectorDBFactory.create(config.vectorDb);
- *   await db.initialize();
+ * ConfigResolver — validates and normalizes host configuration.
+ * It merges host-provided config with environment defaults.
  */
+declare class ConfigResolver {
+    /**
+     * Resolves the final configuration by merging host-provided config with defaults.
+     * @param hostConfig - Partial configuration passed from the host application.
+     */
+    static resolve(hostConfig?: Partial<RagConfig>): RagConfig;
+    /**
+     * Validates the configuration for required fields.
+     */
+    static validate(config: RagConfig): void;
+}
 
-declare class VectorDBFactory {
-    static create(config: VectorDBConfig): IVectorDB;
+/**
+ * BaseVectorProvider — Abstract base class for all vector database providers.
+ * Each provider (Pinecone, Milvus, Redis, etc.) must extend this class.
+ */
+declare abstract class BaseVectorProvider {
+    protected readonly config: VectorDBConfig;
+    protected readonly indexName: string;
+    constructor(config: VectorDBConfig);
+    /**
+     * Initialise the connection (create tables, verify index, etc.)
+     */
+    abstract initialize(): Promise<void>;
+    /**
+     * Insert or update a single vector + content pair.
+     */
+    abstract upsert(doc: UpsertDocument, namespace?: string): Promise<void>;
+    /**
+     * Batch upsert for efficient ingestion of many documents.
+     */
+    abstract batchUpsert(docs: UpsertDocument[], namespace?: string): Promise<void>;
+    /**
+     * Find the top-K most similar vectors to the query vector.
+     */
+    abstract query(vector: number[], topK: number, namespace?: string, filter?: Record<string, unknown>): Promise<VectorMatch[]>;
+    /**
+     * Delete a stored vector by ID.
+     */
+    abstract delete(id: string, namespace?: string): Promise<void>;
+    /**
+     * Delete all vectors in a namespace (full data reset for a project).
+     */
+    abstract deleteNamespace(namespace: string): Promise<void>;
+    /**
+     * Check if the underlying DB is reachable.
+     */
+    abstract ping(): Promise<boolean>;
+    /**
+     * Gracefully close the connection.
+     */
+    abstract disconnect(): Promise<void>;
+}
+
+/**
+ * ProviderRegistry — dynamic provider loader for Vector DBs and LLMs.
+ */
+declare class ProviderRegistry {
+    private static vectorProviders;
+    /**
+     * Register a custom vector provider.
+     */
+    static registerVectorProvider(name: string, providerClass: new (config: VectorDBConfig) => BaseVectorProvider): void;
+    /**
+     * Creates a vector database provider based on the configuration.
+     */
+    static createVectorProvider(config: VectorDBConfig): Promise<BaseVectorProvider>;
+    /**
+     * Creates an LLM provider based on the configuration.
+     */
+    static createLLMProvider(llmConfig: LLMConfig, embeddingConfig?: EmbeddingConfig): ILLMProvider;
 }
 
 /**
- * Pinecone Vector DB Adapter
+ * serverConfig.ts — reads RagConfig from environment variables at runtime.
  *
- * Required options in VectorDBConfig.options:
- *   - apiKey:       string  – Pinecone API key
- *   - environment?: string  – Pinecone environment (legacy, optional for new projects)
+ * This keeps secrets server-side and never exposes them to the browser.
+ * Consumer projects set these env vars in their .env.local file.
  */
 
-declare class PineconeAdapter implements IVectorDB {
+declare function getRagConfig(env?: Record<string, string | undefined>): RagConfig;
+
+declare class PineconeProvider extends BaseVectorProvider {
     private client;
-    private readonly indexName;
     private readonly apiKey;
     constructor(config: VectorDBConfig);
     initialize(): Promise<void>;
@@ -51,21 +160,13 @@ declare class PineconeAdapter implements IVectorDB {
 }
 
 /**
- * pgVector Adapter
- *
- * Stores embeddings in a PostgreSQL table using the pgvector extension.
- * Required options in VectorDBConfig.options:
- *   - connectionString: string  – PostgreSQL connection string
- *   - dimensions?:      number  – vector dimensions (default 1536 for OpenAI ada-002)
- *
- * The adapter auto-creates the table and index on initialize().
+ * PostgreSQLProvider — PostgreSQL implementation using the pgvector extension.
  */
-
-declare class PgVectorAdapter implements IVectorDB {
+declare class PostgreSQLProvider extends BaseVectorProvider {
     private pool;
-    private readonly tableName;
     private readonly dimensions;
     private readonly connectionString;
+    private readonly tableName;
     constructor(config: VectorDBConfig);
     initialize(): Promise<void>;
     upsert(doc: UpsertDocument, namespace?: string): Promise<void>;
@@ -78,33 +179,18 @@ declare class PgVectorAdapter implements IVectorDB {
 }
 
 /**
- * MongoDB Adapter for Atlas Vector Search.
- *
- * Requirements:
- * 1. A MongoDB Atlas cluster (v6.0.11+, v7.0.2+).
- * 2. A Vector Search Index created on the collection.
- *
- * Config Options:
- * - uri: string (Required)
- * - database: string (Required)
- * - collection: string (Required)
- * - indexName: string (Default: 'vector_index')
- * - embeddingKey: string (Default: 'embedding')
- * - contentKey: string (Default: 'content')
- * - metadataKey: string (Default: 'metadata')
- */
-declare class MongoDbAdapter implements IVectorDB {
+ * MongoDBProvider — MongoDB Atlas Vector Search implementation.
+ */
+declare class MongoDBProvider extends BaseVectorProvider {
     private client;
     private db?;
     private collection?;
-    private uri;
     private dbName;
     private collectionName;
-    private indexName;
     private embeddingKey;
     private contentKey;
     private metadataKey;
-    constructor(options: Record<string, unknown>);
+    constructor(config: VectorDBConfig);
     initialize(): Promise<void>;
     upsert(doc: UpsertDocument, namespace?: string): Promise<void>;
     batchUpsert(docs: UpsertDocument[], namespace?: string): Promise<void>;
@@ -115,6 +201,87 @@ declare class MongoDbAdapter implements IVectorDB {
     disconnect(): Promise<void>;
 }
 
+/**
+ * MilvusProvider — implementation for Milvus using its REST API.
+ */
+declare class MilvusProvider extends BaseVectorProvider {
+    private http;
+    constructor(config: VectorDBConfig);
+    initialize(): Promise<void>;
+    upsert(doc: UpsertDocument, namespace?: string): Promise<void>;
+    batchUpsert(docs: UpsertDocument[], namespace?: string): Promise<void>;
+    query(vector: number[], topK: number, namespace?: string, _filter?: Record<string, unknown>): Promise<VectorMatch[]>;
+    delete(id: string, _namespace?: string): Promise<void>;
+    deleteNamespace(_namespace: string): Promise<void>;
+    ping(): Promise<boolean>;
+    disconnect(): Promise<void>;
+}
+
+/**
+ * QdrantProvider — implementation for Qdrant using its REST API.
+ */
+declare class QdrantProvider extends BaseVectorProvider {
+    private http;
+    constructor(config: VectorDBConfig);
+    initialize(): Promise<void>;
+    upsert(doc: UpsertDocument, namespace?: string): Promise<void>;
+    batchUpsert(docs: UpsertDocument[], namespace?: string): Promise<void>;
+    query(vector: number[], topK: number, namespace?: string, _filter?: Record<string, unknown>): Promise<VectorMatch[]>;
+    delete(id: string, _namespace?: string): Promise<void>;
+    deleteNamespace(_namespace: string): Promise<void>;
+    ping(): Promise<boolean>;
+    disconnect(): Promise<void>;
+}
+
+/**
+ * ChromaDBProvider — implementation for ChromaDB using its REST API.
+ */
+declare class ChromaDBProvider extends BaseVectorProvider {
+    private http;
+    private collectionId;
+    constructor(config: VectorDBConfig);
+    initialize(): Promise<void>;
+    upsert(doc: UpsertDocument, namespace?: string): Promise<void>;
+    batchUpsert(docs: UpsertDocument[], namespace?: string): Promise<void>;
+    query(vector: number[], topK: number, namespace?: string, _filter?: Record<string, unknown>): Promise<VectorMatch[]>;
+    delete(id: string, _namespace?: string): Promise<void>;
+    deleteNamespace(_namespace: string): Promise<void>;
+    ping(): Promise<boolean>;
+    disconnect(): Promise<void>;
+}
+
+/**
+ * RedisProvider — implementation for Redis (e.g. Upstash) using its REST API.
+ */
+declare class RedisProvider extends BaseVectorProvider {
+    private http;
+    constructor(config: VectorDBConfig);
+    initialize(): Promise<void>;
+    upsert(doc: UpsertDocument, namespace?: string): Promise<void>;
+    batchUpsert(docs: UpsertDocument[], namespace?: string): Promise<void>;
+    query(vector: number[], topK: number, namespace?: string, _filter?: Record<string, unknown>): Promise<VectorMatch[]>;
+    delete(id: string, namespace?: string): Promise<void>;
+    deleteNamespace(_namespace: string): Promise<void>;
+    ping(): Promise<boolean>;
+    disconnect(): Promise<void>;
+}
+
+/**
+ * WeaviateProvider — implementation for Weaviate using its REST/GraphQL API.
+ */
+declare class WeaviateProvider extends BaseVectorProvider {
+    private http;
+    constructor(config: VectorDBConfig);
+    initialize(): Promise<void>;
+    upsert(doc: UpsertDocument, namespace?: string): Promise<void>;
+    batchUpsert(docs: UpsertDocument[], namespace?: string): Promise<void>;
+    query(vector: number[], topK: number, namespace?: string, _filter?: Record<string, unknown>): Promise<VectorMatch[]>;
+    delete(id: string, _namespace?: string): Promise<void>;
+    deleteNamespace(namespace: string): Promise<void>;
+    ping(): Promise<boolean>;
+    disconnect(): Promise<void>;
+}
+
 /**
  * LLMFactory — instantiates the correct ILLMProvider based on llmConfig.provider.
  *
@@ -208,4 +375,4 @@ declare class OllamaProvider implements ILLMProvider {
     ping(): Promise<boolean>;
 }
 
-export { AnthropicProvider, ChatMessage, ChatOptions, EmbedOptions, EmbeddingConfig, ILLMProvider, IVectorDB, LLMConfig, LLMFactory, MongoDbAdapter, OllamaProvider, OpenAIProvider, PgVectorAdapter, PineconeAdapter, RagConfig, UpsertDocument, VectorDBConfig, VectorDBFactory, VectorMatch, getRagConfig };
+export { AnthropicProvider, BaseVectorProvider, ChatMessage, ChatOptions, ChatResponse, ChromaDBProvider, ConfigResolver, EmbedOptions, EmbeddingConfig, ILLMProvider, IngestDocument, LLMConfig, LLMFactory, MilvusProvider, MongoDBProvider, OllamaProvider, OpenAIProvider, PineconeProvider, Pipeline, PostgreSQLProvider, ProviderRegistry, QdrantProvider, RagConfig, RedisProvider, UpsertDocument, VectorDBConfig, VectorMatch, VectorPlugin, WeaviateProvider, getRagConfig };