@retrivora-ai/rag-engine 0.1.5 → 0.1.6

package/dist/{chunk-S5DRHETN.mjs → chunk-IUTAZ7QR.mjs}

@@ -52,8 +52,37 @@ var PostgreSQLProvider = class extends BaseVectorProvider {
     );
   }
   async batchUpsert(docs, namespace = "") {
-    for (const doc of docs) {
-      await this.upsert(doc, namespace);
+    if (docs.length === 0) return;
+    const client = await this.pool.connect();
+    try {
+      await client.query("BEGIN");
+      const BATCH_SIZE = 50;
+      for (let i = 0; i < docs.length; i += BATCH_SIZE) {
+        const batch = docs.slice(i, i + BATCH_SIZE);
+        const values = [];
+        const valuePlaceholders = batch.map((doc, idx) => {
+          var _a;
+          const offset = idx * 5;
+          values.push(doc.id, namespace, doc.content, JSON.stringify((_a = doc.metadata) != null ? _a : {}), `[${doc.vector.join(",")}]`);
+          return `($${offset + 1}, $${offset + 2}, $${offset + 3}, $${offset + 4}, $${offset + 5}::vector)`;
+        }).join(", ");
+        const query = `
+          INSERT INTO ${this.tableName} (id, namespace, content, metadata, embedding)
+          VALUES ${valuePlaceholders}
+          ON CONFLICT (id) DO UPDATE
+            SET namespace = EXCLUDED.namespace,
+                content   = EXCLUDED.content,
+                metadata  = EXCLUDED.metadata,
+                embedding = EXCLUDED.embedding
+        `;
+        await client.query(query, values);
+      }
+      await client.query("COMMIT");
+    } catch (error) {
+      await client.query("ROLLBACK");
+      throw error;
+    } finally {
+      client.release();
     }
   }
   async query(vector, topK, namespace, filter) {

package/dist/{chunk-AALIF3AL.mjs → chunk-ZM6TYIDH.mjs}

@@ -31,7 +31,7 @@ var PineconeProvider = class extends BaseVectorProvider {
     var _a;
     await this.index(namespace).upsert({
       records: [{
-        id: doc.id,
+        id: String(doc.id),
         values: doc.vector,
         metadata: __spreadValues({ content: doc.content }, (_a = doc.metadata) != null ? _a : {})
       }]
@@ -43,7 +43,7 @@ var PineconeProvider = class extends BaseVectorProvider {
       const records = docs.slice(i, i + BATCH).map((d) => {
         var _a;
         return {
-          id: d.id,
+          id: String(d.id),
           values: d.vector,
           metadata: __spreadValues({ content: d.content }, (_a = d.metadata) != null ? _a : {})
         };
@@ -69,7 +69,7 @@ var PineconeProvider = class extends BaseVectorProvider {
     });
   }
   async delete(id, namespace) {
-    await this.index(namespace).deleteOne({ id });
+    await this.index(namespace).deleteOne({ id: String(id) });
   }
   async deleteNamespace(namespace) {
     await this.client.index(this.indexName).namespace(namespace).deleteAll();

package/dist/handlers/index.d.mts

@@ -1,44 +1,3 @@
-import { a as RagConfig, C as ChatResponse } from '../RagConfig-DG_0f8ka.mjs';
-import { NextRequest, NextResponse } from 'next/server';
-
-/**
- * createChatHandler — factory that returns a Next.js App Router POST handler
- */
-declare function createChatHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<NextResponse<{
-    error: string;
-}> | NextResponse<ChatResponse>>;
-/**
- * createIngestHandler — factory for the document ingestion endpoint.
- */
-declare function createIngestHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<NextResponse<{
-    error: string;
-}> | NextResponse<{
-    results: {
-        docId: string;
-        chunksIngested: number;
-    }[];
-}>>;
-/**
- * createHealthHandler — factory for the health-check endpoint.
- */
-declare function createHealthHandler(config?: Partial<RagConfig>): () => Promise<NextResponse<{
-    timestamp: string;
-    status: string;
-}> | NextResponse<{
-    status: string;
-    error: string;
-}>>;
-/**
- * createUploadHandler — factory for the file upload ingestion endpoint.
- */
-declare function createUploadHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<NextResponse<{
-    error: string;
-}> | NextResponse<{
-    message: string;
-    results: {
-        docId: string;
-        chunksIngested: number;
-    }[];
-}>>;
-
-export { createChatHandler, createHealthHandler, createIngestHandler, createUploadHandler };
+export { c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from '../index-Ck2pt7-8.mjs';
+import '../RagConfig-hBGXJmSx.mjs';
+import 'next/server';

package/dist/handlers/index.d.ts

@@ -1,44 +1,3 @@
-import { a as RagConfig, C as ChatResponse } from '../RagConfig-DG_0f8ka.js';
-import { NextRequest, NextResponse } from 'next/server';
-
-/**
- * createChatHandler — factory that returns a Next.js App Router POST handler
- */
-declare function createChatHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<NextResponse<{
-    error: string;
-}> | NextResponse<ChatResponse>>;
-/**
- * createIngestHandler — factory for the document ingestion endpoint.
- */
-declare function createIngestHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<NextResponse<{
-    error: string;
-}> | NextResponse<{
-    results: {
-        docId: string;
-        chunksIngested: number;
-    }[];
-}>>;
-/**
- * createHealthHandler — factory for the health-check endpoint.
- */
-declare function createHealthHandler(config?: Partial<RagConfig>): () => Promise<NextResponse<{
-    timestamp: string;
-    status: string;
-}> | NextResponse<{
-    status: string;
-    error: string;
-}>>;
-/**
- * createUploadHandler — factory for the file upload ingestion endpoint.
- */
-declare function createUploadHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<NextResponse<{
-    error: string;
-}> | NextResponse<{
-    message: string;
-    results: {
-        docId: string;
-        chunksIngested: number;
-    }[];
-}>>;
-
-export { createChatHandler, createHealthHandler, createIngestHandler, createUploadHandler };
+export { c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from '../index-Bx182KKn.js';
+import '../RagConfig-hBGXJmSx.js';
+import 'next/server';