@retrivora-ai/rag-engine 0.1.4 → 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-GD3QJFNN.mjs → chunk-JI6VD5TJ.mjs}

@@ -1,3 +1,7 @@
+import {
+  buildPayload,
+  resolvePath
+} from "./chunk-UKDXCXW7.mjs";
 import {
   __spreadProps,
   __spreadValues
@@ -192,46 +196,6 @@ ${context}`;
 // src/llm/providers/UniversalLLMAdapter.ts
 import axios2 from "axios";
 
-// src/utils/templateUtils.ts
-function isRecord(value) {
-  return typeof value === "object" && value !== null;
-}
-function resolvePath(obj, path) {
-  if (!path || !obj) return void 0;
-  return path.replace(/\[(\w+)\]/g, ".$1").replace(/^\./, "").split(".").reduce((current, segment) => {
-    if (!isRecord(current) && !Array.isArray(current)) {
-      return void 0;
-    }
-    return current[segment];
-  }, obj);
-}
-function buildPayload(template, vars) {
-  let raw = template;
-  for (const [key, val] of Object.entries(vars)) {
-    const stringifiedVal = val === void 0 ? "null" : JSON.stringify(val);
-    raw = raw.replace(new RegExp(`"\\{\\{${key}\\}\\}"`, "g"), stringifiedVal);
-    raw = raw.replace(new RegExp(`\\{\\{${key}\\}\\}`, "g"), stringifiedVal);
-  }
-  try {
-    return JSON.parse(raw);
-  } catch (err) {
-    throw new Error(`Failed to parse payload template: ${err.message}
-Raw payload: ${raw}`);
-  }
-}
-function mergeDefined(base, override) {
-  const merged = __spreadValues({}, base || {});
-  if (!override) {
-    return merged;
-  }
-  for (const [key, value] of Object.entries(override)) {
-    if (value !== void 0 && value !== null && value !== "") {
-      merged[key] = value;
-    }
-  }
-  return merged;
-}
-
 // src/config/UniversalProfiles.ts
 var OPENAI_BASE = {
   chatPath: "/chat/completions",
@@ -416,7 +380,6 @@ var LLMFactory = class _LLMFactory {
 };
 
 export {
-  mergeDefined,
   OpenAIProvider,
   AnthropicProvider,
   OllamaProvider,

package/dist/chunk-UKDXCXW7.mjs

@@ -0,0 +1,49 @@
+import {
+  __spreadValues
+} from "./chunk-I4E63NIC.mjs";
+
+// src/utils/templateUtils.ts
+function isRecord(value) {
+  return typeof value === "object" && value !== null;
+}
+function resolvePath(obj, path) {
+  if (!path || !obj) return void 0;
+  return path.replace(/\[(\w+)\]/g, ".$1").replace(/^\./, "").split(".").reduce((current, segment) => {
+    if (!isRecord(current) && !Array.isArray(current)) {
+      return void 0;
+    }
+    return current[segment];
+  }, obj);
+}
+function buildPayload(template, vars) {
+  let raw = template;
+  for (const [key, val] of Object.entries(vars)) {
+    const stringifiedVal = val === void 0 ? "null" : JSON.stringify(val);
+    raw = raw.replace(new RegExp(`"\\{\\{${key}\\}\\}"`, "g"), stringifiedVal);
+    raw = raw.replace(new RegExp(`\\{\\{${key}\\}\\}`, "g"), stringifiedVal);
+  }
+  try {
+    return JSON.parse(raw);
+  } catch (err) {
+    throw new Error(`Failed to parse payload template: ${err.message}
+Raw payload: ${raw}`);
+  }
+}
+function mergeDefined(base, override) {
+  const merged = __spreadValues({}, base || {});
+  if (!override) {
+    return merged;
+  }
+  for (const [key, value] of Object.entries(override)) {
+    if (value !== void 0 && value !== null && value !== "") {
+      merged[key] = value;
+    }
+  }
+  return merged;
+}
+
+export {
+  resolvePath,
+  buildPayload,
+  mergeDefined
+};

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 { e as RagConfig, b as ChatResponse } from '../Pipeline-Bo6CUCox.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 { e as RagConfig, b as ChatResponse } from '../Pipeline-Bo6CUCox.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';