@projectservan8n/cnapse 0.5.0 → 0.5.2

package/src/index.tsx

@@ -6,85 +6,94 @@ import { setApiKey, setProvider, setModel, getConfig } from './lib/config.js';
 
 const args = process.argv.slice(2);
 
-// Handle CLI commands
-if (args.length > 0) {
-  const command = args[0];
-
-  switch (command) {
-    case 'auth': {
-      const provider = args[1] as 'openrouter' | 'anthropic' | 'openai';
-      const key = args[2];
-
-      if (!provider || !key) {
-        console.log('Usage: cnapse auth <provider> <api-key>');
-        console.log('Providers: openrouter, anthropic, openai');
-        process.exit(1);
-      }
+async function main() {
+  // Handle CLI commands
+  if (args.length > 0) {
+    const command = args[0];
+
+    switch (command) {
+      case 'auth': {
+        const provider = args[1] as 'openrouter' | 'anthropic' | 'openai';
+        const key = args[2];
 
-      if (!['openrouter', 'anthropic', 'openai'].includes(provider)) {
-        console.log(`Invalid provider: ${provider}`);
-        console.log('Valid providers: openrouter, anthropic, openai');
-        process.exit(1);
-      }
+        if (!provider || !key) {
+          console.log('Usage: cnapse auth <provider> <api-key>');
+          console.log('Providers: openrouter, anthropic, openai');
+          process.exit(1);
+        }
 
-      setApiKey(provider, key);
-      console.log(`✓ ${provider} API key saved`);
-      process.exit(0);
-    }
+        if (!['openrouter', 'anthropic', 'openai'].includes(provider)) {
+          console.log(`Invalid provider: ${provider}`);
+          console.log('Valid providers: openrouter, anthropic, openai');
+          process.exit(1);
+        }
 
-    case 'config': {
-      const subcommand = args[1];
+        setApiKey(provider, key);
+        console.log(`✓ ${provider} API key saved`);
+        process.exit(0);
+      }
 
-      if (subcommand === 'set') {
-        const key = args[2];
-        const value = args[3];
+      case 'config': {
+        const subcommand = args[1];
+
+        // Interactive config TUI if no subcommand
+        if (!subcommand) {
+          const { ConfigUI } = await import('./components/ConfigUI.js');
+          render(<ConfigUI />);
+          return; // Don't render App
+        }
 
-        if (key === 'provider') {
-          if (!['openrouter', 'ollama', 'anthropic', 'openai'].includes(value!)) {
-            console.log('Valid providers: openrouter, ollama, anthropic, openai');
-            process.exit(1);
+        if (subcommand === 'set') {
+          const key = args[2];
+          const value = args[3];
+
+          if (key === 'provider') {
+            if (!['openrouter', 'ollama', 'anthropic', 'openai'].includes(value!)) {
+              console.log('Valid providers: openrouter, ollama, anthropic, openai');
+              process.exit(1);
+            }
+            setProvider(value as any);
+            console.log(`✓ Provider set to: ${value}`);
+          } else if (key === 'model') {
+            setModel(value!);
+            console.log(`✓ Model set to: ${value}`);
+          } else {
+            console.log('Usage: cnapse config set <provider|model> <value>');
           }
-          setProvider(value as any);
-          console.log(`✓ Provider set to: ${value}`);
-        } else if (key === 'model') {
-          setModel(value!);
-          console.log(`✓ Model set to: ${value}`);
-        } else {
-          console.log('Usage: cnapse config set <provider|model> <value>');
+          process.exit(0);
         }
-        process.exit(0);
-      }
 
-      if (subcommand === 'show' || !subcommand) {
-        const config = getConfig();
-        console.log('\nC-napse Configuration:');
-        console.log(`  Provider: ${config.provider}`);
-        console.log(`  Model: ${config.model}`);
-        console.log(`  Ollama Host: ${config.ollamaHost}`);
-        console.log(`  API Keys configured:`);
-        console.log(`    - OpenRouter: ${config.apiKeys.openrouter ? '✓' : '✗'}`);
-        console.log(`    - Anthropic: ${config.apiKeys.anthropic ? '✓' : '✗'}`);
-        console.log(`    - OpenAI: ${config.apiKeys.openai ? '✓' : '✗'}`);
-        console.log('');
-        process.exit(0);
-      }
+        if (subcommand === 'show') {
+          const config = getConfig();
+          console.log('\nC-napse Configuration:');
+          console.log(`  Provider: ${config.provider}`);
+          console.log(`  Model: ${config.model}`);
+          console.log(`  Ollama Host: ${config.ollamaHost}`);
+          console.log(`  API Keys configured:`);
+          console.log(`    - OpenRouter: ${config.apiKeys.openrouter ? '✓' : '✗'}`);
+          console.log(`    - Anthropic: ${config.apiKeys.anthropic ? '✓' : '✗'}`);
+          console.log(`    - OpenAI: ${config.apiKeys.openai ? '✓' : '✗'}`);
+          console.log('');
+          process.exit(0);
+        }
 
-      console.log('Usage: cnapse config [show|set <key> <value>]');
-      process.exit(1);
-    }
+        console.log('Usage: cnapse config [show|set <key> <value>]');
+        process.exit(1);
+      }
 
-    case 'help':
-    case '--help':
-    case '-h': {
-      console.log(`
+      case 'help':
+      case '--help':
+      case '-h': {
+        console.log(`
 C-napse - Autonomous PC Intelligence
 
 Usage:
   cnapse                     Start interactive chat
   cnapse init                Interactive setup wizard
-  cnapse auth <provider> <key>  Set API key
-  cnapse config              Show configuration
+  cnapse config              Interactive configuration
+  cnapse config show         Show current configuration
   cnapse config set <k> <v>  Set config value
+  cnapse auth <provider> <key>  Set API key
   cnapse help                Show this help
 
 Providers:
@@ -95,36 +104,39 @@ Providers:
 
 Quick Start:
   cnapse init                # Interactive setup
+  cnapse config              # Change provider/model
 
 Manual Setup:
   cnapse auth openrouter sk-or-xxxxx
   cnapse config set provider openrouter
   cnapse config set model qwen/qwen-2.5-coder-32b-instruct
 `);
-      process.exit(0);
-    }
+        process.exit(0);
+      }
 
-    case 'version':
-    case '--version':
-    case '-v': {
-      console.log('cnapse v0.2.0');
-      process.exit(0);
-    }
+      case 'version':
+      case '--version':
+      case '-v': {
+        console.log('cnapse v0.5.0');
+        process.exit(0);
+      }
 
-    case 'init': {
-      // Interactive setup with Ink UI
-      const { Setup } = await import('./components/Setup.js');
-      render(<Setup />);
-      process.exit(0);
-    }
+      case 'init': {
+        // Interactive setup with Ink UI
+        const { Setup } = await import('./components/Setup.js');
+        render(<Setup />);
+        return; // Don't render App
+      }
 
-    default: {
-      // Treat as a direct question
-      // For now, just start the app
-      break;
+      default: {
+        // Unknown command - start app anyway
+        break;
+      }
     }
   }
+
+  // Start interactive TUI
+  render(<App />);
 }
 
-// Start interactive TUI
-render(<App />);
+main();

package/src/lib/ollama.ts

@@ -0,0 +1,140 @@
+/**
+ * Ollama utilities - Check status, list models, pull/run models
+ */
+
+import { exec } from 'child_process';
+import { promisify } from 'util';
+
+const execAsync = promisify(exec);
+
+export interface OllamaModel {
+  name: string;
+  size: string;
+  modified: string;
+}
+
+export interface OllamaStatus {
+  installed: boolean;
+  running: boolean;
+  models: OllamaModel[];
+  error?: string;
+}
+
+/**
+ * Check if Ollama is installed and running, list available models
+ */
+export async function checkOllamaStatus(): Promise<OllamaStatus> {
+  try {
+    // Try to list models - this checks both installation and if it's running
+    const { stdout } = await execAsync('ollama list', { timeout: 10000 });
+
+    // Parse the output
+    const lines = stdout.trim().split('\n');
+    const models: OllamaModel[] = [];
+
+    // Skip header line
+    for (let i = 1; i < lines.length; i++) {
+      const line = lines[i];
+      if (!line?.trim()) continue;
+
+      // Parse: NAME    ID    SIZE    MODIFIED
+      const parts = line.split(/\s{2,}/);
+      if (parts.length >= 3) {
+        models.push({
+          name: parts[0]?.trim() || '',
+          size: parts[2]?.trim() || '',
+          modified: parts[3]?.trim() || '',
+        });
+      }
+    }
+
+    return {
+      installed: true,
+      running: true,
+      models,
+    };
+  } catch (err) {
+    const errorMsg = err instanceof Error ? err.message : 'Unknown error';
+
+    // Check if Ollama is installed but not running
+    if (errorMsg.includes('connect') || errorMsg.includes('refused')) {
+      return {
+        installed: true,
+        running: false,
+        models: [],
+        error: 'Ollama is not running. Start it with: ollama serve',
+      };
+    }
+
+    // Ollama not installed
+    if (errorMsg.includes('not found') || errorMsg.includes('not recognized')) {
+      return {
+        installed: false,
+        running: false,
+        models: [],
+        error: 'Ollama not installed. Get it at: https://ollama.ai',
+      };
+    }
+
+    return {
+      installed: false,
+      running: false,
+      models: [],
+      error: errorMsg,
+    };
+  }
+}
+
+/**
+ * Check if a specific model is available
+ */
+export function hasModel(status: OllamaStatus, modelId: string): boolean {
+  const modelName = modelId.split(':')[0]?.toLowerCase() || '';
+  return status.models.some(m => m.name.toLowerCase().startsWith(modelName));
+}
+
+/**
+ * Pull a model (download it)
+ */
+export async function pullModel(modelId: string, onProgress?: (msg: string) => void): Promise<boolean> {
+  try {
+    onProgress?.(`Downloading ${modelId}...`);
+
+    // Pull with a long timeout (models can be large)
+    await execAsync(`ollama pull ${modelId}`, { timeout: 600000 }); // 10 min
+
+    onProgress?.(`Downloaded ${modelId}`);
+    return true;
+  } catch (err) {
+    const errorMsg = err instanceof Error ? err.message : 'Unknown error';
+    onProgress?.(`Failed to download: ${errorMsg}`);
+    return false;
+  }
+}
+
+/**
+ * Run a model to load it into memory
+ */
+export async function runModel(modelId: string): Promise<boolean> {
+  try {
+    // Send a simple prompt to load the model
+    await execAsync(`ollama run ${modelId} "Hi" --nowordwrap`, { timeout: 120000 });
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * Get model size in human-readable format
+ */
+export function getModelInfo(status: OllamaStatus, modelId: string): { available: boolean; size?: string } {
+  const modelName = modelId.split(':')[0]?.toLowerCase() || '';
+  const model = status.models.find(m => m.name.toLowerCase().startsWith(modelName));
+
+  if (model) {
+    return { available: true, size: model.size };
+  }
+
+  return { available: false };
+}