@retrivora-ai/rag-engine 1.8.2 → 1.8.4

package/dist/chunk-LR3VMDVK.mjs

@@ -1,157 +0,0 @@
-import {
-  buildPayload,
-  resolvePath
-} from "./chunk-YLTMFW4M.mjs";
-import {
-  BaseVectorProvider
-} from "./chunk-IMP6FUCY.mjs";
-import {
-  __spreadValues
-} from "./chunk-X4TOT24V.mjs";
-
-// src/providers/vectordb/UniversalVectorProvider.ts
-import axios from "axios";
-var UniversalVectorProvider = class extends BaseVectorProvider {
-  constructor(config) {
-    super(config);
-    this.opts = config.options;
-    if (!this.opts.baseUrl) {
-      throw new Error("[UniversalVectorProvider] options.baseUrl is required");
-    }
-  }
-  async initialize() {
-    var _a;
-    this.http = axios.create({
-      baseURL: this.opts.baseUrl,
-      headers: __spreadValues({
-        "Content-Type": "application/json"
-      }, this.opts.headers),
-      timeout: (_a = this.opts.timeout) != null ? _a : 3e4
-    });
-    if (!await this.ping()) {
-      throw new Error(`[UniversalVectorProvider] Failed to connect to ${this.opts.baseUrl}`);
-    }
-  }
-  async upsert(doc, namespace) {
-    var _a, _b, _c;
-    const endpoint = (_a = this.opts.upsertEndpoint) != null ? _a : "/upsert";
-    const template = (_b = this.opts.upsertTemplate) != null ? _b : JSON.stringify({
-      id: "{{id}}",
-      vector: "{{vector}}",
-      content: "{{content}}",
-      namespace: "{{namespace}}",
-      metadata: "{{metadata}}"
-    });
-    const payload = buildPayload(template, {
-      id: doc.id,
-      vector: doc.vector,
-      content: doc.content,
-      namespace: namespace != null ? namespace : this.indexName,
-      metadata: (_c = doc.metadata) != null ? _c : {}
-    });
-    try {
-      await this.http.post(endpoint, payload);
-    } catch (error) {
-      throw new Error(
-        `[UniversalVectorProvider] Upsert failed: ${error instanceof Error ? error.message : String(error)}`
-      );
-    }
-  }
-  async batchUpsert(docs, namespace) {
-    const batchSize = 50;
-    for (let i = 0; i < docs.length; i += batchSize) {
-      const batch = docs.slice(i, i + batchSize);
-      await Promise.all(
-        batch.map((doc) => this.upsert(doc, namespace))
-      );
-    }
-  }
-  async query(vector, topK, namespace, filter) {
-    var _a, _b;
-    const endpoint = (_a = this.opts.queryEndpoint) != null ? _a : "/query";
-    const template = (_b = this.opts.queryTemplate) != null ? _b : JSON.stringify({
-      vector: "{{vector}}",
-      limit: "{{topK}}",
-      namespace: "{{namespace}}",
-      filter: "{{filter}}"
-    });
-    const sanitizedFilter = this.sanitizeFilter(filter);
-    const payload = buildPayload(template, {
-      vector,
-      topK,
-      namespace: namespace != null ? namespace : this.indexName,
-      filter: sanitizedFilter
-    });
-    try {
-      const response = await this.http.post(endpoint, payload);
-      let results = response.data;
-      if (this.opts.queryResponsePath) {
-        results = resolvePath(response.data, this.opts.queryResponsePath);
-      }
-      if (!Array.isArray(results)) {
-        throw new Error(
-          `[UniversalVectorProvider] Expected array response at "${this.opts.queryResponsePath}", got ${typeof results}`
-        );
-      }
-      return results.map((item) => {
-        var _a2, _b2, _c, _d, _e, _f, _g;
-        return {
-          id: item[(_a2 = this.opts.queryIdField) != null ? _a2 : "id"],
-          score: (_c = item[(_b2 = this.opts.queryScoreField) != null ? _b2 : "score"]) != null ? _c : 0,
-          content: (_e = item[(_d = this.opts.queryContentField) != null ? _d : "content"]) != null ? _e : "",
-          metadata: (_g = item[(_f = this.opts.queryMetadataField) != null ? _f : "metadata"]) != null ? _g : {}
-        };
-      });
-    } catch (error) {
-      throw new Error(
-        `[UniversalVectorProvider] Query failed: ${error instanceof Error ? error.message : String(error)}`
-      );
-    }
-  }
-  async delete(id, namespace) {
-    var _a, _b;
-    const endpoint = (_a = this.opts.deleteEndpoint) != null ? _a : "/delete";
-    const template = (_b = this.opts.deleteTemplate) != null ? _b : JSON.stringify({
-      id: "{{id}}",
-      namespace: "{{namespace}}"
-    });
-    const payload = buildPayload(template, {
-      id,
-      namespace: namespace != null ? namespace : this.indexName
-    });
-    try {
-      await this.http.post(endpoint, payload);
-    } catch (error) {
-      throw new Error(
-        `[UniversalVectorProvider] Delete failed: ${error instanceof Error ? error.message : String(error)}`
-      );
-    }
-  }
-  async deleteNamespace(namespace) {
-    var _a;
-    const endpoint = (_a = this.opts.deleteNamespaceEndpoint) != null ? _a : "/delete-namespace";
-    try {
-      await this.http.post(endpoint, { namespace });
-    } catch (error) {
-      throw new Error(
-        `[UniversalVectorProvider] DeleteNamespace failed: ${error instanceof Error ? error.message : String(error)}`
-      );
-    }
-  }
-  async ping() {
-    var _a;
-    try {
-      const endpoint = (_a = this.opts.pingEndpoint) != null ? _a : "/health";
-      const response = await this.http.get(endpoint, { timeout: 5e3 });
-      return response.status >= 200 && response.status < 300;
-    } catch (e) {
-      return false;
-    }
-  }
-  async disconnect() {
-  }
-};
-
-export {
-  UniversalVectorProvider
-};