mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2026-01-02 01:16:09 +01:00
44 lines
1.9 KiB
Python
44 lines
1.9 KiB
Python
from const.function_calls import GET_FILES, DEV_STEPS, IMPLEMENT_CHANGES, CODE_CHANGES
|
|
from database.models.files import File
|
|
from helpers.files import update_file
|
|
from helpers.AgentConvo import AgentConvo
|
|
from helpers.Agent import Agent
|
|
|
|
class CodeMonkey(Agent):
|
|
def __init__(self, project, developer):
|
|
super().__init__('code_monkey', project)
|
|
self.developer = developer
|
|
|
|
def implement_code_changes(self, convo, code_changes_description, step_index=0):
|
|
if convo == None:
|
|
convo = AgentConvo(self)
|
|
|
|
files_needed = convo.send_message('development/task/request_files_for_code_changes.prompt', {
|
|
"step_description": code_changes_description,
|
|
"directory_tree": self.project.get_directory_tree(True),
|
|
"step_index": step_index,
|
|
"finished_steps": ', '.join(f"#{j}" for j in range(step_index))
|
|
}, GET_FILES)
|
|
|
|
|
|
changes = convo.send_message('development/implement_changes.prompt', {
|
|
"instructions": code_changes_description,
|
|
"directory_tree": self.project.get_directory_tree(),
|
|
"files": self.project.get_files(files_needed),
|
|
}, IMPLEMENT_CHANGES, True)
|
|
|
|
for file_data in changes:
|
|
file_data['full_path'] = self.project.get_full_file_path(file_data['path'], file_data['name'])
|
|
|
|
if file_data['description'] != '':
|
|
(File.insert(app=self.project.app, path=file_data['path'], name=file_data['name'], description=file_data['description'])
|
|
.on_conflict(
|
|
conflict_target=[File.app, File.name, File.path],
|
|
preserve=[],
|
|
update={'description': file_data['description']})
|
|
.execute())
|
|
|
|
update_file(file_data['full_path'], file_data['content'])
|
|
|
|
return convo
|