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

WIP: Refactor send method and remove show_send_progress method in Coder class.

Browse source
This commit is contained in:
Paul Gauthier 2023-05-09 00:05:48 -07:00
parent 5c2cbd301e
commit 1a306d89d9
1 changed files with 2 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

21
coder.py
View file

@ -280,7 +280,7 @@ class Coder:
for line in content: for line in content:
print(role, line) print(role, line)
def send(self, messages, model=None, progress_bar_expected=0, silent=False): def send(self, messages, model=None, silent=False):
# self.show_messages(messages, "all") # self.show_messages(messages, "all")
if not model: if not model:
@ -305,9 +305,7 @@ class Coder:
interrupted = False interrupted = False
try: try:
if progress_bar_expected: if self.pretty and not silent:
self.show_send_progress(completion, progress_bar_expected)
elif self.pretty and not silent:
self.show_send_output_color(completion) self.show_send_output_color(completion)
else: else:
self.show_send_output_plain(completion, silent) self.show_send_output_plain(completion, silent)
@ -316,21 +314,6 @@ class Coder:
return self.resp, interrupted return self.resp, interrupted
def show_send_progress(self, completion, progress_bar_expected):
self.resp = ""
pbar = tqdm(total=progress_bar_expected)
for chunk in completion:
try:
text = chunk.choices[0].delta.content
self.resp += text
except AttributeError:
continue
pbar.update(len(text))
pbar.update(progress_bar_expected)
pbar.close()
def show_send_output_plain(self, completion, silent): def show_send_output_plain(self, completion, silent):
self.resp = "" self.resp = ""