Q i|xddlZddlmZmZmZmZmZmZmZm Z m Z ddl m Z m Z mZmZddlmZddlmZmZddlmZddlmZddlmZdd lmZdd lmZdd lm Z!dd l"m#Z$dd l%m Z dd l&m'Z'dd l(m#Z#ddl)m*Z*ddl+m,Z,m-Z-m.Z.ddl/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5ddl6m7Z7ddl8m9Z9m:Z:e7e;Zde=de=de?fdZ@deee=efdee=effdZAde5de?fdZBde3de?fdZCde=fdZDde=fd ZEd!ee=efdee=effd"ZFed#eeZGGd$d%ZHdS)&N) AnyDictListOptionalTupleTypeTypeVarUnionoverload)parse_qs urlencodeurlparse urlunparse)warn)Redis RedisCluster)ConnectionPool)r)r)AbstractConnection) Connection) SSLConnection)Sentinel) ResponseError __version__)REDIS_URL_ENV_VARSVS_MIN_REDIS_VERSIONSVS_MIN_SEARCH_VERSION) convert_bytesis_cluster_url)AsyncRedisClient RedisClientSyncRedisClient) get_logger)deprecated_argumentdeprecated_functionurlreturnc Nt|}t|j}|ddt |d}t |j|j|j|j ||j f}| }|dd||fS)aRemove 'cluster' parameter from URL query string and kwargs. AsyncRedisCluster doesn't accept 'cluster' parameter, but it might be present in the URL or kwargs for compatibility with other Redis clients. Args: url: Redis URL that might contain cluster parameter **kwargs: Keyword arguments that might contain cluster parameter Returns: Tuple of (cleaned_url, cleaned_kwargs) clusterNT)doseq) rr querypopr rschemenetlocpathparamsfragmentcopy)r&kwargsparsed query_params new_query cleaned_urlcleaned_kwargss C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\redisvl/redis/connection.py"_strip_cluster_from_url_and_kwargsr: s c]]FFL))LY%%%,d333I M M K M  O    K[[]]Ny$'''  &&version1version2cttt|d}ttt|d}t ||D]\}}||krdS||krdSt |t |krdSt |t |krdSdS)a  Check if version1 >= version2. Parameters: version1 (str): The first version string (e.g., "7.2.4"). version2 (str): The second version string (e.g., "6.2.1"). Returns: bool: True if version1 >= version2, False otherwise. .FT)listmapintsplitziplen)r<r=v1_partsv2_partsv1v2s r9is_version_gterJNsCX^^C001122HCX^^C001122Hh))B 7755 "WW44 8}}s8}}$$u XX & &t 4r; module_listcd|DS)zDUnpack a list of Redis modules pulled from the MODULES LIST command.c,i|]}|d|dS)namever).0modules r9 z(unpack_redis_modules..ms" D D DfF6NF5M D D Dr;rP)rKs r9unpack_redis_modulesrTks D D D D DDr;clientcB|d}|dd}t|}|dd}|dd}t |t }|t kp |t k}|o|S)zCheck if Redis server supports SVS-VAMANA. Args: client: Sync Redis client instance Returns: True if SVS-VAMANA is supported, False otherwise server redis_version0.0.0searchr searchlight)infogetRedisConnectionFactory get_modulesrJrrrUr\rXmodules search_versearchlight_verredis_ok modules_oks r9 supports_svsrfps ;;x DHH_g66M$0088GXq))Jkk-33Om-BCCH ,, 5 4 4  " "r;c^K|dd{V}|dd}t|d{V}|dd}|dd}t |t }|t kp |t k}|o|S)zAsync version of _supports_svs. Args: client: Async Redis client instance Returns: True if SVS-VAMANA is supported, False otherwise rWNrXrYrZrr[)r\r]r^get_modules_asyncrJrrr`s r9supports_svs_asyncrisX&& & & & & & &DHH_g66M*<z0convert_index_info_to_schema..sE   DF*R"7"7 ?BIIg   R   r; attributescPt|dkrdSi}d}t|dkr2t|d}|dvr||d<d} t|t|dD]M}|dzt|kr5t||}||dz||<Nn#t t tf$rYnwxYwi}d|vr&t| d|d<nd|vrt|d|d<ntdt|dz D]`}t||d vr5 t||dz|d<n#tt f$rY\wxYwad |vr|d |d <nd |d <d|vr|d|d<nd |d<d |vr|d |d<nId|vr|d|d<n'd|vr|d|d<nd|d<d|vr&|d}ddd}| |||d<d|vr1 t|d|d<n#tt f$rYnwxYwd|vr1 t|d|d<n#tt f$rYnwxYwd|vr1 t|d|d<n#tt f$rYnwxYwd|vr1 t|d|d<n#tt f$rYnwxYwd|vr1 t|d|d<n#tt f$rYnwxYwd|vrdS|S)N)FLATHNSW algorithmr}dimdims)DIMDIMSdistance_metriccosineflat data_typedatatypetypefloat32 compressionLVQ4x4LVQ4) GlobalSQ8 GlobalSQ4reducegraph_max_degreeconstruction_window_sizesearch_window_sizeepsilon) rEstrupperrangelower IndexError TypeErrorrmrBr,r]float) attrs vector_attrs start_pospos6_strikey normalizedcompression_valuecompression_mappings r9parse_vector_attrsz8convert_index_info_to_schema..parse_vector_attrss u::??4   u::>>58}}**,,H+++-5 [)  9c%jj!44 5 5q53u::%%eAh----//C(-a!e L% 5Iz2    D    L !$\%5%5e%<%-> 6*& 3 <  , ,,89J,K,Q,Q,S,SJ( ) )-5J( ) , & &&2;&?&E&E&G&GJ{ # #'-J{ # , & &%1+%>%D%D%F%FJz " " < ' '%1*%=%C%C%E%EJz " " | # #%1&%9%?%?%A%AJz " "&/Jz " L ( ( ,] ; &###  )<(?(?!#4))J} % | # # '*<+A'B'B 8$$ *      - - 14\BT5U1V1V -.. *     & 5 5 9< !;<:: 566 *     < / / 36 !5644 /00 *      $ $ (-l9.E(F(F 9%% *      # #4s A,C C'&C'?FF0/F0J55K K K**K>=K>LL32L3;MM('M(0N NNc }i}ddddddd}dvr |d krd |d <d|D]#\}}|vrd ||<|$ |fd t d t dDn4#t $r'}t d|dt|d}~wwxYw|S)Nno_stemcase_sensitivesortable index_missing index_emptyno_index)NOSTEM CASESENSITIVESORTABLE INDEXMISSING INDEXEMPTYNOINDEXUNFTEXTTunfcXi|]&}||dz'S)r})r)rQrrs r9rSzEconvert_index_info_to_schema..parse_attrs..s1QQQAq!!5Q<QQQr;rrzError parsing index attributes z, )r2removeitemsupdaterrErr)r field_typeoriginal parsed_attrs boolean_attrs redis_attr python_attres` r9 parse_attrsz1convert_index_info_to_schema..parse_attrsmsM::<<  -"+'!    E>>V##&* U# LL   '4':':'<'< ) ) #J U"",0 [) Z((( U   QQQQq#e**a9P9PQQQ     U U USxSS3q66SSTT T Us>=B<< C-"C((C-hash)rNrjsonr/VECTORr)r)rNprefix storage_type stopwords)indexfieldsN)rr@rErappend)rxrzprefixesrrr~ index_fieldsrr schema_fields field_attrsrNfieldr index_dicts r9convert_index_info_to_schemarsL)J,-a0H(D!!c(mmq&8&8A;01!4::<X % %&&{33E}"E'NN([QPPPE'NU####$,WWJ"+ ;  r;Tc eZdZdZeedd ddeedede fd Z e ddeede fd Z e ed d ddeedefd Ze ed d ddeedefd Ze ddeedefdZe ddeedefdZe de defdZe de deeeffdZe dedeeeffdZe dde deeddfdZe ddedeeddfdZe ededeededefdZ e ededee!dede!fdZ e dede"eeee!fdede"ee!ffdZ e d ede#e$e#ee%feeeeeeeffdZ&dS) r^aBuilds connections to a Redis database, supporting both synchronous and asynchronous clients. This class allows for establishing and handling Redis connections using either standard Redis or async Redis clients, based on the provided configuration. connectzBPlease use `get_redis_connection` or `get_async_redis_connection`.NFrn use_asyncr'c T|p t}|r|jn|j}||fi|S)aCreate a connection to the Redis database based on a URL and some connection kwargs. This method sets up either a synchronous or asynchronous Redis client based on the provided parameters. Args: redis_url (Optional[str]): The URL of the Redis server to connect to. If not provided, the environment variable REDIS_URL is used. use_async (bool): If True, an asynchronous client is created. Defaults to False. **kwargs: Additional keyword arguments to be passed to the Redis client constructor. Raises: ValueError: If redis_url is not provided and REDIS_URL environment variable is not set. )roget_async_redis_connectionget_redis_connection)clsrnrr3connection_funcs r9rzRedisConnectionFactory.connectsG27!5!7!7 .7 UC * *S=U y33F333r;c |p t}|drtj|tfi|}n2t |fi|rt j|fi|}nt j|fi|}t| d} | d|n5#t$r(t|dr| |YnwxYw|S)a Creates and returns a synchronous Redis client. Args: url (Optional[str]): The URL of the Redis server. If not provided, the environment variable REDIS_URL is used. **kwargs: Additional keyword arguments to be passed to the Redis client constructor. Returns: Redis: A synchronous Redis client instance. Raises: ValueError: If url is not provided and REDIS_URL environment variable is not set. redis+sentinellib_nameLIB-NAMEecho)ro startswithr^_redis_sentinel_clientrrrfrom_urlrwr]client_setinforhasattrr)rnr3r&rU _lib_names r9rz+RedisConnectionFactory.get_redis_connections(1/11 >>* + + 3+B3XXQWXXFF C * *6 * * 3!*399&99FF^C22622F"&**Z"8"899  '  ! !*i 8 8 8 8 ' ' 'vv&& ' I&&& ' sB++/CCr&cPK|dd}|p|p t}|drtj|t fi|}nRt |fi|r#t|fi|\}}tj |fi|}n"t|fi|\}}t j |fi|}t| d} | d|d{Vn;#t$r.t|dr||d{VYnwxYw|S)aCreates and returns an asynchronous Redis client. NOTE: This method is the future form of `get_async_redis_connection` but is only used internally by the library now. Args: redis_url (Optional[str]): The URL of the Redis server. If neither `redis_url` nor `url` are provided, the environment variable REDIS_URL is used. url (Optional[str]): Former parameter for the URL of the Redis server. Use `redis_url` instead. (Deprecated) **kwargs: Additional keyword arguments to be passed to the async Redis client constructor. Returns: AsyncRedisClient: An asynchronous Redis client instance (either AsyncRedis or AsyncRedisCluster). Raises: ValueError: If url is not provided and REDIS_URL environment variable is not set. r&Nrrrr)r,rorr^r AsyncRedisrr:AsyncRedisClusterrrwr]rrrr)rnr3_deprecated_urlr&rUr7r8rs r9_get_aredis_connectionz-RedisConnectionFactory._get_aredis_connections6!**UD11DD.B.D.D >>* + + H+BZ#)FFC * *6 * * H*L++++ 'K'/ NN~NNFF+M++++ 'K (GGGGF"&**Z"8"899  -'' I>> > > > > > > > > - - -vv&& -kk),,,,,,,,, - sC++5D#"D#c xtdt|dd}|p|p t}|drt j|tfi|St|fi|r"t|fi|\}}tj |fi|St|fi|\}}tj |fi|S)aCreates and returns an asynchronous Redis client. Args: redis_url (Optional[str]): The URL of the Redis server. If neither `redis_url` nor `url` are provided, the environment variable REDIS_URL is used. url (Optional[str]): Former parameter for the URL of the Redis server. Use `redis_url` instead. (Deprecated) **kwargs: Additional keyword arguments to be passed to the async Redis client constructor. Returns: AsyncRedis: An asynchronous Redis client instance. Raises: ValueError: If url is not provided and REDIS_URL environment variable is not set. zGget_async_redis_connection will become async in the next major release.r&Nr) rDeprecationWarningr,rorr^rrrr:rr)rnr3rr&r7r8s r9rz1RedisConnectionFactory.get_async_redis_connectionAs 0 U    !**UD11DD.B.D.D >>* + + F)@Z#) C * *6 * * F*L++++ 'K%-kLL^LL L+M++++ 'K&{EEnEE Er;c F|p t}tj|fi|S)zCCreates and returns a synchronous Redis client for a Redis cluster.)rorr)rnr3r&s r9get_redis_cluster_connectionz3RedisConnectionFactory.get_redis_cluster_connectionqs. 1/11$S33F333r;c f|p t}t|fi|\}}tj|fi|S)zECreates and returns an asynchronous Redis client for a Redis cluster.)ror:rr)rnr3r&r7r8s r9"get_async_redis_cluster_connectionz9RedisConnectionFactory.get_async_redis_cluster_connectionzsI 1/11&H&W&WPV&W&W# ^ )+HHHHHr; redis_clientct|trtdt|tsJ|jjt krtnt}tj tdd|i|jj S)z:Convert a synchronous Redis client to an asynchronous one.z;RedisCluster is not supported for sync-to-async conversion.connection_classrP) rrrmrconnection_poolrrAsyncSSLConnectionAsyncConnectionr from_poolAsyncConnectionPoolconnection_kwargs)rrs r9sync_to_async_redisz*RedisConnectionFactory.sync_to_async_rediss lL 1 1 M  ,..... +< MM     #   !1 .@     r;rUc^tt|SrrTrrKrUs r9r_z"RedisConnectionFactory.get_moduless$#M&2D2D2F2F$G$GHHHr;cnKtt|d{VSrr r s r9rhz(RedisConnectionFactory.get_modules_asyncs:#M8J8J8L8L2L2L2L2L2L2L$M$MNNNr;rc4tt|ttfst dt |} |d|dS#t$r,t|dr| |YdSYdSwxYw)zValidates the sync Redis client. Note: Module validation has been removed. This method now only validates the client type and sets the library name. z=Invalid Redis client instance. Must be Redis or RedisCluster.rrN) issubclassrrrrrwrrrrrrrs r9validate_sync_redisz*RedisConnectionFactory.validate_sync_rediss$|,,ul.CDD O  "(++  -  ' ' I > > > > > - - -|V,, -!!),,,,,, - - - -s A!!/BBcPKtt|ttfst dt |} |d|d{VdS#t$r2t|dr| |d{VYdSYdSwxYw)zValidates the async Redis client. Note: Module validation has been removed. This method now only validates the client type and sets the library name. zOInvalid async Redis client instance. Must be async Redis or async RedisCluster.rNr) rrrrrrwrrrrrs r9validate_async_redisz+RedisConnectionFactory.validate_async_rediss$|,,z;L.MNN a "(++  3--j)DD D D D D D D D D D 3 3 3|V,, 3"'' 222222222222 3 3 3 3s A))5B%$B% redis_classr3c dSrrPrnrr3s r9rz-RedisConnectionFactory._redis_sentinel_clients r;c dSrrPrs r9rz-RedisConnectionFactory._redis_sentinel_clients Sr;c t|\}}}}}i}|r ||d<||d<|r ||d<||d<|r||d<|tkrt|fd|i|} | j|fd|i|St |fd|i|} | j|fd|i|S)aCreate a Redis client connected via Sentinel for high availability. Parses a Sentinel URL and creates a Redis client connected to the master instance discovered by Sentinel. Supports both sync and async clients by using the appropriate Sentinel class. Args: redis_url: Sentinel URL in the format: ``redis+sentinel://[user:pass@]host1:port1[,host2:port2,...][/service][/db]`` Service name defaults to "mymaster" if not specified. redis_class: The Redis client class to use (Redis or AsyncRedis). **kwargs: Additional arguments passed to Sentinel and master_for(). Returns: A Redis client (sync or async) connected to the Sentinel-managed master. Example: >>> client = RedisConnectionFactory._redis_sentinel_client( ... "redis+sentinel://sentinel1:26379,sentinel2:26379/mymaster", ... Redis ... ) usernamepassworddbsentinel_kwargsr)r^_parse_sentinel_urlr AsyncSentinel master_forr) rnrr3 sentinel_list service_namerrrrasync_sentinel sync_sentinels r9rz-RedisConnectionFactory._redis_sentinel_clients36 # 6 6y A A < |R8+-  **2OJ '!)F:   **2OJ '!)F:   F4L * $ $*/>BHN->,*59? %/>BHM,=+*59? r;c:t|}|jdd}|d}g}|D]x}|d}t|dkr1||dt |df[||ddfyd }d }|jr?|jd } | dpd }t| dkr| d}||||j|jfS) a~Parse a Redis Sentinel URL into its components. Args: url: Sentinel URL in the format: ``redis+sentinel://[user:pass@]host1:port1[,host2:port2,...]/service[/db]`` Returns: A tuple containing: - sentinel_list: List of (host, port) tuples for Sentinel nodes - service_name: The Sentinel service name (defaults to "mymaster") - db: The database number (or None if not specified) - username: The username for authentication (or None) - password: The password for authentication (or None) Example: >>> RedisConnectionFactory._parse_sentinel_url( ... "redis+sentinel://user:pass@host1:26379,host2:26380/mymaster/0" ... ) ([('host1', 26379), ('host2', 26380)], 'mymaster', '0', 'user', 'pass') @,:rrr}i gmymasterN/) rr.rCrErrBr/rr) r& parsed_url hosts_partsentinel_hostsr host host_partsr!r path_partss r9rz*RedisConnectionFactory._parse_sentinel_url s%0c]] &,,S11"5 #))#.. " = =DCJ:!##$$jmSA5G5G%HIIII$$jmU%;<<<<!  ? ##..s33J%a=6JL:""]lB 0CZEXXXr;)NFr)'__name__ __module__ __qualname____doc__ classmethodr%rrboolr!r staticmethodr"rr$r rrrrrrr rrr_rhrrr rrrrr rrrBrrPr;r9r^r^sWAF44  49=4 444[46#'$$C=$ $$$\$L ,,#'77C=7 777-,\7r ,,#',F,FC=,F ,F,F,F-,\,F\#'44C=4 444\4#'IIC=I III\I %    \ 4IOIS#XIII\IO(8OT#s(^OOO\O#'--%-3-- ---\-4#'33&33-3 333\30 %)%[rKs SSSSSSSSSSSSSSSSSSSSSSBBBBBBBBBBBB%%%%%%%%??????------CCCCCCRRRRRRBBBBBBHHHHHH<<<<<<************###### >=======HHHHHHHHHH((((((HHHHHHHH H  +' +' 3S#X +'+'+'+'\SCD:Ed4S>&:EtCH~EEEE ##T####:#%5#$####:c &C & & & &qT#s(^qS#Xqqqqh GC ##UYUYUYUYUYUYUYUYUYUYr;