feat(ui): Expose history truncation controls in the Discussion panel
This commit is contained in:
15
gui.py
15
gui.py
@@ -62,6 +62,15 @@ def get_total_token_usage() -> dict:
|
||||
usage[k] += u.get(k, 0) or 0
|
||||
return usage
|
||||
|
||||
def truncate_entries(entries: list[dict], max_pairs: int) -> list[dict]:
|
||||
"""Truncates history to the last N pairs of User/AI messages."""
|
||||
if max_pairs <= 0:
|
||||
return []
|
||||
target_count = max_pairs * 2
|
||||
if len(entries) <= target_count:
|
||||
return entries
|
||||
return entries[-target_count:]
|
||||
|
||||
|
||||
# ------------------------------------------------------------------ comms rendering helpers
|
||||
|
||||
@@ -1344,6 +1353,11 @@ class App:
|
||||
self.disc_entries.clear()
|
||||
self._rebuild_disc_list()
|
||||
|
||||
def cb_disc_truncate(self):
|
||||
self.disc_entries = truncate_entries(self.disc_entries, 2)
|
||||
self._rebuild_disc_list()
|
||||
self._update_status("history truncated")
|
||||
|
||||
def cb_disc_collapse_all(self):
|
||||
for i, entry in enumerate(self.disc_entries):
|
||||
tag = f"disc_content_{i}"
|
||||
@@ -1761,6 +1775,7 @@ class App:
|
||||
dpg.add_button(label="+ Entry", callback=self.cb_disc_append_entry)
|
||||
dpg.add_button(label="-All", callback=self.cb_disc_collapse_all)
|
||||
dpg.add_button(label="+All", callback=self.cb_disc_expand_all)
|
||||
dpg.add_button(label="Truncate", callback=self.cb_disc_truncate)
|
||||
dpg.add_button(label="Clear All", callback=self.cb_disc_clear)
|
||||
dpg.add_button(label="Save", callback=self.cb_disc_save)
|
||||
dpg.add_checkbox(
|
||||
|
||||
22
tests/test_history_truncation.py
Normal file
22
tests/test_history_truncation.py
Normal file
@@ -0,0 +1,22 @@
|
||||
import pytest
|
||||
|
||||
def test_history_truncation():
|
||||
# A dummy test to fulfill the Red Phase for the history truncation controls.
|
||||
# The new function in gui.py should be cb_disc_truncate_history or a related utility.
|
||||
from project_manager import str_to_entry, entry_to_str
|
||||
|
||||
entries = [
|
||||
{"role": "User", "content": "1", "collapsed": False, "ts": "10:00:00"},
|
||||
{"role": "AI", "content": "2", "collapsed": False, "ts": "10:01:00"},
|
||||
{"role": "User", "content": "3", "collapsed": False, "ts": "10:02:00"},
|
||||
{"role": "AI", "content": "4", "collapsed": False, "ts": "10:03:00"}
|
||||
]
|
||||
|
||||
# We expect a new function truncate_entries(entries, max_pairs) to exist
|
||||
from gui import truncate_entries
|
||||
|
||||
truncated = truncate_entries(entries, max_pairs=1)
|
||||
# Keeping the last pair (user + ai)
|
||||
assert len(truncated) == 2
|
||||
assert truncated[0]["content"] == "3"
|
||||
assert truncated[1]["content"] == "4"
|
||||
Reference in New Issue
Block a user