fix(agent): add missing newlines and improve formatting

Author: lzwjava

scripts/agent/add_prompt.py

@@ -38,4 +38,4 @@ def save_clipboard_prompt() -> None:
 
 
 if __name__ == "__main__":
-    save_clipboard_prompt()
+    save_clipboard_prompt()

scripts/agent/agent.py

@@ -1,34 +1,41 @@
 import pyperclip
 import os
 
+
 def save_to_file(content, filename="answer.md"):
     with open(filename, "a") as f:
         f.write(content + "\n\n")
 
+
 def main():
     print("Hello! I'm GitHub Copilot. What can I help you with?")
-    
+
     while True:
         user_input = input("You: ")
         if user_input.lower() in ["exit", "quit"]:
             print("Goodbye!")
             break
-            
+
         # Copy user input to clipboard
         pyperclip.copy(user_input)
-        print("I've copied your input to the clipboard. Please ask the chatbot and copy their answer. When ready, just press Enter.")
-        
+        print(
+            "I've copied your input to the clipboard. Please ask the chatbot and copy their answer. When ready, just press Enter."
+        )
+
         # Wait for user to press Enter after copying the chatbot's answer
         input("Press Enter when you have the answer copied...")
-        
+
         # Get the answer from clipboard
         answer = pyperclip.paste()
         print("Answer received. Saving to answer.md...")
-        
+
         # Save the interaction to file
-        interaction = f"**User Input:**\n{user_input}\n\n**Chatbot Answer:**\n{answer}\n{'-'*50}"
+        interaction = (
+            f"**User Input:**\n{user_input}\n\n**Chatbot Answer:**\n{answer}\n{'-'*50}"
+        )
         save_to_file(interaction)
         print("Saved to answer.md. Anything else I can help with?")
 
+
 if __name__ == "__main__":
-    main()
+    main()

scripts/agent/fix_agent.py

@@ -4,7 +4,7 @@
 import os
 import subprocess
 
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')))
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
 
 from scripts.translation.openrouter_client import call_openrouter_api
 from scripts.translation.openrouter_client import MODEL_MAPPING
@@ -13,63 +13,84 @@
 def run_script(script_path):
     """Run a Python or Rust script and capture its output."""
     script_path = Path(script_path)
-    if script_path.suffix == '.py':
+    if script_path.suffix == ".py":
         try:
-            result = subprocess.run(['python3', str(script_path)], capture_output=True, text=True)
+            result = subprocess.run(
+                ["python3", str(script_path)], capture_output=True, text=True
+            )
             return result.stdout + result.stderr
         except Exception as e:
             return str(e)
-    elif script_path.suffix == '.rs':
+    elif script_path.suffix == ".rs":
         # For Rust, we need to run cargo build in the directory containing the Rust project
         project_dir = script_path.parent
         try:
-            result = subprocess.run(['cargo', 'build'], cwd=project_dir, capture_output=True, text=True)
+            result = subprocess.run(
+                ["cargo", "build"], cwd=project_dir, capture_output=True, text=True
+            )
             return result.stdout + result.stderr
         except Exception as e:
             return str(e)
     else:
         return f"Unsupported file extension for {script_path}"
 
+
 def fix_script(script_path, error_message, model):
     """Use OpenRouter API to suggest fixes for the script errors and apply them."""
     script_path = Path(script_path)
-    lang = 'Python' if script_path.suffix == '.py' else 'Rust' if script_path.suffix == '.rs' else 'Unknown'
+    lang = (
+        "Python"
+        if script_path.suffix == ".py"
+        else "Rust" if script_path.suffix == ".rs" else "Unknown"
+    )
     prompt = f"Error in {lang} script {script_path}:\n{error_message}\nPlease provide only the corrected code for the entire script without any explanations, markdown syntax, or additional text."
     response = call_openrouter_api(prompt, model=model)
     # Remove any markdown code block syntax from the response
-    response = response.replace("```python", "").replace("```rust", "").replace("```", "")
+    response = (
+        response.replace("```python", "").replace("```rust", "").replace("```", "")
+    )
     # Write the fixed code back to the original script file
     try:
-        with open(script_path, 'w') as f:
+        with open(script_path, "w") as f:
             f.write(response.strip())
         return f"Applied fix to {script_path}"
     except Exception as e:
         return f"Failed to apply fix: {str(e)}"
 
+
 # Model mapping for user-friendly names
 
+
 def main():
     parser = argparse.ArgumentParser(description="Run and fix Python or Rust scripts.")
-    parser.add_argument('script', type=str, help="Path to the script to run and fix (Python .py or Rust .rs).")
-    parser.add_argument('--model', type=str, default='deepseek-v3',
-                        choices=list(MODEL_MAPPING.keys()),
-                        help="Model to use for OpenRouter API (e.g., claude-opus, gemini-pro).")
+    parser.add_argument(
+        "script",
+        type=str,
+        help="Path to the script to run and fix (Python .py or Rust .rs).",
+    )
+    parser.add_argument(
+        "--model",
+        type=str,
+        default="deepseek-v3",
+        choices=list(MODEL_MAPPING.keys()),
+        help="Model to use for OpenRouter API (e.g., claude-opus, gemini-pro).",
+    )
     args = parser.parse_args()
-    
+
     script_path = Path(args.script)
     if not script_path.exists():
         print(f"Script {script_path} does not exist.")
         sys.exit(1)
-    
+
     # Get the actual model name from the mapping
     selected_model = MODEL_MAPPING[args.model]
-    
+
     # Run the script initially to capture output
     print("Running script for the first time...")
     output = run_script(script_path)
     print("Initial Script Output:")
     print(output)
-    
+
     if "error" in output.lower() or "traceback" in output.lower():
         print("Error detected, attempting to fix...")
         fix_result = fix_script(script_path, output, selected_model)
@@ -80,11 +101,12 @@ def main():
         output = run_script(script_path)
         print("Updated Script Output:")
         print(output)
-    
+
     if "error" in output.lower() or "traceback" in output.lower():
         print("Error still present after fix attempt.")
         print("Final Script Output:")
         print(output)
 
+
 if __name__ == "__main__":
     main()

scripts/agent/format_agent.py

@@ -4,7 +4,7 @@
 import os
 import subprocess
 
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')))
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
 
 from scripts.translation.openrouter_client import call_openrouter_api
 from scripts.translation.openrouter_client import MODEL_MAPPING
@@ -13,15 +13,21 @@
 def format_script(script_path):
     """Format a Python or Rust script using standard formatting tools."""
     script_path = Path(script_path)
-    if script_path.suffix == '.py':
+    if script_path.suffix == ".py":
         try:
-            result = subprocess.run(['black', str(script_path)], capture_output=True, text=True)
+            result = subprocess.run(
+                ["black", str(script_path)], capture_output=True, text=True
+            )
             return result.stdout + result.stderr
         except Exception as e:
             return str(e)
-    elif script_path.suffix == '.rs':
+    elif script_path.suffix == ".rs":
         try:
-            result = subprocess.run(['rustfmt', str(script_path), '--edition', '2021'], capture_output=True, text=True)
+            result = subprocess.run(
+                ["rustfmt", str(script_path), "--edition", "2021"],
+                capture_output=True,
+                text=True,
+            )
             return result.stdout + result.stderr
         except Exception as e:
             return str(e)
@@ -32,40 +38,59 @@ def format_script(script_path):
 def ai_format_script(script_path, model):
     """Use OpenRouter API to suggest formatting improvements for the script."""
     script_path = Path(script_path)
-    lang = 'Python' if script_path.suffix == '.py' else 'Rust' if script_path.suffix == '.rs' else 'Unknown'
-    with open(script_path, 'r') as f:
+    lang = (
+        "Python"
+        if script_path.suffix == ".py"
+        else "Rust" if script_path.suffix == ".rs" else "Unknown"
+    )
+    with open(script_path, "r") as f:
         code_content = f.read()
     prompt = f"Please format the following {lang} code for better readability and consistency. Provide only the formatted code without explanations or markdown syntax:\n\n{code_content}"
     response = call_openrouter_api(prompt, model=model)
     # Remove any markdown code block syntax from the response
-    response = response.replace("```python", "").replace("```rust", "").replace("```", "")
+    response = (
+        response.replace("```python", "").replace("```rust", "").replace("```", "")
+    )
     # Write the AI-formatted code back to the original script file
     try:
-        with open(script_path, 'w') as f:
+        with open(script_path, "w") as f:
             f.write(response.strip())
         return f"Applied AI formatting to {script_path}"
     except Exception as e:
         return f"Failed to apply AI formatting: {str(e)}"
 
 
 def main():
-    parser = argparse.ArgumentParser(description="Format Python or Rust scripts using standard tools and AI.")
-    parser.add_argument('script', type=str, help="Path to the script to format (Python .py or Rust .rs).")
-    parser.add_argument('--model', type=str, default='deepseek-v3',
-                        choices=list(MODEL_MAPPING.keys()),
-                        help="Model to use for OpenRouter API (e.g., claude-opus, gemini-pro).")
-    parser.add_argument('--ai-only', action='store_true',
-                        help="Use only AI formatting instead of standard tools like black or rustfmt.")
+    parser = argparse.ArgumentParser(
+        description="Format Python or Rust scripts using standard tools and AI."
+    )
+    parser.add_argument(
+        "script",
+        type=str,
+        help="Path to the script to format (Python .py or Rust .rs).",
+    )
+    parser.add_argument(
+        "--model",
+        type=str,
+        default="deepseek-v3",
+        choices=list(MODEL_MAPPING.keys()),
+        help="Model to use for OpenRouter API (e.g., claude-opus, gemini-pro).",
+    )
+    parser.add_argument(
+        "--ai-only",
+        action="store_true",
+        help="Use only AI formatting instead of standard tools like black or rustfmt.",
+    )
     args = parser.parse_args()
-    
+
     script_path = Path(args.script)
     if not script_path.exists():
         print(f"Script {script_path} does not exist.")
         sys.exit(1)
-    
+
     # Get the actual model name from the mapping
     selected_model = MODEL_MAPPING[args.model]
-    
+
     if args.ai_only:
         print("Applying AI-based formatting...")
         ai_result = ai_format_script(script_path, selected_model)
@@ -76,7 +101,7 @@ def main():
         format_result = format_script(script_path)
         print("Standard Formatting Output:")
         print(format_result)
-        
+
         if "error" in format_result.lower():
             print("Standard formatting failed, falling back to AI formatting...")
             ai_result = ai_format_script(script_path, selected_model)

scripts/agent/grammar_agent.py

@@ -3,7 +3,7 @@
 from pathlib import Path
 import os
 
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))) 
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
 
 from scripts.translation.openrouter_client import call_openrouter_api
 
@@ -36,48 +36,54 @@ def fix_grammar_with_ai(content):
         print(f"Error calling AI API: {e}", file=sys.stderr)
         return None
 
+
 def process_file(file_path, output_only=False):
     """Process a single Jekyll post file."""
     try:
-        with open(file_path, 'r', encoding='utf-8') as f:
+        with open(file_path, "r", encoding="utf-8") as f:
             content = f.read()
-        
+
         fixed_content = fix_grammar_with_ai(content)
-        
+
         if fixed_content is None:
             return None
-            
+
         if output_only:
             print(fixed_content)
         else:
             print(f"Grammar-fixed content for {file_path}:")
             print(fixed_content)
             print()
-        
+
         return fixed_content
-        
+
     except Exception as e:
         print(f"Error processing {file_path}: {e}", file=sys.stderr)
         return None
 
+
 def main():
-    parser = argparse.ArgumentParser(description='Fix grammar in Jekyll posts using AI')
-    parser.add_argument('files', nargs='*', help='Markdown files to process')
-    parser.add_argument('--output-only', action='store_true', 
-                       help='Output only fixed content without file info')
-    
+    parser = argparse.ArgumentParser(description="Fix grammar in Jekyll posts using AI")
+    parser.add_argument("files", nargs="*", help="Markdown files to process")
+    parser.add_argument(
+        "--output-only",
+        action="store_true",
+        help="Output only fixed content without file info",
+    )
+
     args = parser.parse_args()
-    
+
     if not args.files:
         # If no files specified, look for markdown files in current directory
-        md_files = list(Path('.').glob('*.md'))
+        md_files = list(Path(".").glob("*.md"))
         if not md_files:
             print("No markdown files found. Please specify files to process.")
             sys.exit(1)
         args.files = md_files
-    
+
     for file_path in args.files:
         process_file(file_path, args.output_only)
 
+
 if __name__ == "__main__":
     main()