quackstack 1.0.12 → 1.0.13

package/dist/cli.cjs

@@ -6,24 +6,51 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const commander_1 = require("commander");
 const chalk_animation_1 = __importDefault(require("chalk-animation"));
+const chalk_1 = __importDefault(require("chalk"));
 const repl_js_1 = require("./repl.js");
 const context_generator_js_1 = require("./lib/context-generator.js");
 const readme_js_1 = require("./commands/readme.js");
+const ai_provider_js_1 = require("./lib/ai-provider.js");
 const path_1 = __importDefault(require("path"));
 const program = new commander_1.Command();
 const PROJECT_NAME = path_1.default.basename(process.cwd());
 program
-    .name("quackstack")
+    .name("QuackStack🐥")
     .description("Your cracked unpaid intern for all things codebase!")
-    .version("1.0.4")
+    .version("1.0.5")
     .option("-r, --reindex", "Force reindex the codebase")
-    .option("-c, --context", "Generate context files for ALL AI coding tools (Cursor, Windsurf, Cline, Continue, Aider)")
-    .option("-d, --docs", "Generate CODEBASE.md - universal documentation for any IDE/editor")
+    .option("-c, --context", "Generate context files for ALL AI coding tools")
+    .option("-d, --docs", "Generate CODEBASE.md")
     .option("--readme", "Generate README.md from your codebase")
-    .option("--cursor", "[DEPRECATED] Use --context instead. Generates .cursorrules only")
+    .option("--cursor", "[DEPRECATED] Use --context instead")
     .option("-w, --watch", "Watch mode: auto-update context files on file changes")
+    .option("-p, --provider <provider>", "AI provider: openai, anthropic, gemini, deepseek, mistral")
+    .option("-m, --model <model>", "Specific model to use")
+    .option("--list-models", "List available providers and models")
     .action(async (options) => {
-    const title = chalk_animation_1.default.rainbow("Welcome to QuackStack! 🐥\n");
+    if (options.listModels) {
+        const client = (0, ai_provider_js_1.getAIClient)();
+        const providers = client.getAvailableProviders();
+        console.log(chalk_1.default.cyan("\nAvailable AI Providers & Models:\n"));
+        if (providers.length === 0) {
+            console.log(chalk_1.default.red("No API keys configured."));
+            process.exit(1);
+        }
+        providers.forEach(provider => {
+            console.log(chalk_1.default.green(`\n${provider.name} (${provider.provider}):`));
+            console.log(chalk_1.default.gray(`  Default: ${provider.defaultModel}`));
+            console.log(chalk_1.default.white("  Available models:"));
+            provider.models.forEach(model => {
+                const isDefault = model === provider.defaultModel;
+                console.log(chalk_1.default.white(`    - ${model}${isDefault ? chalk_1.default.gray(" (default)") : ""}`));
+            });
+        });
+        console.log(chalk_1.default.cyan("\nUsage:"));
+        console.log(chalk_1.default.white("  quackstack --provider anthropic --model claude-sonnet-4-5-20250929"));
+        console.log(chalk_1.default.white("  quackstack -p openai -m gpt-4o\n"));
+        process.exit(0);
+    }
+    const title = chalk_animation_1.default.rainbow("QuackStack\n");
     await new Promise(res => setTimeout(res, 1500));
     title.stop();
     if (options.readme) {
@@ -36,20 +63,19 @@ program
         process.exit(0);
     }
     if (options.cursor) {
-        console.log("⚠️  --cursor is deprecated. Use --context to support all AI tools.\n");
-        console.log("🔍 Generating context for AI assistants...\n");
+        console.log("--cursor is deprecated. Use --context to support all AI tools.\n");
+        console.log("Generating context for AI assistants...\n");
         await (0, context_generator_js_1.generateContextFiles)(PROJECT_NAME);
         await (0, context_generator_js_1.updateGlobalContext)(PROJECT_NAME);
-        console.log("\n✅ Context generation complete!");
-        console.log("💡 Your AI coding assistant will now have codebase context");
+        console.log("\nContext generation complete!");
         process.exit(0);
     }
     if (options.watch) {
-        console.log("👀 Starting watch mode...\n");
+        console.log("Starting watch mode...\n");
         await (0, context_generator_js_1.generateContextFiles)(PROJECT_NAME);
         (0, context_generator_js_1.watchAndUpdateContext)(PROJECT_NAME);
         await new Promise(() => { });
     }
-    await (0, repl_js_1.startREPL)(options.reindex);
+    await (0, repl_js_1.startREPL)(options.reindex, options.provider, options.model);
 });
 program.parse();

package/dist/commands/search.js

@@ -2,7 +2,7 @@
 import { client } from "../lib/database.js";
 import { localEmbeddings } from "../lib/local-embeddings.js";
 import { getAIClient } from "../lib/ai-provider.js";
-export async function search(query, projectName) {
+export async function search(query, projectName, provider, model) {
     const snippets = await client.codeSnippet.findMany({
         where: { projectName },
     });
@@ -28,7 +28,7 @@ export async function search(query, projectName) {
     const context = uniqueResults
         .map((r, i) => `[${i + 1}] ${r.filePath}${r.functionName ? ` (${r.functionName})` : ""}\n${r.content}`)
         .join("\n\n---\n\n");
-    const aiClient = getAIClient();
+    const aiClient = getAIClient(provider, model);
     const answer = await aiClient.generateAnswer(query, context);
     return { answer, sources: uniqueResults };
 }

package/dist/lib/ai-provider.js

@@ -5,54 +5,153 @@ import dotenv from "dotenv";
 dotenv.config({ quiet: true });
 export class AIClient {
     provider;
+    model;
     openai;
     anthropic;
     gemini;
     deepseek;
     mistral;
-    constructor() {
-        const config = this.detectProvider();
+    constructor(providerOverride, modelOverride) {
+        const config = this.detectProvider(providerOverride);
         this.provider = config.provider;
+        this.model = modelOverride || config.model || this.getDefaultModel(config.provider);
+        const availableProvider = this.getAvailableProviders().find(p => p.provider === this.provider);
+        if (availableProvider && !availableProvider.models.includes(this.model)) {
+            throw new Error(`Model '${this.model}' not available for ${this.provider}.\n` +
+                `Available models: ${availableProvider.models.join(', ')}`);
+        }
         this.initializeClient(config);
     }
-    detectProvider() {
-        if (process.env.QUACKSTACK_OPENAI_KEY) {
+    detectProvider(override) {
+        const availableProviders = this.getAvailableProviders();
+        if (availableProviders.length === 0) {
+            throw new Error("No AI API key found. Set one of:\n" +
+                "  QUACKSTACK_OPENAI_KEY\n" +
+                "  QUACKSTACK_ANTHROPIC_KEY\n" +
+                "  QUACKSTACK_GEMINI_KEY\n" +
+                "  QUACKSTACK_DEEPSEEK_KEY\n" +
+                "  QUACKSTACK_MISTRAL_KEY");
+        }
+        if (override) {
+            const providerConfig = availableProviders.find(p => p.provider === override);
+            if (!providerConfig) {
+                throw new Error(`Provider '${override}' not available. Set QUACKSTACK_${override.toUpperCase()}_KEY.\n` +
+                    `Available: ${availableProviders.map(p => p.provider).join(', ')}`);
+            }
             return {
-                provider: "openai",
-                apiKey: process.env.QUACKSTACK_OPENAI_KEY,
+                provider: override,
+                apiKey: this.getApiKey(override),
+                model: providerConfig.defaultModel
             };
         }
+        const first = availableProviders[0];
+        return {
+            provider: first.provider,
+            apiKey: this.getApiKey(first.provider),
+            model: first.defaultModel
+        };
+    }
+    getApiKey(provider) {
+        const keyMap = {
+            openai: process.env.QUACKSTACK_OPENAI_KEY || "",
+            anthropic: process.env.QUACKSTACK_ANTHROPIC_KEY || "",
+            gemini: process.env.QUACKSTACK_GEMINI_KEY || "",
+            deepseek: process.env.QUACKSTACK_DEEPSEEK_KEY || "",
+            mistral: process.env.QUACKSTACK_MISTRAL_KEY || "",
+        };
+        return keyMap[provider] || undefined;
+    }
+    getAvailableProviders() {
+        const providers = [];
+        if (process.env.QUACKSTACK_OPENAI_KEY) {
+            providers.push({
+                provider: "openai",
+                name: "OpenAI",
+                models: ["gpt-4.1",
+                    "gpt-4.1-mini",
+                    "gpt-4.1-nano",
+                    "gpt-4-turbo",
+                    "gpt-3.5-turbo",
+                    "gpt-3.5-turbo-0125"],
+                defaultModel: "gpt-4o-mini"
+            });
+        }
         if (process.env.QUACKSTACK_ANTHROPIC_KEY) {
-            return {
+            providers.push({
                 provider: "anthropic",
-                apiKey: process.env.QUACKSTACK_ANTHROPIC_KEY,
-            };
+                name: "Anthropic",
+                models: ["claude-opus-4",
+                    "claude-sonnet-4",
+                    "claude-3.7-sonnet",
+                    "claude-3.5-haiku"
+                ],
+                defaultModel: "claude-sonnet-4"
+            });
         }
         if (process.env.QUACKSTACK_GEMINI_KEY) {
-            return {
+            providers.push({
                 provider: "gemini",
-                apiKey: process.env.QUACKSTACK_GEMINI_KEY,
-            };
+                name: "Gemini",
+                models: ["gemini-3",
+                    "gemini-2.5-pro",
+                    "gemini-2.5-flash",
+                    "gemini-2.5-flash-lite",
+                    "gemini-2.5-flash-image",
+                    "gemini-2.0-spark",
+                    "gemini-2.0-flash-lite"
+                ],
+                defaultModel: "gemini-2.5-flash"
+            });
         }
         if (process.env.QUACKSTACK_DEEPSEEK_KEY) {
-            return {
+            providers.push({
                 provider: "deepseek",
-                apiKey: process.env.QUACKSTACK_DEEPSEEK_KEY,
-            };
+                name: "DeepSeek",
+                models: ["deepseek-r1",
+                    "deepseek-r1-zero",
+                    "deepseek-r1-70b",
+                    "deepseek-r1-32b",
+                    "deepseek-r1-14b",
+                    "deepseek-r1-8b",
+                    "deepseek-r1-7b",
+                    "deepseek-r1-1.5b",
+                    "deepseek-vl2",
+                    "deepseek-vl2-small",
+                    "deepseek-vl2-tiny"
+                ],
+                defaultModel: "deepseek-chat"
+            });
         }
         if (process.env.QUACKSTACK_MISTRAL_KEY) {
-            return {
+            providers.push({
                 provider: "mistral",
-                apiKey: process.env.QUACKSTACK_MISTRAL_KEY,
-            };
+                name: "Mistral",
+                models: [
+                    "mistral-large",
+                    "mistral-medium",
+                    "mistral-small",
+                    "mistral-7b",
+                    "mixtral-8x7b",
+                    "mixtral-8x22b",
+                    "codestral",
+                    "codestral-mamba",
+                    "mathstral",
+                    "mistral-nemo-12b"
+                ],
+                defaultModel: "mistral-large-latest"
+            });
         }
-        throw new Error("No AI API key found. Please set one of:\n" +
-            "  QUACKSTACK_OPENAI_KEY      - OpenAI GPT-4\n" +
-            "  QUACKSTACK_ANTHROPIC_KEY   - Anthropic Claude\n" +
-            "  QUACKSTACK_GEMINI_KEY      - Google Gemini (free tier available)\n" +
-            "  QUACKSTACK_DEEPSEEK_KEY    - DeepSeek (cheapest option)\n" +
-            "  QUACKSTACK_MISTRAL_KEY     - Mistral AI\n\n" +
-            "Get your API key from the respective provider's website.");
+        return providers;
+    }
+    getDefaultModel(provider) {
+        const defaults = {
+            openai: "gpt-4o-mini",
+            anthropic: "claude-3-5-sonnet-20241022",
+            gemini: "gemini-1.5-flash",
+            deepseek: "deepseek-chat",
+            mistral: "mistral-large-latest",
+        };
+        return defaults[provider];
     }
     initializeClient(config) {
         switch (config.provider) {
@@ -110,7 +209,7 @@ export class AIClient {
         if (!this.openai)
             throw new Error("OpenAI client not initialized");
         const response = await this.openai.chat.completions.create({
-            model: "gpt-4o-mini",
+            model: this.model,
             messages: [
                 { role: "system", content: systemPrompt },
                 { role: "user", content: userPrompt },
@@ -123,7 +222,7 @@ export class AIClient {
         if (!this.anthropic)
             throw new Error("Anthropic client not initialized");
         const response = await this.anthropic.messages.create({
-            model: "claude-3-5-sonnet-20241022",
+            model: this.model,
             max_tokens: 2048,
             system: systemPrompt,
             messages: [{ role: "user", content: userPrompt }],
@@ -136,7 +235,7 @@ export class AIClient {
     async generateGemini(systemPrompt, userPrompt) {
         if (!this.gemini)
             throw new Error("Gemini client not initialized");
-        const model = this.gemini.getGenerativeModel({ model: "gemini-1.5-flash" });
+        const model = this.gemini.getGenerativeModel({ model: this.model });
         const result = await model.generateContent(`${systemPrompt}\n\n${userPrompt}`);
         return result.response.text();
     }
@@ -144,7 +243,7 @@ export class AIClient {
         if (!this.deepseek)
             throw new Error("DeepSeek client not initialized");
         const response = await this.deepseek.chat.completions.create({
-            model: "deepseek-chat",
+            model: this.model,
             messages: [
                 { role: "system", content: systemPrompt },
                 { role: "user", content: userPrompt },
@@ -157,7 +256,7 @@ export class AIClient {
         if (!this.mistral)
             throw new Error("Mistral client not initialized");
         const response = await this.mistral.chat.completions.create({
-            model: "mistral-large-latest",
+            model: this.model,
             messages: [
                 { role: "system", content: systemPrompt },
                 { role: "user", content: userPrompt },
@@ -168,22 +267,36 @@ export class AIClient {
     }
     getProviderName() {
         const names = {
-            openai: "OpenAI GPT",
-            anthropic: "Anthropic Claude",
-            gemini: "Google Gemini",
+            openai: "OpenAI",
+            anthropic: "Anthropic",
+            gemini: "Gemini",
             deepseek: "DeepSeek",
-            mistral: "Mistral AI",
+            mistral: "Mistral",
         };
         return names[this.provider];
     }
     getProvider() {
         return this.provider;
     }
+    getModel() {
+        return this.model;
+    }
+    setModel(model) {
+        const availableProvider = this.getAvailableProviders().find(p => p.provider === this.provider);
+        if (availableProvider && !availableProvider.models.includes(model)) {
+            throw new Error(`Model '${model}' not available for ${this.provider}.\n` +
+                `Available models: ${availableProvider.models.join(', ')}`);
+        }
+        this.model = model;
+    }
 }
 let aiClientInstance = null;
-export function getAIClient() {
-    if (!aiClientInstance) {
-        aiClientInstance = new AIClient();
+export function getAIClient(provider, model) {
+    if (!aiClientInstance || provider || model) {
+        aiClientInstance = new AIClient(provider, model);
     }
     return aiClientInstance;
 }
+export function resetAIClient() {
+    aiClientInstance = null;
+}

package/dist/repl.js

@@ -5,16 +5,20 @@ import { ingest } from "./commands/ingest.js";
 import { client } from "./lib/database.js";
 import path from "path";
 import { detectFileChanges, formatChangeMessage } from "./lib/file-change-detector.js";
+import { getAIClient, resetAIClient } from "./lib/ai-provider.js";
 const PROJECT_NAME = path.basename(process.cwd());
-export async function startREPL(forceReindex = false) {
-    console.log(chalk.cyan("\n💡 Tip: Press Ctrl+C to exit\n"));
+export async function startREPL(forceReindex = false, provider, model) {
+    const aiClient = getAIClient(provider, model);
+    console.log(chalk.cyan(`\nUsing: ${aiClient.getProviderName()} - ${aiClient.getModel()}`));
+    console.log(chalk.gray("Run 'quackstack --list-models' to see all options"));
+    console.log(chalk.cyan("\nPress Ctrl+C to exit\n"));
     if (!forceReindex) {
         const changes = await detectFileChanges(process.cwd(), PROJECT_NAME);
         if (changes && changes.totalChanges > 0) {
-            console.log(chalk.yellow(`\n⚠️  Detected ${changes.totalChanges} file change${changes.totalChanges > 1 ? 's' : ''} since last index:`));
+            console.log(chalk.yellow(`\nDetected ${changes.totalChanges} file change${changes.totalChanges > 1 ? 's' : ''} since last index:`));
             console.log(chalk.yellow(`   ${formatChangeMessage(changes)}`));
             console.log(chalk.yellow(`   Run 'quack --reindex' for best results.\n`));
-            const shouldReindex = await promptUser(chalk.yellow("Would you like to reindex now? (y/n) > "));
+            const shouldReindex = await promptUser(chalk.yellow("Reindex now? (y/n) > "));
             if (shouldReindex.toLowerCase() === 'y') {
                 forceReindex = true;
             }
@@ -25,33 +29,38 @@ export async function startREPL(forceReindex = false) {
     });
     if (existingCount === 0 || forceReindex) {
         if (forceReindex) {
-            console.log(chalk.gray("🗑️  Clearing old index..."));
+            console.log(chalk.gray("Clearing old index..."));
             await client.codeSnippet.deleteMany({
                 where: { projectName: PROJECT_NAME },
             });
         }
-        console.log(chalk.gray("🔍 Indexing your codebase (this may take a moment)..."));
+        console.log(chalk.gray("Indexing your codebase..."));
         await ingest(process.cwd(), PROJECT_NAME, true);
-        console.log(chalk.green("✅ Indexing complete!"));
+        console.log(chalk.green("Indexing complete"));
     }
     const rl = readline.createInterface({
         input: process.stdin,
         output: process.stdout,
-        prompt: chalk.yellow("🐥 Quack! How can I help? > "),
+        prompt: chalk.yellow("quack > "),
     });
     rl.prompt();
     rl.on("line", async (line) => {
         const query = line.trim();
+        if (query.startsWith("/")) {
+            await handleCommand(query, rl);
+            rl.prompt();
+            return;
+        }
         if (!query) {
             rl.prompt();
             return;
         }
         try {
-            const { answer, sources } = await search(query, PROJECT_NAME);
+            const { answer, sources } = await search(query, PROJECT_NAME, provider, model);
             console.log(chalk.white(`\n${answer}\n`));
-            const showDetails = await promptUser(chalk.cyan("💡 Want more details? (y/n) > "));
+            const showDetails = await promptUser(chalk.cyan("Want more details? (y/n) > "));
             if (showDetails.toLowerCase() === "y") {
-                console.log(chalk.blue("\n📚 Relevant Code:\n"));
+                console.log(chalk.blue("\nRelevant Code:\n"));
                 sources.forEach((src, i) => {
                     console.log(chalk.gray(`[${i + 1}] ${src.filePath} (relevance: ${(src.score * 100).toFixed(1)}%)`));
                     console.log(chalk.white(src.content));
@@ -65,10 +74,68 @@ export async function startREPL(forceReindex = false) {
         rl.prompt();
     });
     rl.on("close", () => {
-        console.log(chalk.gray("\n👋 Happy coding!"));
+        console.log(chalk.gray("\nBye"));
         process.exit(0);
     });
 }
+async function handleCommand(command, rl) {
+    const [cmd, ...args] = command.slice(1).split(" ");
+    switch (cmd) {
+        case "model":
+        case "m":
+            if (args.length === 0) {
+                const client = getAIClient();
+                console.log(chalk.cyan(`\nCurrent: ${client.getProviderName()} - ${client.getModel()}\n`));
+            }
+            else {
+                const [newModel] = args;
+                try {
+                    resetAIClient();
+                    const client = getAIClient(undefined, newModel);
+                    console.log(chalk.green(`\nSwitched to: ${client.getProviderName()} - ${client.getModel()}\n`));
+                }
+                catch (error) {
+                    console.log(chalk.red(`\nError: ${error.message}\n`));
+                }
+            }
+            break;
+        case "provider":
+        case "p":
+            if (args.length === 0) {
+                const client = getAIClient();
+                const providers = client.getAvailableProviders();
+                console.log(chalk.cyan("\nAvailable Providers:\n"));
+                providers.forEach(p => {
+                    const current = p.provider === client.getProvider() ? chalk.green(" (current)") : "";
+                    console.log(chalk.white(`  ${p.provider}: ${p.name}${current}`));
+                });
+                console.log();
+            }
+            else {
+                const [newProvider] = args;
+                try {
+                    resetAIClient();
+                    const client = getAIClient(newProvider);
+                    console.log(chalk.green(`\nSwitched to: ${client.getProviderName()} - ${client.getModel()}\n`));
+                }
+                catch (error) {
+                    console.log(chalk.red(`\nError: ${error.message}\n`));
+                }
+            }
+            break;
+        case "help":
+        case "h":
+            console.log(chalk.cyan("\nAvailable Commands:\n"));
+            console.log(chalk.white("  /model, /m [model]    Show or change model"));
+            console.log(chalk.white("  /provider, /p [name]  Show or change provider"));
+            console.log(chalk.white("  /help, /h             Show this help"));
+            console.log();
+            break;
+        default:
+            console.log(chalk.red(`\nUnknown command: ${cmd}\n`));
+            console.log(chalk.gray("Type /help for available commands\n"));
+    }
+}
 function promptUser(question) {
     const rl = readline.createInterface({
         input: process.stdin,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "quackstack",
-  "version": "1.0.12",
+  "version": "1.0.13",
   "description": "Your cracked unpaid intern for all things codebase related! AI-powered codebase search and Q&A.",
   "type": "module",
   "main": "dist/cli.cjs",