feat(trials): stop_trial and clear trials on reset

This commit is contained in:
2026-06-21 12:27:37 +02:00
parent 4ebad1bd6c
commit c53fdd8560
2 changed files with 24 additions and 0 deletions
+12
View File
@@ -469,6 +469,17 @@ class UsageTracker:
finally:
conn.close()
def stop_trial(self, package):
"""End a trial (package became permanent or was disabled)."""
with self._lock:
self._ensure_db()
conn = self._connect()
try:
conn.execute("DELETE FROM trial_packages WHERE package = ?", (package,))
conn.commit()
finally:
conn.close()
def reset(self):
"""Clear all tracked data."""
with self._lock:
@@ -478,6 +489,7 @@ class UsageTracker:
conn.execute("DELETE FROM node_usage")
conn.execute("DELETE FROM prompt_log")
conn.execute("DELETE FROM model_usage")
conn.execute("DELETE FROM trial_packages")
conn.commit()
finally:
conn.close()