prab-cli 1.0.0

package/LICENSE

@@ -0,0 +1,15 @@
+ISC License
+
+Copyright (c) 2024 Prabhanjan Sharma
+
+Permission to use, copy, modify, and/or distribute this software for any
+purpose with or without fee is hereby granted, provided that the above
+copyright notice and this permission notice appear in all copies.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+PERFORMANCE OF THIS SOFTWARE.

package/README.md

@@ -0,0 +1,272 @@
+# Prab CLI
+
+An AI-powered coding assistant for your terminal. Built with Groq's lightning-fast LLMs, featuring autonomous tool execution, syntax-highlighted output, and seamless git integration.
+
+![Version](https://img.shields.io/npm/v/prab-cli)
+![License](https://img.shields.io/npm/l/prab-cli)
+
+## Features
+
+- **AI-Powered Coding Assistant** - Chat with AI that understands your codebase
+- **Autonomous Tool Execution** - AI can read, write, and edit files directly
+- **Git Integration** - Built-in git status, diff, commit, push, and more
+- **Syntax Highlighting** - Beautiful colored output for code blocks and diffs
+- **Multiple Models** - Switch between Groq models on the fly
+- **Task Tracking** - Built-in todo management for complex tasks
+- **Safety Checks** - Confirmation prompts for destructive operations
+- **Session Logging** - Debug and track all AI interactions
+
+## Installation
+
+```bash
+npm install -g prab-cli
+```
+
+## Quick Start
+
+1. **Get a Groq API Key** from [console.groq.com](https://console.groq.com)
+
+2. **Configure your API key:**
+```bash
+groq-cli config
+```
+
+3. **Start chatting:**
+```bash
+groq-cli
+```
+
+## Usage
+
+### Interactive Mode
+
+Simply run `groq-cli` to start an interactive session:
+
+```bash
+groq-cli
+```
+
+You'll see the welcome banner and can start chatting with the AI:
+
+```
+   ____             __      ________    ____
+   / __ \_________ _/ /_    / ____/ /   /  _/
+  / /_/ / ___/ __ `/ __ \  / /   / /    / /
+ / ____/ /  / /_/ / /_/ / / /___/ /____/ /
+/_/   /_/   \__,_/_.___/  \____/_____/___/
+
+  Active Model: llama-3.3-70b-versatile
+  Available Tools: 14
+
+> Help me refactor the authentication module
+```
+
+### Slash Commands
+
+Type `/` during a session to access the command menu:
+
+| Command | Description |
+|---------|-------------|
+| `/` | Open command menu |
+| `/exit` | Exit the CLI |
+| `/clear` | Clear chat history |
+| `/context` | Show loaded file context |
+| `/tools` | List available tools |
+| `/todos` | Show current todo list |
+
+### Command Menu Options
+
+1. **Select Model** - Switch between available Groq models
+2. **Usage** - View current model and session stats
+3. **Tools** - List all available tools
+4. **Todos** - View task list
+5. **Clear Todos** - Reset task list
+6. **Context** - Show file context
+7. **Clear History** - Reset conversation
+8. **API Key** - Update your API key
+9. **Exit** - Quit the CLI
+
+## Available Tools
+
+The AI has access to these tools for autonomous coding:
+
+### File Operations
+| Tool | Description |
+|------|-------------|
+| `read_file` | Read file contents with line numbers |
+| `write_file` | Create or overwrite files |
+| `edit_file` | Make targeted edits to existing files |
+| `glob` | Find files matching patterns |
+| `grep` | Search file contents with regex |
+
+### Git Operations
+| Tool | Description |
+|------|-------------|
+| `git_status` | Show working tree status |
+| `git_diff` | Show changes between commits |
+| `git_log` | View commit history |
+| `git_add` | Stage files for commit |
+| `git_commit` | Create a commit |
+| `git_branch` | List or create branches |
+| `git_push` | Push commits to remote |
+
+### System
+| Tool | Description |
+|------|-------------|
+| `bash` | Execute shell commands |
+| `manage_todos` | Track tasks and progress |
+
+## Colorized Output
+
+The CLI provides rich, colorized output for better readability:
+
+### Git Diff
+- Added lines in **green**
+- Removed lines in **red**
+- Hunk headers in **cyan**
+- File headers in **bold**
+
+### Code Blocks
+- Keywords highlighted in **magenta**
+- Strings in **green**
+- Numbers in **yellow**
+- Comments in **gray**
+- Function calls in **cyan**
+
+### Markdown
+- Headers in **cyan bold**
+- Inline code with **gray background**
+- Bullet points with **yellow** markers
+- Bold and italic text styling
+
+## Configuration
+
+### API Key
+
+Set your Groq API key:
+
+```bash
+# Interactive prompt
+groq-cli config
+
+# Or directly
+groq-cli config YOUR_API_KEY
+```
+
+### Reset API Key
+
+```bash
+groq-cli reset
+```
+
+### Model Management
+
+```bash
+groq-cli model
+```
+
+Options:
+- List available models
+- Show current model
+- Switch to a different model
+
+## Supported Models
+
+The CLI supports all Groq models including:
+
+- `llama-3.3-70b-versatile` (default)
+- `llama-3.1-70b-versatile`
+- `llama-3.1-8b-instant`
+- `mixtral-8x7b-32768`
+- `gemma2-9b-it`
+- And more...
+
+Models are fetched dynamically from the Groq API.
+
+## Session Logging
+
+Debug logs are stored in `~/.prab-cli/logs/`. View them with:
+
+```bash
+# View latest log
+npm run log
+
+# List all logs
+npm run log:list
+```
+
+## Examples
+
+### Read and Edit a File
+
+```
+> Read the package.json and update the version to 2.0.0
+```
+
+The AI will:
+1. Use `read_file` to view package.json
+2. Use `edit_file` to update the version
+3. Show you the diff of changes
+
+### Git Workflow
+
+```
+> Check the git status and commit all changes with a descriptive message
+```
+
+The AI will:
+1. Run `git_status` to see changes
+2. Use `git_add` to stage files
+3. Create a commit with `git_commit`
+
+### Code Search
+
+```
+> Find all files that import the 'express' module
+```
+
+The AI will use `grep` to search for import statements.
+
+## Safety Features
+
+- **Confirmation Prompts** - Destructive operations require approval
+- **Session Memory** - Remember choices for similar operations
+- **Read-Only by Default** - Tools like `read_file` and `git_status` run without confirmation
+
+## Requirements
+
+- Node.js 18+
+- npm or yarn
+- Groq API key
+
+## Development
+
+```bash
+# Clone the repo
+git clone https://github.com/prab002/prab-cli.git
+cd prab-cli
+
+# Install dependencies
+npm install
+
+# Run in development mode
+npm run dev
+
+# Build
+npm run build
+
+# Run production build
+npm start
+```
+
+## License
+
+ISC
+
+## Author
+
+Prabhanjan Sharma
+
+---
+
+Built with [Groq](https://groq.com) | [LangChain](https://langchain.com) | [Chalk](https://github.com/chalk/chalk)

package/dist/index.js

@@ -0,0 +1,374 @@
+#!/usr/bin/env node
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const commander_1 = require("commander");
+const inquirer_1 = __importDefault(require("inquirer"));
+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 ora_1 = __importDefault(require("ora"));
+// Import tool system
+const base_1 = require("./lib/tools/base");
+const executor_1 = require("./lib/tools/executor");
+const file_tools_1 = require("./lib/tools/file-tools");
+const shell_tools_1 = require("./lib/tools/shell-tools");
+const git_tools_1 = require("./lib/tools/git-tools");
+const todo_tool_1 = require("./lib/tools/todo-tool");
+// Import model system
+const groq_provider_1 = require("./lib/models/groq-provider");
+const registry_1 = require("./lib/models/registry");
+const groq_models_1 = require("./lib/groq-models");
+// Import chat handler and safety
+const chat_handler_1 = require("./lib/chat-handler");
+const safety_1 = require("./lib/safety");
+const tracker_1 = require("./lib/tracker");
+// Cache for available models
+let cachedModels = [];
+const program = new commander_1.Command();
+program
+    .name("prab-cli")
+    .description("An AI coding assistant with autonomous tool capabilities")
+    .version("2.0.0");
+// Config command
+program
+    .command("config [key]")
+    .description("Set your Groq API Key")
+    .action(async (key) => {
+    if (key) {
+        (0, config_1.setApiKey)(key.trim());
+        ui_1.log.success("API Key saved successfully!");
+        return;
+    }
+    const { inputKey } = await inquirer_1.default.prompt([
+        {
+            type: "password",
+            name: "inputKey",
+            message: "Enter your Groq API Key:",
+            mask: "*",
+        },
+    ]);
+    (0, config_1.setApiKey)(inputKey.trim());
+    ui_1.log.success("API Key saved successfully!");
+});
+// Reset command
+program
+    .command("reset")
+    .description("Clear your stored API Key")
+    .action(() => {
+    (0, config_1.clearApiKey)();
+    ui_1.log.success("API Key cleared!");
+});
+// Model management commands
+program
+    .command("model")
+    .description("Manage AI models")
+    .action(async () => {
+    const { action } = await inquirer_1.default.prompt([
+        {
+            type: "list",
+            name: "action",
+            message: "Model Management:",
+            choices: [
+                { name: "List available models", value: "list" },
+                { name: "Show current model", value: "current" },
+                { name: "Switch model", value: "switch" },
+                { name: "Cancel", value: "cancel" },
+            ],
+        },
+    ]);
+    if (action === "list") {
+        console.log("\nAvailable Models:\n");
+        console.log((0, registry_1.getModelList)());
+    }
+    else if (action === "current") {
+        const config = (0, config_1.getModelConfig)();
+        ui_1.log.info(`Current model: ${config.modelId}`);
+        ui_1.log.info(`Temperature: ${config.temperature}`);
+    }
+    else if (action === "switch") {
+        const { modelId } = await inquirer_1.default.prompt([
+            {
+                type: "input",
+                name: "modelId",
+                message: "Enter model ID:",
+            },
+        ]);
+        const validation = (0, registry_1.validateModelId)(modelId);
+        if (validation.valid) {
+            (0, config_1.setActiveModel)(modelId);
+            ui_1.log.success(`Switched to model: ${modelId}`);
+        }
+        else {
+            ui_1.log.error(validation.error || "Invalid model");
+            if (validation.suggested) {
+                ui_1.log.info(`Did you mean: ${validation.suggested}?`);
+            }
+        }
+    }
+});
+// Main interactive mode
+program.action(async () => {
+    // Check API Key
+    let apiKey = (0, config_1.getApiKey)();
+    if (!apiKey) {
+        ui_1.log.warning("No API Key found.");
+        const { key } = await inquirer_1.default.prompt([
+            {
+                type: "password",
+                name: "key",
+                message: "Please enter your Groq API Key to get started:",
+                mask: "*",
+            },
+        ]);
+        (0, config_1.setApiKey)(key.trim());
+        apiKey = key.trim();
+    }
+    // Initialize tool registry
+    const toolRegistry = new base_1.ToolRegistry();
+    toolRegistry.register(new file_tools_1.ReadFileTool());
+    toolRegistry.register(new file_tools_1.WriteFileTool());
+    toolRegistry.register(new file_tools_1.EditFileTool());
+    toolRegistry.register(new file_tools_1.GlobTool());
+    toolRegistry.register(new file_tools_1.GrepTool());
+    toolRegistry.register(new shell_tools_1.BashTool());
+    toolRegistry.register(new git_tools_1.GitStatusTool());
+    toolRegistry.register(new git_tools_1.GitAddTool());
+    toolRegistry.register(new git_tools_1.GitDiffTool());
+    toolRegistry.register(new git_tools_1.GitLogTool());
+    toolRegistry.register(new git_tools_1.GitCommitTool());
+    toolRegistry.register(new git_tools_1.GitBranchTool());
+    toolRegistry.register(new git_tools_1.GitPushTool());
+    toolRegistry.register(new todo_tool_1.TodoTool());
+    // Initialize safety checker and tool executor
+    const safetyChecker = new safety_1.SafetyChecker();
+    const toolExecutor = new executor_1.ToolExecutor(toolRegistry, safetyChecker);
+    // Initialize model provider
+    const modelConfig = (0, config_1.getModelConfig)();
+    const modelProvider = new groq_provider_1.GroqProvider(modelConfig.modelId, modelConfig.temperature);
+    try {
+        modelProvider.initialize(apiKey, modelConfig.modelId);
+        tracker_1.tracker.modelInit(modelConfig.modelId, "groq", true);
+        tracker_1.tracker.sessionStart(modelConfig.modelId, toolRegistry.count());
+    }
+    catch (e) {
+        ui_1.log.error("Failed to initialize model.");
+        tracker_1.tracker.modelInit(modelConfig.modelId, "groq", false, e.message);
+        process.exit(1);
+    }
+    // Display banner
+    (0, ui_1.banner)(modelConfig.modelId, toolRegistry.count());
+    // Context Gathering
+    const spinner = (0, ora_1.default)("Checking context...").start();
+    const isRepo = await (0, context_1.isGitRepo)();
+    let contextMessage = "";
+    if (isRepo) {
+        spinner.text = "Git repository detected. Gathering file structure...";
+        const files = await (0, context_1.getFileTree)();
+        contextMessage = `\n\nCurrent Working Directory Context:\nFile List:\n${files.join("\n")}`;
+        spinner.succeed(`Context loaded (${files.length} files detected).`);
+    }
+    else {
+        spinner.succeed("No Git repository detected. Running in standalone mode.");
+    }
+    // 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!');
+    // Display any existing todos
+    const session = (0, config_2.getSessionData)();
+    if (session.todos && session.todos.length > 0) {
+        (0, ui_1.showTodoList)(session.todos);
+    }
+    // Chat Loop
+    while (true) {
+        const { userInput } = await inquirer_1.default.prompt([
+            {
+                type: "input",
+                name: "userInput",
+                message: ">",
+            },
+        ]);
+        // Handle Slash Commands Menu
+        if (userInput.trim() === "/") {
+            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,
+                        });
+                    }
+                }
+                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);
+                }
+                else if (selectedModel === currentModel) {
+                    ui_1.log.info(`Already using ${selectedModel}`);
+                }
+                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();
+                }
+                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("│          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
+        await chatHandler.processUserInput(userInput);
+    }
+});
+program.parse(process.argv);