Introduced new helper methods `reset` and `get_report_sections` in `MessageBuffer` for easier state management
This commit is contained in:
parent
718df34932
commit
9bdf0efa3e
17
cli/main.py
17
cli/main.py
|
|
@ -11,7 +11,6 @@ from rich.columns import Columns
|
|||
from rich.markdown import Markdown
|
||||
from rich.layout import Layout
|
||||
from rich.text import Text
|
||||
from rich.live import Live
|
||||
from rich.table import Table
|
||||
from collections import deque
|
||||
import time
|
||||
|
|
@ -166,6 +165,22 @@ class MessageBuffer:
|
|||
|
||||
self.final_report = "\n\n".join(report_parts) if report_parts else None
|
||||
|
||||
def reset(self):
|
||||
"""Clear all stored messages and reports and reset agent status."""
|
||||
self.messages.clear()
|
||||
self.tool_calls.clear()
|
||||
self.current_report = None
|
||||
self.final_report = None
|
||||
for key in self.report_sections:
|
||||
self.report_sections[key] = None
|
||||
for key in self.agent_status:
|
||||
self.agent_status[key] = "pending"
|
||||
self.current_agent = None
|
||||
|
||||
def get_report_sections(self):
|
||||
"""Return a copy of the current report sections."""
|
||||
return dict(self.report_sections)
|
||||
|
||||
|
||||
message_buffer = MessageBuffer()
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue