hermes-admin/hermes-agent
1
0
Fork 0
Code Issues Pull Requests Actions 28 Packages Projects Releases Wiki Activity

refactor(sessions): export guard always counts every row; export_all reuses _active_clause

Browse Source 
assert_export_safe grew an include_inactive flag whose only production caller
(the console export guard) always passes True, leaving the live-only branch
and its default unused. Drop the parameter and count every row, which is what
the transfer export materializes; the console call and docstring follow. The
existing guard tests seed live rows only and are unchanged.

export_all re-derived the live-row clause by hand; use the existing
_active_clause(include_inactive, False) so "live row" stays defined in one
place. Behaviour is unchanged.

Co-authored-by: joaomarcos <joaomarcosdias444@gmail.com>
This commit is contained in:
kshitijk4poor
2026-09-26 21:18:03 +05:30
committed by kshitij
parent deb7bc7343
commit 41303b379e
3 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
hermes_state_sessions.py
View File

@@ -1401,10 +1401,9 @@ class SessionSessionsMixin:
)
return statuses
def assert_export_safe(self, session_id: str, max_messages: Optional[int] = None,
include_inactive: bool = False) -> int:
"""Row count of this segment — live rows, or every row with ``include_inactive``, matching what
the export materializes — or raise SessionExportTooLargeError (the LIMITed subquery
def assert_export_safe(self, session_id: str, max_messages: Optional[int] = None) -> int:
"""Row count of this segment — every row, archived included, as the transfer export materializes
it — or raise SessionExportTooLargeError (the LIMITed subquery
stops once the bound is exceeded). ``None`` resolves ``sessions.max_export_messages``; 0 disables
the guard."""
from hermes_state import SessionExportTooLargeError, resolved_max_export_messages
@@ -1414,9 +1413,8 @@ class SessionSessionsMixin:
raise ValueError("max_messages must be non-negative")
if max_messages == 0:
return 0
active_clause = "" if include_inactive else " AND active = 1"
row = self._read_one(
f"SELECT COUNT(*) FROM (SELECT 1 FROM messages WHERE session_id = ?{active_clause} LIMIT ?)",
"SELECT COUNT(*) FROM (SELECT 1 FROM messages WHERE session_id = ? LIMIT ?)",
(session_id, max_messages + 1),
)
message_count = int(row[0] if row else 0)