From 3345f37bb30ef014887696d2198789a83c640eda Mon Sep 17 00:00:00 2001 From: Zvonimir Sabljic Date: Thu, 3 Aug 2023 16:21:34 +0200 Subject: [PATCH] Smaller fixes --- euclid/helpers/Project.py | 2 +- euclid/helpers/agents/CodeMonkey.py | 5 +++-- .../development/task/step/write_automated_test.prompt | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/euclid/helpers/Project.py b/euclid/helpers/Project.py index 3b3d68a..091377e 100644 --- a/euclid/helpers/Project.py +++ b/euclid/helpers/Project.py @@ -100,7 +100,7 @@ class Project: f.write(file_snapshot.content) def ask_for_human_verification(self, message, description): - print(colored(message, "orange")) + print(colored(message, "yellow")) print(description) answer = '' while answer != 'continue': diff --git a/euclid/helpers/agents/CodeMonkey.py b/euclid/helpers/agents/CodeMonkey.py index dd8ad2e..3d421e4 100644 --- a/euclid/helpers/agents/CodeMonkey.py +++ b/euclid/helpers/agents/CodeMonkey.py @@ -10,8 +10,9 @@ class CodeMonkey(Agent): super().__init__('code_monkey', project) self.developer = developer - def implement_code_changes(self, code_changes_description): - convo = AgentConvo(self) + def implement_code_changes(self, code_changes_description, convo=None, cb=None): + if convo == None: + convo = AgentConvo(self) steps = convo.send_message('development/task/break_down_code_changes.prompt', { "instructions": code_changes_description, "directory_tree": self.project.get_directory_tree(), diff --git a/euclid/prompts/development/task/step/write_automated_test.prompt b/euclid/prompts/development/task/step/write_automated_test.prompt index e895396..fbf13fd 100644 --- a/euclid/prompts/development/task/step/write_automated_test.prompt +++ b/euclid/prompts/development/task/step/write_automated_test.prompt @@ -1,4 +1,4 @@ -{{ if len(files) > 0}} +{{ if files|length > 0}} Here are the requested files: {% for file in files %} **{{ file.name }}**