dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-03 19:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac utils.split_chat_history_markdown

Browse source
This commit is contained in:
Paul Gauthier 2024-05-11 07:52:06 -07:00
parent 9fac20e432
commit 602a0c7c31
3 changed files with 30 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

29
aider/utils.py
View file

@ -105,3 +105,32 @@ def show_messages(messages, title=None, functions=None):
if functions:
dump(functions)
def split_chat_history_markdown(text):
messages = []
assistant = []
lines = text.splitlines(keepends=True)
for line in lines:
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)
return messages