@ouim/vectoriadb-server 0.1.5 → 0.1.6

package/package.json

@@ -1,11 +1,12 @@
 {
   "name": "@ouim/vectoriadb-server",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "type": "module",
   "main": "index.js",
   "scripts": {
     "start": "node index.js",
-    "demo": "node demo.js"
+    "demo": "node demo.js",
+    "test": "node --test"
   },
   "dependencies": {
     "@huggingface/transformers": "^3.8.1",

package/test/filter-deserialize.test.js

@@ -0,0 +1,46 @@
+import test from 'node:test'
+import assert from 'node:assert/strict'
+import VectoriaDBServer from '../vectoriadb-server.js'
+
+test('server deserializes serialized options.filter (object form) into a function', async () => {
+  const server = new VectoriaDBServer()
+
+  // stub the underlying VectoriaDB method (`search`) to inspect received args
+  server._vectoria = {
+    async search(...args) {
+      // find an arg that contains a `filter` property
+      const opt = args.find(a => a && typeof a === 'object' && Object.prototype.hasOwnProperty.call(a, 'filter'))
+      assert(opt, 'expected an options object containing `filter`')
+      assert.equal(typeof opt.filter, 'function', 'options.filter should be a function after deserialization')
+
+      // sanity-check the deserialized function works
+      const sample = { owner: 'team' }
+      assert.equal(opt.filter(sample), true)
+
+      return { ok: true }
+    },
+  }
+
+  // capture emitted responses from the "socket"
+  let emitted = null
+  const mockSocket = { emit: (evt, payload) => (emitted = { evt, payload }) }
+
+  const payload = {
+    id: 't-filter-1',
+    method: 'search',
+    params: [
+      {
+        // serialized filter object as sent by the client SDK
+        filter: { __isFnString: true, fn: 'function(m){ return !!(m && m.owner === "team") }' },
+      },
+    ],
+  }
+
+  await server._handleRequest(mockSocket, payload)
+
+  assert(emitted, 'expected socket.emit to be called')
+  assert.equal(emitted.evt, 'response')
+  assert.equal(emitted.payload.id, payload.id)
+  assert.deepEqual(emitted.payload.result, { ok: true })
+  assert.equal(emitted.payload.error, null)
+})

package/vectoriadb-server.js

@@ -148,15 +148,21 @@ export default class VectoriaDBServer {
     // Allow passing serialized functions from client: convert stringified functions back to real functions
     const reparsedParams = params.map(p => {
       if (p && typeof p === 'object') {
-        // detect serialized function marker
+        // Top-level serialized function (client forwarded a function as a param)
         if (p.__isFnString && typeof p.fn === 'string') {
           // create function from string - executed in server process (trusted usage only)
           // eslint-disable-next-line no-new-func
           const fn = new Function('return (' + p.fn + ')')()
           return fn
         }
-        // also support options.filter as a string directly
-        if (typeof p.filter === 'string' && p.filter.trim().startsWith('function')) {
+
+        // Support serialized `filter` nested inside an `options` object:
+        // - { filter: { __isFnString: true, fn: 'function(m) { ... }' } }
+        // - or `filter` as a raw function string
+        if (p.filter && typeof p.filter === 'object' && p.filter.__isFnString && typeof p.filter.fn === 'string') {
+          // eslint-disable-next-line no-new-func
+          p.filter = new Function('return (' + p.filter.fn + ')')()
+        } else if (typeof p.filter === 'string' && p.filter.trim().startsWith('function')) {
           // eslint-disable-next-line no-new-func
           p.filter = new Function('return (' + p.filter + ')')()
         }