phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 04:04:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

handle DB and TTL for redis KV store

Browse source
This commit is contained in:
shrinitgoyal 2025-08-19 14:54:10 +05:30
parent 46ff302d87
commit e2d295ae40
2 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/utils/kvstore/redis/redis.py
View file

@ -26,7 +26,10 @@ class RedisKVStoreImpl(KVStore):
async def set(self, key: str, value: str, expiration: datetime | None = None) -> None:
key = self._namespaced_key(key)
await self.redis.set(key, value)
if self.config.ttl:
await self.redis.set(key, value, ex=self.config.ttl)
else:
await self.redis.set(key, value)
if expiration:
await self.redis.expireat(key, expiration)