dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 02:05:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

switch to mdstream

Browse source
This commit is contained in:
Paul Gauthier 2024-01-23 09:20:31 -08:00
parent e50a0e8b09
commit b143bc56ac
2 changed files with 18 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

12
aider/mdstream.py
View file

@ -60,7 +60,7 @@ class MarkdownStream:
self.printed = []
self.when = 0
def update(self, text, final=False, min_delay=0.100):
def update(self, text, final=False, min_delay=0.100, mdargs=None):
now = time.time()
if not final and now - self.when < min_delay:
return
@ -69,7 +69,11 @@ class MarkdownStream:
string_io = io.StringIO()
console = Console(file=string_io, force_terminal=True)
markdown = Markdown(text)
if not mdargs:
mdargs = dict()
markdown = Markdown(text, **mdargs)
console.print(markdown)
output = string_io.getvalue()
@ -78,7 +82,7 @@ class MarkdownStream:
if not final:
num_lines -= 4
if num_lines <= 1:
if num_lines <= 1 and not final:
return
num_printed = len(self.printed)
@ -111,6 +115,6 @@ if __name__ == "__main__":
pm = MarkdownStream()
for i in range(6, len(_text)):
pm.update(_text[:i])
# time.sleep(0.001)
time.sleep(0.001)
pm.update(_text, final=True)