jyapayne/mnemosyne
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: pass real incoming/outgoing bytes to record_turn for context reduction stats

Browse source
This commit is contained in:
Joey Yakimowich-Payne 2026-03-13 12:35:57 -06:00
commit 59cce5c6d2
1 changed files with 20 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

25
src/mnemosyne/gateway.py
View file

@ -1881,7 +1881,12 @@ def create_app(
exc_info=True, exc_info=True,
) )
def _display_turn_status(usage: dict, session: Session) -> None: def _display_turn_status(
usage: dict,
session: Session,
incoming_bytes: int = 0,
outgoing_bytes: int = 0,
) -> None:
"""Post-response status line with cache hit rate.""" """Post-response status line with cache hit rate."""
sid = session.id sid = session.id
ts = session.token_state ts = session.token_state
@ -1898,8 +1903,8 @@ def create_app(
effective_tokens=effective, effective_tokens=effective,
cache_read=cache_read, cache_read=cache_read,
cache_create=cache_create, cache_create=cache_create,
incoming_bytes=0, # filled by telemetry record_request incoming_bytes=incoming_bytes,
outgoing_bytes=0, outgoing_bytes=outgoing_bytes,
) )
cap_str = "" cap_str = ""
@ -2078,7 +2083,12 @@ def create_app(
# Post-response: track usage, display status # Post-response: track usage, display status
if usage: if usage:
_check_token_cap(usage, session) _check_token_cap(usage, session)
_display_turn_status(usage, session) _display_turn_status(
usage,
session,
incoming_bytes=incoming_bytes,
outgoing_bytes=outgoing_bytes,
)
_update_fidelity_pressure(usage, session) _update_fidelity_pressure(usage, session)
emit_event( emit_event(
"response_observed", "response_observed",
@ -2134,7 +2144,12 @@ def create_app(
if usage: if usage:
_check_token_cap(usage, session) _check_token_cap(usage, session)
_display_turn_status(usage, session) _display_turn_status(
usage,
session,
incoming_bytes=incoming_bytes,
outgoing_bytes=outgoing_bytes,
)
_update_fidelity_pressure(usage, session) _update_fidelity_pressure(usage, session)
emit_event( emit_event(