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

refac

Browse source
This commit is contained in:
Paul Gauthier 2023-06-04 11:50:56 -07:00
parent 9d821dbca6
commit cdfc0a38ee
1 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aider/coder.py
View file

@ -14,11 +14,12 @@ from rich.console import Console
from rich.live import Live
from rich.markdown import Markdown
# from aider.dump import dump
from aider import prompts, utils
from aider.commands import Commands
from aider.repomap import RepoMap
from .dump import dump
class MissingAPIKeyError(ValueError):
pass
@ -468,7 +469,11 @@ class Coder:
if live:
live.stop()
def update_files(self, content):
def update_files_gpt35(self, content):
dump(content)
pass
def update_files_gpt4(self, content):
# might raise ValueError for malformed ORIG/UPD blocks
edits = list(utils.find_original_update_blocks(content))
@ -676,8 +681,15 @@ class Coder:
return set(self.get_all_relative_files()) - set(self.get_inchat_relative_files())
def apply_updates(self, content):
if self.main_model == "gpt-4":
method = self.update_files_gpt4
elif self.main_model == "gpt-3.5-turbo":
method = self.update_files_gpt35
else:
raise ValueError(f"apply_updates() doesn't support {self.main_model}")
try:
edited = self.update_files(content)
edited = method(content)
return edited, None
except ValueError as err:
err = err.args[0]