From 41de37ee44dc824eb5a0bd537ae106935e0eab3b Mon Sep 17 00:00:00 2001 From: askdevai-bot Date: Sun, 1 Jun 2025 23:08:23 -0400 Subject: [PATCH] Fix UI callbacks: suppress final answer panel in RichUICallback and CodeUICallback --- tinyagent/hooks/rich_ui_callback.py | 24 +----------------------- 1 file changed, 1 insertion(+), 23 deletions(-) diff --git a/tinyagent/hooks/rich_ui_callback.py b/tinyagent/hooks/rich_ui_callback.py index b499fb4..5820356 100644 --- a/tinyagent/hooks/rich_ui_callback.py +++ b/tinyagent/hooks/rich_ui_callback.py @@ -464,29 +464,7 @@ def _update_display(self) -> None: ) ordered_panels.append(token_panel) - # 8. Final response panel (if we have a response) - if self.response_content: - content = self.response_content - if self.markdown: - self.logger.debug("Converting response to markdown") - escaped_content = escape_markdown_tags(content, self.tags_to_include_in_markdown) - content = Markdown(escaped_content) - - response_panel = create_panel( - content=content, - title=f"Response ({self.timer.elapsed:.1f}s)", - border_style="blue" - ) - ordered_panels.append(response_panel) - - # Add token count panel for final response - response_token_count = self.count_tokens(self.response_content) - token_panel = create_panel( - content=Text(f"Token count: {response_token_count}", style="bold"), - title="Tokens", - border_style="purple" - ) - ordered_panels.append(token_panel) + # 8. Final response panel skipped: final answer display suppressed try: self.logger.debug(f"Updating live display with {len(ordered_panels)} panels")