prab-cli 1.0.0 → 1.1.0

package/dist/index.js

@@ -1,5 +1,38 @@
 #!/usr/bin/env node
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
@@ -10,6 +43,8 @@ const config_1 = require("./lib/config");
 const context_1 = require("./lib/context");
 const ui_1 = require("./lib/ui");
 const config_2 = require("./lib/config");
+const slash_commands_1 = require("./lib/slash-commands");
+const select_1 = __importStar(require("@inquirer/select"));
 const ora_1 = __importDefault(require("ora"));
 // Import tool system
 const base_1 = require("./lib/tools/base");
@@ -176,7 +211,7 @@ program.action(async () => {
     }
     // Initialize chat handler
     const chatHandler = new chat_handler_1.ChatHandler(toolRegistry, toolExecutor, modelProvider, contextMessage);
-    ui_1.log.info('Type "/" for commands menu, or start chatting!');
+    ui_1.log.info('Type "/" for commands (searchable), or start chatting!');
     // Display any existing todos
     const session = (0, config_2.getSessionData)();
     if (session.todos && session.todos.length > 0) {
@@ -191,180 +226,147 @@ program.action(async () => {
                 message: ">",
             },
         ]);
-        // Handle Slash Commands Menu
-        if (userInput.trim() === "/") {
+        // Handle Slash Commands
+        if (userInput.trim() === "/" || userInput.trim().startsWith("/")) {
             const currentModel = modelProvider.modelId;
-            console.log("\n┌────────────────────────────────────────┐");
-            console.log("│           AVAILABLE COMMANDS           │");
-            console.log("└────────────────────────────────────────┘\n");
-            const { action } = await inquirer_1.default.prompt([
-                {
-                    type: "list",
-                    name: "action",
-                    message: "Choose an option:",
-                    choices: [
-                        "1. Select Model",
-                        "2. Usage",
-                        "3. Tools",
-                        "4. Todos",
-                        "5. Clear Todos",
-                        "6. Context",
-                        "7. Clear History",
-                        "8. API Key",
-                        "9. Exit",
-                    ],
-                },
-            ]);
-            // Select Model - Toggle between different models
-            if (action === "1. Select Model") {
-                // Fetch models from Groq API if not cached
-                if (cachedModels.length === 0) {
-                    const spinner = (0, ora_1.default)("Fetching available models from Groq...").start();
-                    cachedModels = await (0, groq_models_1.fetchGroqModels)(apiKey);
-                    spinner.succeed(`Found ${cachedModels.length} models`);
-                }
-                // Group models by owner
-                const grouped = (0, groq_models_1.groupModelsByOwner)(cachedModels);
-                const modelChoices = [];
-                // Build choices grouped by owner
-                for (const [owner, models] of grouped) {
-                    modelChoices.push(new inquirer_1.default.Separator(`─── ${owner} ───`));
-                    for (const model of models) {
-                        const isCurrent = model.id === currentModel;
-                        const ctx = (0, groq_models_1.formatContextWindow)(model.context_window);
-                        modelChoices.push({
-                            name: `${isCurrent ? "✓ " : "  "}${model.id} (${ctx} ctx)`,
-                            value: model.id,
-                            short: model.id,
+            // Show the searchable slash command menu
+            const action = await (0, slash_commands_1.showSlashCommandMenu)();
+            if (!action) {
+                // User cancelled
+                continue;
+            }
+            // Execute the selected command
+            switch (action) {
+                case "model": {
+                    // Fetch models from Groq API if not cached
+                    if (cachedModels.length === 0) {
+                        const spinner = (0, ora_1.default)("Fetching available models from Groq...").start();
+                        cachedModels = await (0, groq_models_1.fetchGroqModels)(apiKey);
+                        spinner.succeed(`Found ${cachedModels.length} models`);
+                    }
+                    // Group models by owner
+                    const grouped = (0, groq_models_1.groupModelsByOwner)(cachedModels);
+                    const modelChoices = [];
+                    // Build choices grouped by owner
+                    for (const [owner, models] of grouped) {
+                        modelChoices.push(new select_1.Separator(`─── ${owner} ───`));
+                        for (const model of models) {
+                            const isCurrent = model.id === currentModel;
+                            const ctx = (0, groq_models_1.formatContextWindow)(model.context_window);
+                            modelChoices.push({
+                                name: `${isCurrent ? "✓ " : "  "}${model.id} (${ctx} ctx)`,
+                                value: model.id,
+                            });
+                        }
+                    }
+                    try {
+                        const selectedModel = await (0, select_1.default)({
+                            message: `Select a model (current: ${currentModel}):`,
+                            choices: modelChoices,
+                            pageSize: 15,
+                            loop: false,
                         });
+                        if (selectedModel && selectedModel !== currentModel) {
+                            const oldModel = currentModel;
+                            (0, config_1.setActiveModel)(selectedModel);
+                            modelProvider.setModel(selectedModel);
+                            ui_1.log.success(`Switched to model: ${selectedModel}`);
+                            tracker_1.tracker.modelSwitch(oldModel, selectedModel, true);
+                        }
+                        else if (selectedModel === currentModel) {
+                            ui_1.log.info(`Already using ${selectedModel}`);
+                        }
+                    }
+                    catch (e) {
+                        // User cancelled with Ctrl+C
                     }
+                    break;
                 }
-                const { selectedModel } = await inquirer_1.default.prompt([
-                    {
-                        type: "list",
-                        name: "selectedModel",
-                        message: `Select a model (current: ${currentModel}):`,
-                        choices: modelChoices,
-                        pageSize: 15,
-                        loop: false,
-                    },
-                ]);
-                if (selectedModel && selectedModel !== currentModel) {
-                    const oldModel = currentModel;
-                    (0, config_1.setActiveModel)(selectedModel);
-                    modelProvider.setModel(selectedModel);
-                    ui_1.log.success(`Switched to model: ${selectedModel}`);
-                    tracker_1.tracker.modelSwitch(oldModel, selectedModel, true);
+                case "usage": {
+                    console.log("\n┌─────────────────────────────────────┐");
+                    console.log("│           MODEL INFO                │");
+                    console.log("└─────────────────────────────────────┘\n");
+                    console.log(`  Current Model: ${currentModel}`);
+                    // Find model in cache and show details
+                    if (cachedModels.length === 0) {
+                        const spinner = (0, ora_1.default)("Fetching model info...").start();
+                        cachedModels = await (0, groq_models_1.fetchGroqModels)(apiKey);
+                        spinner.stop();
+                    }
+                    const modelInfo = cachedModels.find((m) => m.id === currentModel);
+                    if (modelInfo) {
+                        console.log(`  Provider:      ${modelInfo.owned_by}`);
+                        console.log(`  Context:       ${(0, groq_models_1.formatContextWindow)(modelInfo.context_window)} tokens`);
+                        console.log(`  Status:        ${modelInfo.active ? "Active" : "Inactive"}`);
+                    }
+                    console.log("\n┌─────────────────────────────────────┐");
+                    console.log("│         TOKEN CONSUMPTION           │");
+                    console.log("└─────────────────────────────────────┘\n");
+                    const usageStats = chatHandler.getUsageStats();
+                    console.log(`  Prompt Tokens:     ${usageStats.promptTokens.toLocaleString()}`);
+                    console.log(`  Completion Tokens: ${usageStats.completionTokens.toLocaleString()}`);
+                    console.log(`  Total Tokens:      ${usageStats.totalTokens.toLocaleString()}`);
+                    console.log(`  API Requests:      ${usageStats.requestCount}`);
+                    console.log("\n┌─────────────────────────────────────┐");
+                    console.log("│          SESSION STATS              │");
+                    console.log("└─────────────────────────────────────┘\n");
+                    console.log(`  Messages:      ${chatHandler.getMessageCount()}`);
+                    console.log(`  Tools:         ${toolRegistry.count()} available`);
+                    console.log(`  Session ID:    ${tracker_1.tracker.getSessionId()}`);
+                    console.log("");
+                    break;
                 }
-                else if (selectedModel === currentModel) {
-                    ui_1.log.info(`Already using ${selectedModel}`);
+                case "tools": {
+                    console.log("\n┌─────────────────────────────────────┐");
+                    console.log("│         AVAILABLE TOOLS             │");
+                    console.log("└─────────────────────────────────────┘\n");
+                    console.log(toolRegistry.getToolDescriptions());
+                    console.log("");
+                    break;
                 }
-                continue;
-            }
-            // Usage - Show model usage and stats
-            if (action === "2. Usage") {
-                console.log("\n┌─────────────────────────────────────┐");
-                console.log("│           MODEL USAGE               │");
-                console.log("└─────────────────────────────────────┘\n");
-                console.log(`  Current Model: ${currentModel}`);
-                // Find model in cache and show details
-                if (cachedModels.length === 0) {
-                    const spinner = (0, ora_1.default)("Fetching model info...").start();
-                    cachedModels = await (0, groq_models_1.fetchGroqModels)(apiKey);
-                    spinner.stop();
+                case "todos": {
+                    const session = (0, config_2.getSessionData)();
+                    (0, ui_1.showTodoList)(session.todos);
+                    break;
                 }
-                const modelInfo = cachedModels.find((m) => m.id === currentModel);
-                if (modelInfo) {
-                    console.log(`  Provider:      ${modelInfo.owned_by}`);
-                    console.log(`  Context:       ${(0, groq_models_1.formatContextWindow)(modelInfo.context_window)} tokens`);
-                    console.log(`  Status:        ${modelInfo.active ? "Active" : "Inactive"}`);
+                case "clear-todos": {
+                    (0, config_1.clearSessionData)();
+                    ui_1.log.success("Todos cleared.");
+                    break;
+                }
+                case "context": {
+                    console.log("\n┌─────────────────────────────────────┐");
+                    console.log("│          FILE CONTEXT               │");
+                    console.log("└─────────────────────────────────────┘\n");
+                    console.log(contextMessage || "  No context loaded.");
+                    console.log("");
+                    break;
+                }
+                case "clear": {
+                    chatHandler.clearHistory();
+                    ui_1.log.success("Chat history cleared.");
+                    break;
+                }
+                case "api-key": {
+                    const { key } = await inquirer_1.default.prompt([
+                        {
+                            type: "password",
+                            name: "key",
+                            message: "Enter new API Key:",
+                            mask: "*",
+                        },
+                    ]);
+                    (0, config_1.setApiKey)(key.trim());
+                    apiKey = key.trim();
+                    modelProvider.initialize(key.trim(), modelConfig.modelId);
+                    cachedModels = []; // Clear model cache
+                    ui_1.log.success("API Key updated.");
+                    break;
+                }
+                case "exit": {
+                    process.exit(0);
                 }
-                console.log("\n┌─────────────────────────────────────┐");
-                console.log("│          SESSION STATS              │");
-                console.log("└─────────────────────────────────────┘\n");
-                console.log(`  Messages:      ${chatHandler.getMessageCount()}`);
-                console.log(`  Tools:         ${toolRegistry.count()} available`);
-                console.log(`  Session ID:    ${tracker_1.tracker.getSessionId()}`);
-                console.log("");
-                continue;
-            }
-            if (action === "3. Tools") {
-                console.log("\n┌─────────────────────────────────────┐");
-                console.log("│         AVAILABLE TOOLS             │");
-                console.log("└─────────────────────────────────────┘\n");
-                console.log(toolRegistry.getToolDescriptions());
-                console.log("");
-                continue;
-            }
-            if (action === "4. Todos") {
-                const session = (0, config_2.getSessionData)();
-                (0, ui_1.showTodoList)(session.todos);
-                continue;
-            }
-            if (action === "5. Clear Todos") {
-                (0, config_1.clearSessionData)();
-                ui_1.log.success("Todos cleared.");
-                continue;
-            }
-            if (action === "6. Context") {
-                console.log("\n┌─────────────────────────────────────┐");
-                console.log("│          FILE CONTEXT               │");
-                console.log("└─────────────────────────────────────┘\n");
-                console.log(contextMessage || "  No context loaded.");
-                console.log("");
-                continue;
-            }
-            if (action === "7. Clear History") {
-                chatHandler.clearHistory();
-                ui_1.log.success("Chat history cleared.");
-                continue;
-            }
-            if (action === "8. API Key") {
-                const { key } = await inquirer_1.default.prompt([
-                    {
-                        type: "password",
-                        name: "key",
-                        message: "Enter new API Key:",
-                        mask: "*",
-                    },
-                ]);
-                (0, config_1.setApiKey)(key.trim());
-                apiKey = key.trim();
-                modelProvider.initialize(key.trim(), modelConfig.modelId);
-                cachedModels = []; // Clear model cache
-                ui_1.log.success("API Key updated.");
-                continue;
-            }
-            if (action === "9. Exit")
-                process.exit(0);
-            continue;
-        }
-        // Handle direct slash commands
-        if (userInput.startsWith("/")) {
-            const cmd = userInput.trim().toLowerCase();
-            if (cmd === "/exit" || cmd === "/quit")
-                process.exit(0);
-            if (cmd === "/clear") {
-                chatHandler.clearHistory();
-                ui_1.log.success("History cleared.");
-                continue;
-            }
-            if (cmd === "/context") {
-                ui_1.log.info(contextMessage || "No context.");
-                continue;
-            }
-            if (cmd === "/tools") {
-                console.log("\n📦 Available Tools:\n");
-                console.log(toolRegistry.getToolDescriptions());
-                console.log("");
-                continue;
-            }
-            if (cmd === "/todos") {
-                const session = (0, config_2.getSessionData)();
-                (0, ui_1.showTodoList)(session.todos);
-                continue;
             }
-            ui_1.log.warning(`Unknown command: ${cmd}. Type "/" for menu.`);
             continue;
         }
         // Process user input with chat handler

package/dist/lib/chat-handler.js

@@ -12,6 +12,12 @@ class ChatHandler {
     constructor(toolRegistry, toolExecutor, modelProvider, initialContext) {
         this.messages = [];
         this.contextMessage = '';
+        this.usage = {
+            promptTokens: 0,
+            completionTokens: 0,
+            totalTokens: 0,
+            requestCount: 0,
+        };
         this.toolRegistry = toolRegistry;
         this.toolExecutor = toolExecutor;
         this.modelProvider = modelProvider;
@@ -92,7 +98,15 @@ When you need to perform file operations, use the appropriate tools rather than
                         if (chunk.tool_calls && chunk.tool_calls.length > 0) {
                             toolCalls = chunk.tool_calls;
                         }
+                        // Capture usage metadata from the chunk
+                        if (chunk.usage_metadata) {
+                            this.usage.promptTokens += chunk.usage_metadata.input_tokens || 0;
+                            this.usage.completionTokens += chunk.usage_metadata.output_tokens || 0;
+                            this.usage.totalTokens += chunk.usage_metadata.total_tokens || 0;
+                        }
                     }
+                    // Increment request count
+                    this.usage.requestCount++;
                     // Flush any remaining content in the formatter
                     const remaining = formatter.flush();
                     if (remaining) {
@@ -205,6 +219,12 @@ When you need to perform file operations, use the appropriate tools rather than
     getMessageCount() {
         return this.messages.length;
     }
+    /**
+     * Get usage statistics
+     */
+    getUsageStats() {
+        return { ...this.usage };
+    }
     /**
      * Update context (e.g., if file tree changes)
      */

package/dist/lib/slash-commands.js

@@ -0,0 +1,94 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SLASH_COMMANDS = void 0;
+exports.showSlashCommandMenu = showSlashCommandMenu;
+const search_1 = __importDefault(require("@inquirer/search"));
+const chalk_1 = __importDefault(require("chalk"));
+exports.SLASH_COMMANDS = [
+    {
+        name: "/model",
+        description: "Switch between available AI models",
+        shortcut: "m",
+        action: "model",
+    },
+    {
+        name: "/usage",
+        description: "Show model info and session statistics",
+        shortcut: "u",
+        action: "usage",
+    },
+    {
+        name: "/tools",
+        description: "List all available tools",
+        shortcut: "t",
+        action: "tools",
+    },
+    {
+        name: "/todos",
+        description: "Display current todo list",
+        action: "todos",
+    },
+    {
+        name: "/clear-todos",
+        description: "Clear all todos",
+        action: "clear-todos",
+    },
+    {
+        name: "/context",
+        description: "Show file context from current directory",
+        shortcut: "c",
+        action: "context",
+    },
+    {
+        name: "/clear",
+        description: "Clear chat history",
+        action: "clear",
+    },
+    {
+        name: "/api-key",
+        description: "Update your Groq API key",
+        action: "api-key",
+    },
+    {
+        name: "/exit",
+        description: "Exit the application",
+        shortcut: "q",
+        action: "exit",
+    },
+];
+function formatCommandChoice(cmd) {
+    const shortcut = cmd.shortcut ? chalk_1.default.dim(` (${cmd.shortcut})`) : "";
+    return `${chalk_1.default.cyan(cmd.name)}${shortcut}  ${chalk_1.default.dim("─")}  ${cmd.description}`;
+}
+async function showSlashCommandMenu() {
+    try {
+        const answer = await (0, search_1.default)({
+            message: "Command",
+            source: async (input) => {
+                const searchTerm = (input || "").toLowerCase().replace(/^\//, "");
+                return exports.SLASH_COMMANDS.filter((cmd) => {
+                    if (!searchTerm)
+                        return true;
+                    return (cmd.name.toLowerCase().includes(searchTerm) ||
+                        cmd.description.toLowerCase().includes(searchTerm) ||
+                        cmd.shortcut?.toLowerCase() === searchTerm);
+                }).map((cmd) => ({
+                    name: formatCommandChoice(cmd),
+                    value: cmd.action,
+                    description: cmd.description,
+                }));
+            },
+        });
+        return answer;
+    }
+    catch (error) {
+        // User cancelled (Ctrl+C)
+        if (error.message?.includes("cancelled") || error.name === "ExitPromptError") {
+            return null;
+        }
+        throw error;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prab-cli",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "AI-powered coding assistant for your terminal. Built with Groq's lightning-fast LLMs, featuring autonomous tool execution, syntax-highlighted output, and git integration.",
   "main": "dist/index.js",
   "bin": {
@@ -49,6 +49,8 @@
     "LICENSE"
   ],
   "dependencies": {
+    "@inquirer/search": "^4.1.0",
+    "@inquirer/select": "^5.0.4",
     "@langchain/core": "^1.1.11",
     "@langchain/groq": "^1.0.2",
     "@langchain/openai": "^1.2.2",