You've already forked node-redis
mirror of
https://github.com/redis/node-redis.git
synced 2025-08-06 02:15:48 +03:00
feat: add support for vector sets (#2998)
* wip * improve the vadd api * resp3 tests * fix some tests * extract json helper functions in client package * use transformJsonReply * remove the CACHEABLE flag for all vector set commands currently, client side caching is not supported for vector set commands by the server * properly transform vinfo result * add resp3 test for vlinks * add more tests for vrandmember * fix vrem return types * fix vsetattr return type * fix vsim_withscores * implement vlinks_withscores * set minimum docker image version to 8 * align return types * add RAW variant for VEMB -> VEMB_RAW * use the new parseCommand api
This commit is contained in:
committed by
GitHub
parent
b52177752e
commit
c5b4f47975
121
packages/client/lib/commands/VADD.spec.ts
Normal file
121
packages/client/lib/commands/VADD.spec.ts
Normal file
@@ -0,0 +1,121 @@
|
||||
import { strict as assert } from 'node:assert';
|
||||
import testUtils, { GLOBAL } from '../test-utils';
|
||||
import VADD from './VADD';
|
||||
import { BasicCommandParser } from '../client/parser';
|
||||
|
||||
describe('VADD', () => {
|
||||
describe('parseCommand', () => {
|
||||
it('basic usage', () => {
|
||||
const parser = new BasicCommandParser();
|
||||
VADD.parseCommand(parser, 'key', [1.0, 2.0, 3.0], 'element');
|
||||
assert.deepEqual(
|
||||
parser.redisArgs,
|
||||
['VADD', 'key', 'VALUES', '3', '1', '2', '3', 'element']
|
||||
);
|
||||
});
|
||||
|
||||
it('with REDUCE option', () => {
|
||||
const parser = new BasicCommandParser();
|
||||
VADD.parseCommand(parser, 'key', [1.0, 2], 'element', { REDUCE: 50 });
|
||||
assert.deepEqual(
|
||||
parser.redisArgs,
|
||||
['VADD', 'key', 'REDUCE', '50', 'VALUES', '2', '1', '2', 'element']
|
||||
);
|
||||
});
|
||||
|
||||
it('with quantization options', () => {
|
||||
let parser = new BasicCommandParser();
|
||||
VADD.parseCommand(parser, 'key', [1.0, 2.0], 'element', { QUANT: 'Q8' });
|
||||
assert.deepEqual(
|
||||
parser.redisArgs,
|
||||
['VADD', 'key', 'VALUES', '2', '1', '2', 'element', 'Q8']
|
||||
);
|
||||
|
||||
parser = new BasicCommandParser();
|
||||
VADD.parseCommand(parser, 'key', [1.0, 2.0], 'element', { QUANT: 'BIN' });
|
||||
assert.deepEqual(
|
||||
parser.redisArgs,
|
||||
['VADD', 'key', 'VALUES', '2', '1', '2', 'element', 'BIN']
|
||||
);
|
||||
|
||||
parser = new BasicCommandParser();
|
||||
VADD.parseCommand(parser, 'key', [1.0, 2.0], 'element', { QUANT: 'NOQUANT' });
|
||||
assert.deepEqual(
|
||||
parser.redisArgs,
|
||||
['VADD', 'key', 'VALUES', '2', '1', '2', 'element', 'NOQUANT']
|
||||
);
|
||||
});
|
||||
|
||||
it('with all options', () => {
|
||||
const parser = new BasicCommandParser();
|
||||
VADD.parseCommand(parser, 'key', [1.0, 2.0], 'element', {
|
||||
REDUCE: 50,
|
||||
CAS: true,
|
||||
QUANT: 'Q8',
|
||||
EF: 200,
|
||||
SETATTR: { name: 'test', value: 42 },
|
||||
M: 16
|
||||
});
|
||||
assert.deepEqual(
|
||||
parser.redisArgs,
|
||||
[
|
||||
'VADD', 'key', 'REDUCE', '50', 'VALUES', '2', '1', '2', 'element',
|
||||
'CAS', 'Q8', 'EF', '200', 'SETATTR', '{"name":"test","value":42}', 'M', '16'
|
||||
]
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
testUtils.testAll('vAdd', async client => {
|
||||
assert.equal(
|
||||
await client.vAdd('key', [1.0, 2.0, 3.0], 'element'),
|
||||
true
|
||||
);
|
||||
|
||||
// same element should not be added again
|
||||
assert.equal(
|
||||
await client.vAdd('key', [1, 2 , 3], 'element'),
|
||||
false
|
||||
);
|
||||
|
||||
}, {
|
||||
client: { ...GLOBAL.SERVERS.OPEN, minimumDockerVersion: [8, 0] },
|
||||
cluster: { ...GLOBAL.CLUSTERS.OPEN, minimumDockerVersion: [8, 0] },
|
||||
});
|
||||
|
||||
testUtils.testWithClient('vAdd with RESP3', async client => {
|
||||
// Test basic functionality with RESP3
|
||||
assert.equal(
|
||||
await client.vAdd('resp3-key', [1.5, 2.5, 3.5], 'resp3-element'),
|
||||
true
|
||||
);
|
||||
|
||||
// same element should not be added again
|
||||
assert.equal(
|
||||
await client.vAdd('resp3-key', [1, 2 , 3], 'resp3-element'),
|
||||
false
|
||||
);
|
||||
|
||||
// Test with options to ensure complex parameters work with RESP3
|
||||
assert.equal(
|
||||
await client.vAdd('resp3-key', [4.0, 5.0, 6.0], 'resp3-element2', {
|
||||
QUANT: 'Q8',
|
||||
CAS: true,
|
||||
SETATTR: { type: 'test', value: 123 }
|
||||
}),
|
||||
true
|
||||
);
|
||||
|
||||
// Verify the vector set was created correctly
|
||||
assert.equal(
|
||||
await client.vCard('resp3-key'),
|
||||
2
|
||||
);
|
||||
}, {
|
||||
...GLOBAL.SERVERS.OPEN,
|
||||
clientOptions: {
|
||||
RESP: 3
|
||||
},
|
||||
minimumDockerVersion: [8, 0]
|
||||
});
|
||||
});
|
Reference in New Issue
Block a user