@retrivora-ai/rag-engine 1.9.6 → 1.9.7

package/dist/handlers/index.mjs

@@ -6,9 +6,6 @@ var __getOwnPropSymbols = Object.getOwnPropertySymbols;
 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 = {}))
@@ -56,34 +53,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
@@ -2298,26 +2267,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 ConfigurationException = class extends RetrivoraError {
-  constructor(message, details) {
-    super(message, "CONFIGURATION_ERROR", details);
-  }
-};
-
 // src/core/ConfigResolver.ts
 var ConfigResolver = class {
   /**
@@ -2363,13 +2312,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) {
@@ -3410,6 +3359,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 {
@@ -3455,7 +3476,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);
@@ -3479,9 +3500,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()
+          }
         );
       }
     }
@@ -6410,7 +6434,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.`
       );
     }
@@ -7129,6 +7153,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);
   }
   /**
@@ -7162,31 +7188,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);
+    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");
+    }
   }
 };
 
@@ -7536,10 +7624,67 @@ function createSuggestionsHandler(configOrPlugin) {
     }
   };
 }
+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 NextResponse.json({ error: `Not Found: POST segment "${segment}" not supported.` }, { status: 404 });
+    }
+  }
+  async function routeGetRequest(req, segment) {
+    if (segment === "health") {
+      return healthHandler();
+    }
+    return 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 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 NextResponse.json({ error: msg }, { status: 500 });
+      }
+    }
+  };
+}
 export {
   createChatHandler,
   createHealthHandler,
   createIngestHandler,
+  createRagHandler,
   createStreamHandler,
   createSuggestionsHandler,
   createUploadHandler,

package/dist/{index-CjQdL0cX.d.ts → index-B9J_XEh0.d.ts}

@@ -1,4 +1,4 @@
-import { k as RagConfig, I as ILLMProvider, g as IngestDocument, C as ChatMessage, d as ChatResponse, u as RetrievalResult, n as UniversalRagConfig } from './ILLMProvider-DNhyOYoK.js';
+import { k as RagConfig, I as ILLMProvider, g as IngestDocument, C as ChatMessage, d as ChatResponse, u as RetrievalResult, n as UniversalRagConfig } from './ILLMProvider-Bhk6zJOK.js';
 
 /**
  * DocumentChunker — splits long text into overlapping chunks
@@ -179,5 +179,9 @@ declare class ConfigurationException extends RetrivoraError {
 declare class AuthenticationException extends RetrivoraError {
     constructor(message?: string, details?: unknown);
 }
+/**
+ * Wraps any unknown error into an appropriate RetrivoraError subclass.
+ */
+declare function wrapError(err: unknown, defaultCode: RetrivoraErrorCode, defaultMessage?: string): RetrivoraError;
 
-export { AuthenticationException as A, type Chunk as C, DocumentChunker as D, EmbeddingFailedException as E, ProviderNotFoundException as P, RateLimitException as R, type ChunkOptions as a, ConfigurationException as b, RetrievalException as c, Retrivora as d, RetrivoraError as e, type RetrivoraErrorCode as f, Pipeline as g };
+export { AuthenticationException as A, type Chunk as C, DocumentChunker as D, EmbeddingFailedException as E, ProviderNotFoundException as P, RateLimitException as R, type ChunkOptions as a, ConfigurationException as b, RetrievalException as c, Retrivora as d, RetrivoraError as e, type RetrivoraErrorCode as f, Pipeline as g, wrapError as w };

package/dist/{index-C9v7-tWd.d.mts → index-BJ4cd-t5.d.mts}

@@ -1,4 +1,4 @@
-import { k as RagConfig, I as ILLMProvider, g as IngestDocument, C as ChatMessage, d as ChatResponse, u as RetrievalResult, n as UniversalRagConfig } from './ILLMProvider-DNhyOYoK.mjs';
+import { k as RagConfig, I as ILLMProvider, g as IngestDocument, C as ChatMessage, d as ChatResponse, u as RetrievalResult, n as UniversalRagConfig } from './ILLMProvider-Bhk6zJOK.mjs';
 
 /**
  * DocumentChunker — splits long text into overlapping chunks
@@ -179,5 +179,9 @@ declare class ConfigurationException extends RetrivoraError {
 declare class AuthenticationException extends RetrivoraError {
     constructor(message?: string, details?: unknown);
 }
+/**
+ * Wraps any unknown error into an appropriate RetrivoraError subclass.
+ */
+declare function wrapError(err: unknown, defaultCode: RetrivoraErrorCode, defaultMessage?: string): RetrivoraError;
 
-export { AuthenticationException as A, type Chunk as C, DocumentChunker as D, EmbeddingFailedException as E, ProviderNotFoundException as P, RateLimitException as R, type ChunkOptions as a, ConfigurationException as b, RetrievalException as c, Retrivora as d, RetrivoraError as e, type RetrivoraErrorCode as f, Pipeline as g };
+export { AuthenticationException as A, type Chunk as C, DocumentChunker as D, EmbeddingFailedException as E, ProviderNotFoundException as P, RateLimitException as R, type ChunkOptions as a, ConfigurationException as b, RetrievalException as c, Retrivora as d, RetrivoraError as e, type RetrivoraErrorCode as f, Pipeline as g, wrapError as w };

package/dist/{index-Hgbwl9X4.d.ts → index-Bu7T6xgr.d.ts}

@@ -1,4 +1,4 @@
-import { k as RagConfig, I as ILLMProvider, C as ChatMessage, d as ChatResponse, g as IngestDocument } from './ILLMProvider-DNhyOYoK.js';
+import { k as RagConfig, I as ILLMProvider, C as ChatMessage, d as ChatResponse, g as IngestDocument } from './ILLMProvider-Bhk6zJOK.js';
 import { NextRequest, NextResponse } from 'next/server';
 
 interface ValidationError {
@@ -110,7 +110,7 @@ declare class VectorPlugin {
     /**
      * Run a streaming chat query.
      */
-    chatStream(message: string, history?: ChatMessage[], namespace?: string): AsyncGenerator<string | ChatResponse, void, any>;
+    chatStream(message: string, history?: ChatMessage[], namespace?: string): AsyncGenerator<string | ChatResponse, void, unknown>;
     /**
      * Ingest documents into the vector database.
      */
@@ -224,5 +224,22 @@ declare function createSuggestionsHandler(configOrPlugin?: Partial<RagConfig> |
 }> | NextResponse<{
     suggestions: string[];
 }>>;
+/**
+ * createRagHandler — factory that returns GET and POST handlers for a catch-all route
+ * (e.g. `src/app/api/retrivora/[[...retrivora]]/route.ts`).
+ */
+declare function createRagHandler(configOrPlugin?: Partial<RagConfig> | VectorPlugin): {
+    GET: (req: NextRequest, context: any) => Promise<NextResponse<{
+        timestamp: string;
+        vectorDb: HealthCheckResult;
+        llm: HealthCheckResult;
+        embedding?: HealthCheckResult;
+        allHealthy: boolean;
+        status: string;
+    }> | NextResponse<{
+        error: string;
+    }>>;
+    POST: (req: NextRequest, context: any) => Promise<Response>;
+};
 
-export { ConfigValidator as C, type HealthCheckResult as H, type IProviderValidator as I, type ValidationError as V, type IProviderHealthChecker as a, VectorPlugin as b, createChatHandler as c, createHealthHandler as d, createIngestHandler as e, createStreamHandler as f, createUploadHandler as g, sseFrame as h, sseMetaFrame as i, sseTextFrame as j, createSuggestionsHandler as k, sseObservabilityFrame as l, sseUIFrame as m, sseErrorFrame as s };
+export { ConfigValidator as C, type HealthCheckResult as H, type IProviderValidator as I, type ValidationError as V, type IProviderHealthChecker as a, VectorPlugin as b, createChatHandler as c, createHealthHandler as d, createIngestHandler as e, createRagHandler as f, createStreamHandler as g, createUploadHandler as h, sseFrame as i, sseMetaFrame as j, sseTextFrame as k, createSuggestionsHandler as l, sseObservabilityFrame as m, sseUIFrame as n, sseErrorFrame as s };

package/dist/{index-CHL1jdYm.d.mts → index-C3SVtPYg.d.mts}

@@ -1,4 +1,4 @@
-import { k as RagConfig, I as ILLMProvider, C as ChatMessage, d as ChatResponse, g as IngestDocument } from './ILLMProvider-DNhyOYoK.mjs';
+import { k as RagConfig, I as ILLMProvider, C as ChatMessage, d as ChatResponse, g as IngestDocument } from './ILLMProvider-Bhk6zJOK.mjs';
 import { NextRequest, NextResponse } from 'next/server';
 
 interface ValidationError {
@@ -110,7 +110,7 @@ declare class VectorPlugin {
     /**
      * Run a streaming chat query.
      */
-    chatStream(message: string, history?: ChatMessage[], namespace?: string): AsyncGenerator<string | ChatResponse, void, any>;
+    chatStream(message: string, history?: ChatMessage[], namespace?: string): AsyncGenerator<string | ChatResponse, void, unknown>;
     /**
      * Ingest documents into the vector database.
      */
@@ -224,5 +224,22 @@ declare function createSuggestionsHandler(configOrPlugin?: Partial<RagConfig> |
 }> | NextResponse<{
     suggestions: string[];
 }>>;
+/**
+ * createRagHandler — factory that returns GET and POST handlers for a catch-all route
+ * (e.g. `src/app/api/retrivora/[[...retrivora]]/route.ts`).
+ */
+declare function createRagHandler(configOrPlugin?: Partial<RagConfig> | VectorPlugin): {
+    GET: (req: NextRequest, context: any) => Promise<NextResponse<{
+        timestamp: string;
+        vectorDb: HealthCheckResult;
+        llm: HealthCheckResult;
+        embedding?: HealthCheckResult;
+        allHealthy: boolean;
+        status: string;
+    }> | NextResponse<{
+        error: string;
+    }>>;
+    POST: (req: NextRequest, context: any) => Promise<Response>;
+};
 
-export { ConfigValidator as C, type HealthCheckResult as H, type IProviderValidator as I, type ValidationError as V, type IProviderHealthChecker as a, VectorPlugin as b, createChatHandler as c, createHealthHandler as d, createIngestHandler as e, createStreamHandler as f, createUploadHandler as g, sseFrame as h, sseMetaFrame as i, sseTextFrame as j, createSuggestionsHandler as k, sseObservabilityFrame as l, sseUIFrame as m, sseErrorFrame as s };
+export { ConfigValidator as C, type HealthCheckResult as H, type IProviderValidator as I, type ValidationError as V, type IProviderHealthChecker as a, VectorPlugin as b, createChatHandler as c, createHealthHandler as d, createIngestHandler as e, createRagHandler as f, createStreamHandler as g, createUploadHandler as h, sseFrame as i, sseMetaFrame as j, sseTextFrame as k, createSuggestionsHandler as l, sseObservabilityFrame as m, sseUIFrame as n, sseErrorFrame as s };