@retrivora-ai/rag-engine 0.1.2 → 0.1.4

package/dist/index.mjs

@@ -1,7 +1,15 @@
 import {
-  __spreadValues,
+  ConfigResolver,
+  Pipeline,
+  ProviderRegistry,
+  VectorPlugin
+} from "./chunk-AIAB2IEE.mjs";
+import {
   mergeDefined
-} from "./chunk-ZPXLQR5Q.mjs";
+} from "./chunk-GD3QJFNN.mjs";
+import {
+  __spreadValues
+} from "./chunk-I4E63NIC.mjs";
 
 // src/components/ChatWidget.tsx
 import { useState as useState4 } from "react";
@@ -545,8 +553,12 @@ export {
   ChatWidget,
   ChatWindow,
   ConfigProvider,
+  ConfigResolver,
   MessageBubble,
+  Pipeline,
+  ProviderRegistry,
   SourceCard,
+  VectorPlugin,
   useConfig,
   useRagChat
 };

package/dist/server.d.mts

@@ -1,6 +1,7 @@
-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 { e as RagConfig, g as VectorDBConfig, f as UpsertDocument, V as VectorMatch, L as LLMConfig, c as EmbeddingConfig, I as ILLMProvider, C as ChatMessage, a as ChatOptions, E as EmbedOptions } from './Pipeline-Bo6CUCox.mjs';
+export { b as ChatResponse, d as IngestDocument, P as Pipeline, R as RAGConfig, U as UIConfig } from './Pipeline-Bo6CUCox.mjs';
+import { B as BaseVectorProvider } from './DocumentChunker-D9-fObJp.mjs';
+export { C as Chunk, a as ChunkOptions, b as ConfigResolver, D as DocumentChunker, P as ProviderRegistry, V as VectorPlugin } from './DocumentChunker-D9-fObJp.mjs';
 export { createChatHandler, createHealthHandler, createIngestHandler, createUploadHandler } from './handlers/index.mjs';
 import 'next/server';
 
@@ -13,30 +14,8 @@ import 'next/server';
 
 declare function getRagConfig(env?: Record<string, string | undefined>): RagConfig;
 
-/**
- * VectorDBFactory — instantiates the correct IVectorDB adapter
- * based on the VectorDBConfig.provider value.
- *
- * Usage:
- *   const db = VectorDBFactory.create(config.vectorDb);
- *   await db.initialize();
- */
-
-declare class VectorDBFactory {
-    static create(config: VectorDBConfig): IVectorDB;
-}
-
-/**
- * Pinecone Vector DB Adapter
- *
- * Required options in VectorDBConfig.options:
- *   - apiKey:       string  – Pinecone API key
- *   - environment?: string  – Pinecone environment (legacy, optional for new projects)
- */
-
-declare class PineconeAdapter implements IVectorDB {
+declare class PineconeProvider extends BaseVectorProvider {
     private client;
-    private readonly indexName;
     private readonly apiKey;
     constructor(config: VectorDBConfig);
     initialize(): Promise<void>;
@@ -51,21 +30,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 +49,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')
+ * MongoDBProvider — MongoDB Atlas Vector Search implementation.
  */
-declare class MongoDbAdapter implements IVectorDB {
+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 +71,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 +245,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, ChromaDBProvider, EmbedOptions, EmbeddingConfig, ILLMProvider, LLMConfig, LLMFactory, MilvusProvider, MongoDBProvider, OllamaProvider, OpenAIProvider, PineconeProvider, PostgreSQLProvider, QdrantProvider, RagConfig, RedisProvider, UpsertDocument, VectorDBConfig, VectorMatch, WeaviateProvider, getRagConfig };

package/dist/server.d.ts

@@ -1,6 +1,7 @@
-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.js';
-export { b as ChatResponse, e as IngestDocument, R as RAGConfig, i as RAGPipeline, U as UIConfig } from './RAGPipeline-BmkIv1HD.js';
-export { C as Chunk, a as ChunkOptions, D as DocumentChunker } from './DocumentChunker-BUrIrcPk.js';
+import { e as RagConfig, g as VectorDBConfig, f as UpsertDocument, V as VectorMatch, L as LLMConfig, c as EmbeddingConfig, I as ILLMProvider, C as ChatMessage, a as ChatOptions, E as EmbedOptions } from './Pipeline-Bo6CUCox.js';
+export { b as ChatResponse, d as IngestDocument, P as Pipeline, R as RAGConfig, U as UIConfig } from './Pipeline-Bo6CUCox.js';
+import { B as BaseVectorProvider } from './DocumentChunker-BQ5kQD7B.js';
+export { C as Chunk, a as ChunkOptions, b as ConfigResolver, D as DocumentChunker, P as ProviderRegistry, V as VectorPlugin } from './DocumentChunker-BQ5kQD7B.js';
 export { createChatHandler, createHealthHandler, createIngestHandler, createUploadHandler } from './handlers/index.js';
 import 'next/server';
 
@@ -13,30 +14,8 @@ import 'next/server';
 
 declare function getRagConfig(env?: Record<string, string | undefined>): RagConfig;
 
-/**
- * VectorDBFactory — instantiates the correct IVectorDB adapter
- * based on the VectorDBConfig.provider value.
- *
- * Usage:
- *   const db = VectorDBFactory.create(config.vectorDb);
- *   await db.initialize();
- */
-
-declare class VectorDBFactory {
-    static create(config: VectorDBConfig): IVectorDB;
-}
-
-/**
- * Pinecone Vector DB Adapter
- *
- * Required options in VectorDBConfig.options:
- *   - apiKey:       string  – Pinecone API key
- *   - environment?: string  – Pinecone environment (legacy, optional for new projects)
- */
-
-declare class PineconeAdapter implements IVectorDB {
+declare class PineconeProvider extends BaseVectorProvider {
     private client;
-    private readonly indexName;
     private readonly apiKey;
     constructor(config: VectorDBConfig);
     initialize(): Promise<void>;
@@ -51,21 +30,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 +49,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')
+ * MongoDBProvider — MongoDB Atlas Vector Search implementation.
  */
-declare class MongoDbAdapter implements IVectorDB {
+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 +71,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 +245,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, ChromaDBProvider, EmbedOptions, EmbeddingConfig, ILLMProvider, LLMConfig, LLMFactory, MilvusProvider, MongoDBProvider, OllamaProvider, OpenAIProvider, PineconeProvider, PostgreSQLProvider, QdrantProvider, RagConfig, RedisProvider, UpsertDocument, VectorDBConfig, VectorMatch, WeaviateProvider, getRagConfig };