omni-agent-cli 2.0.0

package/README.md

@@ -0,0 +1,161 @@
+# ◈ OmniAgent — Universal AI CLI Agent
+
+A powerful, beautiful CLI agent that works with **any AI provider** and has full filesystem + shell capabilities.
+
+```
+  ██████╗ ███╗   ███╗███╗   ██╗██╗     ███████╗ ██████╗ ███████╗███╗   ██╗████████╗
+ ██╔═══██╗████╗ ████║████╗  ██║██║     ██╔════╝██╔════╝ ██╔════╝████╗  ██║╚══██╔══╝
+ ██║   ██║██╔████╔██║██╔██╗ ██║██║     ███████╗██║  ███╗█████╗  ██╔██╗ ██║   ██║   
+ ██║   ██║██║╚██╔╝██║██║╚██╗██║██║     ╚════██║██║   ██║██╔══╝  ██║╚██╗██║   ██║   
+ ╚██████╔╝██║ ╚═╝ ██║██║ ╚████║██║     ███████║╚██████╔╝███████╗██║ ╚████║   ██║   
+  ╚═════╝ ╚═╝     ╚═╝╚═╝  ╚═══╝╚═╝     ╚══════╝ ╚═════╝ ╚══════╝╚═╝  ╚═══╝   ╚═╝   
+```
+
+---
+
+## 🚀 Quick Start
+
+```bash
+# Install dependencies
+npm install
+
+# First run — guided setup wizard
+node index.js
+
+# Or set up separately
+node index.js --setup
+```
+
+---
+
+## 🔌 Supported Providers
+
+| Provider | Format | Notes |
+|----------|--------|-------|
+| **Anthropic** (Claude) | Native | claude-opus, sonnet, haiku |
+| **OpenAI** | OpenAI | gpt-4o, o1, o3-mini |
+| **Groq** | OpenAI | Ultra fast inference |
+| **xAI** (Grok) | OpenAI | grok-2, grok-beta |
+| **DeepSeek** | OpenAI | deepseek-chat, reasoner |
+| **Mistral AI** | OpenAI | mistral-large |
+| **Together AI** | OpenAI | Llama, Qwen, etc. |
+| **OpenRouter** | OpenAI | All models via one key |
+| **io.net** | OpenAI | Decentralized inference |
+| **Ollama** | OpenAI | Local models |
+| **Custom** | OpenAI | Any OpenAI-compatible API (kiai.ai, etc.) |
+
+---
+
+## 💻 Usage
+
+```bash
+# Start in current directory
+node index.js
+
+# Start in specific directory
+node index.js /path/to/project
+
+# Global install (optional)
+npm link
+omni-agent /path/to/project
+```
+
+---
+
+## ⌨️ Commands
+
+| Command | Description |
+|---------|-------------|
+| `/help` | Show all commands |
+| `/clear` | Clear screen + conversation history |
+| `/stats` | Show current session statistics |
+| `/cd <dir>` | Change working directory |
+| `/model <name>` | Switch model on the fly |
+| `/setup` | Reconfigure provider/API key/model |
+| `/reset` | Start new conversation (clear history) |
+| `/history` | Show conversation history |
+| `/exit` | Exit and show session statistics |
+
+---
+
+## 🛠️ Agent Tools
+
+The agent has access to **13 filesystem and shell tools**:
+
+| Tool | Description |
+|------|-------------|
+| `list_directory` | List files with metadata |
+| `read_file` | Read file content |
+| `write_file` | Create/overwrite files |
+| `append_to_file` | Append to files |
+| `patch_file` | Surgically edit text in files |
+| `delete_path` | Delete files/directories |
+| `copy_path` | Copy files |
+| `move_path` | Move/rename files |
+| `create_directory` | Create directories recursively |
+| `get_file_info` | Get file metadata |
+| `find_files` | Find files by glob pattern |
+| `search_in_files` | Grep search in files |
+| `execute_command` | Run any shell command |
+
+---
+
+## ⚡ Token Efficiency
+
+OmniAgent uses **parallel tool calling** — when the AI needs multiple independent pieces of information, it fetches them ALL in a single API call instead of making sequential requests.
+
+Example: "List the src/ folder and read package.json"
+- ❌ Naive: 3 API calls (list → read → answer)  
+- ✅ OmniAgent: 2 API calls (list+read simultaneously → answer)
+
+---
+
+## 📊 Session Statistics
+
+On exit (or `/stats`), OmniAgent shows:
+- Session duration
+- Messages sent/received  
+- API requests made
+- Input/output token counts
+- Estimated cost
+- Tool usage breakdown with visual bars
+
+---
+
+## ⚙️ Config File
+
+Config is saved to `~/.omni-agent.json`:
+
+```json
+{
+  "providerKey": "groq",
+  "baseURL": "https://api.groq.com/openai/v1",
+  "apiKey": "your-key-here",
+  "model": "llama-3.3-70b-versatile",
+  "format": "openai",
+  "maxTokens": 8192,
+  "temperature": 0.3
+}
+```
+
+---
+
+## 💡 Example Prompts
+
+```
+⟩ Refactor all the .js files in src/ to use async/await instead of callbacks
+
+⟩ Find all TODO comments in this project and create a TODO.md file
+
+⟩ Run the tests and fix any failures
+
+⟩ Create a REST API with Express for a simple todo app
+
+⟩ Analyze this codebase and suggest improvements
+
+⟩ Set up a Python virtual environment and install requirements.txt
+```
+
+---
+
+*Built with Node.js · chalk · gradient-string · figlet · ora · inquirer · boxen*

package/index.js

@@ -0,0 +1,364 @@
+#!/usr/bin/env node
+
+import { createInterface } from 'readline';
+import chalk from 'chalk';
+import path from 'path';
+import { existsSync } from 'fs';
+
+import {
+  displayBanner, displayHelp, displayMessage,
+  displayToolCall, displayToolResult, displayError,
+  displayStats, createSpinner, displayThinking,
+  displayFilePreview, displayConfirmPrompt, C,
+} from './src/ui.js';
+import { loadConfig, saveConfig, setupWizard, PROVIDERS } from './src/config.js';
+import { Agent } from './src/agent.js';
+import { SessionStats } from './src/stats.js';
+import { fetchProviderModels, fetchAllProviderModels, formatModelList, formatAllModels } from './src/model-fetcher.js';
+
+// ─── Main ──────────────────────────────────────────────────────────────────────
+async function main() {
+  await displayBanner();
+
+  // Config
+  let config = loadConfig();
+  const needsSetup = !config || process.argv.includes('--setup') || process.argv.includes('-s');
+  if (needsSetup) {
+    config = await setupWizard(config);
+    saveConfig(config);
+    if (process.argv.includes('--setup') || process.argv.includes('-s')) {
+      console.log(chalk.green('\n  ✅ Setup complete. Run: node index.js\n'));
+      process.exit(0);
+    }
+  }
+
+  // Working directory
+  const argDir = process.argv.find((a) => !a.startsWith('-') && a !== process.argv[0] && a !== process.argv[1]);
+  let workdir = argDir ? path.resolve(argDir) : process.cwd();
+  if (argDir && !existsSync(workdir)) { displayError(`Dir not found: ${workdir}`); workdir = process.cwd(); }
+
+  // Agent + Stats
+  const stats = new SessionStats();
+  const agent = new Agent(config, stats);
+  agent.setWorkdir(workdir);
+
+  // Confirmation mode (can be toggled with /confirm off)
+  let confirmEnabled = true;
+  // Store last fetched model list for selection by number
+  let lastModelList = [];
+
+  // Info bar
+  console.log(
+    `  📁 ${C.primary('Dir:')}    ${C.dim(workdir)}\n` +
+    `  🤖 ${C.secondary('Provider:')} ${chalk.magenta(config.providerKey)} ${C.dim('›')} ${C.accent(config.model)}\n` +
+    `  💡 ${C.dim('Type')} ${chalk.white('/help')} ${C.dim('for commands, /exit to quit')}\n` +
+    C.dim('  ' + '─'.repeat(56) + '\n')
+  );
+
+  // ── Readline ──────────────────────────────────────────────────────────────────
+  const rl = createInterface({ input: process.stdin, output: process.stdout, terminal: true });
+  let isProcessing = false;
+
+  // ── Readline helper to read a single key/line ─────────────────────────────────
+  const readLine = (prompt) => new Promise((resolve) => {
+    rl.question(prompt, resolve);
+  });
+
+  const readKey = () => new Promise((resolve) => {
+    process.stdin.setRawMode?.(true);
+    process.stdin.resume();
+    process.stdin.once('data', (buf) => {
+      process.stdin.setRawMode?.(false);
+      resolve(buf.toString());
+    });
+  });
+
+  // ── Exit ──────────────────────────────────────────────────────────────────────
+  const handleExit = async () => {
+    console.log('\n');
+    await displayStats(stats.getStats());
+    rl.close();
+    process.exit(0);
+  };
+  rl.on('close', async () => { if (!isProcessing) await displayStats(stats.getStats()); process.exit(0); });
+  process.on('SIGINT', handleExit);
+  process.on('SIGTERM', handleExit);
+
+  // Safety net — prevents any unhandled rejection from killing the process
+  process.on('unhandledRejection', (err) => {
+    try { displayError('Unhandled error: ' + (err?.message || err)); } catch {}
+    isProcessing = false;
+    showPrompt();
+  });
+  process.on('uncaughtException', (err) => {
+    try { displayError('Uncaught error: ' + (err?.message || err)); } catch {}
+    isProcessing = false;
+    showPrompt();
+  });
+
+  // ── Prompt ────────────────────────────────────────────────────────────────────
+  const showPrompt = () => {
+    const dirName = path.basename(workdir);
+    const confirmTag = confirmEnabled ? C.dim('') : C.warning(' [no-confirm]');
+    rl.question(`\n  ${C.dim('[')}${C.primary(dirName)}${C.dim(']')}${confirmTag} ${C.secondary('⟩')} `, handleInput);
+  };
+
+  // ── Confirmation handler ──────────────────────────────────────────────────────
+  const handleConfirmation = async (toolName, input) => {
+    displayFilePreview(toolName, input);
+    await displayConfirmPrompt(toolName);
+
+    process.stdout.write('\n  ' + C.dim('Waiting for your choice [1/2/3/4]... '));
+
+    return new Promise((resolve) => {
+      // Try raw mode first (single keypress)
+      if (process.stdin.setRawMode) {
+        process.stdin.setRawMode(true);
+        process.stdin.resume();
+        const onKey = (buf) => {
+          const key = buf.toString();
+          process.stdin.setRawMode(false);
+          process.stdin.removeListener('data', onKey);
+          if (key === '1' || key === '\r' || key === '\n') {
+            console.log(C.success('1') + C.dim(' — Yes, once'));
+            resolve('yes');
+          } else if (key === '2') {
+            console.log(C.success('2') + C.dim(' — Yes, allow always'));
+            resolve('always');
+          } else if (key === '3' || key === '\x1b') {
+            console.log(C.warning('3') + C.dim(' — Skipped'));
+            resolve('skip');
+          } else if (key === '4') {
+            console.log(C.error('4') + C.dim(' — Suggest changes'));
+            resolve('modify');
+          } else {
+            console.log(C.success('1') + C.dim(' — Yes (default)'));
+            resolve('yes');
+          }
+        };
+        process.stdin.on('data', onKey);
+      } else {
+        // Fallback: readline for Termux/environments without raw mode
+        rl.question('', (ans) => {
+          const n = ans.trim();
+          if (n === '2')      resolve('always');
+          else if (n === '3') resolve('skip');
+          else if (n === '4') resolve('modify');
+          else                resolve('yes');
+        });
+      }
+    });
+  };
+
+  // ── Input handler ─────────────────────────────────────────────────────────────
+  const handleInput = async (input) => {
+    const trimmed = input.trim();
+    if (!trimmed) { showPrompt(); return; }
+
+    // Bare number input after /models → select model by number
+    if (/^\d+$/.test(trimmed) && lastModelList.length > 0) {
+      const n = parseInt(trimmed, 10);
+      if (n >= 1 && n <= lastModelList.length) {
+        config.model = lastModelList[n - 1]; // full name, not truncated
+        agent.updateConfig(config); saveConfig(config);
+        console.log(C.success('\n  Model: ' + C.accent(config.model) + '\n'));
+      } else {
+        console.log(C.warning('\n  Number out of range (1-' + lastModelList.length + ')\n'));
+      }
+      showPrompt(); return;
+    }
+
+    // ── Commands ───────────────────────────────────────────────────────────────
+    if (trimmed === '/exit' || trimmed === '/quit' || trimmed === '/q') { await handleExit(); return; }
+    if (trimmed === '/help'  || trimmed === '/h')  { displayHelp();   showPrompt(); return; }
+    if (trimmed === '/clear' || trimmed === '/cls') {
+      await displayBanner(); agent.clearHistory();
+      console.log(C.dim('  🔄 Cleared.\n')); showPrompt(); return;
+    }
+    if (trimmed === '/stats') { await displayStats(stats.getStats()); showPrompt(); return; }
+    if (trimmed === '/reset' || trimmed === '/new') {
+      agent.clearHistory();
+      console.log(C.success('\n  ✅ New conversation.\n')); showPrompt(); return;
+    }
+    if (trimmed.startsWith('/cd ')) {
+      const newDir = trimmed.slice(4).trim();
+      const resolved = path.isAbsolute(newDir) ? newDir : path.resolve(workdir, newDir);
+      if (existsSync(resolved)) {
+        workdir = resolved; agent.setWorkdir(workdir); process.chdir(workdir);
+        console.log(C.success(`\n  ✅ ${C.primary(workdir)}\n`));
+      } else { displayError(`Not found: ${resolved}`); }
+      showPrompt(); return;
+    }
+    if (trimmed === '/setup') {
+      config = await setupWizard(config); saveConfig(config); agent.updateConfig(config);
+      showPrompt(); return;
+    }
+    if (trimmed.startsWith('/model ')) {
+      const modelArg = trimmed.slice(7).trim();
+      // Support /model 3 (pick by number from last /models list)
+      const asNum = parseInt(modelArg, 10);
+      if (!isNaN(asNum) && asNum >= 1 && lastModelList.length >= asNum) {
+        config.model = lastModelList[asNum - 1];
+      } else {
+        config.model = modelArg;
+      }
+      agent.updateConfig(config); saveConfig(config);
+      console.log(C.success(`\n  ✅ Model: ${C.accent(config.model)}\n`)); showPrompt(); return;
+    }
+    if (trimmed.startsWith('/confirm')) {
+      const val = trimmed.split(' ')[1];
+      if (val === 'off') { confirmEnabled = false; agent.allowAllTools = true; console.log(C.warning('\n  ⚠ Confirmations OFF — agent will run freely\n')); }
+      else               { confirmEnabled = true;  agent.allowAllTools = false; console.log(C.success('\n  ✅ Confirmations ON\n')); }
+      showPrompt(); return;
+    }
+    if (trimmed === '/models' || trimmed.startsWith('/models ')) {
+      const arg = trimmed.slice(7).trim();
+
+      if (arg === 'all') {
+        process.stdout.write(C.dim('  Fetching all providers...\n'));
+        const apiConfigs = { [config.providerKey]: { apiKey: config.apiKey, baseURL: config.baseURL } };
+        let allResults;
+        try { allResults = await fetchAllProviderModels(apiConfigs); }
+        catch(e) { displayError(e.message); showPrompt(); return; }
+        console.log('\n' + C.primary('  -- ALL PROVIDERS & MODELS --') + '\n');
+        console.log(formatAllModels(allResults));
+        lastModelList = Object.values(allResults).flatMap(r => r.models);
+        console.log('\n' + C.dim('  /model <number> to switch') + '\n');
+        showPrompt(); return;
+      }
+
+      const targetKey = arg || config.providerKey;
+      const provider = PROVIDERS[targetKey];
+      if (!provider) {
+        displayError('Unknown: ' + targetKey + ' | Available: ' + Object.keys(PROVIDERS).join(', '));
+        showPrompt(); return;
+      }
+      process.stdout.write(C.dim('  Fetching ' + provider.label + '...\n'));
+      let modResult;
+      try {
+        const isSame = (targetKey === config.providerKey);
+        modResult = await fetchProviderModels(
+          targetKey,
+          isSame ? config.apiKey  : null,
+          isSame ? config.baseURL : provider.baseURL
+        );
+      } catch(e) { displayError(e.message); showPrompt(); return; }
+      console.log('\n' + formatModelList(targetKey, modResult) + '\n');
+      lastModelList = modResult.models;
+      if (modResult.models.length > 0) console.log(C.dim('  /model <number> to switch\n'));
+      showPrompt(); return;
+    }
+    if (trimmed === '/providers') {
+      console.log('\n');
+      for (const [key, p] of Object.entries(PROVIDERS)) {
+        const active = key === config.providerKey;
+        console.log(`  ${active ? C.accent(key.padEnd(14)) : C.dim(key.padEnd(14))} ${p.label}${active ? C.success(' ◄') : ''}`);
+      }
+      console.log('\n' + C.dim('  /setup to switch\n')); showPrompt(); return;
+    }
+    if (trimmed === '/history') {
+      for (const m of agent.history) {
+        const preview = typeof m.content === 'string' ? m.content.slice(0, 80).replace(/\n/g,' ') : '[structured]';
+        console.log(`  ${C.accent(m.role.padEnd(12))} ${C.dim(preview)}`);
+      }
+      console.log(''); showPrompt(); return;
+    }
+
+    // ── Send to agent ──────────────────────────────────────────────────────────
+    isProcessing = true;
+    stats.addMessage('user');
+
+    process.stdout.write(C.dim('  ⟳ Thinking...')  );
+    let _dotTimer = setInterval(() => process.stdout.write(C.dim('.')), 1500);
+    let _stopped = false;
+    const spinner = { stop: () => { if(_stopped) return; _stopped=true; clearInterval(_dotTimer); process.stdout.write('\n'); }, text: '', start: () => {} };
+
+    try {
+      const finalContent = await agent.chat(trimmed, workdir, async (event) => {
+
+        // ── Thinking bullets ─────────────────────────────────────────────────
+        if (event.type === 'thinking') {
+          spinner.stop();
+          displayThinking(event.bullets);
+          spinner.start();
+          spinner.text = C.dim('Planning…');
+          return;
+        }
+
+        // ── Confirmation required ────────────────────────────────────────────
+        if (event.type === 'confirm_needed') {
+          spinner.stop();
+          if (confirmEnabled) {
+            const decision = await handleConfirmation(event.tool, event.input);
+            event.resolve(decision);
+          } else {
+            // Confirmations off — auto approve, but still show preview
+            displayFilePreview(event.tool, event.input);
+            console.log(C.dim('  ✓ Auto-approved (confirm off)\n'));
+            event.resolve('yes');
+          }
+          spinner.start();
+          spinner.text = C.dim('Executing…');
+          return;
+        }
+
+        // ── Feedback request (option 4) ──────────────────────────────────────
+        if (event.type === 'request_feedback') {
+          spinner.stop();
+          const fb = await readLine('\n  ' + C.accent('What should be changed? ') + C.dim('→ '));
+          event.resolve(fb);
+          spinner.start();
+          return;
+        }
+
+        // ── Tool start (non-confirm tools) ───────────────────────────────────
+        if (event.type === 'tool_start') {
+          spinner.stop();
+          displayToolCall(event.tool, event.input);
+          spinner.start();
+          spinner.text = C.dim(`Running ${event.tool}…`);
+          return;
+        }
+
+        // ── Tool done ────────────────────────────────────────────────────────
+        if (event.type === 'tool_done') {
+          spinner.stop();
+          displayToolResult(event.tool, event.result, event.durationMs);
+          spinner.start();
+          spinner.text = C.dim('Thinking…');
+          return;
+        }
+
+        // ── Token counter ────────────────────────────────────────────────────
+        if (event.type === 'tokens') {
+          stats.addTokens(event.inputTokens || 0, event.outputTokens || 0);
+          const total = (event.inputTokens || 0) + (event.outputTokens || 0);
+          spinner.text = C.dim(`Thinking… `) + C.dim(`[${total.toLocaleString()} tok]`);
+        }
+      });
+
+      spinner.stop();
+      stats.addMessage('assistant');
+      displayMessage('assistant', finalContent);
+
+    } catch (err) {
+      spinner.stop();
+      stats.addError();
+      displayError(err.message || 'Unexpected error');
+      if (err.message?.includes('401') || err.message?.includes('Unauthorized'))
+        console.log(C.dim(`\n  💡 Check API key: /setup\n`));
+      else if (err.message?.includes('model'))
+        console.log(C.dim(`\n  💡 Try: /model <name>\n`));
+    }
+
+    isProcessing = false;
+    showPrompt();
+  };
+
+  showPrompt();
+}
+
+main().catch((err) => {
+  console.error(chalk.red('\n  Fatal:'), err.message);
+  process.exit(1);
+});

package/package.json

@@ -0,0 +1,30 @@
+{
+  "name": "omni-agent-cli",
+  "version": "2.0.0",
+  "description": "Universal AI CLI Agent — any provider, any model, full filesystem power",
+  "type": "module",
+  "bin": {
+    "omni-agent": "./index.js",
+    "oa": "./index.js"
+  },
+  "scripts": {
+    "start": "node index.js",
+    "setup": "node index.js --setup"
+  },
+  "dependencies": {
+    "axios": "^1.7.2",
+    "boxen": "^7.1.1",
+    "chalk": "^5.3.0",
+    "cli-spinners": "^2.9.2",
+    "glob": "^11.0.0",
+    "gradient-string": "^2.0.2",
+    "inquirer": "^9.3.7",
+    "mime-types": "^2.1.35",
+    "ora": "^8.1.1",
+    "strip-ansi": "^7.1.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "preferGlobal": true
+}