coder-agent 2.8.0 → 2.8.2

package/dist/agent.js

@@ -563,6 +563,12 @@ export class Agent {
     setModel(model) {
         this.model = model;
     }
+    getApiKey() {
+        return this.apiKey;
+    }
+    setApiKey(key) {
+        this.apiKey = key;
+    }
     async chat(userMessage, signal) {
         try {
             if (signal?.aborted) {
@@ -831,9 +837,10 @@ ${JSON.stringify(allMessages.slice(1).map(m => ({ role: m.role, content: m.conte
 
 Instructions:
 1. Review the conversation history and extract any important new project setup details, style preferences, build/test commands, package quirks, or persistent instructions that should be remembered for future sessions.
-2. Integrate these new learnings into the existing memory structure. Keep existing useful learnings/preferences, but clean up duplicates or obsolete info.
-3. Keep the content concise, clean, and formatted in Markdown.
-4. If there are no new learnings, setup details, or instructions in the conversation, output EXACTLY the existing memory content. Do not add conversational text, just output the updated/existing markdown content.`;
+2. Maintain and update the "## File-to-Data Map" section. If any files were read, modified, created, or discussed, verify if they are listed in the map. If not, add them with their relative path and a brief 1-sentence description of the data, logic, or functions they contain. If they are already listed but their content/purpose changed, update their description.
+3. Integrate these new learnings and file catalog updates into the existing memory structure. Keep existing useful learnings/preferences, but clean up duplicates or obsolete info.
+4. Keep the content concise, clean, and formatted in Markdown.
+5. If there are no new learnings, setup details, file updates, or instructions in the conversation, output EXACTLY the existing memory content. Do not add conversational text, just output the updated/existing markdown content.`;
             const responseObj = await callGeminiAPIWithRotation(this.apiKey, {
                 model: this.model,
                 messages: [{ role: "user", content: prompt }],

package/dist/index.js

@@ -180,9 +180,15 @@ async function main() {
         console.log(chalk.dim(`✓ Default model set to: ${tempModel}`));
         process.exit(0);
     }
-    // Bootstrap API Key if missing
-    if (!apiKey) {
-        apiKey = await promptApiKey();
+    // Load API Key
+    apiKey = apiKey || "";
+    // Single-Shot Mode key check
+    if (queryArgs.length > 0 && !apiKey) {
+        console.log(chalk.hex('#ff453a')('✕ error'));
+        console.log(chalk.dim("  Gemini API Key is required but missing."));
+        console.log(chalk.dim("  Please set it using the GEMINI_API_KEY environment variable or run:"));
+        console.log(`  ${chalk.cyan("coder-agent --set-key <key>")}`);
+        process.exit(1);
     }
     let currentAbortController = null;
     let originalListeners = [];
@@ -279,7 +285,12 @@ async function main() {
     // Save/update last used info immediately
     await saveLastUsedInfo(CURRENT_VERSION, Date.now());
     printBanner(modelToUse);
-    if (showWelcomeCommands) {
+    if (!agent.getApiKey()) {
+        console.log(chalk.hex('#ff9f0a')('  🔑 Gemini API Key is missing!'));
+        console.log(`     Please type ${chalk.cyan('/key <your_api_key>')} to set and save it globally.`);
+        console.log(chalk.dim("     Get a free key at https://aistudio.google.com\n"));
+    }
+    else if (showWelcomeCommands) {
         console.log(chalk.white.bold("  💡 Get started with interactive slash commands:"));
         console.log(`    ${chalk.hex('#0a84ff')('/model')} ${chalk.gray('[name]')} — View active model or switch to [name]`);
         console.log(`    ${chalk.hex('#0a84ff')('/clear')}        — Wipe conversation memory`);
@@ -288,11 +299,15 @@ async function main() {
         console.log(`    ${chalk.hex('#0a84ff')('/exit')}         — Exit Coder`);
         console.log();
     }
+    let isRlClosed = false;
     rl = readline.createInterface({
         input: process.stdin,
         output: process.stdout,
         terminal: true,
     });
+    rl.on("close", () => {
+        isRlClosed = true;
+    });
     process.stdin.on("keypress", (char, key) => {
         if (isHijacked || !rl)
             return;
@@ -301,6 +316,7 @@ async function main() {
                 console.log();
                 console.log(chalk.dim("  Available Commands:"));
                 console.log(`  ${chalk.hex('#0a84ff')('/model')} ${chalk.gray('[name]')} ${chalk.dim('— View active model or switch to [name]')}`);
+                console.log(`  ${chalk.hex('#0a84ff')('/key')}   ${chalk.gray('[api_key]')} ${chalk.dim('— Set/save your Gemini API Key globally')}`);
                 console.log(`  ${chalk.hex('#0a84ff')('/clear')}        ${chalk.dim('— Wipe conversation memory')}`);
                 console.log(`  ${chalk.hex('#0a84ff')('/status')}       ${chalk.dim('— Show active model and memory usage')}`);
                 console.log(`  ${chalk.hex('#0a84ff')('/help')}         ${chalk.dim('— Show help screen')}`);
@@ -314,24 +330,42 @@ async function main() {
     let pasteTimeout = null;
     let lineCountInBurst = 0;
     async function executeAgentChat(trimmed) {
+        if (isRlClosed)
+            return;
         // Pause standard input processing during agent thinking & updates
-        rl.pause();
+        try {
+            rl.pause();
+        }
+        catch { }
         // Built-in slash commands
         if (trimmed === "/exit" || trimmed === "/quit") {
-            rl.close();
+            try {
+                rl.close();
+            }
+            catch { }
             process.exit(0);
         }
         if (trimmed === "/clear") {
             agent.clearMemory();
             console.log(chalk.hex('#30d158')('✓') + ' ' + chalk.gray('Memory cleared'));
-            rl.resume();
-            rl.prompt();
+            if (!isRlClosed) {
+                try {
+                    rl.resume();
+                    rl.prompt();
+                }
+                catch { }
+            }
             return;
         }
         if (trimmed === "/status") {
             console.log(chalk.dim(`session · ${agent.memoryStatus()}`));
-            rl.resume();
-            rl.prompt();
+            if (!isRlClosed) {
+                try {
+                    rl.resume();
+                    rl.prompt();
+                }
+                catch { }
+            }
             return;
         }
         if (trimmed.startsWith("/model")) {
@@ -351,22 +385,77 @@ async function main() {
                     console.log(chalk.dim(`  Model must be one of: ${VALID_MODELS.join(" · ")}`));
                 }
             }
-            rl.resume();
-            rl.prompt();
+            if (!isRlClosed) {
+                try {
+                    rl.resume();
+                    rl.prompt();
+                }
+                catch { }
+            }
+            return;
+        }
+        if (trimmed.startsWith("/key")) {
+            const parts = trimmed.split(/\s+/);
+            if (parts.length === 1) {
+                const keyToCheck = agent.getApiKey();
+                if (keyToCheck) {
+                    const masked = keyToCheck.slice(0, 7) + "..." + keyToCheck.slice(-4);
+                    console.log(chalk.dim(`  Gemini API Key is set: `) + chalk.gray(masked));
+                }
+                else {
+                    console.log(chalk.hex('#ff453a')('✕ error') + chalk.dim(' — Gemini API Key is missing. Set it using: /key <your_api_key>'));
+                }
+            }
+            else {
+                const newKey = parts[1].trim();
+                if (!newKey) {
+                    console.log(chalk.hex('#ff453a')('✕ error') + chalk.dim(' — API Key cannot be empty.'));
+                }
+                else {
+                    await saveApiKey(newKey);
+                    agent.setApiKey(newKey);
+                    console.log(chalk.hex('#30d158')('✓') + ' ' + chalk.gray('Gemini API Key saved and set successfully.'));
+                }
+            }
+            if (!isRlClosed) {
+                try {
+                    rl.resume();
+                    rl.prompt();
+                }
+                catch { }
+            }
             return;
         }
         if (trimmed === "/help") {
             console.log(chalk.white.bold("\n  Interactive Commands:"));
             console.log(chalk.gray(`    /model [name] — View active model or switch to [name]
+    /key [api_key]— Set/save your Gemini API Key globally (hides the key from the LLM)
     /clear        — Wipe conversation memory
     /status       — Show active model and memory usage
     /exit         — Exit Coder`));
             console.log(chalk.white.bold("\n  API Keys & Configuration:"));
             console.log(chalk.gray(`    • Stored at: ~/.coder-config.json
-    • To change key: Exit and run 'coder-agent --set-key <key>'
+    • To change key: Type '/key <api_key>' or run 'coder-agent --set-key <key>'
     • Env variable option: GEMINI_API_KEY`));
-            rl.resume();
-            rl.prompt();
+            if (!isRlClosed) {
+                try {
+                    rl.resume();
+                    rl.prompt();
+                }
+                catch { }
+            }
+            return;
+        }
+        // Block logic execution if API Key is missing
+        if (!agent.getApiKey()) {
+            console.log(chalk.hex('#ff453a')('✕ error') + chalk.dim(' — Gemini API Key is missing. Please set it using: /key <your_api_key>'));
+            if (!isRlClosed) {
+                try {
+                    rl.resume();
+                    rl.prompt();
+                }
+                catch { }
+            }
             return;
         }
         currentAbortController = new AbortController();
@@ -395,9 +484,14 @@ async function main() {
             stopHijack();
             currentAbortController = null;
         }
-        rl.resume();
-        console.log(chalk.dim('────────────────────────────────────────────────'));
-        rl.prompt();
+        if (!isRlClosed) {
+            try {
+                rl.resume();
+                console.log(chalk.dim('────────────────────────────────────────────────'));
+                rl.prompt();
+            }
+            catch { }
+        }
     }
     console.log(chalk.dim('────────────────────────────────────────────────'));
     rl.setPrompt(chalk.hex('#0a84ff')('›') + ' ');

package/dist/memory.js

@@ -315,7 +315,7 @@ export async function loadAgentMemoryPrompt(agentType, scope) {
                 }
                 catch {
                     // Create default template
-                    const defaultTemplate = `# Persistent Agent Memory\n\nUse this section to write general rules, project setup details, style preferences, or learnings. Update this file using your write_file/patch_file tools to persist memories.\n\n## Learnings & Guidelines\n- (No memories stored yet. Add your learnings here.)\n`;
+                    const defaultTemplate = `# Persistent Agent Memory\n\nUse this section to write general rules, project setup details, style preferences, or learnings. Update this file using your write_file/patch_file tools to persist memories.\n\n## Learnings & Guidelines\n- (No memories stored yet. Add your learnings here.)\n\n## File-to-Data Map\n- (List of files and what data/logic they contain. This helps the agent know what file has what data.)\n`;
                     try {
                         await fs.writeFile(memoryFile, defaultTemplate, "utf-8");
                         memoryContent = defaultTemplate;
@@ -333,13 +333,13 @@ export async function loadAgentMemoryPrompt(agentType, scope) {
     let scopeNote = "";
     switch (scope) {
         case "user":
-            scopeNote = "- Since this memory has 'user' scope, keep learnings general since they apply across all of the user's projects.";
+            scopeNote = "- Since this memory has 'user' scope, keep learnings and cataloging general since they apply across all of the user's projects.";
             break;
         case "project":
-            scopeNote = "- Since this memory has 'project' scope and is shared with the team via version control, tailor memories specifically to this project.";
+            scopeNote = "- Since this memory has 'project' scope and is shared with the team via version control, tailor memories and file cataloging specifically to this project.";
             break;
         case "local":
-            scopeNote = "- Since this memory has 'local' scope (not checked into version control), tailor memories specifically to this machine and project.";
+            scopeNote = "- Since this memory has 'local' scope (not checked into version control), tailor memories and file cataloging specifically to this machine and project.";
             break;
     }
     return `
@@ -349,8 +349,9 @@ This memory is loaded at the start of every session and injected into your syste
 
 ### Guidelines for Memory Usage:
 1. Read the memory contents below to see existing learnings, project instructions, or user preferences.
-2. If you learn something new that would be useful for future sessions (e.g., build commands, package quirks, style guidelines, API details, tool workarounds), update this file directly using your \`write_file\` or \`patch_file\` tools targeting the file path: \`${memoryFile}\`.
-3. Keep the content concise, clean, and organized.
+2. Refer to the '## File-to-Data Map' section to quickly find which files contain what data, features, logic, or configurations.
+3. If you learn something new that would be useful for future sessions (e.g., build commands, package quirks, style guidelines, API details, tool workarounds), or if you create/modify files, update this file directly using your \`write_file\` or \`patch_file\` tools targeting the file path: \`${memoryFile}\`.
+4. Keep the content concise, clean, and organized.
 ${scopeNote}
 
 ### Current Memory Contents:

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "coder-agent",
-  "version": "2.8.0",
+  "version": "2.8.2",
   "description": "CLI coding agent powered by Google Gemini",
   "type": "module",
   "main": "dist/index.js",