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
57
packages/client/lib/commands/VEMB_RAW.ts
Normal file
57
packages/client/lib/commands/VEMB_RAW.ts
Normal file
@@ -0,0 +1,57 @@
|
||||
import { CommandParser } from '../client/parser';
|
||||
import {
|
||||
RedisArgument,
|
||||
Command,
|
||||
BlobStringReply,
|
||||
SimpleStringReply,
|
||||
DoubleReply
|
||||
} from '../RESP/types';
|
||||
import { transformDoubleReply } from './generic-transformers';
|
||||
import VEMB from './VEMB';
|
||||
|
||||
type RawVembReply = {
|
||||
quantization: SimpleStringReply;
|
||||
raw: BlobStringReply;
|
||||
l2Norm: DoubleReply;
|
||||
quantizationRange?: DoubleReply;
|
||||
};
|
||||
|
||||
const transformRawVembReply = {
|
||||
2: (reply: any[]): RawVembReply => {
|
||||
return {
|
||||
quantization: reply[0],
|
||||
raw: reply[1],
|
||||
l2Norm: transformDoubleReply[2](reply[2]),
|
||||
...(reply[3] !== undefined && { quantizationRange: transformDoubleReply[2](reply[3]) })
|
||||
};
|
||||
},
|
||||
3: (reply: any[]): RawVembReply => {
|
||||
return {
|
||||
quantization: reply[0],
|
||||
raw: reply[1],
|
||||
l2Norm: reply[2],
|
||||
quantizationRange: reply[3]
|
||||
};
|
||||
},
|
||||
};
|
||||
|
||||
export default {
|
||||
IS_READ_ONLY: true,
|
||||
/**
|
||||
* Retrieve the RAW approximate vector associated with a vector set element
|
||||
*
|
||||
* @param parser - The command parser
|
||||
* @param key - The key of the vector set
|
||||
* @param element - The name of the element to retrieve the vector for
|
||||
* @see https://redis.io/commands/vemb/
|
||||
*/
|
||||
parseCommand(
|
||||
parser: CommandParser,
|
||||
key: RedisArgument,
|
||||
element: RedisArgument
|
||||
) {
|
||||
VEMB.parseCommand(parser, key, element);
|
||||
parser.push('RAW');
|
||||
},
|
||||
transformReply: transformRawVembReply
|
||||
} as const satisfies Command;
|
Reference in New Issue
Block a user