@vierratale/ai 0.1.0-beta.1 → 0.1.0-beta.3

package/README.md

@@ -40,9 +40,28 @@ vierrataleai --model vierratale-pro   # Use specific model
 - `/model [name]` - Switch model
 - `/provider [name]` - Switch provider
 - `/models` - List available models
+- `/search <query>` - Search the web and summarize with AI
 - `/clear` - Clear screen
+- `/new` - Start a new conversation (clear memory)
 - `/quit` - Exit
 
+## Smart Search
+
+Asking a question (Who/What/When/Where/Why/How...?) automatically searches the
+web and summarizes the results with AI. For example:
+
+```
+> Who is Cristiano Ronaldo?
+```
+
+triggers a live web search and answers from the results. Use `/search <query>`
+to force a search explicitly.
+
+## Conversation Memory
+
+Your conversation is saved automatically and restored on the next launch, so
+you can continue where you left off. Use `/new` to start fresh.
+
 ## Configuration
 
 Config file: `~/.config/vierrataleai/config.json`

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vierratale/ai",
-  "version": "0.1.0-beta.1",
+  "version": "0.1.0-beta.3",
   "description": "VierrataleAI - Intelligent terminal assistant",
   "type": "module",
   "bin": {

package/src/cli.js

@@ -6,6 +6,9 @@ import { ProviderFactory } from './providers/index.js';
 import { Branding } from './ui/branding.js';
 import { Terminal } from './ui/terminal.js';
 import { loadSystemPrompt, showBanner } from './ui/banner.js';
+import { WebSearch } from './utils/websearch.js';
+import { detectIntent } from './utils/intents.js';
+import { Session } from './session.js';
 
 function parseArgs(args) {
   const parsed = { provider: null, model: null, clear: false, version: false, help: false };
@@ -37,15 +40,61 @@ Commands (in chat):
   /model [name]           Switch model
   /provider [name]        Switch provider
   /models                 List available models
+  /search <query>         Search the web and summarize with AI
   /clear                  Clear screen
+  /new                    Start a new conversation (clear memory)
   /quit                   Exit
 `);
 }
 
+async function answerWithSearch(messages, provider, query) {
+  Terminal.printInfo('Searching the web...');
+  const results = await WebSearch.search(query);
+
+  if (results.length === 0) {
+    Terminal.printError('No results found for the query.');
+    return;
+  }
+
+  Terminal.printSearchResults(results);
+
+  const searchContext = results.map((r, i) =>
+    `${i + 1}. ${r.title}\n   URL: ${r.url}\n   ${r.snippet}`
+  ).join('\n\n');
+
+  messages.push({
+    role: 'user',
+    content: `Web search results for "${query}":\n\n${searchContext}\n\nPlease summarize these results and answer based on them. Be concise.`,
+  });
+  Session.save(messages);
+
+  Terminal.printAIStart();
+  let response = '';
+  try {
+    for await (const chunk of provider.stream(messages, {
+      model: Config.get('model'),
+      systemPrompt,
+    })) {
+      Terminal.printAIChunk(chunk);
+      response += chunk;
+    }
+  } catch (err) {
+    Terminal.printError(err.message || 'Stream error');
+  }
+  Terminal.printAIEnd();
+  if (response) {
+    messages.push({ role: 'assistant', content: response });
+  }
+}
+
 async function chat(provider, systemPrompt) {
-  const messages = [];
+  const messages = Session.load();
   let exited = false;
 
+  if (messages.length > 0) {
+    Terminal.printInfo(`Resumed ${messages.filter((m) => m.role === 'user').length} previous message(s).`);
+  }
+
   const rl = createInterface({
     input: process.stdin,
     output: process.stdout,
@@ -61,7 +110,7 @@ async function chat(provider, systemPrompt) {
 
   const prompt = () => new Promise((resolve) => {
     rl.question(
-      `${Branding.colors.bold}${Branding.colors.primary}${Branding.USER_PROMPT} >${Branding.colors.reset} `,
+      `\n${Branding.colors.bold}${Branding.colors.primary}┌ ${Branding.USER_PROMPT}${Branding.colors.reset} `,
       resolve
     );
   });
@@ -90,7 +139,7 @@ async function chat(provider, systemPrompt) {
 
     if (cmd === '/clear') {
       Terminal.clear();
-      showBanner(provider.name, Config.get('model'));
+      showBanner(provider.name, Config.get('model'), provider.displayName);
       continue;
     }
 
@@ -99,6 +148,13 @@ async function chat(provider, systemPrompt) {
       continue;
     }
 
+    if (cmd === '/new') {
+      Session.clear();
+      messages.length = 0;
+      Terminal.printSuccess('Started a new conversation.');
+      continue;
+    }
+
     if (cmd === '/models') {
       const models = provider.name === 'ollama'
         ? Catalog.getLocalModels()
@@ -146,7 +202,25 @@ async function chat(provider, systemPrompt) {
       continue;
     }
 
+    if (cmd.startsWith('/search ')) {
+      const query = trimmed.slice(8).trim();
+      if (!query) {
+        Terminal.printWarning('Usage: /search <query>');
+        continue;
+      }
+      await answerWithSearch(messages, provider, query);
+      continue;
+    }
+
+    const intent = detectIntent(trimmed);
+    if (intent.type === 'knowledge') {
+      Terminal.printSuccess(`Auto-search: "${trimmed}"`);
+      await answerWithSearch(messages, provider, trimmed);
+      continue;
+    }
+
     messages.push({ role: 'user', content: trimmed });
+    Session.save(messages);
     Terminal.printAIStart();
 
     let response = '';
@@ -166,6 +240,7 @@ async function chat(provider, systemPrompt) {
     if (response) {
       messages.push({ role: 'assistant', content: response });
     }
+    Session.save(messages);
   }
 }
 
@@ -198,7 +273,7 @@ export async function run() {
   Config.save({ provider: provider.name });
 
   if (args.clear) Terminal.clear();
-  showBanner(provider.name, Config.get('model'));
+  showBanner(provider.name, Config.get('model'), provider.displayName);
 
   const systemPrompt = loadSystemPrompt();
   await chat(provider, systemPrompt);

package/src/providers/base.js

@@ -3,6 +3,10 @@ export class BaseProvider {
     this.name = name;
   }
 
+  get displayName() {
+    return 'Cortex';
+  }
+
   async *stream(messages, options = {}) {
     throw new Error('stream() must be implemented');
   }

package/src/providers/ollama.js

@@ -8,6 +8,10 @@ export class OllamaProvider extends BaseProvider {
     this.host = Config.get('ollamaHost');
   }
 
+  get displayName() {
+    return 'Cortex';
+  }
+
   async isAvailable() {
     try {
       const resp = await fetch(`${this.host}/api/tags`, { signal: AbortSignal.timeout(3000) });

package/src/providers/openai.js

@@ -7,6 +7,10 @@ export class OpenAIProvider extends BaseProvider {
     super('openai');
   }
 
+  get displayName() {
+    return 'Nebula';
+  }
+
   async isAvailable() {
     const key = Config.get('openaiApiKey');
     if (!key) return false;

package/src/session.js

@@ -0,0 +1,41 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'fs';
+import { join } from 'path';
+import { Config } from './config.js';
+
+const SESSION_FILE = join(Config.getConfigDir(), 'history.json');
+const MAX_MESSAGES = 20;
+
+export const Session = {
+  load() {
+    try {
+      if (existsSync(SESSION_FILE)) {
+        const raw = readFileSync(SESSION_FILE, 'utf-8');
+        const data = JSON.parse(raw || '[]');
+        if (Array.isArray(data)) return data.slice(-MAX_MESSAGES);
+      }
+    } catch {}
+    return [];
+  },
+
+  save(messages) {
+    try {
+      const dir = Config.getConfigDir();
+      if (!existsSync(dir)) mkdirSync(dir, { recursive: true });
+      writeFileSync(SESSION_FILE, JSON.stringify(messages.slice(-MAX_MESSAGES), null, 2));
+    } catch {}
+  },
+
+  clear() {
+    try {
+      if (existsSync(SESSION_FILE)) writeFileSync(SESSION_FILE, '[]');
+    } catch {}
+  },
+
+  hasHistory() {
+    try {
+      return existsSync(SESSION_FILE) && (JSON.parse(readFileSync(SESSION_FILE, 'utf-8') || '[]').length > 0);
+    } catch {
+      return false;
+    }
+  },
+};

package/src/ui/banner.js

@@ -2,6 +2,7 @@ import { readFileSync } from 'fs';
 import { join, dirname } from 'path';
 import { fileURLToPath } from 'url';
 import { Branding } from './branding.js';
+import { Terminal } from './terminal.js';
 
 const __dirname = dirname(fileURLToPath(import.meta.url));
 
@@ -15,17 +16,22 @@ export function loadSystemPrompt() {
   }
 }
 
-export function showBanner(provider, model) {
-  console.log(Branding.colors.primary + Branding.BANNER + Branding.colors.reset);
-  console.log(
-    `  ${Branding.colors.dim}Intelligent Terminal Assistant${Branding.colors.reset}`
-  );
-  console.log(
-    `  ${Branding.colors.accent}Model:${Branding.colors.reset} ${model}  ` +
-    `${Branding.colors.accent}Engine:${Branding.colors.reset} ${provider}`
-  );
-  console.log(
-    `  ${Branding.colors.dim}Type /help for commands, /quit to exit${Branding.colors.reset}`
-  );
+export function showBanner(providerName, model, providerDisplayName) {
+  const C = Branding.colors;
+  const width = 44;
+  const border = `  ╔${'═'.repeat(width)}╗`;
+
+  console.log();
+  console.log(`${C.primary}${Branding.BANNER}${C.reset}`);
+  console.log(`  ${C.bold}${C.accent}▸ Vierratale AI ▸ Cortex Engine${C.reset}`);
+  console.log();
+  console.log(border);
+  console.log(`  ║  ${C.bold}${C.primary}Model${C.reset}${' '.repeat(width - 12)}║`);
+  console.log(`  ║   ${C.accent}${model}${C.reset}${' '.repeat(Math.max(1, width - 5 - model.length))}║`);
+  console.log(`  ║  ${C.bold}${C.primary}Engine${C.reset}${' '.repeat(width - 13)}║`);
+  console.log(`  ║   ${C.accent}${providerDisplayName || 'Cortex'}${C.reset}${' '.repeat(Math.max(1, width - 7 - (providerDisplayName || 'Cortex').length))}║`);
+  console.log(`  ╚${'═'.repeat(width)}╝`);
+  console.log();
+  console.log(`  ${C.dim}Type /help for commands · /quit to exit${C.reset}`);
   console.log();
 }

package/src/ui/branding.js

@@ -1,6 +1,6 @@
 export const Branding = {
   APP_NAME: 'VierrataleAI',
-  VERSION: '0.1.0-beta.1',
+  VERSION: '0.1.0-beta.3',
 
   colors: {
     primary: '\x1b[38;2;124;58;237m',

package/src/ui/terminal.js

@@ -3,25 +3,56 @@ import { Branding } from './branding.js';
 const C = Branding.colors;
 
 export const Terminal = {
+  divider(char = '─', length = 50) {
+    return `${C.dim}${char.repeat(length)}${C.reset}`;
+  },
+
   printUser(text) {
-    console.log(`${C.bold}${C.primary}${Branding.USER_PROMPT} >${C.reset} ${text}`);
+    const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' });
+    console.log();
+    console.log(`${this.divider('─', 4)} ${C.bold}${C.primary}YOU${C.reset} ${C.dim}${time}${C.reset} ${this.divider('─', 34)}`);
+    console.log(`${C.primary}┃${C.reset} ${text}`);
   },
 
   printAIStart() {
-    process.stdout.write(`${C.bold}${C.accent}${Branding.AI_PROMPT} >${C.reset} `);
+    const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' });
+    this.spinner = { frame: 0, timer: null };
+    console.log();
+    console.log(`${this.divider('─', 4)} ${C.bold}${C.accent}${Branding.AI_PROMPT}${C.reset} ${C.dim}${time}${C.reset} ${this.divider('─', 34)}`);
+    this.curLine = '';
+    this._printPrefix();
+  },
+
+  _printPrefix() {
+    process.stdout.write(`${C.accent}┃ ${C.reset}`);
   },
 
   printAIChunk(text) {
+    this.curLine = (this.curLine || '') + text;
     process.stdout.write(text);
+    this._maybeWrap();
+  },
+
+  _maybeWrap() {
+    if (this.curLine && this.curLine.length > 100) {
+      const words = this.curLine.split(' ');
+      if (words.length > 1) {
+        const lastWord = words[words.length - 1];
+        const rest = this.curLine.slice(0, this.curLine.length - lastWord.length).trimEnd();
+        process.stdout.write(`\n${C.accent}┃ ${C.reset}`);
+        this.curLine = lastWord;
+      }
+    }
   },
 
   printAIEnd() {
-    console.log();
+    process.stdout.write('\n');
+    console.log(`${C.accent}┗${C.reset}${this.divider('─', 44)}`);
     console.log();
   },
 
   printError(msg) {
-    console.error(`${C.error}Error: ${msg}${C.reset}`);
+    console.log(`${C.error}✖ ${msg}${C.reset}`);
   },
 
   printInfo(msg) {
@@ -29,11 +60,23 @@ export const Terminal = {
   },
 
   printSuccess(msg) {
-    console.log(`${C.success}${msg}${C.reset}`);
+    console.log(`${C.success}✔ ${msg}${C.reset}`);
   },
 
   printWarning(msg) {
-    console.log(`${C.warning}${msg}${C.reset}`);
+    console.log(`${C.warning}⚠ ${msg}${C.reset}`);
+  },
+
+  printSearchResults(results) {
+    console.log(`${C.warning}┌── Web Search Results ──${C.reset}`);
+    results.forEach((r, i) => {
+      console.log(
+        `  ${C.warning}${i + 1}.${C.reset} ${C.bold}${r.title}${C.reset}`
+      );
+      console.log(`     ${C.dim}${r.url}${C.reset}`);
+      if (r.snippet) console.log(`     ${r.snippet.slice(0, 120)}`);
+    });
+    console.log(`${C.warning}└──${C.reset}`);
   },
 
   clear() {
@@ -44,6 +87,6 @@ export const Terminal = {
     return text
       .replace(/\*\*(.*?)\*\*/g, `${C.bold}$1${C.reset}`)
       .replace(/`([^`]+)`/g, `${C.accent}\`$1\`${C.reset}`)
-      .replace(/```(\w*)\n([\s\S]*?)```/g, `${C.dim}┌─$1─┐${C.reset}\n$2${C.dim}└─────┘${C.reset}`);
+      .replace(/```(\w*)\n([\s\S]*?)```/g, `${C.dim}┌─${C.reset}\n$2${C.dim}└─────${C.reset}`);
   },
 };

package/src/utils/intents.js

@@ -0,0 +1,30 @@
+const QUESTION_PREFIXES = [
+  'who', 'what', 'when', 'where', 'why', 'how',
+  'which', 'whose', 'whom', 'is', 'are', 'was', 'were',
+  'can', 'could', 'does', 'do', 'did', 'would', 'should', 'will',
+];
+
+const KNOWLEDGE_PREFIXES = [
+  'tell me about', "what is", "what's", "who is", "who's",
+  'what are', 'who was', 'when did', 'when was', 'where is',
+  'where was', 'how did', 'how does', 'how to', 'why did', 'why is',
+  'latest', 'news', 'current',
+];
+
+export function detectIntent(text) {
+  const lower = text.toLowerCase().trim();
+  if (lower.startsWith('/')) return { type: 'command' };
+
+  const isQuestion = /\?$/.test(lower);
+  const firstWord = lower.split(/\s+/)[0] || '';
+
+  let needsWeb = false;
+  if (isQuestion && QUESTION_PREFIXES.includes(firstWord)) {
+    needsWeb = true;
+  }
+  if (KNOWLEDGE_PREFIXES.some((p) => lower.startsWith(p + ' ') || lower === p)) {
+    needsWeb = true;
+  }
+
+  return { type: needsWeb ? 'knowledge' : 'chat' };
+}

package/src/utils/websearch.js

@@ -0,0 +1,93 @@
+const DDG_SEARCH = 'https://html.duckduckgo.com/html/';
+const WIKI_SEARCH = 'https://en.wikipedia.org/w/api.php';
+const UA = 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36';
+
+export class WebSearch {
+  static async search(query, maxResults = 5) {
+    let results = await this._searchDuckDuckGo(query, maxResults);
+    if (results.length === 0) {
+      results = await this._searchWikipedia(query, maxResults);
+    }
+    return results;
+  }
+
+  static async _searchDuckDuckGo(query, maxResults) {
+    const url = `${DDG_SEARCH}?q=${encodeURIComponent(query)}`;
+    try {
+      const resp = await fetch(url, {
+        headers: { 'User-Agent': UA },
+        signal: AbortSignal.timeout(10000),
+      });
+      if (!resp.ok) return [];
+      const html = await resp.text();
+      return this._parseDuckDuckGo(html, maxResults);
+    } catch {
+      return [];
+    }
+  }
+
+  static async _searchWikipedia(query, maxResults) {
+    const params = new URLSearchParams({
+      action: 'query',
+      list: 'search',
+      srsearch: query,
+      srlimit: String(maxResults),
+      format: 'json',
+      utf8: '1',
+    });
+    const url = `${WIKI_SEARCH}?${params.toString()}`;
+    try {
+      const resp = await fetch(url, {
+        headers: { 'User-Agent': UA },
+        signal: AbortSignal.timeout(10000),
+      });
+      if (!resp.ok) return [];
+      const data = await resp.json();
+      const search = data?.query?.search || [];
+      return search.map((r) => ({
+        title: r.title,
+        url: `https://en.wikipedia.org/wiki/${encodeURIComponent(r.title.replace(/ /g, '_'))}`,
+        snippet: this._stripTags(r.snippet || ''),
+        source: 'wikipedia',
+      }));
+    } catch {
+      return [];
+    }
+  }
+
+  static _parseDuckDuckGo(html, maxResults) {
+    const results = [];
+    const resultRegex = /<a[^>]*class="result__a"[^>]*href="([^"]*)"[^>]*>(.*?)<\/a>[\s\S]*?<a[^>]*class="result__snippet"[^>]*>(.*?)<\/a>/gi;
+    let match;
+    let count = 0;
+    while ((match = resultRegex.exec(html)) !== null && count < maxResults) {
+      let url = match[1];
+      const snippet = this._stripTags(match[3] || '');
+      const title = this._stripTags(match[2] || '');
+      url = this._decodeDdgUrl(url);
+      if (url && !url.includes('duckduckgo.com/y.js')) {
+        results.push({ title, url, snippet, source: 'duckduckgo' });
+        count++;
+      }
+    }
+    return results;
+  }
+
+  static _decodeDdgUrl(url) {
+    const match = url.match(/uddg=([^&]+)/);
+    return match ? decodeURIComponent(match[1]) : url;
+  }
+
+  static _stripTags(text) {
+    const map = {
+      '&amp;': '&', '&lt;': '<', '&gt;': '>', '&quot;': '"',
+      '&#039;': "'", '&#39;': "'", '&apos;': "'", '&nbsp;': ' ',
+      '&#x27;': "'", '&hellip;': '...',
+    };
+    return String(text)
+      .replace(/<[^>]*>/g, '')
+      .replace(/&[a-zA-Z0-9#]+;/g, (m) => map[m] ?? '')
+      .replace(/\s{2,}/g, ' ')
+      .trim();
+  }
+}