tech-debt-visualizer 0.2.0 → 0.2.1

package/dist/cli.js

@@ -30,9 +30,10 @@ program
     .option("-f, --format <type>", "Output format: cli | html | json | markdown", "cli")
     .option("--no-llm", "Skip LLM-powered insights")
     .option("--llm", "Enable LLM (default). Use with --llm-key and/or --llm-model")
-    .option("--llm-key <key>", "API key for LLM (overrides GEMINI_API_KEY / OPENAI_API_KEY / OPENROUTER_API_KEY)")
-    .option("--llm-model <model>", "Model name (e.g. gemini-1.5-flash, gpt-4o-mini)")
-    .option("--llm-max-tokens <n>", "Max tokens for LLM responses (default: 2048 for overall, 1024 per-file)", (v) => parseInt(v, 10))
+    .option("--llm-key <key>", "API key (overrides env: GEMINI_API_KEY, OPENAI_API_KEY, OPENROUTER_API_KEY)")
+    .option("--llm-endpoint <url>", "OpenAI-compatible API base URL (e.g. https://api.openai.com/v1 or proxy)")
+    .option("--llm-model <model>", "Model name (e.g. gpt-4o-mini, gemini-2.5-flash)")
+    .option("--llm-max-tokens <n>", "Max tokens per response", (v) => parseInt(v, 10))
     .option("--ci", "CI mode: minimal output, exit with non-zero if debt score is high")
     .action(async (path, opts) => {
     const repoPath = join(process.cwd(), path);
@@ -41,6 +42,7 @@ program
     const outputPath = opts.output ?? (format === "html" ? "tech-debt-report.html" : undefined);
     const llmConfigOverrides = {
         apiKey: opts.llmKey,
+        baseURL: opts.llmEndpoint,
         model: opts.llmModel,
         ...(opts.llmMaxTokens != null && opts.llmMaxTokens > 0 ? { maxTokens: opts.llmMaxTokens } : {}),
     };
@@ -82,9 +84,9 @@ program
             if (!llmConfig) {
                 progress.update(totalSteps, { task: "Skipping LLM (no key)" });
                 progress.stop();
-                process.stderr.write(chalk.yellow("  No LLM API key found. Set GEMINI_API_KEY or OPENAI_API_KEY (or use --llm-key <key>).\n" +
-                    "  Example: export GEMINI_API_KEY=your_key   or   --llm-key your_key\n" +
-                    "  Or add GEMINI_API_KEY=your_key to a .env file in the current directory.\n" +
+                process.stderr.write(chalk.yellow("  No LLM API key found. Use --llm-key <key> or set one of:\n" +
+                    "  GEMINI_API_KEY, OPENAI_API_KEY, OPENROUTER_API_KEY (or .env).\n" +
+                    "  For a custom endpoint: --llm-endpoint <url> --llm-key <key>\n" +
                     "  Skipping AI insights for this run.\n\n"));
             }
             else {

package/dist/llm.d.ts

@@ -18,7 +18,7 @@ export interface LLMConfig {
     maxTokens?: number;
 }
 export type LLMProvider = "openai" | "openrouter" | "gemini";
-/** Resolve provider and auth from config + env. When --llm-key is used, provider is inferred from key format so a Gemini key is not sent to OpenRouter. */
+/** Resolve provider and auth from config + env. Explicit baseURL = OpenAI-compatible; else key format or env picks provider. */
 export declare function resolveLLMConfig(config?: LLMConfig): {
     provider: LLMProvider;
     apiKey: string;

package/dist/llm.js

@@ -68,13 +68,19 @@ function parseFileAssessmentResponse(raw) {
         code: code || undefined,
     };
 }
-/** Resolve provider and auth from config + env. When --llm-key is used, provider is inferred from key format so a Gemini key is not sent to OpenRouter. */
+/** Resolve provider and auth from config + env. Explicit baseURL = OpenAI-compatible; else key format or env picks provider. */
 export function resolveLLMConfig(config = {}) {
+    const explicitBase = (config.baseURL ?? process.env.OPENAI_BASE_URL)?.replace(/\/$/, "");
     const cliKey = config.apiKey;
-    const openRouterKey = cliKey ?? process.env.OPENROUTER_API_KEY;
-    const geminiKey = cliKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_GENAI_API_KEY;
     const openaiKey = cliKey ?? process.env.OPENAI_API_KEY ?? process.env.ANTHROPIC_API_KEY;
-    // When a single key is passed (e.g. --llm-key), pick provider by key format so we don't send a Gemini key to OpenRouter (401 "No cookie auth").
+    if (explicitBase && openaiKey) {
+        return {
+            provider: "openai",
+            apiKey: openaiKey,
+            baseURL: explicitBase,
+            model: config.model ?? process.env.OPENAI_MODEL ?? OPENAI_DEFAULT_MODEL,
+        };
+    }
     if (cliKey) {
         if (cliKey.startsWith("AIza")) {
             return {
@@ -88,23 +94,25 @@ export function resolveLLMConfig(config = {}) {
             return {
                 provider: "openai",
                 apiKey: cliKey,
-                baseURL: config.baseURL ?? process.env.OPENAI_BASE_URL ?? "",
+                baseURL: (config.baseURL ?? process.env.OPENAI_BASE_URL ?? "").replace(/\/$/, ""),
                 model: config.model ?? process.env.OPENAI_MODEL ?? OPENAI_DEFAULT_MODEL,
             };
         }
     }
-    if (openRouterKey) {
+    if (cliKey ?? process.env.OPENROUTER_API_KEY) {
+        const key = cliKey ?? process.env.OPENROUTER_API_KEY;
         return {
             provider: "openrouter",
-            apiKey: openRouterKey,
+            apiKey: key,
             baseURL: config.baseURL ?? process.env.OPENROUTER_BASE_URL ?? OPENROUTER_BASE,
             model: config.model ?? process.env.OPENROUTER_MODEL ?? OPENROUTER_DEFAULT_MODEL,
         };
     }
-    if (geminiKey) {
+    if (cliKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_GENAI_API_KEY) {
+        const key = cliKey ?? process.env.GEMINI_API_KEY ?? process.env.GOOGLE_GENAI_API_KEY;
         return {
             provider: "gemini",
-            apiKey: geminiKey,
+            apiKey: key,
             baseURL: GEMINI_BASE,
             model: config.model ?? process.env.GEMINI_MODEL ?? GEMINI_DEFAULT_MODEL,
         };
@@ -113,7 +121,7 @@ export function resolveLLMConfig(config = {}) {
         return {
             provider: "openai",
             apiKey: openaiKey,
-            baseURL: config.baseURL ?? process.env.OPENAI_BASE_URL ?? "",
+            baseURL: (config.baseURL ?? process.env.OPENAI_BASE_URL ?? "").replace(/\/$/, ""),
             model: config.model ?? process.env.OPENAI_MODEL ?? OPENAI_DEFAULT_MODEL,
         };
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tech-debt-visualizer",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Language-agnostic CLI that analyzes repos and generates interactive technical debt visualizations with AI-powered insights",
   "type": "module",
   "main": "dist/index.js",