@retrivora-ai/rag-engine 1.7.9 → 1.8.1

package/dist/server.d.mts

@@ -1,8 +1,7 @@
-import { j as VectorDBConfig, L as LLMConfig, E as EmbeddingConfig, h as RagConfig, I as IngestDocument, C as ChatMessage, d as ChatResponse, l as RetrievalResult, i as UpsertDocument, V as VectorMatch, G as GraphDBConfig, m as GraphNode, n as Edge, o as GraphSearchResult, k as VectorDBProvider, f as LLMProvider, e as EmbeddingProvider, g as RAGConfig, U as UIConfig, c as ChatOptions } from './index-wCRqMtdX.mjs';
-import { I as ILLMProvider, E as EmbedOptions } from './DocumentChunker-DCuxrOdM.mjs';
-export { C as Chunk, a as ChunkOptions, D as DocumentChunker } from './DocumentChunker-DCuxrOdM.mjs';
-import { H as HealthCheckResult, I as IProviderValidator, a as IProviderHealthChecker } from './index-DhsG2o5q.mjs';
-export { C as ConfigValidator, V as ValidationError, b as VectorPlugin, c as createChatHandler, d as createHealthHandler, e as createIngestHandler, f as createStreamHandler, g as createUploadHandler, s as sseErrorFrame, h as sseFrame, i as sseMetaFrame, j as sseTextFrame } from './index-DhsG2o5q.mjs';
+import { l as VectorDBConfig, L as LLMConfig, e as EmbeddingConfig, j as RagConfig, I as ILLMProvider, g as IngestDocument, C as ChatMessage, d as ChatResponse, n as RetrievalResult, k as UpsertDocument, V as VectorMatch, G as GraphDBConfig, o as GraphNode, p as Edge, q as GraphSearchResult, m as VectorDBProvider, h as LLMProvider, f as EmbeddingProvider, i as RAGConfig, U as UIConfig, c as ChatOptions, E as EmbedOptions } from './ILLMProvider-BfRgI1Xh.mjs';
+export { C as Chunk, a as ChunkOptions, D as DocumentChunker } from './DocumentChunker-Dh9TvmGG.mjs';
+import { H as HealthCheckResult, I as IProviderValidator, a as IProviderHealthChecker } from './index-BV0z5mb6.mjs';
+export { C as ConfigValidator, V as ValidationError, b as VectorPlugin, c as createChatHandler, d as createHealthHandler, e as createIngestHandler, f as createStreamHandler, g as createUploadHandler, s as sseErrorFrame, h as sseFrame, i as sseMetaFrame, j as sseTextFrame } from './index-BV0z5mb6.mjs';
 import 'next/server';
 
 /**
@@ -122,6 +121,11 @@ declare class Pipeline {
     private embeddingCache;
     private initialised;
     constructor(config: RagConfig);
+    /**
+     * Expose the underlying LLM provider (set after initialize()).
+     * Used by the stream handler to pass to UITransformer.analyzeAndDecide().
+     */
+    getLLMProvider(): ILLMProvider | undefined;
     initialize(): Promise<void>;
     /**
      * Ingest documents with automatic chunking, embedding, and batch upsert.
@@ -158,6 +162,7 @@ declare class Pipeline {
      * Universal retrieval method combining all enabled providers.
      * Uses an LRU-bounded embedding cache to avoid re-embedding the same query.
      */
+    private generateUiTransformation;
     retrieve(query: string, options: {
         namespace?: string;
         topK?: number;
@@ -871,6 +876,31 @@ declare class UniversalVectorProvider extends BaseVectorProvider {
  * LLMFactory — instantiates the correct ILLMProvider based on llmConfig.provider.
  */
 declare class LLMFactory {
+    /**
+     * Register a custom LLM provider factory at runtime.
+     *
+     * Use this to add support for any LLM backend (Azure OpenAI, Cohere, Mistral,
+     * Bedrock, etc.) without modifying this library's source code.
+     *
+     * @example
+     * // In your Next.js app initialization:
+     * import { LLMFactory } from '@retrivora-ai/rag-engine/server';
+     * import { MyCustomProvider } from './providers/MyCustomProvider';
+     *
+     * LLMFactory.register('my-provider', (config) => new MyCustomProvider(config));
+     *
+     * // Then set in your .env.local:
+     * // LLM_PROVIDER=my-provider
+     */
+    static register(name: string, factory: (config: LLMConfig) => ILLMProvider): void;
+    /**
+     * Unregister a previously registered custom provider.
+     */
+    static unregister(name: string): void;
+    /**
+     * List all registered provider names (built-in + custom).
+     */
+    static listProviders(): string[];
     static create(llmConfig: LLMConfig, embeddingConfig?: EmbeddingConfig): ILLMProvider;
     static getValidator(provider: LLMProvider): IProviderValidator | null;
     static getHealthChecker(provider: LLMProvider): IProviderHealthChecker | null;

package/dist/server.d.ts

@@ -1,8 +1,7 @@
-import { j as VectorDBConfig, L as LLMConfig, E as EmbeddingConfig, h as RagConfig, I as IngestDocument, C as ChatMessage, d as ChatResponse, l as RetrievalResult, i as UpsertDocument, V as VectorMatch, G as GraphDBConfig, m as GraphNode, n as Edge, o as GraphSearchResult, k as VectorDBProvider, f as LLMProvider, e as EmbeddingProvider, g as RAGConfig, U as UIConfig, c as ChatOptions } from './index-wCRqMtdX.js';
-import { I as ILLMProvider, E as EmbedOptions } from './DocumentChunker-Bmscbh-X.js';
-export { C as Chunk, a as ChunkOptions, D as DocumentChunker } from './DocumentChunker-Bmscbh-X.js';
-import { H as HealthCheckResult, I as IProviderValidator, a as IProviderHealthChecker } from './index-BDqz2_Yu.js';
-export { C as ConfigValidator, V as ValidationError, b as VectorPlugin, c as createChatHandler, d as createHealthHandler, e as createIngestHandler, f as createStreamHandler, g as createUploadHandler, s as sseErrorFrame, h as sseFrame, i as sseMetaFrame, j as sseTextFrame } from './index-BDqz2_Yu.js';
+import { l as VectorDBConfig, L as LLMConfig, e as EmbeddingConfig, j as RagConfig, I as ILLMProvider, g as IngestDocument, C as ChatMessage, d as ChatResponse, n as RetrievalResult, k as UpsertDocument, V as VectorMatch, G as GraphDBConfig, o as GraphNode, p as Edge, q as GraphSearchResult, m as VectorDBProvider, h as LLMProvider, f as EmbeddingProvider, i as RAGConfig, U as UIConfig, c as ChatOptions, E as EmbedOptions } from './ILLMProvider-BfRgI1Xh.js';
+export { C as Chunk, a as ChunkOptions, D as DocumentChunker } from './DocumentChunker-Dh9TvmGG.js';
+import { H as HealthCheckResult, I as IProviderValidator, a as IProviderHealthChecker } from './index-1Z4GuYBi.js';
+export { C as ConfigValidator, V as ValidationError, b as VectorPlugin, c as createChatHandler, d as createHealthHandler, e as createIngestHandler, f as createStreamHandler, g as createUploadHandler, s as sseErrorFrame, h as sseFrame, i as sseMetaFrame, j as sseTextFrame } from './index-1Z4GuYBi.js';
 import 'next/server';
 
 /**
@@ -122,6 +121,11 @@ declare class Pipeline {
     private embeddingCache;
     private initialised;
     constructor(config: RagConfig);
+    /**
+     * Expose the underlying LLM provider (set after initialize()).
+     * Used by the stream handler to pass to UITransformer.analyzeAndDecide().
+     */
+    getLLMProvider(): ILLMProvider | undefined;
     initialize(): Promise<void>;
     /**
      * Ingest documents with automatic chunking, embedding, and batch upsert.
@@ -158,6 +162,7 @@ declare class Pipeline {
      * Universal retrieval method combining all enabled providers.
      * Uses an LRU-bounded embedding cache to avoid re-embedding the same query.
      */
+    private generateUiTransformation;
     retrieve(query: string, options: {
         namespace?: string;
         topK?: number;
@@ -871,6 +876,31 @@ declare class UniversalVectorProvider extends BaseVectorProvider {
  * LLMFactory — instantiates the correct ILLMProvider based on llmConfig.provider.
  */
 declare class LLMFactory {
+    /**
+     * Register a custom LLM provider factory at runtime.
+     *
+     * Use this to add support for any LLM backend (Azure OpenAI, Cohere, Mistral,
+     * Bedrock, etc.) without modifying this library's source code.
+     *
+     * @example
+     * // In your Next.js app initialization:
+     * import { LLMFactory } from '@retrivora-ai/rag-engine/server';
+     * import { MyCustomProvider } from './providers/MyCustomProvider';
+     *
+     * LLMFactory.register('my-provider', (config) => new MyCustomProvider(config));
+     *
+     * // Then set in your .env.local:
+     * // LLM_PROVIDER=my-provider
+     */
+    static register(name: string, factory: (config: LLMConfig) => ILLMProvider): void;
+    /**
+     * Unregister a previously registered custom provider.
+     */
+    static unregister(name: string): void;
+    /**
+     * List all registered provider names (built-in + custom).
+     */
+    static listProviders(): string[];
     static create(llmConfig: LLMConfig, embeddingConfig?: EmbeddingConfig): ILLMProvider;
     static getValidator(provider: LLMProvider): IProviderValidator | null;
     static getHealthChecker(provider: LLMProvider): IProviderHealthChecker | null;