dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 09:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

aider: Add tqdm progress bar to summarize_chat_history_markdown loop.

Browse source 
# Aider chat conversation:

USER: add a tqdm to `summarize_chat_history_markdown` loop

ASSISTANT: ```diff
--- aider/history.py
+++ aider/history.py
@@ ... @@
 from aider.sendchat import simple_send_with_retries
+from tqdm import tqdm

 class ChatSummary:
     def __init__(self, model=None, max_tokens=1024):
@@ ... @@
     def summarize_chat_history_markdown(self, text):
         messages = []
         assistant = []
+        lines = text.splitlines(keepends=True)
+        for line in tqdm(lines, desc="Summarizing chat history"):
-        for line in text.splitlines(keepends=True):
             if line.startswith("# "):
                 continue
             if line.startswith(">"):
                 continue
             if line.startswith("#### /"):
                 continue

             if line.startswith("#### "):
                 if assistant:
                     assistant = "".join(assistant)
                     if assistant.strip():
                         messages.append(dict(role="assistant", content=assistant))
                     assistant = []

                 content = line[5:]
                 if content.strip() and content.strip() != "<blank>":
                     messages.append(dict(role="user", content=line[5:]))
                 continue

             assistant.append(line)

         summary = self.summarize(messages[-40:])
         return summary
```

This diff adds the `tqdm` progress bar to the `summarize_chat_history_markdown` method in the `aider/history.py` file. It wraps the loop that processes the chat history lines with `tqdm`, providing a description for the progress bar.
This commit is contained in:
Paul Gauthier 2024-05-06 09:31:32 -07:00
parent 64470955d4
commit e51e0219ee
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
aider/history.py
View file

@ -4,6 +4,7 @@ from aider import models, prompts
from aider.dump import dump # noqa: F401 from aider.dump import dump # noqa: F401
from aider.sendchat import simple_send_with_retries from aider.sendchat import simple_send_with_retries
from tqdm import tqdm
class ChatSummary: class ChatSummary:
def __init__(self, model=None, max_tokens=1024): def __init__(self, model=None, max_tokens=1024):
@ -93,7 +94,8 @@ class ChatSummary:
def summarize_chat_history_markdown(self, text): def summarize_chat_history_markdown(self, text):
messages = [] messages = []
assistant = [] assistant = []
for line in text.splitlines(keepends=True): lines = text.splitlines(keepends=True)
for line in tqdm(lines, desc="Summarizing chat history"):
if line.startswith("# "): if line.startswith("# "):
continue continue
if line.startswith(">"): if line.startswith(">"):