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

chore: default values for session list expiration

Browse source
This commit is contained in:
hajdul88 2025-10-15 13:26:06 +02:00
parent 678d7d22d7
commit c4545cdaa3
1 changed files with 4 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -72,7 +72,7 @@ class RedisAdapter(CacheDBInterface):
question: str, question: str,
context: str, context: str,
answer: str, answer: str,
ttl: int | None = None, ttl: int | None = 86400,
): ):
""" """
Add a Q/A/context triplet to a Redis list for this session. Add a Q/A/context triplet to a Redis list for this session.
@ -100,7 +100,7 @@ class RedisAdapter(CacheDBInterface):
if ttl is not None: if ttl is not None:
await self.async_redis.expire(session_key, ttl) await self.async_redis.expire(session_key, ttl)
async def get_latest_qa(self, user_id: str, session_id: str, last_n: int = 1): async def get_latest_qa(self, user_id: str, session_id: str, last_n: int = 10):
""" """
Retrieve the most recent Q/A/context triplet(s) for the given session. Retrieve the most recent Q/A/context triplet(s) for the given session.
""" """
@ -142,7 +142,6 @@ async def main():
"What is Redis?", "What is Redis?",
"Basic DB context", "Basic DB context",
"Redis is an in-memory data store.", "Redis is an in-memory data store.",
ttl=10,
) )
await adapter.add_qa( await adapter.add_qa(
user_id, user_id,
@ -150,7 +149,6 @@ async def main():
"Who created Redis?", "Who created Redis?",
"Historical context", "Historical context",
"Salvatore Sanfilippo (antirez).", "Salvatore Sanfilippo (antirez).",
ttl=10,
) )
print("\nLatest QA:") print("\nLatest QA:")
@ -168,8 +166,7 @@ async def main():
session_id, session_id,
"What is Redis?", "What is Redis?",
"Database context", "Database context",
"Redis is an in-memory data store.", "Redis is an in-memory data store."
ttl=3600,
) )
await adapter.add_qa( await adapter.add_qa(
@ -177,8 +174,7 @@ async def main():
session_id, session_id,
"Who created Redis?", "Who created Redis?",
"History context", "History context",
"Salvatore Sanfilippo (antirez).", "Salvatore Sanfilippo (antirez)."
ttl=3600,
) )
print(await adapter.get_all_qas(user_id, session_id)) print(await adapter.get_all_qas(user_id, session_id))