Q i$dZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZmZmZGdd Zd S) zBase cache interface for RedisVL. This module defines the abstract base cache interface that is implemented by specific cache types such as LLM caches and embedding caches. )Mapping)AnyDictOptionalUnion)Redis) RedisCluster)RedisConnectionFactory)AsyncRedisClientSyncRedisClientSyncRedisClusterc~eZdZUdZeeed<eeed<ddddifdedee deed eed ed e ee ff d Z d efdZ ded efdZed ee fdZddee d dfdZd efdZd efdZddedee d dfdZddedee d dfdZddZddZddZddZdS) BaseCachezBase abstract cache interface for all RedisVL caches. This class defines common functionality shared by all cache implementations, including TTL management, connection handling, and basic cache operations. _redis_client_async_redis_clientNzredis://localhost:6379namettl redis_clientasync_redis_client redis_urlconnection_kwargsc||_d|_|||||d|_||_||_|s|r d|_dSd|_dS)a?Initialize a base cache. Args: name (str): The name of the cache. ttl (Optional[int], optional): The time-to-live for records cached in Redis. Defaults to None. redis_client (Optional[SyncRedisClient], optional): A redis client connection instance. Defaults to None. redis_url (str, optional): The redis url. Defaults to redis://localhost:6379. connection_kwargs (Dict[str, Any]): The connection arguments for the redis client. Defaults to empty {}. N)rrrFT)r_ttlset_ttl redis_kwargsrr_owns_redis_client)selfrrrrrrs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\redisvl/extensions/cache/base.py__init__zBaseCache.__init__s|* #'  S)"!2  $6 )  +- +&+D # # #&*D # # #returnc|jdS)zpGet the key prefix for Redis keys. Returns: str: The prefix to use for Redis keys. :)rrs r _get_prefixzBaseCache._get_prefixCs )r entry_idc2||S)zGenerate a full Redis key for the given entry ID. Args: entry_id (str): The unique entry ID. Returns: str: The full Redis key including prefix. )r%)rr&s r _make_keyzBaseCache._make_keyKs!""$$0h000r c|jS)z6The default TTL, in seconds, for entries in the cache.)rr$s rrz BaseCache.ttlVs yr c|r=t|tstd|t||_dSd|_dS)a#Set the default TTL, in seconds, for entries in the cache. Args: ttl (Optional[int], optional): The optional time-to-live expiration for the cache, in seconds. Raises: ValueError: If the time-to-live value is not an integer. z"TTL must be an integer value, got N) isinstanceint ValueErrorr)rrs rrzBaseCache.set_ttl[sR  c3'' M !Kc!K!KLLLCDIIIDIIIr c|j1|jd}|jd}tj|fi||_|jS)znGet or create a Redis client. Returns: SyncRedisClient: A Redis client instance. Nrr)rrrfrom_url)rurlkwargss r_get_redis_clientzBaseCache._get_redis_clientlsK   %#K0C&':;F!&!>!>v!>!>D !!r cKt|dr|j|jd}|r6t |t t frtj||_nt|jd}|jdi}t |ts$tdt|j tj|fi||_|jS)z}Get or create an async Redis client. Returns: AsyncRedisClient: An async Redis client instance. rNrrrzaExpected `connection_kwargs` to be a dictionary (e.g. {'decode_responses': True}), but got type: )hasattrrrgetr+rr r sync_to_async_redisstrr TypeErrortype__name__get_async_redis_connection)rclientr0r1s r_get_async_redis_clientz!BaseCache._get_async_redis_clientys t233 t7O7W&**>::F *Ve\-BCC +A+U,,(($+K899*../BBGG!&'22#A)-f)>AA +EcTTVTT(''r keycx||n|j}|r,|}|||dSdS)aSet or refresh the expiration time for a key in the cache. Args: key (str): The Redis key to set the expiration on. ttl (Optional[int], optional): The time-to-live in seconds. If None, uses the default TTL configured for this cache instance. Defaults to None. Note: If neither the provided TTL nor the default TTL is set (both are None), this method will have no effect. N)rr2expirerr>rrr<s rr@zBaseCache.expiresPoss49  %++--F MM#t $ $ $ $ $ % %r cK||n|j}|r8|d{V}|||d{VdSdS)aAsynchronously set or refresh the expiration time for a key in the cache. Args: key (str): The Redis key to set the expiration on. ttl (Optional[int], optional): The time-to-live in seconds. If None, uses the default TTL configured for this cache instance. Defaults to None. Note: If neither the provided TTL nor the default TTL is set (both are None), this method will have no effect. N)rr=r@rAs raexpirezBaseCache.aexpiresxoss49  +7799999999F--T** * * * * * * * * * + +r cf|}|}d} |||dd\}}|r |j||dkrdSt |t r=t |}td|DrdSn|})zClear the cache of all keys.rT*dcursormatchcountc3"K|] }|dkV dSrN.0vs r z"BaseCache.clear..&55!qAv555555r N) r2r%scandeleter+rlistvaluesallrr<prefixrH cursor_intkeys cursor_valuess rclearzBaseCache.clears''))!!## $%{{&6 TW{XX J % t$$QJ00 $ $Z%6%6%8%8 9 9 55}55555E$ $r cK|d{V}|}d} |||ddd{V\}}|r|j|d{V|dkrdSt |t r=t |}td|DrdSn|})z"Async clear the cache of all keys.NrTrErFrGc3"K|] }|dkV dSrLrMrNs rrQz#BaseCache.aclear..rRr ) r=r%rSrTr+rrUrVrWrXs raclearzBaseCache.aclears)3355555555!!## $%+[[\\\&1&&       J +#fmT********QJ00 $ $Z%6%6%8%8 9 9 55}55555E$ $r c||jdurdS|jr)|jd|_d|_dSdS)zDisconnect from Redis.FN)rrcloserr$s r disconnectzBaseCache.disconnectsW  "e + + F   ,   $ $ & & &!%D (,D $ $ $  , ,r cK|jdurdS|jr |jd|_t|dr/|jr*|jd{Vd|_dSdSdS)zAsync disconnect from Redis.FNr)rrrbr4racloser$s r adisconnectzBaseCache.adisconnects  "e + + F   &   $ $ & & &!%D  4. / / ,D4L ,*1133 3 3 3 3 3 3 3'+D $ $ $ , , , ,r )N)r!N)r: __module__ __qualname____doc__rr __annotations__r r7r,rrrr%r(propertyrrr2r=r@rCr]r`rcrfrMr rrrs? O,,,,!"23333 "269=1,.&+&+&+c]&+/ &+ %%56 &+  &+ S>&+&+&+&+PS 1# 1# 1 1 1 1Xc]X8C=D" "? " " " "(/?((((4%%#%HSM%T%%%%$+++8C=+D++++$$$$$,$$$$0 , , , , , , , , , ,r rN)ricollections.abcrtypingrrrrredisr redis.clusterr redisvl.redis.connectionr redisvl.typesr r r rrMr rrrs $#####------------&&&&&&;;;;;;MMMMMMMMMMl,l,l,l,l,l,l,l,l,l,r