@adaas/a-frame 0.0.6 → 0.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adaas/a-frame",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "description": "A-frame is index build engine for A-Concept ecosystem. It provides a way to build index structures for A-Concept basics",
   "license": "Apache-2.0",
   "main": "./dist/index.cjs",
@@ -61,7 +61,7 @@
     "build": "tsup --config tsup.config.ts"
   },
   "dependencies": {
-    "@adaas/a-concept": "^0.2.4"
+    "@adaas/a-concept": "^0.2.6"
   },
   "devDependencies": {
     "@types/chai": "^4.3.14",

package/src/lib/components/Search/A_FrameSearch.component.ts

@@ -0,0 +1,66 @@
+import { A_Component } from "@adaas/a-concept"
+import { A_FrameVector } from "../../entities/A_FrameVector/A_FrameVector.entity"
+
+export class A_FrameSearch extends A_Component {
+
+    static cosineSimilarity(
+        a: A_FrameVector,
+        b: A_FrameVector
+    ): number {
+        const dot = a.dot(b)
+        const mag = a.magnitude() * b.magnitude()
+        return mag === 0 ? 0 : dot / mag
+    }
+
+    static euclideanDistance(
+        a: A_FrameVector,
+        b: A_FrameVector
+    ): number {
+        A_FrameSearch.assertSameLength(a, b)
+
+        let sum = 0
+        for (let i = 0; i < a.length; i++) {
+            const diff = a.values[i] - b.values[i]
+            sum += diff * diff
+        }
+        return Math.sqrt(sum)
+    }
+
+    static nearest(
+        query: A_FrameVector,
+        vectors: readonly A_FrameVector[],
+        limit = 1
+    ): A_FrameVector[] {
+        return [...vectors]
+            .map(v => ({
+                vector: v,
+                score: A_FrameSearch.cosineSimilarity(query, v),
+            }))
+            .sort((a, b) => b.score - a.score)
+            .slice(0, limit)
+            .map(r => r.vector)
+    }
+
+    static rank(
+        query: A_FrameVector,
+        vectors: readonly A_FrameVector[]
+    ): Array<{ vector: A_FrameVector; score: number }> {
+        return vectors
+            .map(v => ({
+                vector: v,
+                score: A_FrameSearch.cosineSimilarity(query, v),
+            }))
+            .sort((a, b) => b.score - a.score)
+    }
+
+    private static assertSameLength(
+        a: A_FrameVector,
+        b: A_FrameVector
+    ) {
+        if (a.length !== b.length) {
+            throw new Error(
+                `Embedding dimension mismatch: ${a.length} vs ${b.length}`
+            )
+        }
+    }
+}

package/src/lib/entities/A_FrameVector/A_FrameVector.entity.ts

@@ -0,0 +1,54 @@
+import { A_Entity } from "@adaas/a-concept"
+
+export class A_FrameVector extends A_Entity {
+    readonly values: Float32Array
+
+    constructor(values: number[] | Float32Array) {
+        super();
+        this.values = values instanceof Float32Array
+            ? values
+            : new Float32Array(values)
+    }
+
+    get length(): number {
+        return this.values.length
+    }
+
+    magnitude(): number {
+        let sum = 0
+        for (let i = 0; i < this.values.length; i++) {
+            sum += this.values[i] ** 2
+        }
+        return Math.sqrt(sum)
+    }
+
+    normalize(): A_FrameVector {
+        const mag = this.magnitude()
+        if (mag === 0) return this
+
+        const normalized = new Float32Array(this.length)
+        for (let i = 0; i < this.length; i++) {
+            normalized[i] = this.values[i] / mag
+        }
+
+        return new A_FrameVector(normalized)
+    }
+
+    dot(other: A_FrameVector): number {
+        this.assertSameLength(other)
+
+        let sum = 0
+        for (let i = 0; i < this.length; i++) {
+            sum += this.values[i] * other.values[i]
+        }
+        return sum
+    }
+
+    private assertSameLength(other: A_FrameVector) {
+        if (this.length !== other.length) {
+            throw new Error(
+                `Embedding dimension mismatch: ${this.length} vs ${other.length}`
+            )
+        }
+    }
+}