Linter agent (WIP)

Author: paolorechia

code_it/agents/linter.py

@@ -0,0 +1,56 @@
+from code_it.agents.base import BaseAgent
+
+
+class Linter(BaseAgent):
+    def __init__(self, llm) -> None:
+        super().__init__(llm)
+        self.start_string = "NewCode:"
+        self.stop_string = "ExistingSourceCode:"
+        self.prompt_template = """"You're AI tool that applies a linter to a Python Code and fixes the issues found. Think step-by-step.
+You should fulfill your role like in the example below:
+
+ExistingSourceCode:
+import os
+import os
+files = os.listdir()
+************* Module persistent_source
+persistent_source.py:3:0: C0304: Final newline missing (missing-final-newline)
+persistent_source.py:1:0: C0114: Missing module docstring (missing-module-docstring)
+persistent_source.py:2:0: W0404: Reimport 'os' (imported line 1) (reimported)
+
+------------------------------------------------------------------
+Your code has been rated at 0.00/10 (previous run: 0.00/10, +0.00)
+
+
+Programmer AI: I need to perform some modifications on the source code
+1. There is a misssing newline.
+2. Missing module docstring, I should add one
+3. There are duplicated imports, I should remove one
+NewCode:
+\"\"\"Module to list files in current directory \"\"\"
+import os
+files = os.listdir()
+
+ExistingSourceCode:
+
+Notice that you once you finish the subtask, you should add the word 'ExistingSourceCode:' in a new line,
+like in the example above.
+
+Now please help with the subtask below.
+
+ExistingSourceCode: {source_code}
+Lint Stdout: {stdout}
+Programmer AI:
+"""
+
+    def parse_output(self, result):
+        print("Linter result", result)
+        if self.stop_string not in result:
+            result = result.split(self.stop_string)[0]
+
+        if self.start_string not in result:
+            return None
+
+        result = result.split(self.start_string)[1]
+        print("Parsed linter", result)
+        return result

code_it/task_executor.py

@@ -89,11 +89,17 @@ def execute(self, task: str):
                 for _ in range(self.config.dependency_samples):
                     dep = self.dependency_tracker.execute_task(plan="\n".join(plan))
                     for d in dep:
-                        d = d.replace("-", "")
-                        if d in DEPENDENCY_BLACKLIST:
+                        d = d.replace("-", "").strip()
+                        if " " in d:
+                            d = d.split(" ")[0]
+
+                        if len(d) < 2 or d in DEPENDENCY_BLACKLIST:
                             continue
                         dependencies.append(d)
 
+                if not dependencies:
+                    break
+
                 dependencies = list(set(dependencies))
                 logger.info("Dependencies: %s", dependencies)
 

task.txt

@@ -1,3 +1 @@
-Your job is to plot an example chart using matplotlib. Create your own random data.
-Run this code only when you're finished.
-DO NOT add code and run into a single step.
+Your job is to generate cat jokes and save in a file called 'cat_jokes.txt'. Be creative!

task_chuck_norris.txt

@@ -0,0 +1,5 @@
+Question: The endpoint https://api.chucknorris.io/jokes/random returns a joke about Chuck Norries
+
+1. Write a python Program that fetches a joke from this endpoint.
+2. Extract the joke from the response. Access the 'value' in the JSON to extract it.
+3. Prints the joke to the screen.

task_plot.txt

@@ -0,0 +1,3 @@
+Your job is to plot an example chart using matplotlib. Create your own random data.
+Run this code only when you're finished.
+DO NOT add code and run into a single step.