@retrivora-ai/rag-engine 0.1.3 → 0.1.5

package/dist/chunk-7NXI6ZWX.mjs

@@ -0,0 +1,89 @@
+import {
+  BaseVectorProvider
+} from "./chunk-VOIWNO5O.mjs";
+import {
+  __spreadValues
+} from "./chunk-I4E63NIC.mjs";
+
+// src/providers/vectordb/MilvusProvider.ts
+import axios from "axios";
+var MilvusProvider = class extends BaseVectorProvider {
+  constructor(config) {
+    super(config);
+    const opts = config.options;
+    const baseUrl = opts.baseUrl || opts.uri;
+    if (!baseUrl) throw new Error("[MilvusProvider] baseUrl/uri is required");
+    this.http = axios.create({
+      baseURL: baseUrl,
+      headers: __spreadValues(__spreadValues({
+        "Content-Type": "application/json"
+      }, opts.headers || {}), opts.apiKey ? { Authorization: `Bearer ${opts.apiKey}` } : {})
+    });
+  }
+  async initialize() {
+    await this.ping();
+  }
+  async upsert(doc, namespace) {
+    const payload = {
+      collectionName: this.indexName,
+      data: [__spreadValues({
+        vector: doc.vector,
+        content: doc.content,
+        metadata: doc.metadata || {}
+      }, namespace ? { namespace } : {})]
+    };
+    await this.http.post("/v1/vector/upsert", payload);
+  }
+  async batchUpsert(docs, namespace) {
+    const payload = {
+      collectionName: this.indexName,
+      data: docs.map((doc) => __spreadValues({
+        vector: doc.vector,
+        content: doc.content,
+        metadata: doc.metadata || {}
+      }, namespace ? { namespace } : {}))
+    };
+    await this.http.post("/v1/vector/upsert", payload);
+  }
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async query(vector, topK, namespace, _filter) {
+    const payload = {
+      collectionName: this.indexName,
+      vector,
+      limit: topK,
+      filter: namespace ? `namespace == "${namespace}"` : void 0,
+      outputFields: ["content", "metadata"]
+    };
+    const { data } = await this.http.post("/v1/vector/search", payload);
+    return (data.data || []).map((res) => ({
+      id: String(res["id"]),
+      score: res["distance"],
+      content: res["content"],
+      metadata: res["metadata"]
+    }));
+  }
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async delete(id, _namespace) {
+    await this.http.post("/v1/vector/delete", {
+      collectionName: this.indexName,
+      id
+    });
+  }
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async deleteNamespace(_namespace) {
+  }
+  async ping() {
+    try {
+      await this.http.get("/v1/health");
+      return true;
+    } catch (e) {
+      return false;
+    }
+  }
+  async disconnect() {
+  }
+};
+
+export {
+  MilvusProvider
+};

package/dist/chunk-AALIF3AL.mjs

@@ -0,0 +1,91 @@
+import {
+  BaseVectorProvider
+} from "./chunk-VOIWNO5O.mjs";
+import {
+  __spreadValues
+} from "./chunk-I4E63NIC.mjs";
+
+// src/providers/vectordb/PineconeProvider.ts
+import { Pinecone } from "@pinecone-database/pinecone";
+var PineconeProvider = class extends BaseVectorProvider {
+  constructor(config) {
+    super(config);
+    const opts = config.options;
+    if (!opts.apiKey) throw new Error("[PineconeProvider] options.apiKey is required");
+    this.apiKey = opts.apiKey;
+  }
+  async initialize() {
+    var _a, _b;
+    this.client = new Pinecone({ apiKey: this.apiKey });
+    const indexes = await this.client.listIndexes();
+    const names = (_b = (_a = indexes.indexes) == null ? void 0 : _a.map((i) => i.name)) != null ? _b : [];
+    if (!names.includes(this.indexName)) {
+      throw new Error(`[PineconeProvider] Index "${this.indexName}" not found.`);
+    }
+  }
+  index(namespace) {
+    const idx = this.client.index(this.indexName);
+    return namespace ? idx.namespace(namespace) : idx.namespace("");
+  }
+  async upsert(doc, namespace) {
+    var _a;
+    await this.index(namespace).upsert({
+      records: [{
+        id: doc.id,
+        values: doc.vector,
+        metadata: __spreadValues({ content: doc.content }, (_a = doc.metadata) != null ? _a : {})
+      }]
+    });
+  }
+  async batchUpsert(docs, namespace) {
+    const BATCH = 100;
+    for (let i = 0; i < docs.length; i += BATCH) {
+      const records = docs.slice(i, i + BATCH).map((d) => {
+        var _a;
+        return {
+          id: d.id,
+          values: d.vector,
+          metadata: __spreadValues({ content: d.content }, (_a = d.metadata) != null ? _a : {})
+        };
+      });
+      await this.index(namespace).upsert({ records });
+    }
+  }
+  async query(vector, topK, namespace, filter) {
+    var _a;
+    const result = await this.index(namespace).query(__spreadValues({
+      vector,
+      topK,
+      includeMetadata: true
+    }, filter ? { filter } : {}));
+    return ((_a = result.matches) != null ? _a : []).map((m) => {
+      var _a2, _b;
+      return {
+        id: m.id,
+        score: (_a2 = m.score) != null ? _a2 : 0,
+        content: ((_b = m.metadata) == null ? void 0 : _b.content) || "",
+        metadata: m.metadata
+      };
+    });
+  }
+  async delete(id, namespace) {
+    await this.index(namespace).deleteOne({ id });
+  }
+  async deleteNamespace(namespace) {
+    await this.client.index(this.indexName).namespace(namespace).deleteAll();
+  }
+  async ping() {
+    try {
+      await this.client.listIndexes();
+      return true;
+    } catch (e) {
+      return false;
+    }
+  }
+  async disconnect() {
+  }
+};
+
+export {
+  PineconeProvider
+};