title: "Source Code Comment Generator",
description: `Add comments to source code to make it more understandable for AI systems or human developers.`,
description: "Format source code command",
description: "Build command",
const { format, build } = env.vars
// Get files from environment or modified files from Git if none provided
files = await git.listFiles("staged", { askStageOnEmpty: true })
if (!files.length) files = await git.listFiles("modified-base")
// custom filter to only process code files
/\.(py|m?ts|m?js|cs|java|c|cpp|h|hpp)$/.test(filename) && // known languages only
!/\.test/.test(filename) // ignore test files
files = files.sort(() => Math.random() - 0.5)
console.log(YAML.stringify(files.map((f) => f.filename)))
// Process each file separately to avoid context explosion
const jobs = host.promiseQueue(5)
await jobs.mapAll(files, processFile)
async function processFile(file: WorkspaceFile) {
console.log(`processing ${file.filename}`)
if (!file.content) console.log(`empty file, continue`)
const newContent = await addComments(file)
// Save modified content if different
if (newContent && file.content !== newContent) {
console.log(`updating ${file.filename}`)
await workspace.writeText(file.filename, newContent)
const formatRes = await host.exec(`${format} ${file.filename}`)
if (formatRes.exitCode !== 0) {
const buildRes = await host.exec(`${build} ${file.filename}`)
if (buildRes.exitCode !== 0) {
// last LLM as judge check
if (!revert) revert = await checkModifications(file.filename)
console.error(`reverting ${file.filename}...`)
await workspace.writeText(file.filename, file.content)
console.error(`error: ${e}`)
// Function to add comments to code
async function addComments(file: WorkspaceFile): Promise<string | undefined> {
let { filename, content } = file
if (parsers.tokens(file) > 20000) return undefined // too big
const res = await runPrompt(
// Define code snippet for AI context with line numbers
// AI prompt to add comments for better understanding
ctx.def("FILE", code, { detectPromptInjection: "available" })
ctx.$`You are an expert developer at all programming languages.
You are tasked with adding comments to code in FILE to make it more understandable for AI systems or human developers.
You should analyze it, and add/update appropriate comments as needed.
To add or update comments to this code, follow these steps:
1. Analyze the code to understand its structure and functionality.
- If you are not familiar with the programming language, emit an empty file.
- If there is no code, emit an empty file.
2. Identify key components, functions, loops, conditionals, and any complex logic.
3. Add comments that explain:
- The purpose of functions or code blocks using the best comment format for that programming language.
- How complex algorithms or logic work
- Any assumptions or limitations in the code
- The meaning of important variables or data structures
- Any potential edge cases or error handling
- All function arguments and return value
- A Top level file comment that describes the code in the file
When adding or updating comments, follow these guidelines:
- Use clear and concise language
- Avoid stating the obvious (e.g., don't just restate what the code does)
- Focus on the "why" and "how" rather than just the "what"
- Use single-line comments for brief explanations
- Use multi-line comments for longer explanations or function/class descriptions
- Always place comments above the code they refer to.
- If comments already exist, review and update them as needed.
- Minimize changes to existing comments.
- For TypeScript functions, classes and fields, use JSDoc comments. do NOT add type annotations in comments.
- For Python functions and classes, use docstrings.
- do NOT modify comments with TODOs.
- do NOT modify comments with URLs or links as they are reference to external resources.
- do NOT add comments to imports
Your output should be the original code with your added comments. Make sure to preserve ALL the original code's formatting and structure. DO NOT BE LAZY.
Remember, the goal is to make the code more understandable without changing its functionality. DO NOT MODIFY THE CODE ITSELF.
Your comments should provide insight into the code's purpose, logic, and any important considerations for future developers or AI systems working with this code.
"system.safety_jailbreak",
"system.safety_harmful_content",
"system.safety_validate_harmful_content",
label: `comment ${filename}`,
const { text, fences } = res
const newContent = fences?.[0]?.content ?? text
async function checkModifications(filename: string): Promise<boolean> {
const diff = await git.diff({ paths: filename })
const res = await runPrompt(
ctx.def("DIFF", diff, { language: "diff" })
ctx.$`You are an expert developer at all programming languages.
Your task is to analyze the changes in DIFF and make sure that only comments are modified.
Report all changes that are not comments or spacing and print <MOD>;
otherwise, print <NO_MOD>.
system: ["system.assistant", "system.safety_jailbreak"],
label: `check comments in ${filename}`,
res.text?.includes("<MOD>") || !res.text?.includes("<NO_MOD>")