add docs command

Author: morganpartee

codegpt/gpt_interface.py

@@ -52,9 +52,7 @@ def send_iffy_edit(prompt: str, code: Dict[str, str], clipboard: bool = False, y
         > <the code to be output line 1>
         > <the code to be output, line n...>
         
-        You must include an explanation of what you did, and the code to be output, regardless of the format or file.
-        
-        OUTPUT:""")
+        You must include an explanation of what you did, and the code to be output, regardless of the format or file.""")
 
     else:
         full_prompt += dedent("""
@@ -71,9 +69,7 @@ def send_iffy_edit(prompt: str, code: Dict[str, str], clipboard: bool = False, y
         > <code line 1>
         > <code line n...>
         
-        You must include the filename, an explanation of what you did, and the code for the file to be output, regardless of the format or file.
-        
-        OUTPUT:""")
+        You must include the filename, an explanation of what you did, and the code for the file to be output, regardless of the format or file.""")
 
     max_tokens = confirm_send(full_prompt, yes=yes, silent=clipboard)
 

codegpt/main.py

@@ -88,7 +88,7 @@ def edit_file(
         return
 
     if raw_out:
-        print(result['code'])
+        print(result.get('code') or result)
         return
 
     files.write_text(result, backup)

codegpt/parse.py

@@ -17,9 +17,10 @@ def parse_resp(response:dict):
         # If the line doesn't start with '>', it's a key
         if line[0] != '>':
             # Strip leading/trailing whitespace and remove ':' from the key
-            key = line.strip().replace(":", '')
+            key = line.strip().replace(":", '').lower()
             # Initialize an empty value for this key in the current dictionary
             curr_dict[key] = ""
+        else:
             curr_dict[key] += line.strip().strip('> ').strip('>') + '\n'
 
     # Add the final dictionary to the output list

tests/test_edit_file.py

@@ -12,7 +12,7 @@ def test_edit_file_add_comment():
 
     # Call the function being tested
     result = run.invoke(app, ['do', instruction, "--yes", "--raw-out", "--raw-code", f"'{raw_code}'"])
-
+    
     # Use a regex to match the output of the function
     assert "#" in result.stdout
     assert "def foo():" in result.stdout
@@ -34,6 +34,8 @@ def test_varnames():
     # Call the function being tested
     result = run.invoke(app, ['do', instruction, "--yes", "--raw-out", "--raw-code", f"'{raw_code}'"])
 
+    print(result.stdout)
+    
     assert 'v0 = 0.2' in result.stdout
     assert 'a = 2' not in result.stdout
     assert 'n = 21' not in result.stdout