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

Merge 5a0d71452e into d266c59c2a

Browse source
This commit is contained in:
Shrinit Goyal 2025-10-03 14:11:23 +02:00 • committed by GitHub
commit 80c04e26d5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/utils/kvstore/config.py
View file

@ -31,10 +31,12 @@ class RedisKVStoreConfig(CommonConfig):
type: Literal["redis"] = KVStoreType.redis.value type: Literal["redis"] = KVStoreType.redis.value
host: str = "localhost" host: str = "localhost"
port: int = 6379 port: int = 6379
db: int = 0
ttl: int = None
@property @property
def url(self) -> str: def url(self) -> str:
return f"redis://{self.host}:{self.port}" return f"redis://{self.host}:{self.port}/{self.db}"
@classmethod @classmethod
def pip_packages(cls) -> list[str]: def pip_packages(cls) -> list[str]:

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: async def set(self, key: str, value: str, expiration: datetime | None = None) -> None:
key = self._namespaced_key(key) 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: if expiration:
await self.redis.expireat(key, expiration) await self.redis.expireat(key, expiration)