@retrivora-ai/rag-engine 0.1.2 → 0.1.4

package/src/test-refactor.ts

@@ -0,0 +1,59 @@
+import { VectorPlugin } from './core/VectorPlugin';
+
+async function testRefactor() {
+  console.log('🚀 Starting VectorPlugin Refactor Test...\n');
+
+  // Test 1: Initialization with Empty Config (should use env defaults or fallbacks)
+  console.log('--- Test 1: Default Config ---');
+  try {
+    const pluginDefault = new VectorPlugin();
+    const config = pluginDefault.getConfig();
+    console.log('✅ Resolved Project ID:', config.projectId);
+    console.log('✅ Resolved Vector DB Provider:', config.vectorDb.provider);
+  } catch (err) {
+    console.error('❌ Test 1 Failed:', (err as Error).message);
+  }
+
+  // Test 2: Overriding Config from Host
+  console.log('\n--- Test 2: Host Override Config ---');
+  try {
+    const pluginOverride = new VectorPlugin({
+      projectId: 'host-override-test',
+      vectorDb: {
+        provider: 'milvus',
+        indexName: 'milvus-index',
+        options: { uri: 'http://localhost:19530' }
+      }
+    });
+    const config = pluginOverride.getConfig();
+    console.log('✅ Resolved Project ID (Overridden):', config.projectId);
+    console.log('✅ Resolved Vector DB Provider (Overridden):', config.vectorDb.provider);
+  } catch (err) {
+    console.error('❌ Test 2 Failed:', (err as Error).message);
+  }
+
+  // Test 3: Provider Registry Loading
+  console.log('\n--- Test 3: Provider Registry Loading ---');
+  try {
+    const { ProviderRegistry } = await import('./core/ProviderRegistry');
+    const milvusProvider = await ProviderRegistry.createVectorProvider({
+      provider: 'milvus',
+      indexName: 'test',
+      options: { uri: 'http://localhost' }
+    });
+    console.log('✅ Milvus Provider Instantiated:', milvusProvider.constructor.name);
+    
+    const redisProvider = await ProviderRegistry.createVectorProvider({
+      provider: 'redis',
+      indexName: 'test',
+      options: { baseUrl: 'http://localhost', apiKey: 'test' }
+    });
+    console.log('✅ Redis Provider Instantiated:', redisProvider.constructor.name);
+  } catch (err) {
+    console.error('❌ Test 3 Failed:', (err as Error).message);
+  }
+
+  console.log('\n✨ Refactor Test Completed.');
+}
+
+testRefactor().catch(console.error);

package/src/types/index.ts

@@ -0,0 +1,13 @@
+export interface VectorMatch {
+  id: string;
+  score: number;
+  content: string;
+  metadata?: Record<string, unknown>;
+}
+
+export interface UpsertDocument {
+  id: string;
+  vector: number[];
+  content: string;
+  metadata?: Record<string, unknown>;
+}

package/src/utils/templateUtils.ts

@@ -58,19 +58,19 @@ export function buildPayload(template: string, vars: Record<string, unknown>): u
 /**
  * Merges two objects, ignoring undefined, null, or empty string values in the override.
  */
-export function mergeDefined<T extends Record<string, unknown>>(
-  base: T,
+export function mergeDefined<T extends object>(
+  base: T | undefined | null,
   override?: Partial<T>
 ): T {
+  const merged = { ...(base || {}) } as T;
+
   if (!override) {
-    return { ...base };
+    return merged;
   }
 
-  const merged = { ...base };
-
   for (const [key, value] of Object.entries(override)) {
     if (value !== undefined && value !== null && value !== '') {
-      merged[key as keyof T] = value as T[keyof T];
+      (merged as Record<string, unknown>)[key] = value;
     }
   }
 

package/dist/DocumentChunker-BUrIrcPk.d.mts

@@ -1,43 +0,0 @@
-/**
- * DocumentChunker — splits long text into overlapping chunks
- * suitable for vector embedding and retrieval.
- *
- * Strategy: sentence-aware sliding window
- *   1. Split on sentence boundaries
- *   2. Accumulate sentences into chunks up to `chunkSize` characters
- *   3. Carry over `chunkOverlap` characters from the previous chunk
- */
-interface Chunk {
-    id: string;
-    content: string;
-    metadata?: Record<string, unknown>;
-}
-interface ChunkOptions {
-    /** Target chunk size in characters (default 1000) */
-    chunkSize?: number;
-    /** Overlap between consecutive chunks in characters (default 200) */
-    chunkOverlap?: number;
-    /** Source document identifier used as ID prefix */
-    docId?: string;
-    /** Extra metadata to attach to every chunk */
-    metadata?: Record<string, unknown>;
-}
-declare class DocumentChunker {
-    private readonly chunkSize;
-    private readonly chunkOverlap;
-    constructor(chunkSize?: number, chunkOverlap?: number);
-    /**
-     * Split a single text string into overlapping chunks.
-     */
-    chunk(text: string, options?: ChunkOptions): Chunk[];
-    /**
-     * Chunk multiple documents at once.
-     */
-    chunkMany(documents: Array<{
-        content: string;
-        docId?: string;
-        metadata?: Record<string, unknown>;
-    }>): Chunk[];
-}
-
-export { type Chunk as C, DocumentChunker as D, type ChunkOptions as a };

package/dist/DocumentChunker-BUrIrcPk.d.ts

@@ -1,43 +0,0 @@
-/**
- * DocumentChunker — splits long text into overlapping chunks
- * suitable for vector embedding and retrieval.
- *
- * Strategy: sentence-aware sliding window
- *   1. Split on sentence boundaries
- *   2. Accumulate sentences into chunks up to `chunkSize` characters
- *   3. Carry over `chunkOverlap` characters from the previous chunk
- */
-interface Chunk {
-    id: string;
-    content: string;
-    metadata?: Record<string, unknown>;
-}
-interface ChunkOptions {
-    /** Target chunk size in characters (default 1000) */
-    chunkSize?: number;
-    /** Overlap between consecutive chunks in characters (default 200) */
-    chunkOverlap?: number;
-    /** Source document identifier used as ID prefix */
-    docId?: string;
-    /** Extra metadata to attach to every chunk */
-    metadata?: Record<string, unknown>;
-}
-declare class DocumentChunker {
-    private readonly chunkSize;
-    private readonly chunkOverlap;
-    constructor(chunkSize?: number, chunkOverlap?: number);
-    /**
-     * Split a single text string into overlapping chunks.
-     */
-    chunk(text: string, options?: ChunkOptions): Chunk[];
-    /**
-     * Chunk multiple documents at once.
-     */
-    chunkMany(documents: Array<{
-        content: string;
-        docId?: string;
-        metadata?: Record<string, unknown>;
-    }>): Chunk[];
-}
-
-export { type Chunk as C, DocumentChunker as D, type ChunkOptions as a };