@ekodb/ekodb-client 0.26.1 → 0.26.2

package/dist/search-wire.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/search-wire.test.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const node_fs_1 = require("node:fs");
+const node_path_1 = require("node:path");
+const vitest_1 = require("vitest");
+const search_1 = require("./search");
+const query_builder_1 = require("./query-builder");
+const golden = JSON.parse((0, node_fs_1.readFileSync)((0, node_path_1.resolve)(__dirname, "../../test-fixtures/search-requests.json"), "utf8"));
+const vector = [0.1, 0.2, 0.3];
+const base = (query = "") => new search_1.SearchQueryBuilder(query)
+    .bypassRipple(false)
+    .bypassCache(false)
+    .limit(10);
+(0, vitest_1.describe)("cross-client search wire cases", () => {
+    const cases = {
+        filtered_vector: base()
+            .vector(vector)
+            .vectorMetric("cosine")
+            .vectorK(10)
+            .vectorThreshold(0.25)
+            .filters(new query_builder_1.QueryBuilder().eq("category", "ml").build().filter)
+            .build(),
+        hybrid_custom_weights: base("machine learning")
+            .vector(vector)
+            .textWeight(0.7)
+            .vectorWeight(0.3)
+            .build(),
+        named_vector_field: base()
+            .vector(vector)
+            .vectorField("title_embedding")
+            .vectorMetric("dotproduct")
+            .vectorK(10)
+            .build(),
+        full_text: base("enginer")
+            .fields(["title", "body"])
+            .weights({ title: 2.5, body: 1.5 })
+            .fuzzy(true)
+            .maxEditDistance(2)
+            .build(),
+    };
+    for (const [name, query] of Object.entries(cases)) {
+        (0, vitest_1.it)(name, () => (0, vitest_1.expect)(JSON.parse(JSON.stringify(query))).toEqual(golden[name]));
+    }
+    (0, vitest_1.it)("omits unset options", () => {
+        (0, vitest_1.expect)(JSON.parse(JSON.stringify(new search_1.SearchQueryBuilder("").build()))).toEqual({ query: "" });
+    });
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ekodb/ekodb-client",
-  "version": "0.26.1",
+  "version": "0.26.2",
   "description": "Official TypeScript/JavaScript client for ekoDB",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/search-wire.test.ts

@@ -0,0 +1,57 @@
+import { readFileSync } from "node:fs";
+import { resolve } from "node:path";
+import { describe, expect, it } from "vitest";
+import { SearchQueryBuilder } from "./search";
+import { QueryBuilder } from "./query-builder";
+
+const golden = JSON.parse(
+  readFileSync(
+    resolve(__dirname, "../../test-fixtures/search-requests.json"),
+    "utf8",
+  ),
+);
+const vector = [0.1, 0.2, 0.3];
+const base = (query = "") =>
+  new SearchQueryBuilder(query)
+    .bypassRipple(false)
+    .bypassCache(false)
+    .limit(10);
+
+describe("cross-client search wire cases", () => {
+  const cases = {
+    filtered_vector: base()
+      .vector(vector)
+      .vectorMetric("cosine")
+      .vectorK(10)
+      .vectorThreshold(0.25)
+      .filters(new QueryBuilder().eq("category", "ml").build().filter)
+      .build(),
+    hybrid_custom_weights: base("machine learning")
+      .vector(vector)
+      .textWeight(0.7)
+      .vectorWeight(0.3)
+      .build(),
+    named_vector_field: base()
+      .vector(vector)
+      .vectorField("title_embedding")
+      .vectorMetric("dotproduct")
+      .vectorK(10)
+      .build(),
+    full_text: base("enginer")
+      .fields(["title", "body"])
+      .weights({ title: 2.5, body: 1.5 })
+      .fuzzy(true)
+      .maxEditDistance(2)
+      .build(),
+  };
+  for (const [name, query] of Object.entries(cases)) {
+    it(name, () =>
+      expect(JSON.parse(JSON.stringify(query))).toEqual(golden[name]),
+    );
+  }
+  it("omits unset options", () => {
+    expect(
+      JSON.parse(JSON.stringify(new SearchQueryBuilder("").build())),
+    ).toEqual({ query: "" });
+  });
+});