flowseeker 0.1.7 → 0.1.9

package/dist/index/vectorStore.js

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InMemoryVectorStore = void 0;
+exports.cosineSimilarity = cosineSimilarity;
+class InMemoryVectorStore {
+    constructor() {
+        this.records = new Map();
+    }
+    upsert(record) {
+        this.records.set(record.id, record);
+    }
+    search(vector, limit) {
+        return Array.from(this.records.values())
+            .map((record) => ({ record, score: cosineSimilarity(vector, record.vector) }))
+            .filter((item) => Number.isFinite(item.score))
+            .sort((left, right) => right.score - left.score)
+            .slice(0, limit);
+    }
+    size() {
+        return this.records.size;
+    }
+}
+exports.InMemoryVectorStore = InMemoryVectorStore;
+function cosineSimilarity(left, right) {
+    if (left.length === 0 || left.length !== right.length) {
+        return 0;
+    }
+    let dot = 0;
+    let leftNorm = 0;
+    let rightNorm = 0;
+    for (let index = 0; index < left.length; index += 1) {
+        dot += left[index] * right[index];
+        leftNorm += left[index] * left[index];
+        rightNorm += right[index] * right[index];
+    }
+    if (leftNorm === 0 || rightNorm === 0) {
+        return 0;
+    }
+    return dot / (Math.sqrt(leftNorm) * Math.sqrt(rightNorm));
+}
+//# sourceMappingURL=vectorStore.js.map