Spaces:
Sleeping
Sleeping
| """Throttled platform UI updates driven by transcript rendering.""" | |
| import time | |
| from collections.abc import Callable | |
| from loguru import logger | |
| from .platforms.ports import OutboundMessenger | |
| from .safe_diagnostics import format_exception_for_log | |
| from .transcript import RenderCtx, TranscriptBuffer | |
| class ThrottledTranscriptEditor: | |
| """Rate-limited status message edits from a growing transcript.""" | |
| def __init__( | |
| self, | |
| *, | |
| outbound: OutboundMessenger, | |
| parse_mode: str | None, | |
| get_limit_chars: Callable[[], int], | |
| transcript: TranscriptBuffer, | |
| render_ctx: RenderCtx, | |
| node_id: str, | |
| chat_id: str, | |
| status_msg_id: str, | |
| debug_platform_edits: bool, | |
| log_messaging_error_details: bool = False, | |
| ) -> None: | |
| self._outbound = outbound | |
| self._parse_mode = parse_mode | |
| self._get_limit_chars = get_limit_chars | |
| self._transcript = transcript | |
| self._render_ctx = render_ctx | |
| self._node_id = node_id | |
| self._chat_id = chat_id | |
| self._status_msg_id = status_msg_id | |
| self._debug_platform_edits = debug_platform_edits | |
| self._log_messaging_error_details = log_messaging_error_details | |
| self._last_ui_update = 0.0 | |
| self._last_displayed_text: str | None = None | |
| self._last_status: str | None = None | |
| def last_status(self) -> str | None: | |
| return self._last_status | |
| async def update(self, status: str | None = None, *, force: bool = False) -> None: | |
| """Render transcript + optional status line and edit the platform message.""" | |
| now = time.time() | |
| if not force and now - self._last_ui_update < 1.0: | |
| return | |
| self._last_ui_update = now | |
| if status is not None: | |
| self._last_status = status | |
| try: | |
| display = self._transcript.render( | |
| self._render_ctx, | |
| limit_chars=self._get_limit_chars(), | |
| status=status, | |
| ) | |
| except Exception as e: | |
| logger.warning( | |
| "Transcript render failed for node {}: {}", | |
| self._node_id, | |
| format_exception_for_log( | |
| e, log_full_message=self._log_messaging_error_details | |
| ), | |
| ) | |
| return | |
| if display and display != self._last_displayed_text: | |
| logger.debug( | |
| "PLATFORM_EDIT: node_id={} chat_id={} msg_id={} force={} status={!r} chars={}", | |
| self._node_id, | |
| self._chat_id, | |
| self._status_msg_id, | |
| bool(force), | |
| status, | |
| len(display), | |
| ) | |
| if self._debug_platform_edits: | |
| logger.debug("PLATFORM_EDIT_TEXT:\n{}", display) | |
| self._last_displayed_text = display | |
| try: | |
| await self._outbound.queue_edit_message( | |
| self._chat_id, | |
| self._status_msg_id, | |
| display, | |
| parse_mode=self._parse_mode, | |
| ) | |
| except Exception as e: | |
| logger.warning( | |
| "Failed to update platform for node {}: {}", | |
| self._node_id, | |
| format_exception_for_log( | |
| e, log_full_message=self._log_messaging_error_details | |
| ), | |
| ) | |