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

refactored summarize_chat_history_markdown

Browse source
This commit is contained in:
Paul Gauthier 2024-05-05 18:58:16 -07:00
parent 3bb237bdc1
commit cd84cc0d3e
1 changed files with 33 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

30
aider/history.py
View file

@ -1,6 +1,6 @@
import argparse
from aider import prompts
from aider import models, prompts
from aider.dump import dump # noqa: F401
from aider.sendchat import simple_send_with_retries
@ -90,15 +90,7 @@ class ChatSummary:
return [dict(role="user", content=summary)]
def main():
parser = argparse.ArgumentParser()
parser.add_argument("filename", help="Markdown file to parse")
args = parser.parse_args()
with open(args.filename, "r") as f:
text = f.read()
def summarize_chat_history_markdown(self, text):
messages = []
assistant = []
for line in text.splitlines(keepends=True):
@ -123,8 +115,22 @@ def main():
assistant.append(line)
summarizer = ChatSummary("gpt-3.5-turbo", weak_model=False)
summary = summarizer.summarize(messages[-40:])
summary = self.summarize(messages[-40:])
return summary
def main():
parser = argparse.ArgumentParser()
parser.add_argument("filename", help="Markdown file to parse")
args = parser.parse_args()
model = models.Model("gpt-3.5-turbo")
summarizer = ChatSummary(model)
with open(args.filename, "r") as f:
text = f.read()
summary = summarizer.summarize_chat_history_markdown(text)
dump(summary)