Ih0ddlZddlmZddlmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZmZmZmZdZd Zd Zd Zd Zd ZdZdZdZdZdZGddeZGddeZ GddeZ!y)N)Enum) AwaitableDictListOptionalUnion) NEVER_DECODE)get_protocol_version) DataError)CommandsProtocol EncodableTKeyTNumberVADDVSIMVREMVDIMVCARDVEMBVLINKSVINFOVSETATTRVGETATTR VRANDMEMBERceZdZdZdZdZdZy)QuantizationOptionsz*Quantization options for the VADD command.NOQUANTBINQ8N)__name__ __module__ __qualname____doc__rrrZ/var/www/html/planif/env/lib/python3.12/site-packages/redis/commands/vectorset/commands.pyrrs4G C Br%rc eZdZdZdZdZdZdZy)CallbacksOptionsz2Options that can be set for the commands callbacksRAW WITHSCORESALLOW_DECODINGRESP3N)r r!r"r#r)r*r+r,r$r%r&r(r(s< CJ%N Er%r(ceZdZdZ d dedeeeefde de e de e de e d e ed e eee fd e e d eee e ffd Z d!dedeeeee fde e de e d e ede e de e de e de e d eee eeeeeeeffe eeeeeeeffffdZded eee e ffdZded eee e ffdZdede d eee e ffdZ d"dede de e d eee eeeee effe eeeee effffdZ d"dede de e d eee eeeee efeee efeffe eeeee efeee efeffffdZded eeeeffdZ d#dede d e eee fd eee e ffdZdede d ee eee effdZ d#dede e d eee eee e fe eee e fffdZy)$VectorSetCommandszRedis VectorSet commandsNkeyvectorelement reduce_dimcas quantizationef attributesnumlinksreturnc |r|s tdg} |r| jd|gg} t|tr| jd|gn-| jdt |g| j|| j| | j ||r| j d|r| j |j |r| jd|g|r;t|trtj|} n|} | jd| g| r| jd| g|jt|g| S) a Add vector ``vector`` for element ``element`` to a vector set ``key``. ``reduce_dim`` sets the dimensions to reduce the vector to. If not provided, the vector is not reduced. ``cas`` is a boolean flag that indicates whether to use CAS (check-and-set style) when adding the vector. If not provided, CAS is not used. ``quantization`` sets the quantization type to use. If not provided, int8 quantization is used. The options are: - NOQUANT: No quantization - BIN: Binary quantization - Q8: Signed 8-bit quantization ``ef`` sets the exploration factor to use. If not provided, the default exploration factor is used. ``attributes`` is a dictionary or json string that contains the attributes to set for the vector. If not provided, no attributes are set. ``numlinks`` sets the number of links to create for the vector. If not provided, the default number of links is used. For more information see https://redis.io/commands/vadd z(Both vector and element must be providedREDUCEFP32VALUESCASEFSETATTRM) r extend isinstancebyteslenappendvaluedictjsondumpsexecute_commandVADD_CMD) selfr/r0r1r2r3r4r5r6r7pieces values_piecesattributes_jsons r&vaddzVectorSetCommands.vadd+s#NWFG G  MM8Z0 1 fe $  &&!1 2  (CK!8 9   ( m$ g  MM%  MM,,, - MM4* % *d+"&**Z"8", MM9o6 7  MM3/ *#t##Hc;F;;r%input with_scorescountfilter filter_eftruth no_threadc |s tdg} i} t|tr| jd|gnQt|tr.| jdt |g| j|n| jd|g|r.| j dd| tjj<|r| jd|g|r| jd|g|r| jd |g|r| jd |g|r| j d | r| j d |jt|g| i| S) a Compare a vector or element ``input`` with the other vectors in a vector set ``key``. ``with_scores`` sets if the results should be returned with the similarity scores of the elements in the result. ``count`` sets the number of results to return. ``ef`` sets the exploration factor. ``filter`` sets filter that should be applied for the search. ``filter_ef`` sets the max filtering effort. ``truth`` when enabled forces the command to perform linear scan. ``no_thread`` when enabled forces the command to execute the search on the data structure in the main thread. For more information see https://redis.io/commands/vsim z'input' should be providedr;r<ELEr*TCOUNTr>FILTERz FILTER-EFTRUTHNOTHREAD) r rBrCrAlistrDrEr(r*rFrJVSIM_CMD) rLr/rQrRrSr5rTrUrVrWrMoptionss r&vsimzVectorSetCommands.vsimys%J89 9 eU # MM65/ * t $ MM8SZ0 1 MM% MM5%. )  MM, '9=G$//55 6  MM7E* + MM4* %  MM8V, -  MM; 2 3  MM' "  MM* %#t##HcFFFgFFr%c.|jt|S)a Get the dimension of a vector set. In the case of vectors that were populated using the `REDUCE` option, for random projection, the vector set will report the size of the projected (reduced) dimension. Raises `redis.exceptions.ResponseError` if the vector set doesn't exist. For more information see https://redis.io/commands/vdim )rJVDIM_CMDrLr/s r&vdimzVectorSetCommands.vdims##Hc22r%c.|jt|S)z Get the cardinality(the number of elements) of a vector set with key ``key``. Raises `redis.exceptions.ResponseError` if the vector set doesn't exist. For more information see https://redis.io/commands/vcard )rJ VCARD_CMDrds r&vcardzVectorSetCommands.vcards##Is33r%c0|jt||S)zw Remove an element from a vector set. For more information see https://redis.io/commands/vrem )rJVREM_CMDrLr/r1s r&vremzVectorSetCommands.vrems ##Hc7;;r%rawc\i}g}|j||gt|jdvrd|tjj <|r|j dd|t<t|jdr#|jjjds9t|jdr@|jjjdrd|tjj <d|tjj <|jtg|i|S)a' Get the approximated vector of an element ``element`` from vector set ``key``. ``raw`` is a boolean flag that indicates whether to return the interal representation used by the vector. For more information see https://redis.io/commands/vembed )3Tr)connection_pooldecode_responses nodes_manager)rAr clientr(r,rFrEr hasattrrqconnection_kwargsrsr+r)rJVEMB_CMD)rLr/r1rmr`rMs r&vembzVectorSetCommands.vembs sGn%  , 848G$**00 1  MM% $(GL ! %67KK//AABTU _5KK--??@RS BF(77==>26G$((.. /#t##HAvAAAr%ci}g}|j||g|r.|jdd|tjj<|j t g|i|S)a Returns the neighbors for each level the element ``element`` exists in the vector set ``key``. The result is a list of lists, where each list contains the neighbors for one level. If the element does not exist, or if the vector set does not exist, None is returned. If the ``WITHSCORES`` option is provided, the result is a list of dicts, where each dict contains the neighbors for one level, with the scores as values. For more information see https://redis.io/commands/vlinks r*T)rArEr(r*rFrJ VLINKS_CMD)rLr/r1rRr`rMs r&vlinkszVectorSetCommands.vlinkssa* sGn%  MM, '9=G$//55 6#t##JCC7CCr%c.|jt|S)zw Get information about a vector set. For more information see https://redis.io/commands/vinfo )rJ VINFO_CMDrds r&vinfozVectorSetCommands.vinfo-s ##Is33r%c|d}n(t|trtj|}n|}|j t |||S)z Associate or remove JSON attributes ``attributes`` of element ``element`` for vector set ``key``. For more information see https://redis.io/commands/vsetattr z{})rBrGrHrIrJ VSETATTR_CMD)rLr/r1r6rOs r&vsetattrzVectorSetCommands.vsetattr5sC  "O  D )"jj4O(O##L#wPPr%c0|jt||S)a  Retrieve the JSON attributes of an element ``elemet`` for vector set ``key``. If the element does not exist, or if the vector set does not exist, None is returned. For more information see https://redis.io/commands/vgetattr )rJ VGETATTR_CMDrks r&vgetattrzVectorSetCommands.vgetattrHs##L#w??r%c|g}|j|||j||jtg|S)a Returns random elements from a vector set ``key``. ``count`` is the number of elements to return. If ``count`` is not provided, a single element is returned as a single string. If ``count`` is positive(smaller than the number of elements in the vector set), the command returns a list with up to ``count`` distinct elements from the vector set If ``count`` is negative, the command returns a list with ``count`` random elements, potentially with duplicates. If ``count`` is greater than the number of elements in the vector set, only the entire set is returned as a list. If the vector set does not exist, ``None`` is returned. For more information see https://redis.io/commands/vrandmember )rErJVRANDMEMBER_CMD)rLr/rSrMs r& vrandmemberzVectorSetCommands.vrandmemberUs?, c   MM% #t##O=f==r%)NFNNNN)FNNNNFF)F)N)r r!r"r#rrrfloatrCstrrintboolrrrGrrPr rrarerhrlrxr{r~rrrr$r%r&r.r.(s"%)#6:#15"&L< L<d5k5()L< L< SM L< d^ L<23L< V L<U49-.L<3-L< y~s" #LIG (4d:&6Z=O8P&P QRSTeD,d:v3E.FFGHI K IGV 3 3y~s':!; 344% #(;"<4<C(B(B"%(B,4TN(B (5j!14Z3H!HIJKtJ'c:o)>>?@ B (BVFKDD"%D444% $(=">4QUQQ"%Q3;E$)> ( > (5cC012HU49c>=R4SS >r%r.)"rHenumrtypingrrrrr redis.clientr redis.commands.helpersr redis.exceptionsr redis.typingr r rrrKr_rjrcrgrwrzr}rrrrr(r.r$r%r&rs~ 99%7&CC           $tG>(G>r%