osai-agent 4.1.14 → 4.2.2

package/README.md

@@ -167,6 +167,7 @@ Upgrade to Pro for higher daily limits (500 msgs/day, 30 req/min, 5 connections)
 osai-agent config                       Show current configuration
 osai-agent config set-os <os>           Override OS detection
 osai-agent skills                       Manage installed skills
+osai-agent account delete               Permanently delete your account and all data
 osai-agent stop-subagent                Stop a running sub-agent
 osai-agent --version                    Show version
 osai-agent --help                       Show help

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "osai-agent",
-  "version": "4.1.14",
+  "version": "4.2.2",
   "type": "module",
   "description": "OS AI Agent - YOUR AI AGENT",
   "main": "src/index.js",

package/src/agent/context.js

@@ -1,9 +1,4 @@
-// =============================================================================
-// OS AI Agent — Context Builder
-// =============================================================================
-// Builds structured context for the agent loop, including device information,
-// OS detection, and device capabilities. Used primarily in network/SSH mode.
-// =============================================================================
+// Context builder — device info, OS, capabilities
 
 /**
  * Build a structured context object for the agent

package/src/agent/prompt.js

@@ -1,6 +1,4 @@
-// =============================================================================
-// OS AI Agent — Agent Prompt Builder (v4.0 — Coding Mode + Anti-Hallucination)
-// =============================================================================
+// Agent prompt builder — mode-specific prompts
 
 import fs from 'fs/promises';
 import path from 'path';

package/src/agent/react-loop.js

@@ -1,9 +1,4 @@
-// =============================================================================
-// OS AI Agent — Agent Loop (v4.0 — Coding Mode + Todos + Web Tools)
-// =============================================================================
-// Core agent loop: GOAL -> PLAN -> EXECUTE -> OBSERVE -> REFLECT -> REPEAT
-// Supports GENERAL, NETWORK, and CODING modes with full tool dispatch.
-// =============================================================================
+// Agent loop — GOAL→PLAN→EXEC→OBSERVE→REFLECT
 
 import {
   executeLocal, readFile, writeFile, editFile, listDir, detectOS, getSystemInfo, resolvePath,

package/src/agent/subagent.js

@@ -1,9 +1,4 @@
-// =============================================================================
-// OS AI Agent — Subagent Runner (max 1 concurrent)
-// =============================================================================
-// Spawns a read-only child agent for exploration. Results are returned to the
-// parent via structured [SUBAGENT_RESULT] tool output.
-// =============================================================================
+// Subagent runner — read-only child agent (max 1)
 
 import { AgentLoop } from './react-loop.js';
 import { SUBAGENT_MAX_ITERATIONS } from '../utils/constants.js';

package/src/commands/account.js

@@ -0,0 +1,107 @@
+import Conf from 'conf';
+import chalk from 'chalk';
+import boxen from 'boxen';
+import ora from 'ora';
+import { createInterface } from 'node:readline';
+import { rmSync, existsSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import { toHttpUrl, DEFAULT_SERVER_URL } from '../services/server-url.js';
+import { APP_VERSION } from '../utils/constants.js';
+
+export const deleteAccount = async ({ server: serverArg }) => {
+  const config = new Conf({ projectName: 'osai-agent' });
+  const token = config.get('token');
+  const serverUrl = serverArg || config.get('server') || DEFAULT_SERVER_URL;
+  const httpUrl = toHttpUrl(serverUrl);
+
+  const headerContent = [
+    ` ${chalk.hex('#4a9eff').bold('OS AI AGENT')} ${chalk.gray(`v${APP_VERSION}`)}`,
+    '',
+    ` ${chalk.hex('#8ab4d8')('Your AI sysadmin, network engineer & senior developer')}`,
+    '',
+    ` ${token ? chalk.green('●') : chalk.red('○')} ${chalk.hex('#8ab4d8')('Auth:')} ${chalk.white(token ? 'Authenticated' : 'Not authenticated')}`,
+  ].join('\n');
+
+  console.log(boxen(headerContent, {
+    padding: { left: 3, right: 3 },
+    margin: { top: 1, bottom: 1 },
+    borderStyle: 'round',
+    borderColor: '#4a9eff',
+    float: 'center',
+  }));
+
+  if (!token) {
+    console.log(chalk.red('\nYou need to be logged in to delete your account.'));
+    console.log(chalk.yellow('Run "osai-agent register" or "osai-agent login" first.\n'));
+    process.exit(1);
+  }
+
+  console.log(boxen(
+    chalk.hex('#f7768e').bold('⚠  WARNING') +
+    chalk.hex('#c0caf5')('\n\nThis will permanently delete your account and all associated data.') +
+    chalk.hex('#c0caf5')('\nThis includes: devices, logs, sessions, provider settings, pro key, and more.') +
+    chalk.hex('#ff9e64')('\n\nThis action CANNOT be undone.') +
+    chalk.hex('#9aa5ce')('\n\nYou will not be able to create a new account') +
+    chalk.hex('#9aa5ce')('\nwith the same email or from this machine for 24 hours.'),
+    {
+      padding: 2,
+      borderStyle: 'round',
+      borderColor: '#f7768e',
+      textAlignment: 'center',
+    }
+  ));
+
+  const rl = createInterface({ input: process.stdin, output: process.stdout });
+  const answer = await new Promise((resolve) => {
+    rl.question(chalk.hex('#ff9e64').bold('\nType "yes" to confirm: '), resolve);
+  });
+  rl.close();
+
+  if (answer.trim().toLowerCase() !== 'yes') {
+    console.log(chalk.yellow('\nAccount deletion cancelled.\n'));
+    process.exit(0);
+  }
+
+  const spinner = ora({ text: 'Deleting account...', color: 'red' }).start();
+
+  try {
+    const response = await fetch(`${httpUrl}/auth/account`, {
+      method: 'DELETE',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${token}`,
+      },
+    });
+
+    if (!response.ok) {
+      const errorData = await response.json().catch(() => null);
+      const errorMsg = errorData?.error || response.statusText;
+      spinner.fail(`Failed to delete account: ${response.status} - ${errorMsg}`);
+      process.exit(1);
+    }
+
+    // Clear local config
+    config.clear();
+
+    // Remove entire config directory
+    const configDir = join(homedir(), '.config', 'osai-agent');
+    if (existsSync(configDir)) {
+      try { rmSync(configDir, { recursive: true, force: true }); } catch {}
+    }
+
+    spinner.succeed('Account deleted successfully');
+
+    console.log(boxen(
+      chalk.hex('#7aa2f7').bold('  Account Deleted') +
+      chalk.hex('#c0caf5')('\n\n  All your data has been permanently removed.') +
+      chalk.hex('#9aa5ce')('\n  You will be able to create a new account') +
+      chalk.hex('#9aa5ce')('\n  with the same email or from this machine') +
+      chalk.hex('#9aa5ce')('\n  after 24 hours.'),
+      { padding: 2, borderStyle: 'round', borderColor: '#4a9eff', textAlignment: 'center' }
+    ));
+  } catch (err) {
+    spinner.fail(`Failed to delete account: ${err.message}`);
+    process.exit(1);
+  }
+};

package/src/commands/admin.js

@@ -0,0 +1,82 @@
+import chalk from 'chalk';
+import ora from 'ora';
+import { createInterface } from 'node:readline';
+import { toHttpUrl, DEFAULT_SERVER_URL } from '../services/server-url.js';
+import Conf from 'conf';
+
+const promptSecret = () => new Promise((resolve) => {
+  const rl = createInterface({ input: process.stdin, output: process.stdout });
+  rl.question(chalk.hex('#ff9e64')('Enter admin secret: '), (answer) => {
+    rl.close();
+    resolve(answer.trim());
+  });
+});
+
+const adminFetch = async (serverArg, path, method = 'GET', body = null) => {
+  const config = new Conf({ projectName: 'osai-agent' });
+  const serverUrl = serverArg || config.get('server') || DEFAULT_SERVER_URL;
+  const httpUrl = toHttpUrl(serverUrl);
+
+  const secret = await promptSecret();
+  if (!secret) {
+    console.log(chalk.red('Admin secret is required.'));
+    process.exit(1);
+  }
+
+  const options = {
+    method,
+    headers: { 'Authorization': `Bearer ${secret}` },
+  };
+  if (body) {
+    options.headers['Content-Type'] = 'application/json';
+    options.body = JSON.stringify(body);
+  }
+
+  const response = await fetch(`${httpUrl}/auth${path}`, options);
+  const data = await response.json().catch(() => null);
+
+  if (!response.ok) {
+    console.log(chalk.red(`Error ${response.status}: ${data?.error || response.statusText}`));
+    process.exit(1);
+  }
+
+  return data;
+};
+
+export const listUsers = async ({ server }) => {
+  const data = await adminFetch(server, '/admin/users');
+  const users = data.users || [];
+
+  if (users.length === 0) {
+    console.log(chalk.yellow('No users found.'));
+    return;
+  }
+
+  console.log(chalk.hex('#4a9eff').bold(`\n  Users (${users.length}):\n`));
+  for (const u of users) {
+    const statusBadge = u.status === 'disabled'
+      ? chalk.red('disabled')
+      : chalk.green('active');
+    const planBadge = u.plan === 'pro' ? chalk.hex('#ff9e64')('pro') : chalk.gray('free');
+    const created = new Date(u.created_at).toLocaleDateString();
+    const machines = (u.machine_ids || []).length;
+    console.log(
+      `  ${chalk.white(u.email)}` +
+      `  ${statusBadge}` +
+      `  ${planBadge}` +
+      `  ${chalk.gray(`${machines} machine(s)`)}` +
+      `  ${chalk.gray(`created ${created}`)}`
+    );
+  }
+  console.log();
+};
+
+export const disableUser = async ({ server, email }) => {
+  const data = await adminFetch(server, '/admin/disable-user', 'POST', { email });
+  const spinner = ora().succeed(chalk.green(`User disabled: ${data.email}`));
+};
+
+export const enableUser = async ({ server, email }) => {
+  const data = await adminFetch(server, '/admin/enable-user', 'POST', { email });
+  const spinner = ora().succeed(chalk.green(`User enabled: ${data.email}`));
+};

package/src/index.js

@@ -25,6 +25,8 @@ import { manageSkills } from './commands/skills.js';
 import { handleMcpCommand } from './commands/mcp.js';
 import { stopSubagent } from './commands/stop-subagent.js';
 import { claimProKey } from './commands/pro.js';
+import { deleteAccount } from './commands/account.js';
+import { listUsers, disableUser, enableUser } from './commands/admin.js';
 import minimist from 'minimist';
 import updateNotifier from 'update-notifier';
 import { createRequire } from 'module';
@@ -133,6 +135,37 @@ switch (cmd) {
     await claimProKey({ key: args.key || args._[1], server: args.server });
     break;
 
+  case 'account':
+    if (args._[1] === 'delete') {
+      await deleteAccount({ server: args.server });
+    } else {
+      console.error('Usage: osai-agent account delete');
+      process.exit(1);
+    }
+    break;
+
+  case 'admin':
+    if (args._[1] === 'users') {
+      await listUsers({ server: args.server });
+    } else if (args._[1] === 'disable') {
+      if (!args._[2]) { console.error('Usage: osai-agent admin disable <email>'); process.exit(1); }
+      await disableUser({ server: args.server, email: args._[2] });
+    } else if (args._[1] === 'enable') {
+      if (!args._[2]) { console.error('Usage: osai-agent admin enable <email>'); process.exit(1); }
+      await enableUser({ server: args.server, email: args._[2] });
+    } else {
+      console.log();
+      console.log('Usage: osai-agent admin <subcommand> [args]');
+      console.log();
+      console.log('Subcommands:');
+      console.log('  users                     List all users');
+      console.log('  disable <email>           Disable a user account');
+      console.log('  enable <email>            Re-enable a user account');
+      console.log();
+      process.exit(1);
+    }
+    break;
+
   default:
     if (cmd) {
       console.error(`Unknown command: ${cmd}`);
@@ -156,6 +189,7 @@ switch (cmd) {
     console.log('  mcp             Manage MCP servers');
     console.log('  stop-subagent   Stop a running subagent');
     console.log('  pro             Activate a pro access code');
+    console.log('  account delete  Permanently delete your account and all data');
     console.log();
     process.exit(cmd ? 1 : 0);
 }

package/src/memory/store.js

@@ -1,16 +1,4 @@
-// =============================================================================
-// OS AI Agent — Memory Store
-// =============================================================================
-// Lightweight JSON-based memory system that stores actions, results, and
-// failures across sessions. Enables the agent to learn from past interactions.
-// Implements a 6-layer memory model inspired by Claude Code's architecture:
-//   1. System Prompt (static)
-//   2. Session Messages (current conversation)
-//   3. Compact History (auto-trimmed)
-//   4. Project Knowledge (instructions.md)
-//   5. Persistent Memory (this store - actions/failures/preferences)
-//   6. Semantic Search (future — keyword-based for now)
-// =============================================================================
+// Memory store — persistent action/result history
 import fs from 'fs/promises';
 import path from 'path';
 import os from 'os';

package/src/parser/dependencies.js

@@ -1,9 +1,4 @@
-// =============================================================================
-// OS AI Agent — Dependency Analysis Utilities
-// =============================================================================
-// Pure functions for extracting imports and exports from source code.
-// Used by the GET_DEPENDENCIES tool and dependency auto-resolution.
-// =============================================================================
+// Dependency analysis — import/export extractor
 
 /**
  * Extract exported symbol names from JS/TS source content.

package/src/parser/markdown.js

@@ -1,10 +1,4 @@
-// =============================================================================
-// OS AI Agent — Markdown Renderer v4.0
-// =============================================================================
-// Two modes:
-//   renderMarkdown(text)        — full buffered render (après réception complète)
-//   StreamMarkdown              — classe stateful pour le streaming token-by-token
-// =============================================================================
+// Markdown renderer — buffered + streaming
 
 import chalk from 'chalk';
 

package/src/safety/check.js

@@ -1,9 +1,4 @@
-// =============================================================================
-// OS AI Agent — Safety Check System (v4.0 — Coding Mode aware)
-// =============================================================================
-// Four-tier safety classification: READ, WRITE, DANGEROUS, ASK
-// In CODING mode, WRITE tools auto-approved with --no-confirm.
-// =============================================================================
+// Safety checker — READ/WRITE/DANGEROUS/ASK tiers
 
 const TIER = {
   READ: 'READ',

package/src/services/executor.js

@@ -1,9 +1,4 @@
-// =============================================================================
-// OS AI Agent — Command Executor
-// =============================================================================
-// Unified command executor that dispatches to local or SSH based on device.
-// Acts as a bridge between the agent loop and the tool system.
-// =============================================================================
+// Command executor — local or SSH dispatch
 import { exec } from 'child_process';
 import { connectSSH, execSSH, closeSSH } from './ssh.js';
 import { DEFAULTS } from '../utils/constants.js';

package/src/services/session.js

@@ -1,7 +1,4 @@
-// =============================================================================
-// OS AI Agent — Session Manager
-// Supports both LOCAL (JSON file) and CLOUD (MongoDB via server API) storage.
-// =============================================================================
+// Session manager — local JSON + cloud MongoDB
 import fs   from 'fs/promises';
 import { mkdirSync, writeFileSync } from 'fs';
 import path from 'path';

package/src/services/ssh.js

@@ -1,6 +1,4 @@
-// =============================================================================
-// OS AI Agent — SSH Connection Manager
-// =============================================================================
+// SSH connection manager
 import { Client } from 'ssh2';
 import { DEFAULTS } from '../utils/constants.js';
 import { logger } from '../utils/logger.js';

package/src/services/websocket.js

@@ -1,10 +1,4 @@
-// =============================================================================
-// OS AI Agent — WebSocket Client (Standalone)
-// =============================================================================
-// Standalone WebSocket client with auto-reconnect and exponential backoff.
-// Used as a lower-level utility that can be imported by the agent loop or
-// other modules needing persistent WebSocket connections.
-// =============================================================================
+// WebSocket client — auto-reconnect + backoff
 import WebSocket from 'ws';
 import { DEFAULTS } from '../utils/constants.js';
 import { logger } from '../utils/logger.js';

package/src/skills/loader.js

@@ -1,8 +1,4 @@
-// =============================================================================
-// OS AI Agent — Skills Loader
-// =============================================================================
-// Discovers and manages SKILL.md files from personal and project directories.
-// =============================================================================
+// Skills loader — discovers SKILL.md files
 
 import fs from 'fs/promises';
 import path from 'path';

package/src/tools/browser.js

@@ -1,10 +1,4 @@
-// =============================================================================
-// OS AI Agent — Browser Automation Module (Playwright + HTTP Fallback)
-// =============================================================================
-// Safe, stateless, per-request browser automation for the AI agent.
-// Each call is fully isolated — launch → close in finally block.
-// Falls back to HTTP fetch if Playwright/Chromium is unavailable.
-// =============================================================================
+// Browser automation — Playwright + HTTP fallback
 
 import { logger } from '../utils/logger.js';
 import { searchDDG, searchDDGHttp } from './search-providers.js';

package/src/tools/local.js

@@ -1,6 +1,4 @@
-// =============================================================================
-// OS AI Agent — Local Tool System (v4.0 — Coding Mode + Todos)
-// =============================================================================
+// Local tool system — command execution + file ops
 import { exec, spawn } from 'child_process';
 import { promisify } from 'util';
 import fs from 'fs/promises';
@@ -937,9 +935,7 @@ export const validateLocalCommand = (command) => {
   return { safe: true };
 };
 
-// =============================================================================
-// NEW TOOLS — GLOB, GREP, GIT, READ_FILE_RANGE, ASK_USER, DIAG_POST_EDIT
-// =============================================================================
+// Glob, Grep, Git, ReadFileRange, AskUser, DiagPostEdit
 
 import fastGlob from 'fast-glob';
 import { execFile } from 'child_process';

package/src/tools/registry.js

@@ -1,10 +1,4 @@
-// =============================================================================
-// OS AI Agent — Tool Registry
-// =============================================================================
-// Central registry for all tools the agent can use. Each tool is an independent
-// module with a standard interface: name, description, parameters, execute(),
-// and safety tier.
-// =============================================================================
+// Tool registry — central tool catalog
 
 import { logger } from '../utils/logger.js';
 import { TOOLS, SAFETY_TIERS } from '../utils/constants.js';

package/src/tools/ssh.js

@@ -1,6 +1,4 @@
-// =============================================================================
-// OS AI Agent — SSH Tool (Connection Pool + Dispatch)
-// =============================================================================
+// SSH tool — connection pool + dispatch
 import { connectSSH, execSSH, closeSSH } from '../services/ssh.js';
 import { logger } from '../utils/logger.js';
 

package/src/ui/components/ProviderMenu.js

@@ -544,7 +544,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
   useInput((input, key) => {
     if (!visible) return;
 
-    // === MENU PHASE ===
+    // MENU PHASE
     if (phase === 'menu') {
       if (key.escape) { resetAll(); onCancel(); return; }
       if (key.upArrow) { setMenuCursor(c => (c > 0 ? c - 1 : filteredMenu.length - 1)); return; }
@@ -574,7 +574,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
       return;
     }
 
-    // === CATALOG PHASE ===
+    // CATALOG PHASE
     if (phase === 'catalog') {
       if (key.escape) { setPhase('menu'); setCatalogQuery(''); setCatalogCursor(0); return; }
       if (key.upArrow) { setCatalogCursor(c => (c > 0 ? c - 1 : filteredCatalog.length - 1)); return; }
@@ -611,7 +611,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
       return;
     }
 
-    // === BASE URL PHASE (local mode only) ===
+    // BASE URL PHASE (local mode only)
     if (phase === 'base_url') {
       if (key.escape) { setPhase('catalog'); setLocalBaseUrlInput(''); setApiKeyInput(''); return; }
       if (key.return) {
@@ -631,7 +631,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
       return;
     }
 
-    // === CONFIGURED PHASE ===
+    // CONFIGURED PHASE
     if (phase === 'configured') {
       if (key.escape) { resetAll(); return; }
       if (key.upArrow) { setConfiguredCursor(c => (c > 0 ? c - 1 : filteredConfigured.length - 1)); return; }
@@ -648,7 +648,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
       return;
     }
 
-    // === MODELS PHASE (full setup) ===
+    // MODELS PHASE (full setup)
     if (phase === 'models') {
       if (modelsLoading) {
         if (key.escape) { setPhase('catalog'); setModelsQuery(''); setModelsCursor(0); setSelectedProvider(null); setModelsData(null); return; }
@@ -669,7 +669,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
       return;
     }
 
-    // === SWITCH MODELS PHASE (model selection for switch_model flow) ===
+    // SWITCH MODELS PHASE (model selection for switch_model flow)
     if (phase === 'switch_models') {
       if (key.escape) { setPhase('configured'); setModelsQuery(''); setModelsCursor(0); setSelectedProvider(null); return; }
       if (key.upArrow) { setModelsCursor(c => (c > 0 ? c - 1 : allModels.length - 1)); return; }
@@ -686,7 +686,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
       return;
     }
 
-    // === API KEY PHASE (full setup: server mode after model select, local mode before model fetch) ===
+    // API KEY PHASE (full setup: server mode after model select, local mode before model fetch)
     if (phase === 'api_key') {
       if (key.escape) {
         if (isLocal) {
@@ -716,7 +716,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
       return;
     }
 
-    // === UPDATE KEY INPUT PHASE ===
+    // UPDATE KEY INPUT PHASE
     if (phase === 'update_key_input') {
       if (key.escape) { setPhase('configured'); setApiKeyInput(''); setSettingError(null); return; }
       if (key.return) { handleUpdateKeySubmit(); return; }
@@ -732,7 +732,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
   const separator = '─'.repeat(52);
   const { rows } = useWindowSize();
 
-  // === SETTING UP VIEW ===
+  // SETTING UP VIEW
   if (settingUp) {
     return h(Box, { flexDirection: 'column', borderStyle: 'round', borderColor: '#2a2e3f', paddingX: 1, paddingY: 0, marginY: 1 },
       h(Text, { color: '#7aa2f7', bold: true }, TITLE),
@@ -743,7 +743,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
     );
   }
 
-  // === BASE URL INPUT (local mode) ===
+  // BASE URL INPUT (local mode)
   if (phase === 'base_url') {
     return h(Box, { flexDirection: 'column', borderStyle: 'round', borderColor: '#2a2e3f', paddingX: 1, paddingY: 0, marginY: 1 },
       h(Box, { flexDirection: 'row', alignItems: 'center' },
@@ -766,7 +766,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
     );
   }
 
-  // === API KEY VIEW ===
+  // API KEY VIEW
   if (phase === 'api_key') {
     const masked = apiKeyInput ? '\u2022'.repeat(apiKeyInput.length) : '';
     return h(Box, { flexDirection: 'column', borderStyle: 'round', borderColor: '#2a2e3f', paddingX: 1, paddingY: 0, marginY: 1 },
@@ -794,7 +794,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
     );
   }
 
-  // === UPDATE KEY INPUT VIEW ===
+  // UPDATE KEY INPUT VIEW
   if (phase === 'update_key_input') {
     const masked = apiKeyInput ? '\u2022'.repeat(apiKeyInput.length) : '';
     return h(Box, { flexDirection: 'column', borderStyle: 'round', borderColor: '#2a2e3f', paddingX: 1, paddingY: 0, marginY: 1 },
@@ -822,7 +822,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
     );
   }
 
-  // === MODELS VIEW (reused for both full setup and switch_model) ===
+  // MODELS VIEW (reused for both full setup and switch_model)
   const isSwitchModels = phase === 'switch_models';
   if (phase === 'models' || isSwitchModels) {
     if (modelsLoading) {
@@ -882,7 +882,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
     );
   }
 
-  // === CONFIGURED VIEW ===
+  // CONFIGURED VIEW
   if (phase === 'configured') {
     if (configuredLoading) {
       return h(Box, { flexDirection: 'column', borderStyle: 'round', borderColor: '#2a2e3f', paddingX: 1, paddingY: 0, marginY: 1 },
@@ -965,7 +965,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
     );
   }
 
-  // === CATALOG VIEW ===
+  // CATALOG VIEW
   if (phase === 'catalog') {
     if (catalogLoading) {
       return h(Box, { flexDirection: 'column', borderStyle: 'round', borderColor: '#2a2e3f', paddingX: 1, paddingY: 0, marginY: 1 },
@@ -1055,7 +1055,7 @@ export function ProviderMenu({ visible, onSelect, onCancel, serverUrl, token, cu
     );
   }
 
-  // === MENU VIEW ===
+  // MENU VIEW
   return h(Box, { flexDirection: 'column', borderStyle: 'round', borderColor: '#2a2e3f', paddingX: 1, paddingY: 0, marginY: 1 },
     h(Box, { flexDirection: 'row', alignItems: 'center' },
       h(Text, { color: '#7aa2f7', bold: true }, TITLE),

package/src/utils/constants.js

@@ -1,6 +1,4 @@
-// =============================================================================
-// OS AI Agent — Constants & Configuration
-// =============================================================================
+// Constants & configuration
 
 export const APP_NAME = 'osai-agent';
 import { createRequire } from 'module';

package/src/utils/helpers.js

@@ -1,6 +1,4 @@
-// =============================================================================
-// OS AI Agent — Shared Helpers
-// =============================================================================
+// Shared helpers
 
 /**
  * Estimate token count from text (rough: ~4 chars per token)

package/src/utils/logger.js

@@ -1,6 +1,4 @@
-// =============================================================================
-// OS AI Agent — Logger
-// =============================================================================
+// Logger
 import chalk from 'chalk';
 import { ENV_VARS } from './constants.js';
 

package/src/utils/sound.js

@@ -1,6 +1,4 @@
-// =============================================================================
-// OS AI Agent — System Sound Notification
-// =============================================================================
+// System sound notifications
 
 /**
  * Play a system beep sound (cross-platform: Windows, macOS, Linux)