@khiem_enhance/ai-doc-agent 0.1.0 → 0.1.3

package/dist/cli.js

@@ -1,6 +1,7 @@
 #!/usr/bin/env node
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+require("dotenv/config");
 const commander_1 = require("commander");
 const generate_1 = require("./commands/generate");
 const program = new commander_1.Command();
@@ -12,5 +13,20 @@ program
     .command("generate")
     .option("--since <commit>", "Only analyze changes since commit")
     .option("--output <dir>", "Docs output directory", "docs")
-    .action(generate_1.generateDocs);
+    .option("--only <part>", "architecture|modules|all", "all")
+    .option("--max-files <n>", "Max files included per LLM request", "8")
+    .option("--max-chars <n>", "Max characters included per LLM request", "60000")
+    .action(async (opts) => {
+    // normalize options
+    const only = String(opts.only ?? "all");
+    const maxFiles = Number(opts.maxFiles ?? 8);
+    const maxChars = Number(opts.maxChars ?? 60000);
+    await (0, generate_1.generateDocs)({
+        since: opts.since,
+        output: opts.output,
+        only: only ?? "all",
+        maxFiles: Number.isFinite(maxFiles) ? maxFiles : 8,
+        maxChars: Number.isFinite(maxChars) ? maxChars : 60000,
+    });
+});
 program.parse();

package/dist/commands/generate.js

@@ -12,34 +12,43 @@ const modules_1 = require("../analyzers/modules");
 const markdownWriter_1 = require("../writers/markdownWriter");
 const gitUtils_1 = require("../git/gitUtils");
 const moduleDetector_1 = require("../scanner/moduleDetector");
+const truncate = (s, maxChars) => s.length > maxChars ? s.slice(0, maxChars) + "\n\n...<truncated>" : s;
 async function generateDocs(options) {
     const root = process.cwd();
     const files = options.since
-        ? (0, gitUtils_1.getChangedFiles)(options.since).map(f => path_1.default.join(root, f))
+        ? (0, gitUtils_1.getChangedFiles)(options.since).map((f) => path_1.default.join(root, f))
         : await (0, fileScanner_1.scanProject)(root);
+    const only = options.only ?? "all";
+    const maxFiles = Math.max(1, options.maxFiles ?? 8);
+    const maxChars = Math.max(5000, options.maxChars ?? 60000);
     // ---------- Architecture ----------
-    const tree = files
-        .map(f => path_1.default.relative(root, f))
-        .join("\n");
-    const architectureSource = files
-        .slice(0, 25)
-        .map(f => `FILE: ${f}\n${(0, contentReader_1.readFile)(f)}`)
-        .join("\n\n");
-    const architecture = await (0, architecture_1.generateArchitectureDoc)(tree, architectureSource);
-    (0, markdownWriter_1.writeDoc)(options.output, "architecture.md", architecture);
-    // ---------- Modules ----------
-    const modules = (0, moduleDetector_1.detectModules)(files, root);
-    for (const [moduleName, moduleFiles] of Object.entries(modules)) {
-        const fileList = moduleFiles
-            .map(f => path_1.default.relative(root, f))
-            .join("\n");
-        const source = moduleFiles
-            .slice(0, 20)
-            .map(f => `FILE: ${f}\n${(0, contentReader_1.readFile)(f)}`)
+    if (only === "architecture" || only === "all") {
+        const tree = files.map((f) => path_1.default.relative(root, f)).join("\n");
+        // IMPORTANT: limit payload by file count + maxChars
+        const architectureSourceRaw = files
+            .slice(0, maxFiles)
+            .map((f) => `FILE: ${f}\n${(0, contentReader_1.readFile)(f)}`)
             .join("\n\n");
-        const doc = await (0, modules_1.generateModuleDocs)(moduleName, fileList, source);
-        (0, markdownWriter_1.writeDoc)(path_1.default.join(options.output, "modules"), `${moduleName}.md`, doc);
-        console.log(`📄 Module doc generated: ${moduleName}`);
+        const architectureSource = truncate(architectureSourceRaw, maxChars);
+        const architecture = await (0, architecture_1.generateArchitectureDoc)(tree, architectureSource);
+        (0, markdownWriter_1.writeDoc)(options.output, "architecture.md", architecture);
+        console.log("📄 Architecture doc generated");
+    }
+    // ---------- Modules ----------
+    if (only === "modules" || only === "all") {
+        const modules = (0, moduleDetector_1.detectModules)(files, root);
+        for (const [moduleName, moduleFiles] of Object.entries(modules)) {
+            const fileList = moduleFiles.map((f) => path_1.default.relative(root, f)).join("\n");
+            // IMPORTANT: limit payload by file count + maxChars
+            const sourceRaw = moduleFiles
+                .slice(0, maxFiles)
+                .map((f) => `FILE: ${f}\n${(0, contentReader_1.readFile)(f)}`)
+                .join("\n\n");
+            const source = truncate(sourceRaw, maxChars);
+            const doc = await (0, modules_1.generateModuleDocs)(moduleName, fileList, source);
+            (0, markdownWriter_1.writeDoc)(path_1.default.join(options.output, "modules"), `${moduleName}.md`, doc);
+            console.log(`📄 Module doc generated: ${moduleName}`);
+        }
     }
     console.log("✅ Docs generation completed");
 }

package/dist/llm/openaiClient.js

@@ -6,14 +6,60 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.openai = void 0;
 exports.askLLM = askLLM;
 const openai_1 = __importDefault(require("openai"));
-const env_1 = require("../config/env");
-exports.openai = new openai_1.default({
-    apiKey: env_1.env.openaiKey
-});
-async function askLLM(prompt) {
-    const res = await exports.openai.chat.completions.create({
-        model: env_1.env.model,
-        messages: [{ role: "user", content: prompt }]
-    });
-    return res.choices[0].message.content;
+const apiKey = process.env.OPENAI_API_KEY;
+if (!apiKey) {
+    throw new Error("Missing OPENAI_API_KEY.\n" +
+        "Set it before running:\n" +
+        '  export OPENAI_API_KEY="sk-xxxx"\n');
+}
+exports.openai = new openai_1.default({ apiKey });
+const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
+/**
+ * askLLM with:
+ * - retry on 429 using retry-after headers
+ * - fail gracefully if retry-after is extremely long (token budget exhausted)
+ */
+async function askLLM(prompt, maxRetries = 5) {
+    const model = process.env.MODEL || "gpt-4.1-mini";
+    let attempt = 0;
+    while (true) {
+        try {
+            const res = await exports.openai.chat.completions.create({
+                model,
+                messages: [{ role: "user", content: prompt }],
+            });
+            return res.choices[0].message.content ?? "";
+        }
+        catch (err) {
+            attempt++;
+            const status = err?.status;
+            const headers = err?.headers || {};
+            const retryAfterMs = Number(headers["retry-after-ms"]) ||
+                (Number(headers["retry-after"]) || 0) * 1000;
+            if (status === 429) {
+                // If retry-after is huge => org quota/token budget exhausted for a long window.
+                if (retryAfterMs && retryAfterMs > 120000) {
+                    throw new Error([
+                        "Rate limited (429) due to token/request limits.",
+                        `Retry-After is very long: ~${Math.ceil(retryAfterMs / 1000)}s.`,
+                        "",
+                        "Fix suggestions:",
+                        "- Reduce input size:",
+                        "  ai-doc-agent generate --only architecture --max-files 5 --max-chars 30000",
+                        "- Or run modules separately:",
+                        "  ai-doc-agent generate --only modules --max-files 4 --max-chars 25000",
+                        "- Or increase your OpenAI limits / add billing on your OpenAI account.",
+                    ].join("\n"));
+                }
+                if (attempt <= maxRetries) {
+                    const wait = retryAfterMs || 20000;
+                    console.warn(`⚠️ Rate limited (429). Retrying in ${Math.ceil(wait / 1000)}s... (${attempt}/${maxRetries})`);
+                    await sleep(wait);
+                    continue;
+                }
+            }
+            // Non-429 or exceeded retries
+            throw err;
+        }
+    }
 }

package/package.json

@@ -1,13 +1,17 @@
 {
   "name": "@khiem_enhance/ai-doc-agent",
-  "version": "0.1.0",
+  "version": "0.1.3",
   "description": "AI-powered documentation generator from source code",
   "license": "MIT",
   "bin": {
     "ai-doc-agent": "dist/cli.js"
   },
   "main": "dist/cli.js",
-  "files": ["dist", "README.md", "LICENSE"],
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
   "scripts": {
     "build": "tsc",
     "prepublishOnly": "npm run build"