gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: adds log key to RedisAdapter

Browse source
This commit is contained in:
hajdul88 2026-01-15 11:56:01 +01:00
parent 8b49f892ce
commit ace34b9a91
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/infrastructure/databases/cache/redis/RedisAdapter.py vendored
View file

@ -17,13 +17,14 @@ class RedisAdapter(CacheDBInterface):
host, host,
port, port,
lock_name="default_lock", lock_name="default_lock",
log_key="usage_logs",
username=None, username=None,
password=None, password=None,
timeout=240, timeout=240,
blocking_timeout=300, blocking_timeout=300,
connection_timeout=30, connection_timeout=30,
): ):
super().__init__(host, port, lock_name) super().__init__(host, port, lock_name, log_key)
self.host = host self.host = host
self.port = port self.port = port
@ -195,7 +196,7 @@ class RedisAdapter(CacheDBInterface):
CacheConnectionError: If Redis connection fails or times out. CacheConnectionError: If Redis connection fails or times out.
""" """
try: try:
usage_logs_key = f"usage_logs:{user_id}" usage_logs_key = f"{self.log_key}:{user_id}"
await self.async_redis.rpush(usage_logs_key, json.dumps(log_entry)) await self.async_redis.rpush(usage_logs_key, json.dumps(log_entry))
@ -223,7 +224,7 @@ class RedisAdapter(CacheDBInterface):
List of usage log entries, most recent first. List of usage log entries, most recent first.
""" """
try: try:
usage_logs_key = f"usage_logs:{user_id}" usage_logs_key = f"{self.log_key}:{user_id}"
entries = await self.async_redis.lrange(usage_logs_key, -limit, -1) entries = await self.async_redis.lrange(usage_logs_key, -limit, -1)
return [json.loads(e) for e in reversed(entries)] if entries else [] return [json.loads(e) for e in reversed(entries)] if entries else []
except (redis.ConnectionError, redis.TimeoutError) as e: except (redis.ConnectionError, redis.TimeoutError) as e: