Add debugging advice to AICodeBot

🤖 Add debugging advice to AICodeBot

Usage:

aicodebot debug $command_that_fails

Author: TechNickAI

README.md

@@ -4,7 +4,7 @@
 
 AICodeBot is a coding assistant designed to make your coding life easier. With capabilities to perform code reviews, manage dependencies, and even suggest improvements, think of it as your AI version of a pair programmer.
 
-⚠️ Status: This project is in its infancy with very limited features. Sometimes it does dumb things. 
+⚠️ Status: This project is in its infancy with very limited features. Sometimes it does dumb things.
 
 Right now, it only generates commit messages for you, but you can look at the Features list below to get an idea of where it is going. Give the project a star and follow along while we build out more of the foundation.
 
@@ -27,6 +27,7 @@ Options:
 Commands:
   alignment  Get a message about Heart-Centered AI Alignment ❤ + 🤖.
   commit     Generate a git commit message and commit changes after you...
+  debug      Run a command and get debugging advice if it fails.
   fun-fact   Tell me something interesting about programming or AI.
   version    Print the version number.
 ```
@@ -36,9 +37,9 @@ Commands:
 ### Code Workflow Improvements
 
 - [X] **Assisted Git Commit**: Automatically generate a commit message.
+- [X] **Assisted Debugging**: Run a command with aicodebot and it captures the log message and tries to figure out what's going on from the error message.  Eventually, it could also suggest fixes for the error and make the changes for you. Try it out with `aicodebot debug $command`
 - [ ] **Code Review**: Provides feedback on potential issues in code, such as style violations, potential bugs, and performance issues. It could also suggest best practices for code improvement. Eventually: FIX the code automatically and notify the team.
 - [ ] **Dependency Management**: Updating dependencies to their latest versions with pull requests that run tests.
-- [ ] **Assisted Debugging**: Run a command with aicodebot and it captures the log message and tries to figure out what's going on from the error message.  Eventually, it could also suggest fixes for the error and make the changes for you.
 - [ ] **Documentation Generation**: Generates comprehensive documentation for code, including docstrings, README files, and wiki pages.
 - [ ] **Performance Optimization Suggestions**: Suggests potential performance optimizations for code.
 - [ ] **Code Formatting**: Automatically formats code according to a specified style guide.

aicodebot/cli.py

@@ -13,6 +13,7 @@
 # Create a Console object
 console = Console()
 bot_style = Style(color="#30D5C8")
+DEFAULT_MAX_TOKENS = 1024
 
 
 def setup_environment():
@@ -69,7 +70,7 @@ def alignment(verbose):
     prompt = load_prompt(Path(__file__).parent / "prompts" / "alignment.yaml")
 
     # Set up the language model
-    llm = OpenAI(temperature=1, max_tokens=1024)
+    llm = OpenAI(temperature=1, max_tokens=DEFAULT_MAX_TOKENS)
 
     # Set up the chain
     chain = LLMChain(llm=llm, prompt=prompt, verbose=verbose)
@@ -142,6 +143,45 @@ def commit(verbose, max_tokens, yes):
     Path.unlink(temp_file_name)
 
 
+@cli.command(context_settings={"ignore_unknown_options": True})
+@click.argument("command", nargs=-1)
+@click.option("-v", "--verbose", count=True)
+def debug(command, verbose):
+    """Run a command and get debugging advice if it fails."""
+    setup_environment()
+
+    # Load the prompt
+    prompt = load_prompt(Path(__file__).parent / "prompts" / "debug.yaml")
+
+    # Set up the language model
+    llm = OpenAI(temperature=0.1, max_tokens=DEFAULT_MAX_TOKENS)
+
+    # Set up the chain
+    chat_chain = LLMChain(llm=llm, prompt=prompt, verbose=verbose)
+    command_str = " ".join(command)
+
+    # Run the command and capture its output
+    console.print(f"Running:\n{command_str}")
+    process = subprocess.run(command_str, shell=True, capture_output=True, text=True)  # noqa: S602
+
+    # Print the output of the command
+    output = f"Standard Output:\n{process.stdout}\nStandard Error:\n{process.stderr}"
+    console.print(f"Output:\n{output}")
+
+    # Print a message about the exit status
+    if process.returncode == 0:
+        console.print("The command completed successfully.")
+    else:
+        console.print(f"The command exited with status {process.returncode}.")
+
+    # If the command failed, send its output to ChatGPT for analysis
+    if process.returncode != 0:
+        error_output = process.stderr
+        with console.status("Thinking", spinner="point"):
+            response = chat_chain.run(error_output)
+            console.print(response, style=bot_style)
+
+
 @cli.command()
 @click.option("-v", "--verbose", count=True)
 def fun_fact(verbose):
@@ -152,7 +192,7 @@ def fun_fact(verbose):
     prompt = load_prompt(Path(__file__).parent / "prompts" / "fun_fact.yaml")
 
     # Set up the language model
-    llm = ChatOpenAI(temperature=1, max_tokens=1024)
+    llm = ChatOpenAI(temperature=1, max_tokens=DEFAULT_MAX_TOKENS)
 
     # Set up the chain
     chat_chain = LLMChain(llm=llm, prompt=prompt, verbose=verbose)

aicodebot/prompts/debug.yaml

@@ -0,0 +1,13 @@
+_type: prompt
+template_format: f-string
+input_variables: ["command_output"]
+template: |
+    I ran a command my terminal, and it failed.
+
+    Here's the output:
+
+    BEGIN OUTPUT
+    {command_output}
+    END OUTPUT
+
+    Help me understand what happened and how might I be able to fix it.