@aethermind/setup 0.1.4 → 0.1.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aethermind/setup",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "CLI setup tool for Aethermind AgentOS — configures your project with one command",
   "bin": {
     "aethermind-setup": "./bin/setup.js"

package/src/detect.js

@@ -15,15 +15,26 @@ function checkNodeVersion() {
   return version;
 }
 
+/**
+ * Detect the project language/runtime.
+ * Returns 'node', 'python', or 'unknown'.
+ */
+function detectLanguage(cwd) {
+  if (fs.existsSync(path.join(cwd, 'package.json'))) return 'node';
+  if (
+    fs.existsSync(path.join(cwd, 'requirements.txt')) ||
+    fs.existsSync(path.join(cwd, 'pyproject.toml')) ||
+    fs.existsSync(path.join(cwd, 'setup.py')) ||
+    fs.existsSync(path.join(cwd, 'Pipfile'))
+  ) return 'python';
+  return 'unknown';
+}
+
 function detectPackageJson(cwd) {
   const pkgPath = path.join(cwd, 'package.json');
-  if (!fs.existsSync(pkgPath)) {
-    fail('No se encontro package.json en el directorio actual');
-    console.log('  Ejecuta este comando desde la raiz de tu proyecto.');
-    process.exit(1);
-  }
+  if (!fs.existsSync(pkgPath)) return null;
   const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
-  ok(`Proyecto: ${pkg.name || path.basename(cwd)} (package.json encontrado)`);
+  ok(`Proyecto: ${pkg.name || path.basename(cwd)}`);
   return pkg;
 }
 
@@ -49,7 +60,7 @@ function detectInstalledSDKs(cwd) {
   if (found.length > 0) {
     ok(`SDKs detectados: ${found.join(', ')}`);
   } else {
-    info('No se detectaron SDKs de IA instalados (openai, anthropic, gemini, ollama)');
+    info('No se detectaron SDKs de IA instalados');
   }
   return found;
 }
@@ -61,4 +72,32 @@ function detectPackageManager(cwd) {
   return 'npm';
 }
 
-module.exports = { checkNodeVersion, detectPackageJson, detectInstalledSDKs, detectPackageManager };
+/**
+ * Detect if Ollama is installed (binary on PATH or running on default port).
+ */
+function detectOllama() {
+  const { execSync } = require('child_process');
+
+  // Check if ollama binary exists
+  try {
+    execSync('ollama --version', { stdio: 'pipe', timeout: 5000 });
+    ok('Ollama detectado');
+    return true;
+  } catch (_) {
+    // not on PATH
+  }
+
+  // Check if something is listening on port 11434 (default Ollama port)
+  const http = require('http');
+  return new Promise((resolve) => {
+    const req = http.get('http://localhost:11434', { timeout: 2000 }, (res) => {
+      req.destroy();
+      ok('Ollama detectado (puerto 11434 activo)');
+      resolve(true);
+    });
+    req.on('error', () => resolve(false));
+    req.on('timeout', () => { req.destroy(); resolve(false); });
+  });
+}
+
+module.exports = { checkNodeVersion, detectLanguage, detectPackageJson, detectInstalledSDKs, detectPackageManager, detectOllama };

package/src/env.js

@@ -4,61 +4,100 @@ const fs = require('fs');
 const path = require('path');
 const { ok, info } = require('./ui.js');
 
-const ENV_KEYS = {
-  AETHERMIND_API_KEY: null,
-  NODE_OPTIONS: '--require @aethermind/providers/register',
-};
+const GATEWAY_URL = 'https://aethermind-agentos-production.up.railway.app/gateway/v1';
 
-function updateEnvFile(cwd, apiKey) {
+/**
+ * Update .env file for a Node.js project.
+ * Adds AETHERMIND_API_KEY and NODE_OPTIONS=--require @aethermind/agent/register
+ */
+function updateEnvForNode(cwd, apiKey) {
   const envPath = path.join(cwd, '.env');
-  const backupPath = path.join(cwd, '.env.aethermind.backup');
-  let existing = '';
+  backupEnv(cwd, envPath);
+
+  let content = readEnvOrEmpty(envPath);
+
+  content = setEnvVar(content, 'AETHERMIND_API_KEY', apiKey);
+  content = setEnvVar(content, 'NODE_OPTIONS', buildNodeOptions(content));
+
+  fs.writeFileSync(envPath, content, 'utf-8');
+  ok('.env actualizado (AETHERMIND_API_KEY + NODE_OPTIONS)');
+}
+
+/**
+ * Add Ollama proxy env vars.
+ * Sets OLLAMA_HOST to point to the local Aethermind proxy instead of Ollama directly.
+ */
+function addOllamaEnv(cwd) {
+  const envPath = path.join(cwd, '.env');
+  let content = readEnvOrEmpty(envPath);
+
+  content = setEnvVar(content, 'OLLAMA_HOST', 'http://localhost:11435');
+
+  fs.writeFileSync(envPath, content, 'utf-8');
+  ok('.env actualizado (OLLAMA_HOST → proxy en puerto 11435)');
+}
+
+/**
+ * Update .env file for a Python or generic project.
+ * Sets OPENAI_BASE_URL, OPENAI_API_KEY (= SDK key), and ANTHROPIC_BASE_URL.
+ * The gateway authenticates via the SDK key and uses stored provider keys.
+ */
+function updateEnvForGateway(cwd, apiKey) {
+  const envPath = path.join(cwd, '.env');
+  backupEnv(cwd, envPath);
+
+  let content = readEnvOrEmpty(envPath);
+
+  content = setEnvVar(content, 'AETHERMIND_API_KEY', apiKey);
+  content = setEnvVar(content, 'OPENAI_BASE_URL', GATEWAY_URL);
+  content = setEnvVar(content, 'OPENAI_API_KEY', apiKey);
+  content = setEnvVar(content, 'ANTHROPIC_BASE_URL', GATEWAY_URL);
+  content = setEnvVar(content, 'ANTHROPIC_API_KEY', apiKey);
 
-  // Backup existing .env
+  fs.writeFileSync(envPath, content, 'utf-8');
+  ok('.env actualizado (OPENAI_BASE_URL + ANTHROPIC_BASE_URL + API keys)');
+}
+
+// ─── Helpers ────────────────────────────────────────
+
+function readEnvOrEmpty(envPath) {
+  if (fs.existsSync(envPath)) {
+    return fs.readFileSync(envPath, 'utf-8');
+  }
+  return '';
+}
+
+function backupEnv(cwd, envPath) {
+  const backupPath = path.join(cwd, '.env.aethermind.backup');
   if (fs.existsSync(envPath)) {
-    existing = fs.readFileSync(envPath, 'utf-8');
     fs.copyFileSync(envPath, backupPath);
     ok('Backup guardado en .env.aethermind.backup');
   }
+}
 
-  const lines = existing ? existing.split('\n') : [];
-  const updates = {
-    AETHERMIND_API_KEY: apiKey,
-    NODE_OPTIONS: buildNodeOptions(lines),
-  };
-
-  let content = existing;
-
-  for (const [key, value] of Object.entries(updates)) {
-    const regex = new RegExp(`^${key}=.*$`, 'm');
-    const newLine = `${key}=${value}`;
-
-    if (regex.test(content)) {
-      content = content.replace(regex, newLine);
-    } else {
-      if (content && !content.endsWith('\n')) {
-        content += '\n';
-      }
-      content += newLine + '\n';
-    }
-  }
+function setEnvVar(content, key, value) {
+  const regex = new RegExp(`^${key}=.*$`, 'm');
+  const newLine = `${key}=${value}`;
 
-  fs.writeFileSync(envPath, content, 'utf-8');
-  ok('.env actualizado');
+  if (regex.test(content)) {
+    return content.replace(regex, newLine);
+  }
+  if (content && !content.endsWith('\n')) {
+    content += '\n';
+  }
+  return content + newLine + '\n';
 }
 
-function buildNodeOptions(lines) {
-  const requireFlag = '--require @aethermind/providers/register';
+function buildNodeOptions(content) {
+  const requireFlag = '--require @aethermind/agent/register';
+  const lines = content.split('\n');
   const existingLine = lines.find((l) => l.startsWith('NODE_OPTIONS='));
 
   if (existingLine) {
     const existingValue = existingLine.replace('NODE_OPTIONS=', '');
-    if (existingValue.includes(requireFlag)) {
-      return existingValue;
-    }
+    if (existingValue.includes(requireFlag)) return existingValue;
     return existingValue ? `${existingValue} ${requireFlag}` : requireFlag;
   }
-
   return requireFlag;
 }
 
@@ -71,13 +110,12 @@ function restoreEnvFile(cwd) {
     fs.unlinkSync(backupPath);
     ok('.env restaurado desde backup');
   } else if (fs.existsSync(envPath)) {
-    // Remove our additions from .env
     let content = fs.readFileSync(envPath, 'utf-8');
     content = content.replace(/^AETHERMIND_API_KEY=.*\n?/m, '');
-    // Remove our require from NODE_OPTIONS
-    content = content.replace(/ ?--require @aethermind\/providers\/register/g, '');
-    // Remove empty NODE_OPTIONS line
+    content = content.replace(/ ?--require @aethermind\/agent\/register/g, '');
     content = content.replace(/^NODE_OPTIONS=\s*\n?/m, '');
+    content = content.replace(/^OPENAI_BASE_URL=.*\n?/m, '');
+    content = content.replace(/^ANTHROPIC_BASE_URL=.*\n?/m, '');
     fs.writeFileSync(envPath, content, 'utf-8');
     ok('Entradas de Aethermind removidas del .env');
   } else {
@@ -85,4 +123,4 @@ function restoreEnvFile(cwd) {
   }
 }
 
-module.exports = { updateEnvFile, restoreEnvFile };
+module.exports = { updateEnvForNode, updateEnvForGateway, addOllamaEnv, restoreEnvFile };

package/src/index.js

@@ -2,10 +2,10 @@
 
 const readline = require('readline');
 const { banner, ok, info, separator, summary, c } = require('./ui.js');
-const { checkNodeVersion, detectPackageJson, detectInstalledSDKs, detectPackageManager } = require('./detect.js');
+const { checkNodeVersion, detectLanguage, detectPackageJson, detectInstalledSDKs, detectPackageManager, detectOllama } = require('./detect.js');
 const { validateApiKey } = require('./validate.js');
-const { installProviders, uninstallProviders } = require('./install.js');
-const { updateEnvFile, restoreEnvFile } = require('./env.js');
+const { installAgent, uninstallAgent, installOllamaProxy } = require('./install.js');
+const { updateEnvForNode, updateEnvForGateway, addOllamaEnv, restoreEnvFile } = require('./env.js');
 
 function prompt(question) {
   return new Promise((resolve) => {
@@ -21,7 +21,7 @@ function prompt(question) {
 }
 
 function confirm(question) {
-  return prompt(`${question} (s/n): `).then((a) => a.toLowerCase() === 's' || a.toLowerCase() === 'y');
+  return prompt(`${question} (y/n): `).then((a) => a.toLowerCase() === 's' || a.toLowerCase() === 'y');
 }
 
 async function runSetup() {
@@ -29,26 +29,30 @@ async function runSetup() {
 
   banner();
 
-  // 1. Check Node version
+  // 1. Check Node version (needed to run this script)
   checkNodeVersion();
 
-  // 2. Detect package.json
-  const pkg = detectPackageJson(cwd);
+  // 2. Detect project language
+  const language = detectLanguage(cwd);
+  ok(`Proyecto detectado: ${language === 'node' ? 'Node.js' : language === 'python' ? 'Python' : 'Otro'}`);
 
-  // 3. Detect installed SDKs
-  const sdks = detectInstalledSDKs(cwd);
+  // 3. Detect project details
+  if (language === 'node') {
+    detectPackageJson(cwd);
+    detectInstalledSDKs(cwd);
+  }
 
-  // 4. Detect package manager
-  const pm = detectPackageManager(cwd);
+  // 4. Detect Ollama
+  const hasOllama = await detectOllama();
 
   separator();
 
   // 5. Ask for API key
-  const apiKey = await prompt(c.cyan('  ? ') + 'Ingresa tu AETHERMIND_API_KEY: ');
+  const apiKey = await prompt(c.cyan('  ? ') + 'Pega tu SDK API Key (la encontras en el dashboard): ');
 
   if (!apiKey) {
     console.log(c.red('\n  API key no puede estar vacia.'));
-    console.log('  Obtene tu key en: https://aethermind-agent-os-dashboard.vercel.app/settings\n');
+    console.log('  Obtene tu key en: https://aethermind-agent-os-dashboard.vercel.app\n');
     process.exit(1);
   }
 
@@ -57,12 +61,31 @@ async function runSetup() {
 
   separator();
 
-  // 7. Show what will be modified and confirm
+  // 7. Show plan and confirm
   console.log('');
   console.log(c.bold('  Se realizaran los siguientes cambios:'));
-  console.log(`    - Instalar @aethermind/providers (via ${pm})`);
-  console.log('    - Agregar AETHERMIND_API_KEY al .env');
-  console.log('    - Agregar --require @aethermind/providers/register a NODE_OPTIONS');
+
+  if (language === 'node') {
+    const pm = detectPackageManager(cwd);
+    console.log(`    - Instalar @aethermind/agent (via ${pm})`);
+    console.log('    - Agregar AETHERMIND_API_KEY al .env');
+    console.log('    - Agregar --require @aethermind/agent/register a NODE_OPTIONS');
+  } else {
+    console.log('    - Agregar AETHERMIND_API_KEY al .env');
+    console.log('    - Configurar OPENAI_BASE_URL -> Aethermind Gateway');
+    console.log('    - Configurar ANTHROPIC_BASE_URL -> Aethermind Gateway');
+  }
+
+  if (hasOllama) {
+    console.log('    - Configurar OLLAMA_HOST -> proxy local (puerto 11435)');
+    if (language === 'node') {
+      const pm = detectPackageManager(cwd);
+      console.log(`    - Instalar @aethermind/proxy (via ${pm})`);
+    }
+  }
+
+  console.log('');
+  console.log(c.dim('  Tu codigo NO se modifica.'));
   console.log('');
 
   const proceed = await confirm(c.cyan('  ? ') + 'Continuar?');
@@ -73,25 +96,35 @@ async function runSetup() {
 
   console.log('');
 
-  // 8. Install @aethermind/providers
-  installProviders(pm);
-
-  // 9. Update .env
-  updateEnvFile(cwd, apiKey);
+  // 8. Execute
+  if (language === 'node') {
+    const pm = detectPackageManager(cwd);
+    installAgent(pm);
+    updateEnvForNode(cwd, apiKey);
+    if (hasOllama) {
+      installOllamaProxy(pm);
+      addOllamaEnv(cwd);
+    }
+  } else {
+    updateEnvForGateway(cwd, apiKey);
+    if (hasOllama) {
+      addOllamaEnv(cwd);
+    }
+  }
 
-  // 10. Show summary
-  summary();
+  // 9. Done
+  summary(language, hasOllama);
 }
 
 async function runUninstall() {
   const cwd = process.cwd();
 
   banner();
-  console.log(c.yellow('  Desinstalando Aethermind AgentOS...\n'));
+  console.log(c.yellow('  Desinstalando Aethermind...\n'));
 
-  const pm = detectPackageManager(cwd);
+  const language = detectLanguage(cwd);
 
-  const proceed = await confirm(c.cyan('  ? ') + 'Desinstalar @aethermind/providers y restaurar .env?');
+  const proceed = await confirm(c.cyan('  ? ') + 'Desinstalar Aethermind y restaurar .env?');
   if (!proceed) {
     console.log('\n  Cancelado.\n');
     process.exit(0);
@@ -99,13 +132,14 @@ async function runUninstall() {
 
   console.log('');
 
-  // Uninstall package
-  uninstallProviders(pm);
+  if (language === 'node') {
+    const pm = detectPackageManager(cwd);
+    uninstallAgent(pm);
+  }
 
-  // Restore .env
   restoreEnvFile(cwd);
 
-  console.log(c.green('\n  Aethermind AgentOS desinstalado correctamente.\n'));
+  console.log(c.green('\n  Aethermind desinstalado correctamente.\n'));
 }
 
 async function main({ uninstall = false } = {}) {

package/src/install.js

@@ -3,42 +3,61 @@
 const { execSync } = require('child_process');
 const { ok, fail } = require('./ui.js');
 
-function installProviders(packageManager) {
+function installAgent(packageManager) {
   const cmds = {
-    npm: 'npm install @aethermind/providers --legacy-peer-deps',
-    pnpm: 'pnpm add @aethermind/providers',
-    yarn: 'yarn add @aethermind/providers',
-    bun: 'bun add @aethermind/providers',
+    npm: 'npm install @aethermind/agent --legacy-peer-deps',
+    pnpm: 'pnpm add @aethermind/agent',
+    yarn: 'yarn add @aethermind/agent',
+    bun: 'bun add @aethermind/agent',
   };
 
   const cmd = cmds[packageManager] || cmds.npm;
 
   try {
     execSync(cmd, { stdio: 'pipe', timeout: 120000 });
-    ok('@aethermind/providers instalado');
+    ok('@aethermind/agent instalado');
   } catch (err) {
-    fail(`Error instalando @aethermind/providers con ${packageManager}`);
+    fail(`Error instalando @aethermind/agent con ${packageManager}`);
     console.log(`  Intenta manualmente: ${cmd}`);
     throw new Error(`Fallo la instalacion: ${err.message}`);
   }
 }
 
-function uninstallProviders(packageManager) {
+function uninstallAgent(packageManager) {
   const cmds = {
-    npm: 'npm uninstall @aethermind/providers',
-    pnpm: 'pnpm remove @aethermind/providers',
-    yarn: 'yarn remove @aethermind/providers',
-    bun: 'bun remove @aethermind/providers',
+    npm: 'npm uninstall @aethermind/agent',
+    pnpm: 'pnpm remove @aethermind/agent',
+    yarn: 'yarn remove @aethermind/agent',
+    bun: 'bun remove @aethermind/agent',
   };
 
   const cmd = cmds[packageManager] || cmds.npm;
 
   try {
     execSync(cmd, { stdio: 'pipe', timeout: 120000 });
-    ok('@aethermind/providers desinstalado');
+    ok('@aethermind/agent desinstalado');
   } catch (_) {
     // may not be installed, that's ok
   }
 }
 
-module.exports = { installProviders, uninstallProviders };
+function installOllamaProxy(packageManager) {
+  const cmds = {
+    npm: 'npm install @aethermind/proxy --legacy-peer-deps',
+    pnpm: 'pnpm add @aethermind/proxy',
+    yarn: 'yarn add @aethermind/proxy',
+    bun: 'bun add @aethermind/proxy',
+  };
+
+  const cmd = cmds[packageManager] || cmds.npm;
+
+  try {
+    execSync(cmd, { stdio: 'pipe', timeout: 120000 });
+    ok('@aethermind/proxy instalado');
+  } catch (err) {
+    fail(`Error instalando @aethermind/proxy con ${packageManager}`);
+    console.log(`  Intenta manualmente: ${cmd}`);
+  }
+}
+
+module.exports = { installAgent, uninstallAgent, installOllamaProxy };

package/src/ui.js

@@ -36,17 +36,28 @@ function separator() {
   console.log(c.dim('  ' + '-'.repeat(35)));
 }
 
-function summary() {
+function summary(language, hasOllama) {
   console.log('');
-  console.log(c.dim('  ' + '='.repeat(35)));
-  console.log(c.green(c.bold('  Listo. Tu app esta siendo monitoreada.')));
+  console.log(c.dim('  ' + '='.repeat(40)));
+  console.log(c.green(c.bold('  Listo! Tu app esta siendo monitoreada.')));
   console.log('');
   console.log('  Corre tu app normalmente:');
-  console.log(c.cyan('    node index.js'));
+  if (language === 'python') {
+    console.log(c.cyan('    python app.py'));
+  } else {
+    console.log(c.cyan('    node app.js'));
+  }
+  if (hasOllama) {
+    console.log('');
+    console.log('  Para modelos Ollama, antes inicia el proxy:');
+    console.log(c.cyan('    npx @aethermind/proxy'));
+  }
+  console.log('');
+  console.log('  No necesitas cambiar tu codigo.');
   console.log('');
   console.log('  Mira tu dashboard:');
   console.log(c.cyan('    https://aethermind-agent-os-dashboard.vercel.app'));
-  console.log(c.dim('  ' + '='.repeat(35)));
+  console.log(c.dim('  ' + '='.repeat(40)));
   console.log('');
 }