You've already forked node-redis
mirror of
https://github.com/redis/node-redis.git
synced 2025-08-03 04:01:40 +03:00
c5b4f47975
* 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
58 lines
1.4 KiB
TypeScript
58 lines
1.4 KiB
TypeScript
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;
|