devsplain 2.0.1 → 2.1.1

package/bin/cli.js

@@ -10,7 +10,7 @@ const { execSync } = require('child_process');
 let rl;
 let askQuestion;
 
-/** Checks if the Git repository is dirty [ds] */
+// Checks if the Git working tree is dirty [ds]
 function isGitDirty() {
     try {
         const gitDir = execSync('git rev-parse --is-inside-work-tree', { stdio: ['ignore', 'pipe', 'ignore'], encoding: 'utf8' }).trim();
@@ -23,10 +23,13 @@ function isGitDirty() {
     return false;
 }
 
-/** Checks if a line of code is inside a string [ds] */
+// Checks if a line of code is inside a string literal [ds]
 function isLineInsideString(lines, targetLineIndex, ext = '') {
     const isPython = ext.toLowerCase() === '.py';
     const isHTML = ['.html', '.vue', '.svelte'].includes(ext.toLowerCase());
+    const isRustOrSwift = ['.rs', '.swift'].includes(ext.toLowerCase());
+    const isCpp = ['.cpp', '.cc', '.cxx', '.c', '.h', '.hpp'].includes(ext.toLowerCase());
+    const isJS = ['.js', '.jsx', '.ts', '.tsx'].includes(ext.toLowerCase());
     let inBacktick = false;
     let inTripleDouble = false;
     let inTripleSingle = false;
@@ -34,19 +37,55 @@ function isLineInsideString(lines, targetLineIndex, ext = '') {
     let inDouble = false;
     let inBlockJS = false;
     let inBlockHTML = false;
+    let blockDepthJS = 0;
+    let inCppRawString = false;
+    let cppRawDelimiter = '';
+    let inRegex = false;
     for (let i = 0; i < targetLineIndex; i++) {
         const line = lines[i];
         let j = 0;
         while (j < line.length) {
             if (inBlockJS) {
+                if (line.slice(j, j + 2) === '/*') {
+                    if (isRustOrSwift) blockDepthJS++;
+                    j += 2;
+                    continue;
+                }
                 if (line.slice(j, j + 2) === '*/') {
-                    inBlockJS = false;
+                    if (isRustOrSwift && blockDepthJS > 1) {
+                        blockDepthJS--;
+                    } else {
+                        inBlockJS = false;
+                        blockDepthJS = 0;
+                    }
                     j += 2;
                     continue;
                 }
                 j++;
                 continue;
             }
+            if (inCppRawString) {
+                if (line.slice(j, j + 2 + cppRawDelimiter.length) === ')' + cppRawDelimiter + '"') {
+                    inCppRawString = false;
+                    j += 2 + cppRawDelimiter.length;
+                    continue;
+                }
+                j++;
+                continue;
+            }
+            if (inRegex) {
+                let escaped = false;
+                let k = j - 1;
+                while (k >= 0 && line[k] === '\\') {
+                    escaped = !escaped;
+                    k--;
+                }
+                if (line[j] === '/' && !escaped) {
+                    inRegex = false;
+                }
+                j++;
+                continue;
+            }
             if (inBlockHTML) {
                 if (line.slice(j, j + 3) === '-->') {
                     inBlockHTML = false;
@@ -65,7 +104,6 @@ function isLineInsideString(lines, targetLineIndex, ext = '') {
                 } else if (isHTML) {
                     if (line.slice(j, j + 4) === '<!--') {
                         inBlockHTML = true;
-                        // Check if current character is a backtick [ds]
                         j += 4;
                         continue;
                     }
@@ -78,17 +116,56 @@ function isLineInsideString(lines, targetLineIndex, ext = '') {
                         break; // Ignore rest of line
                     }
                 } else {
-                    if (line.slice(j, j + 2) === '//') {
-                        break; // Ignore rest of line
-                    }
-                    if (line.slice(j, j + 2) === '/*') {
-                        inBlockJS = true;
-                        j += 2;
-                        continue;
-                    }
-                    const isShellOrRuby = ['.sh', '.rb', '.php'].includes(ext.toLowerCase());
-                    if (isShellOrRuby && line[j] === '#') {
-                        break; // Ignore rest of line
+                    const isShellOrRuby = ['.sh', '.rb'].includes(ext.toLowerCase());
+                    if (isShellOrRuby) {
+                        if (line[j] === '#') {
+                            break; // Ignore rest of line
+                        }
+                    } else {
+                        if (line.slice(j, j + 2) === '//') {
+                            break; // Ignore rest of line
+                        }
+                        if (line.slice(j, j + 2) === '/*') {
+                            inBlockJS = true;
+                            blockDepthJS = 1;
+                            j += 2;
+                            continue;
+                        }
+                        if (ext.toLowerCase() === '.php' && line[j] === '#') {
+                            break; // Ignore rest of line
+                        }
+                        if (isCpp && line[j] === 'R' && line[j+1] === '"') {
+                            const match = line.slice(j).match(/^R"([^()\\\s]{0,16})\(/);
+                            if (match) {
+                                cppRawDelimiter = match[1];
+                                inCppRawString = true;
+                                j += match[0].length;
+                                continue;
+                            }
+                        }
+                        if (isJS && line[j] === '/') {
+                            let k = j - 1;
+                            while (k >= 0 && /\s/.test(line[k])) k--;
+                            let isRegex = false;
+                            if (k < 0) {
+                                isRegex = true;
+                            } else {
+                                const prevChar = line[k];
+                                if (/[=({\[:,;!+*&|?<>-]/.test(prevChar)) {
+                                    isRegex = true;
+                                } else {
+                                    const prefix = line.slice(0, k + 1);
+                                    if (/(?:return|typeof|yield|await|throw)\s*$/.test(prefix)) {
+                                        isRegex = true;
+                                    }
+                                }
+                            }
+                            if (isRegex) {
+                                inRegex = true;
+                                j++;
+                                continue;
+                            }
+                        }
                     }
                 }
             }
@@ -125,7 +202,6 @@ function isLineInsideString(lines, targetLineIndex, ext = '') {
                 }
                 }
                 if (!inBacktick) {
-                    // Check if current character is a double quote [ds]
                     if (line[j] === '"' && !inSingle) {
                         let escaped = false;
                         let k = j - 1;
@@ -157,13 +233,16 @@ function isLineInsideString(lines, targetLineIndex, ext = '') {
             inDouble = false;
         }
     }
-    return inBacktick || inTripleDouble || inTripleSingle || inSingle || inDouble;
+    return inBacktick || inTripleDouble || inTripleSingle || inSingle || inDouble || inCppRawString || inRegex;
 }
 
-/** Analyzes comments in a given list of lines [ds] */
+// Analyzes comments in a given set of code lines [ds]
 function analyzeComments(lines, ext = '') {
     const isPython = ext.toLowerCase() === '.py';
     const isHTML = ['.html', '.vue', '.svelte'].includes(ext.toLowerCase());
+    const isRustOrSwift = ['.rs', '.swift'].includes(ext.toLowerCase());
+    const isCpp = ['.cpp', '.cc', '.cxx', '.c', '.h', '.hpp'].includes(ext.toLowerCase());
+    const isJS = ['.js', '.jsx', '.ts', '.tsx'].includes(ext.toLowerCase());
     const analysis = [];
     let inBacktick = false;
     let inTripleDouble = false;
@@ -172,6 +251,10 @@ function analyzeComments(lines, ext = '') {
     let inDouble = false;
     let inBlockJS = false;
     let inBlockHTML = false;
+    let blockDepthJS = 0;
+    let inCppRawString = false;
+    let cppRawDelimiter = '';
+    let inRegex = false;
     for (let i = 0; i < lines.length; i++) {
         const line = lines[i];
         let commentStartIndex = -1;
@@ -179,14 +262,46 @@ function analyzeComments(lines, ext = '') {
         let j = 0;
         while (j < line.length) {
             if (inBlockJS) {
+                if (line.slice(j, j + 2) === '/*') {
+                    if (isRustOrSwift) blockDepthJS++;
+                    j += 2;
+                    continue;
+                }
                 if (line.slice(j, j + 2) === '*/') {
-                    inBlockJS = false;
+                    if (isRustOrSwift && blockDepthJS > 1) {
+                        blockDepthJS--;
+                    } else {
+                        inBlockJS = false;
+                        blockDepthJS = 0;
+                    }
                     j += 2;
                     continue;
                 }
                 j++;
                 continue;
             }
+            if (inCppRawString) {
+                if (line.slice(j, j + 2 + cppRawDelimiter.length) === ')' + cppRawDelimiter + '"') {
+                    inCppRawString = false;
+                    j += 2 + cppRawDelimiter.length;
+                    continue;
+                }
+                j++;
+                continue;
+            }
+            if (inRegex) {
+                let escaped = false;
+                let k = j - 1;
+                while (k >= 0 && line[k] === '\\') {
+                    escaped = !escaped;
+                    k--;
+                }
+                if (line[j] === '/' && !escaped) {
+                    inRegex = false;
+                }
+                j++;
+                continue;
+            }
             if (inBlockHTML) {
                 if (line.slice(j, j + 3) === '-->') {
                     inBlockHTML = false;
@@ -220,20 +335,60 @@ function analyzeComments(lines, ext = '') {
                         break;
                     }
                 } else {
-                    if (line.slice(j, j + 2) === '//') {
-                        commentStartIndex = j;
-                        break;
-                    }
-                    if (line.slice(j, j + 2) === '/*') {
-                        commentStartIndex = j;
-                        inBlockJS = true;
-                        j += 2;
-                        continue;
-                    }
-                    const isShellOrRuby = ['.sh', '.rb', '.php'].includes(ext.toLowerCase());
-                    if (isShellOrRuby && line[j] === '#') {
-                        commentStartIndex = j;
-                        break;
+                    const isShellOrRuby = ['.sh', '.rb'].includes(ext.toLowerCase());
+                    if (isShellOrRuby) {
+                        if (line[j] === '#') {
+                            commentStartIndex = j;
+                            break;
+                        }
+                    } else {
+                        if (line.slice(j, j + 2) === '//') {
+                            commentStartIndex = j;
+                            break;
+                        }
+                        if (line.slice(j, j + 2) === '/*') {
+                            commentStartIndex = j;
+                            inBlockJS = true;
+                            blockDepthJS = 1;
+                            j += 2;
+                            continue;
+                        }
+                        if (ext.toLowerCase() === '.php' && line[j] === '#') {
+                            commentStartIndex = j;
+                            break;
+                        }
+                        if (isCpp && line[j] === 'R' && line[j+1] === '"') {
+                            const match = line.slice(j).match(/^R"([^()\\\s]{0,16})\(/);
+                            if (match) {
+                                cppRawDelimiter = match[1];
+                                inCppRawString = true;
+                                j += match[0].length;
+                                continue;
+                            }
+                        }
+                        if (isJS && line[j] === '/') {
+                            let k = j - 1;
+                            while (k >= 0 && /\s/.test(line[k])) k--;
+                            let isRegex = false;
+                            if (k < 0) {
+                                isRegex = true;
+                            } else {
+                                const prevChar = line[k];
+                                if (/[=({\[:,;!+*&|?<>-]/.test(prevChar)) {
+                                    isRegex = true;
+                                } else {
+                                    const prefix = line.slice(0, k + 1);
+                                    if (/(?:return|typeof|yield|await|throw)\s*$/.test(prefix)) {
+                                        isRegex = true;
+                                    }
+                                }
+                            }
+                            if (isRegex) {
+                                inRegex = true;
+                                j++;
+                                continue;
+                            }
+                        }
                     }
                 }
             }
@@ -281,7 +436,6 @@ function analyzeComments(lines, ext = '') {
                             inDouble = !inDouble;
                         }
                     }
-                    // Check if current character is a single quote [ds]
                     else if (line[j] === "'" && !inDouble) {
                         let escaped = false;
                         let k = j - 1;
@@ -323,7 +477,7 @@ function analyzeComments(lines, ext = '') {
     return analysis;
 }
 
-/** Splices comments into the given data [ds] */
+// Splices comments into the given code data [ds]
 function spliceComments(data, comments, mode = 'default', ext = '') {
     const hasCRLF = data.includes('\r\n');
     const lineEnding = hasCRLF ? '\r\n' : '\n';
@@ -372,7 +526,6 @@ function spliceComments(data, comments, mode = 'default', ext = '') {
                 continue;
             }
 
-            // Prune mode: remove all comments from the line [ds]
             if (mode === 'prune') {
                 if (lineAnalysis.isPureComment) {
                     finalDeletions.add(lineNum);
@@ -447,7 +600,6 @@ function spliceComments(data, comments, mode = 'default', ext = '') {
             const indentation = indentMatch ? indentMatch[1] : '';
 
             const commentLines = c.comment.split(/\r?\n/).map((line, idx) => {
-                // Add the comment to the list of comments [ds]
                 let trimmed = line.trimStart();
                 if (!trimmed) return '';
 
@@ -479,7 +631,6 @@ function spliceComments(data, comments, mode = 'default', ext = '') {
     const filteredText = filtered.map(line => line.text);
     const filteredIndices = filtered.map(line => line.originalIndex);
 
-    // Check if the filtered text is equal to the original text [ds]
     const textEqual = filteredText.every((text, idx) => {
         const origIdx = filteredIndices[idx];
         const originalLine = originalLines[origIdx];
@@ -518,7 +669,7 @@ function spliceComments(data, comments, mode = 'default', ext = '') {
     return annotated.map(line => line.text).join(lineEnding);
 }
 
-/** Runs the CLI interface [ds] */
+// Runs the CLI interface for the application [ds]
 async function runCLI() {
     rl = readline.createInterface({ input: process.stdin, output: process.stdout });
     askQuestion = (query) => new Promise((resolve) => rl.question(query, resolve));
@@ -662,7 +813,6 @@ Options:
     ];
     const allIgnored = [...defaultIgnoredFolders, ...userIgnorePatterns];
 
-    /** Checks if a given path is ignored [ds] */
     function isPathIgnored(targetPath) {
         const filename = path.basename(targetPath);
         for (const pattern of allIgnored) {
@@ -673,7 +823,7 @@ Options:
         return false;
     }
 
-    /** Processes a given path [ds] */
+    // Processes a given file path, either a file or directory [ds]
     async function processPath(targetPath) {
         const stats = fs.statSync(targetPath);
 
@@ -731,6 +881,7 @@ Options:
                         fs.renameSync(tempPath, targetPath);
                         console.log(` Successfully saved ${targetPath}`);
                     } else {
+                        // Asks the user if they want to save the commented code to a file [ds]
                         console.log(` Skipped ${targetPath}`);
                     }
                 } else {
@@ -763,6 +914,7 @@ Options:
     rl.close();
 }
 
+// Starts the CLI application if this script is run directly [ds]
 if (require.main === module) {
     runCLI().catch(err => {
         console.error(err);

package/lib/config.js

@@ -4,7 +4,7 @@ const os = require('os');
 const readline = require('readline');
 const configPath = path.join(os.homedir(), '.devsplainrc');
 
-/** Prompts the user for a secret input, hiding the input from the console [ds] */
+/** Asks the user for a secret input (e.g., an API key) without echoing the input to the console. [ds] */
 function askSecret(query) {
   return new Promise((resolve) => {
     const rl = readline.createInterface({
@@ -33,7 +33,7 @@ function askSecret(query) {
   });
 }
 
-/** Retrieves the configuration, either from environment variables or by prompting the user [ds] */
+/** Attempts to fetch the configuration settings, either from environment variables or by prompting the user. [ds] */
 async function getConfig(forceWizard = false) {
   if (process.env.DEVSPLAIN_API_KEY || process.env.DEVSPLAIN_PROVIDER) {
     const provider = process.env.DEVSPLAIN_PROVIDER || 'gemini';
@@ -46,8 +46,8 @@ async function getConfig(forceWizard = false) {
       baseUrl
     };
   }
+// If the configuration file doesn't exist, or if we're forcing the wizard, let's prompt the user for configuration. [ds]
 
-  // If the configuration file does not exist or forceWizard is true, prompt the user to configure [ds]
   if (!fs.existsSync(configPath) || forceWizard) {
     let rl = readline.createInterface({
       input: process.stdin,
@@ -58,13 +58,13 @@ async function getConfig(forceWizard = false) {
     let config = null;
     let confirmed = false;
 
-    // Continuously prompt the user until the configuration is confirmed [ds]
+    // Keep prompting the user until they confirm their configuration settings. [ds]
     while (!confirmed) {
       let baseUrl = "";
       let model = "";
       let provider = "";
 
-      // Display the list of available AI providers [ds]
+      // Display the list of available AI providers to the user. [ds]
       console.log("\nWhich AI Provider Do You want to use?");
       console.log("1. Groq (Free, Fast, Llama-3)");
       console.log("2. Gemini (Free Tier)");
@@ -115,7 +115,7 @@ async function getConfig(forceWizard = false) {
         continue;
       }
 
-      // Prompt the user for an API key [ds]
+      // Get the API key from the user. [ds]
       let apiKey = '';
       while (true) {
         const promptMsg = provider === 'custom' 
@@ -142,7 +142,7 @@ async function getConfig(forceWizard = false) {
         console.log(`API key is required for provider '${provider}'.`);
       }
 
-      // Ask the user if they want to enable auto-pruning of existing comments [ds]
+      // Ask the user if they want to enable auto-pruning of existing human/AI comments. [ds]
       let autoPrune = false;
       while (true) {
         const pruneAns = (await askQuestion("Do you want devsplain to aggressively prune (overwrite) existing human/AI comments? (y/n, default: n): ")).trim().toLowerCase();
@@ -156,7 +156,7 @@ async function getConfig(forceWizard = false) {
         console.log("Invalid choice. Please enter 'y' or 'n'.");
       }
 
-      // Display a summary of the configuration [ds]
+      // Display a summary of the user's configuration settings. [ds]
       console.log("\n--- Configuration Summary ---");
       console.log(`Provider:     ${provider}`);
       console.log(`Model:        ${model}`);
@@ -165,7 +165,7 @@ async function getConfig(forceWizard = false) {
       console.log(`Auto-Prune:   ${autoPrune ? 'Yes' : 'No'}`);
       console.log("-----------------------------\n");
 
-      // Confirm the configuration with the user [ds]
+      // Confirm the configuration settings with the user. [ds]
       while (true) {
         const confirm = (await askQuestion("Does this look correct? (y/n, default: y): ")).trim().toLowerCase();
         if (confirm === '' || confirm === 'y' || confirm === 'yes') {
@@ -187,10 +187,9 @@ async function getConfig(forceWizard = false) {
 
     rl.close();
 
-    // Write the configuration to the config file [ds]
+    // Write the configuration settings to the config file. [ds]
     fs.writeFileSync(configPath, JSON.stringify(config, null, 2));
     try {
-      // Set the permissions of the config file to prevent other users from reading it [ds]
       if (process.platform !== 'win32') {
         fs.chmodSync(configPath, 0o600);
       }
@@ -204,4 +203,5 @@ async function getConfig(forceWizard = false) {
   }
 }
 
+// Export the getConfig function for use in other modules. [ds]
 module.exports = { getConfig };

package/lib/llm.js

@@ -1,6 +1,7 @@
-/** Fetches a URL with retry functionality [ds] */
+// Asynchronously fetches data from the specified URL with a retry mechanism [ds]
 async function fetchWithRetry(url, options, maxRetries = 3, initialDelay = 1000) {
     let lastError;
+    // Loop through each attempt, up to the maximum number of retries [ds]
     for (let attempt = 0; attempt < maxRetries; attempt++) {
         const controller = new AbortController();
         const timeoutId = setTimeout(() => controller.abort(), 15000);
@@ -36,66 +37,70 @@ async function fetchWithRetry(url, options, maxRetries = 3, initialDelay = 1000)
     throw lastError;
 }
 
-/** Gets comments for the given code [ds] */
+// Retrieves comments for the given code, language, and configuration [ds]
 async function getComments(code, language, config, mode = 'default') {
     const lines = code.split(/\r?\n/);
     const numberedCode = lines.map((line, index) => `${index + 1}: ${line}`).join('\n');
 
-    let prompt = "";
-    if (mode === 'clean') {
-        prompt = `
-You are a code documentation scrubber. Analyze the following ${language} code which has line numbers prepended to it.
-Your goal is to identify all lines containing comments (both block/JSDoc and inline comments) and return their line numbers to delete them.
-
-CRITICAL RULES:
-1. You MUST respond with ONLY a raw, valid JSON array of objects. NO markdown formatting, NO backticks, NO explanations, NO text before or after the JSON.
-2. Each object must have exactly two properties: "line" (the integer line number of the comment line to delete) and "action" (which must be the string "delete").
-3. Do NOT include the original code in your response.
-4. If no comments are found, return an empty array: [].
-5. For block or JSDoc comments (e.g., starting with /* and ending with */), you MUST identify and return the line numbers of ALL lines in that block, including the opening /*, all intermediate lines, and the closing */. Do NOT leave trailing comment delimiters behind.
-
-Example Output:
-[
-  { "line": 4, "action": "delete" },
-  { "line": 5, "action": "delete" }
-]
+    // Extract the file extension from the language string [ds]
+    const extMatch = language.match(/\.[0-9a-z]+$/i);
+    const ext = extMatch ? extMatch[0].toLowerCase() : '';
+    const isPython = ext === '.py';
+    const isRubyOrShell = ['.rb', '.sh', '.php'].includes(ext);
+    const isHTML = ['.html', '.vue', '.svelte'].includes(ext);
+    const isSql = ext === '.sql';
+    
+    // Define the single-line comment token and examples [ds]
+    let singleLineToken = '//';
+    let blockExample = '/** Calculates the total price */';
+    let inlineExample = '// Check for null values';
+
+    if (isPython || isRubyOrShell) {
+        singleLineToken = '#';
+        blockExample = '# Calculates the total price';
+        inlineExample = '# Check for null values';
+    } else if (isHTML) {
+        singleLineToken = '<!--';
+        blockExample = '<!-- Calculates the total price -->';
+        inlineExample = '<!-- Check for null values -->';
+    } else if (isSql) {
+        singleLineToken = '--';
+        blockExample = '-- Calculates the total price';
+        inlineExample = '-- Check for null values';
+    }
 
-Here is the source code:
-${numberedCode}
-        `.trim();
-    } else {
-        let instruction = "Provide JSDoc/docstrings block comments above functions and sparse inline comments for complex logic.";
-        if (mode === 'light') {
-            instruction = "Provide ONLY JSDoc/docstrings above functions. Keep it minimal.";
-        } else if (mode === 'full') {
-            instruction = "Provide highly detailed JSDoc/docstrings above functions, and exhaustive step-by-step inline comments (using standard comment syntax like // or #) explaining every conditional branch, loop, variable assignment, and logical block inside function bodies. Do not be sparse; explain the code's execution flow in detail.";
-        }
+    // Provide instructions based on the mode [ds]
+    let instruction = `Provide block comments above functions and sparse inline comments for complex logic.`;
+    if (mode === 'light') {
+        instruction = `Provide ONLY block comments above functions. Keep it minimal.`;
+    } else if (mode === 'full') {
+        instruction = `Provide highly detailed block comments above functions, and exhaustive step-by-step inline comments explaining every conditional branch, loop, variable assignment, and logical block inside function bodies. Do not be sparse; explain the code's execution flow in detail.`;
+    }
 
-        prompt = `
+    let prompt = `
 You are a code documentation engine. Analyze the following ${language} code which has line numbers prepended to it.
 ${instruction}
 
 CRITICAL RULES:
 1. You MUST respond with ONLY a raw, valid JSON array of objects. NO markdown formatting, NO backticks, NO explanations, NO text before or after the JSON.
-2. Each object must have exactly two properties: "line" (the integer line number where the comment should be inserted ABOVE) and "comment" (the text of the comment itself, including standard comment syntax like // or /** */).
+2. Each object must have exactly two properties: "line" (the integer line number where the comment should be inserted ABOVE) and "comment" (the text of the comment itself).
 3. Do NOT include the original code in your response.
 4. If no comments are needed, return an empty array: [].
+5. IMPORTANT: Use ONLY ${singleLineToken} for comments. DO NOT use docstrings or multi-line string literals like """ or ''' for comments.
 
 Example Output:
 [
-  { "line": 4, "comment": "/** Calculates the total price */" },
-  { "line": 12, "comment": "// Check for null values" }
+  { "line": 4, "comment": "${blockExample}" },
+  { "line": 12, "comment": "${inlineExample}" }
 ]
 
 Here is the source code:
 ${numberedCode}
-        `.trim();
-    }
+    `.trim();
 
-    // Initialize text response [ds]
     let textResponse = "";
 
-    // Check if provider is Gemini [ds]
+    // Handle API requests based on the provider [ds]
     if (config.provider === 'gemini') {
         const url = `https://generativelanguage.googleapis.com/v1beta/models/${config.model}:generateContent?key=${config.apiKey}`;
         let data;
@@ -118,6 +123,7 @@ ${numberedCode}
         }
         textResponse = data.candidates[0].content.parts[0].text;
     } else if (config.provider === 'claude') {
+        // Handle Claude API requests [ds]
         const url = `${config.baseUrl}/v1/messages`;
         let data;
         try {
@@ -146,7 +152,7 @@ ${numberedCode}
         }
         textResponse = data.content[0].text;
     } 
-    // Otherwise, use an OpenAI-compatible provider [ds]
+    // Handle other API requests [ds]
     else {
         const url = `${config.baseUrl}/v1/chat/completions`;
         let data;
@@ -176,7 +182,6 @@ ${numberedCode}
         textResponse = data.choices[0].message.content;
     }
 
-    // Clean up the text response [ds]
     let cleanText = textResponse.trim();
     const start = cleanText.indexOf('[');
     const end = cleanText.lastIndexOf(']');
@@ -184,9 +189,9 @@ ${numberedCode}
         cleanText = cleanText.substring(start, end + 1);
     }
 
-    // Parse the response as JSON [ds]
     let parsed;
     try {
+        // Attempt to parse the response as JSON [ds]
         parsed = JSON.parse(cleanText);
     } catch (e) {
         throw new Error(`Parsing Error: Failed to parse LLM response as JSON. Raw response was:\n${textResponse}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "devsplain",
-  "version": "2.0.1",
+  "version": "2.1.1",
   "description": "An agent-agnostic CLI tool that automatically adds JSDoc and inline comments to your code using free LLMs.",
   "author": "mwahaj36",
   "license": "MIT",