@retrivora-ai/rag-engine 1.9.6 → 1.9.7

package/dist/server.js

@@ -10,9 +10,6 @@ var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __propIsEnum = Object.prototype.propertyIsEnumerable;
 var __knownSymbol = (name, symbol) => (symbol = Symbol[name]) ? symbol : /* @__PURE__ */ Symbol.for("Symbol." + name);
-var __typeError = (msg) => {
-  throw TypeError(msg);
-};
 var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
 var __spreadValues = (a, b) => {
   for (var prop in b || (b = {}))
@@ -77,34 +74,6 @@ var __asyncGenerator = (__this, __arguments, generator) => {
   }, method = (k, call, wait, clear) => it[k] = (x) => (call = new Promise((yes, no, run) => (run = () => resume(k, x, yes, no), q ? q.then(run) : run())), clear = () => q === wait && (q = 0), q = wait = call.then(clear, clear), call), q, it = {};
   return generator = generator.apply(__this, __arguments), it[__knownSymbol("asyncIterator")] = () => it, method("next"), method("throw"), method("return"), it;
 };
-var __yieldStar = (value) => {
-  var obj = value[__knownSymbol("asyncIterator")], isAwait = false, method, it = {};
-  if (obj == null) {
-    obj = value[__knownSymbol("iterator")]();
-    method = (k) => it[k] = (x) => obj[k](x);
-  } else {
-    obj = obj.call(value);
-    method = (k) => it[k] = (v) => {
-      if (isAwait) {
-        isAwait = false;
-        if (k === "throw") throw v;
-        return v;
-      }
-      isAwait = true;
-      return {
-        done: false,
-        value: new __await(new Promise((resolve) => {
-          var x = obj[k](v);
-          if (!(x instanceof Object)) __typeError("Object expected");
-          resolve(x);
-        }), 1)
-      };
-    };
-  }
-  return it[__knownSymbol("iterator")] = () => it, method("next"), "throw" in obj ? method("throw") : it.throw = (x) => {
-    throw x;
-  }, "return" in obj && method("return"), it;
-};
 var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")]) ? it.call(obj) : (obj = obj[__knownSymbol("iterator")](), it = {}, method = (key, fn) => (fn = obj[key]) && (it[key] = (arg) => new Promise((yes, no, done) => (arg = fn.call(obj, arg), done = arg.done, Promise.resolve(arg.value).then((value) => yes({ value, done }), no)))), method("next"), method("return"), it);
 
 // src/utils/templateUtils.ts
@@ -2144,13 +2113,15 @@ __export(server_exports, {
   createFromPreset: () => createFromPreset,
   createHealthHandler: () => createHealthHandler,
   createIngestHandler: () => createIngestHandler,
+  createRagHandler: () => createRagHandler,
   createStreamHandler: () => createStreamHandler,
   createUploadHandler: () => createUploadHandler,
   getRagConfig: () => getRagConfig,
   sseErrorFrame: () => sseErrorFrame,
   sseFrame: () => sseFrame,
   sseMetaFrame: () => sseMetaFrame,
-  sseTextFrame: () => sseTextFrame
+  sseTextFrame: () => sseTextFrame,
+  wrapError: () => wrapError
 });
 module.exports = __toCommonJS(server_exports);
 
@@ -2375,46 +2346,6 @@ function getEnvConfig(env = process.env, base) {
   } : {});
 }
 
-// src/exceptions/index.ts
-var RetrivoraError = class extends Error {
-  constructor(message, code, details) {
-    super(message);
-    this.name = new.target.name;
-    this.code = code;
-    this.details = details;
-  }
-};
-var ProviderNotFoundException = class extends RetrivoraError {
-  constructor(providerType, provider, details) {
-    super(`Unsupported ${providerType} provider: ${provider}`, "PROVIDER_NOT_FOUND", details);
-  }
-};
-var EmbeddingFailedException = class extends RetrivoraError {
-  constructor(message = "Embedding generation failed", details) {
-    super(message, "EMBEDDING_FAILED", details);
-  }
-};
-var RetrievalException = class extends RetrivoraError {
-  constructor(message = "Retrieval failed", details) {
-    super(message, "RETRIEVAL_FAILED", details);
-  }
-};
-var RateLimitException = class extends RetrivoraError {
-  constructor(message = "Provider rate limit exceeded", details) {
-    super(message, "RATE_LIMITED", details);
-  }
-};
-var ConfigurationException = class extends RetrivoraError {
-  constructor(message, details) {
-    super(message, "CONFIGURATION_ERROR", details);
-  }
-};
-var AuthenticationException = class extends RetrivoraError {
-  constructor(message = "Provider authentication failed", details) {
-    super(message, "AUTHENTICATION_ERROR", details);
-  }
-};
-
 // src/core/ConfigResolver.ts
 var ConfigResolver = class {
   /**
@@ -2460,13 +2391,13 @@ var ConfigResolver = class {
    */
   static validate(config) {
     if (!config.projectId) {
-      throw new ConfigurationException("[ConfigResolver] projectId is required");
+      throw new Error("[ConfigResolver] projectId is required");
     }
     if (!config.vectorDb.provider) {
-      throw new ConfigurationException("[ConfigResolver] vectorDb.provider is required");
+      throw new Error("[ConfigResolver] vectorDb.provider is required");
     }
     if (!config.llm.provider) {
-      throw new ConfigurationException("[ConfigResolver] llm.provider is required");
+      throw new Error("[ConfigResolver] llm.provider is required");
     }
   }
   static mergeRetrievalWorkflow(rag, retrieval, workflow) {
@@ -3546,6 +3477,78 @@ ${context != null ? context : "None"}` },
   }
 };
 
+// src/exceptions/index.ts
+var RetrivoraError = class extends Error {
+  constructor(message, code, details) {
+    super(message);
+    this.name = new.target.name;
+    this.code = code;
+    this.details = details;
+  }
+};
+var ProviderNotFoundException = class extends RetrivoraError {
+  constructor(providerType, provider, details) {
+    super(`Unsupported ${providerType} provider: ${provider}`, "PROVIDER_NOT_FOUND", details);
+  }
+};
+var EmbeddingFailedException = class extends RetrivoraError {
+  constructor(message = "Embedding generation failed", details) {
+    super(message, "EMBEDDING_FAILED", details);
+  }
+};
+var RetrievalException = class extends RetrivoraError {
+  constructor(message = "Retrieval failed", details) {
+    super(message, "RETRIEVAL_FAILED", details);
+  }
+};
+var RateLimitException = class extends RetrivoraError {
+  constructor(message = "Provider rate limit exceeded", details) {
+    super(message, "RATE_LIMITED", details);
+  }
+};
+var ConfigurationException = class extends RetrivoraError {
+  constructor(message, details) {
+    super(message, "CONFIGURATION_ERROR", details);
+  }
+};
+var AuthenticationException = class extends RetrivoraError {
+  constructor(message = "Provider authentication failed", details) {
+    super(message, "AUTHENTICATION_ERROR", details);
+  }
+};
+function wrapError(err, defaultCode, defaultMessage) {
+  var _a;
+  if (err instanceof RetrivoraError) {
+    return err;
+  }
+  const error = err;
+  const message = (error == null ? void 0 : error.message) || defaultMessage || String(err);
+  const status = (error == null ? void 0 : error.status) || (error == null ? void 0 : error.statusCode) || ((_a = error == null ? void 0 : error.response) == null ? void 0 : _a.status);
+  const code = error == null ? void 0 : error.code;
+  if (status === 429 || /rate[- ]?limit/i.test(message) || code === "RATE_LIMIT_EXCEEDED") {
+    return new RateLimitException(message, err);
+  }
+  if (status === 401 || status === 403 || /unauthorized|auth|api[- ]?key/i.test(message) || code === "INVALID_API_KEY") {
+    return new AuthenticationException(message, err);
+  }
+  switch (defaultCode) {
+    case "PROVIDER_NOT_FOUND":
+      return new ProviderNotFoundException("provider", message, err);
+    case "EMBEDDING_FAILED":
+      return new EmbeddingFailedException(message, err);
+    case "RETRIEVAL_FAILED":
+      return new RetrievalException(message, err);
+    case "RATE_LIMITED":
+      return new RateLimitException(message, err);
+    case "CONFIGURATION_ERROR":
+      return new ConfigurationException(message, err);
+    case "AUTHENTICATION_ERROR":
+      return new AuthenticationException(message, err);
+    default:
+      return new RetrivoraError(message, defaultCode, err);
+  }
+}
+
 // src/llm/LLMFactory.ts
 var customProviders = /* @__PURE__ */ new Map();
 var LLMFactory = class _LLMFactory {
@@ -3591,7 +3594,7 @@ var LLMFactory = class _LLMFactory {
     ];
   }
   static create(llmConfig, embeddingConfig) {
-    var _a, _b;
+    var _a, _b, _c;
     switch (llmConfig.provider) {
       case "openai":
         return new OpenAIProvider(llmConfig, embeddingConfig);
@@ -3615,9 +3618,12 @@ var LLMFactory = class _LLMFactory {
           return new UniversalLLMAdapter(llmConfig);
         }
         throw new ProviderNotFoundException(
-          "LLM",
-          String(llmConfig.provider),
-          `[LLMFactory] Unknown provider "${llmConfig.provider}". Built-in providers: ${_LLMFactory.listProviders().join(", ")}. Register a custom provider with LLMFactory.register().`
+          "llm",
+          (_c = llmConfig.provider) != null ? _c : "undefined",
+          {
+            message: `Unknown provider "${llmConfig.provider}". Register a custom provider with LLMFactory.register().`,
+            available: _LLMFactory.listProviders()
+          }
         );
       }
     }
@@ -6552,7 +6558,7 @@ var Pipeline = class {
       embedBatchOptions
     );
     if (vectors.length !== chunks.length) {
-      throw new Error(
+      throw new EmbeddingFailedException(
         `[Pipeline] Embedding mismatch: got ${vectors.length} vectors for ${chunks.length} chunks. Check embedding provider logs for individual chunk failures.`
       );
     }
@@ -7163,6 +7169,82 @@ Suggestions:`;
   }
 };
 
+// src/core/Retrivora.ts
+var Retrivora = class {
+  constructor(config) {
+    this.config = ConfigResolver.resolveUniversal(config);
+    this.pipeline = new Pipeline(this.config);
+  }
+  async initialize() {
+    try {
+      await ConfigValidator.validateAndThrow(this.config);
+    } catch (err) {
+      throw wrapError(err, "CONFIGURATION_ERROR");
+    }
+    try {
+      await this.pipeline.initialize();
+    } catch (err) {
+      throw wrapError(err, "AUTHENTICATION_ERROR");
+    }
+  }
+  async ingest(documents, namespace) {
+    try {
+      return await this.pipeline.ingest(documents, namespace);
+    } catch (err) {
+      const msg = String(err);
+      let defaultCode = "RETRIEVAL_FAILED";
+      if (msg.includes("Embed") || msg.includes("embed")) {
+        defaultCode = "EMBEDDING_FAILED";
+      }
+      throw wrapError(err, defaultCode);
+    }
+  }
+  async ask(question, history = [], namespace) {
+    try {
+      return await this.pipeline.ask(question, history, namespace);
+    } catch (err) {
+      const msg = String(err);
+      let defaultCode = "RETRIEVAL_FAILED";
+      if (msg.includes("Embed") || msg.includes("embed")) {
+        defaultCode = "EMBEDDING_FAILED";
+      }
+      throw wrapError(err, defaultCode);
+    }
+  }
+  askStream(_0) {
+    return __asyncGenerator(this, arguments, function* (question, history = [], namespace) {
+      try {
+        const stream = this.pipeline.askStream(question, history, namespace);
+        try {
+          for (var iter = __forAwait(stream), more, temp, error; more = !(temp = yield new __await(iter.next())).done; more = false) {
+            const chunk = temp.value;
+            yield chunk;
+          }
+        } catch (temp) {
+          error = [temp];
+        } finally {
+          try {
+            more && (temp = iter.return) && (yield new __await(temp.call(iter)));
+          } finally {
+            if (error)
+              throw error[0];
+          }
+        }
+      } catch (err) {
+        const msg = String(err);
+        let defaultCode = "RETRIEVAL_FAILED";
+        if (msg.includes("Embed") || msg.includes("embed")) {
+          defaultCode = "EMBEDDING_FAILED";
+        }
+        throw wrapError(err, defaultCode);
+      }
+    });
+  }
+  getPipeline() {
+    return this.pipeline;
+  }
+};
+
 // src/core/ProviderHealthCheck.ts
 var ProviderHealthCheck = class {
   /**
@@ -7271,6 +7353,8 @@ var VectorPlugin = class {
   constructor(hostConfig) {
     this.config = ConfigResolver.resolve(hostConfig);
     this.validationPromise = ConfigValidator.validateAndThrow(this.config);
+    this.validationPromise.catch(() => {
+    });
     this.pipeline = new Pipeline(this.config);
   }
   /**
@@ -7304,55 +7388,93 @@ var VectorPlugin = class {
    * Run a chat query.
    */
   async chat(message, history = [], namespace) {
-    await this.validationPromise;
-    return this.pipeline.ask(message, history, namespace);
+    try {
+      await this.validationPromise;
+    } catch (err) {
+      throw wrapError(err, "CONFIGURATION_ERROR");
+    }
+    try {
+      return await this.pipeline.ask(message, history, namespace);
+    } catch (err) {
+      const msg = String(err);
+      let defaultCode = "RETRIEVAL_FAILED";
+      if (msg.includes("Embed") || msg.includes("embed")) {
+        defaultCode = "EMBEDDING_FAILED";
+      }
+      throw wrapError(err, defaultCode);
+    }
   }
   /**
    * Run a streaming chat query.
    */
   chatStream(_0) {
     return __asyncGenerator(this, arguments, function* (message, history = [], namespace) {
-      yield new __await(this.validationPromise);
-      yield* __yieldStar(this.pipeline.askStream(message, history, namespace));
+      try {
+        yield new __await(this.validationPromise);
+      } catch (err) {
+        throw wrapError(err, "CONFIGURATION_ERROR");
+      }
+      try {
+        const stream = this.pipeline.askStream(message, history, namespace);
+        try {
+          for (var iter = __forAwait(stream), more, temp, error; more = !(temp = yield new __await(iter.next())).done; more = false) {
+            const chunk = temp.value;
+            yield chunk;
+          }
+        } catch (temp) {
+          error = [temp];
+        } finally {
+          try {
+            more && (temp = iter.return) && (yield new __await(temp.call(iter)));
+          } finally {
+            if (error)
+              throw error[0];
+          }
+        }
+      } catch (err) {
+        const msg = String(err);
+        let defaultCode = "RETRIEVAL_FAILED";
+        if (msg.includes("Embed") || msg.includes("embed")) {
+          defaultCode = "EMBEDDING_FAILED";
+        }
+        throw wrapError(err, defaultCode);
+      }
     });
   }
   /**
    * Ingest documents into the vector database.
    */
   async ingest(documents, namespace) {
-    await this.validationPromise;
-    return this.pipeline.ingest(documents, namespace);
+    try {
+      await this.validationPromise;
+    } catch (err) {
+      throw wrapError(err, "CONFIGURATION_ERROR");
+    }
+    try {
+      return await this.pipeline.ingest(documents, namespace);
+    } catch (err) {
+      const msg = String(err);
+      let defaultCode = "RETRIEVAL_FAILED";
+      if (msg.includes("Embed") || msg.includes("embed")) {
+        defaultCode = "EMBEDDING_FAILED";
+      }
+      throw wrapError(err, defaultCode);
+    }
   }
   /**
    * Get auto-suggestions based on a query prefix.
    */
   async getSuggestions(query, namespace) {
-    await this.validationPromise;
-    return this.pipeline.getSuggestions(query, namespace);
-  }
-};
-
-// src/core/Retrivora.ts
-var Retrivora = class {
-  constructor(config) {
-    this.config = ConfigResolver.resolveUniversal(config);
-    this.pipeline = new Pipeline(this.config);
-  }
-  async initialize() {
-    await ConfigValidator.validateAndThrow(this.config);
-    await this.pipeline.initialize();
-  }
-  async ingest(documents, namespace) {
-    return this.pipeline.ingest(documents, namespace);
-  }
-  async ask(question, history = [], namespace) {
-    return this.pipeline.ask(question, history, namespace);
-  }
-  askStream(question, history = [], namespace) {
-    return this.pipeline.askStream(question, history, namespace);
-  }
-  getPipeline() {
-    return this.pipeline;
+    try {
+      await this.validationPromise;
+    } catch (err) {
+      throw wrapError(err, "CONFIGURATION_ERROR");
+    }
+    try {
+      return await this.pipeline.getSuggestions(query, namespace);
+    } catch (err) {
+      throw wrapError(err, "RETRIEVAL_FAILED");
+    }
   }
 };
 
@@ -7899,6 +8021,79 @@ function createUploadHandler(configOrPlugin) {
     }
   };
 }
+function createSuggestionsHandler(configOrPlugin) {
+  const plugin = configOrPlugin instanceof VectorPlugin ? configOrPlugin : new VectorPlugin(configOrPlugin);
+  return async function POST(req) {
+    try {
+      const body = await req.json();
+      const { query, namespace } = body;
+      if (typeof query !== "string") {
+        return import_server.NextResponse.json({ error: "query is required" }, { status: 400 });
+      }
+      const suggestions = await plugin.getSuggestions(query, namespace);
+      return import_server.NextResponse.json({ suggestions });
+    } catch (err) {
+      const message = err instanceof Error ? err.message : "Internal server error";
+      return import_server.NextResponse.json({ error: message }, { status: 500 });
+    }
+  };
+}
+function createRagHandler(configOrPlugin) {
+  const plugin = configOrPlugin instanceof VectorPlugin ? configOrPlugin : new VectorPlugin(configOrPlugin);
+  const chatHandler = createChatHandler(plugin);
+  const streamHandler = createStreamHandler(plugin);
+  const uploadHandler = createUploadHandler(plugin);
+  const healthHandler = createHealthHandler(plugin);
+  const suggestionsHandler = createSuggestionsHandler(plugin);
+  async function routePostRequest(req, segment) {
+    switch (segment) {
+      case "chat":
+        return streamHandler(req);
+      case "chat-sync":
+        return chatHandler(req);
+      case "upload":
+        return uploadHandler(req);
+      case "suggestions":
+        return suggestionsHandler(req);
+      case "health":
+        return healthHandler();
+      default:
+        return import_server.NextResponse.json({ error: `Not Found: POST segment "${segment}" not supported.` }, { status: 404 });
+    }
+  }
+  async function routeGetRequest(req, segment) {
+    if (segment === "health") {
+      return healthHandler();
+    }
+    return import_server.NextResponse.json({ error: `Method Not Allowed: GET is only supported for "health" segment.` }, { status: 405 });
+  }
+  async function getSegment(context) {
+    var _a;
+    const resolvedParams = typeof ((_a = context == null ? void 0 : context.params) == null ? void 0 : _a.then) === "function" ? await context.params : context == null ? void 0 : context.params;
+    const segments = (resolvedParams == null ? void 0 : resolvedParams.retrivora) || [];
+    return segments[0] || "chat";
+  }
+  return {
+    GET: async (req, context) => {
+      try {
+        const segment = await getSegment(context);
+        return await routeGetRequest(req, segment);
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : "GET Routing failed";
+        return import_server.NextResponse.json({ error: msg }, { status: 500 });
+      }
+    },
+    POST: async (req, context) => {
+      try {
+        const segment = await getSegment(context);
+        return await routePostRequest(req, segment);
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : "POST Routing failed";
+        return import_server.NextResponse.json({ error: msg }, { status: 500 });
+      }
+    }
+  };
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AnthropicProvider,
@@ -7944,11 +8139,13 @@ function createUploadHandler(configOrPlugin) {
   createFromPreset,
   createHealthHandler,
   createIngestHandler,
+  createRagHandler,
   createStreamHandler,
   createUploadHandler,
   getRagConfig,
   sseErrorFrame,
   sseFrame,
   sseMetaFrame,
-  sseTextFrame
+  sseTextFrame,
+  wrapError
 });