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

refac

Browse source
This commit is contained in:
Paul Gauthier 2023-04-08 14:14:30 -07:00
parent 2fbf57e22c
commit 9d7643bacb
1 changed files with 22 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

32
coder.py
View file

@ -237,10 +237,28 @@ MAKE ANY CHANGES BASED OFF THESE FILES!
temperature=0,
stream = True,
)
resp = []
if show_progress:
pbar = tqdm(total = show_progress)
return self.show_send_progress(completion)
else:
return self.show_send_output(completion)
def show_send_progress(self, completion):
resp = []
pbar = tqdm(total = show_progress)
for chunk in completion:
try:
text = chunk.choices[0].delta.content
resp.append(text)
except AttributeError:
continue
pbar.update(len(text))
resp = ''.join(resp)
return resp
def show_send_output(self, completion):
resp = []
for chunk in completion:
try:
@ -248,15 +266,9 @@ MAKE ANY CHANGES BASED OFF THESE FILES!
resp.append(text)
except AttributeError:
continue
if show_progress:
pbar.update(len(text))
else:
sys.stdout.write(text)
sys.stdout.flush()
if show_progress:
pbar.update(show_progress)
pbar.close()
sys.stdout.write(text)
sys.stdout.flush()
resp = ''.join(resp)
return resp