a bgB@sPddlmZmZmZmZmZmZmZddlm Z ddl m Z Gddde Z dS))AnyIteratorListOptionalSequenceTuplecast) ByteStore) get_clientc@seZdZdZddddddeeeeeeeeeddddZ eeddd Z e ee ee d d d Ze eee fdd ddZe edd ddZddeeeedddZdS) RedisStoreaBaseStore implementation using Redis as the underlying store. Examples: Create a RedisStore instance and perform operations on it: .. code-block:: python # Instantiate the RedisStore with a Redis connection from langchain_community.storage import RedisStore from langchain_community.utilities.redis import get_client client = get_client('redis://localhost:6379') redis_store = RedisStore(client=client) # Set values for keys redis_store.mset([("key1", b"value1"), ("key2", b"value2")]) # Get values for keys values = redis_store.mget(["key1", "key2"]) # [b"value1", b"value2"] # Delete keys redis_store.mdelete(["key1"]) # Iterate over keys for key in redis_store.yield_keys(): print(key) # noqa: T201 N)client redis_url client_kwargsttl namespace)r r rrrreturnc Cszddlm}Wn.ty>}ztd|WYd}~n d}~00|rT|sL|rTtd|sd|sdtd|rt||stdt|jd|}n |stdt|fi|pi}||_ t|t s|durtd t|d||_ ||_ dS) aInitialize the RedisStore with a Redis connection. Must provide either a Redis client or a redis_url with optional client_kwargs. Args: client: A Redis connection instance redis_url: redis url client_kwargs: Keyword arguments to pass to the Redis client ttl: time to expire keys in seconds if provided, if None keys will never expire namespace: if provided, all keys will be prefixed with this namespace r)RediszLThe RedisStore requires the redis library to be installed. pip install redisNz`Either a Redis client or a redis_url with optional client_kwargs must be provided, but not both.z6Either a Redis client or a redis_url must be provided.zExpected Redis client, got z instead.z$Expected int or None, got type(ttl)=) Zredisr ImportError ValueError isinstance TypeErrortype__name__r r intrr) selfr r rrrreZ_clientro/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/storage/redis.py__init__&s<  zRedisStore.__init__)keyrcCs d}|jr|j||S|S)zGet the key with the namespace prefix. Args: key (str): The original key. Returns: str: The key with the namespace prefix. /)r)rr delimiterrrr_get_prefixed_keyas zRedisStore._get_prefixed_key)keysrcs(ttttjfdd|DS)z.Get the values associated with the given keys.csg|]}|qSrr".0rrrr sz#RedisStore.mget..)rrrbytesr mget)rr#rr'rr+os zRedisStore.mget)key_value_pairsrcCs<|j}|D] \}}|j||||jdq|dS)zSet the given key-value pairs.)exN)r Zpipelinesetr"rexecute)rr,pipervaluerrrmsetvs  zRedisStore.msetcs"fdd|D}jj|dS)zDelete the given keys.csg|]}|qSrr$r%r'rrr(r)z&RedisStore.mdelete..N)r delete)rr#Z_keysrr'rmdelete~szRedisStore.mdelete)prefix)r5rccst|r||}n |d}ttt|jj|d}|D]8}|d}|jrh|t|jdd}|Vq6|Vq6dS)zYield keys in the store.*)matchzutf-8N) r"rrr*r scan_iterdecoderlen)rr5patternr9rZ decoded_keyZ relative_keyrrr yield_keyss   zRedisStore.yield_keys)r __module__ __qualname____doc__rrstrdictrrr"rrr*r+rr2r4rr=rrrrr s&  ;r N) typingrrrrrrrZlangchain_core.storesr Z#langchain_community.utilities.redisr r rrrrs$