From c4b9dd2921fb06959b4e0d7a458ff23010ca50e9 Mon Sep 17 00:00:00 2001 From: Karl Hudgell Date: Sun, 20 Apr 2025 10:05:43 +0100 Subject: [PATCH] last 7 prompts not 7 days --- Update.sh | 0 lib.py | 12 +++++------- 2 files changed, 5 insertions(+), 7 deletions(-) create mode 100644 Update.sh diff --git a/Update.sh b/Update.sh new file mode 100644 index 0000000..e69de29 diff --git a/lib.py b/lib.py index 1c88b4b..0fea8df 100644 --- a/lib.py +++ b/lib.py @@ -17,7 +17,7 @@ from tenacity import ( ) import nest_asyncio import json -from datetime import datetime, timedelta +from datetime import datetime nest_asyncio.apply() @@ -26,17 +26,15 @@ logging.basicConfig(level=logging.INFO) LOG_FILE = "./prompts_log.jsonl" -def load_recent_prompts(days=7): +def load_recent_prompts(count=7): recent_prompts = [] - cutoff_date = datetime.now().date() - timedelta(days=days) try: with open(LOG_FILE, "r") as f: - for line in f: + lines = f.readlines() + for line in lines[-count:]: data = json.loads(line.strip()) - prompt_date = datetime.strptime(data["date"], "%Y-%m-%d").date() - if prompt_date >= cutoff_date: - recent_prompts.append(data["prompt"]) + recent_prompts.append(data["prompt"]) except FileNotFoundError: pass # No prompts yet