Merge pull request #870 from ATheorell/refactor

Refactor
AntonOsika · Nov 26, 2023 · ab5e3e3 · ab5e3e3
2 parents 54b6dd9 + b665604
commit ab5e3e3
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/gpt_engineer/__init__.py b/gpt_engineer/__init__.py
@@ -3,5 +3,6 @@
     ai,
     chat_to_files,
 )
-from gpt_engineer.tools import code_vector_repository
-from gpt_engineer.core.default import on_disk_repository
+
+# from gpt_engineer.tools import code_vector_repository
+# from gpt_engineer.core.default import on_disk_repository
diff --git a/gpt_engineer/applications/cli/collect.py b/gpt_engineer/applications/cli/collect.py
@@ -162,4 +162,4 @@ def collect_and_send_human_review(
     """Collects and stores human review of the code"""
     review = human_review_input()
     collect_learnings(prompt, model, temperature, config, memory, review)
-    return []
+    # return []
diff --git a/gpt_engineer/applications/cli/main.py b/gpt_engineer/applications/cli/main.py
@@ -178,7 +178,7 @@ def main(
     else:
         execution_fn = execute_entrypoint
 
-    if --vector_improve_mode:
+    if vector_improve_mode:
         improve_fn = vector_improve
     else:
         improve_fn = improve
@@ -198,7 +198,7 @@ def main(
         preprompts_holder=preprompts_holder,
     )
     if improve_mode:
-        if --vector_improve_mode:
+        if vector_improve_mode:
             code = get_all_code(project_path)
         else:
             code = ask_for_files(project_path)

diff --git a/gpt_engineer/core/default/lean_agent.py b/gpt_engineer/core/default/lean_agent.py
@@ -56,7 +56,7 @@ def init(self, prompt: str) -> Code:
         code = gen_code(self.ai, prompt, self.memory, self.preprompts_holder)
         entrypoint = gen_entrypoint(self.ai, code, self.memory, self.preprompts_holder)
         code = Code(code | entrypoint)
-        code = self.execution_env.execute_program(code)
+        self.execution_env.execute_program(code)
         return code
 
     def improve(
@@ -68,5 +68,5 @@ def improve(
                 self.ai, code, self.memory, self.preprompts_holder
             )
             code = Code(code | entrypoint)
-        code = self.execution_env.execute_program(code)
+        self.execution_env.execute_program(code)
         return code