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

package/README.md

@@ -40,6 +40,7 @@ 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
 - `/quit` - Exit
 

package/package.json

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

package/src/cli.js

@@ -6,6 +6,7 @@ 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';
 
 function parseArgs(args) {
   const parsed = { provider: null, model: null, clear: false, version: false, help: false };
@@ -37,6 +38,7 @@ 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
   /quit                   Exit
 `);
@@ -61,7 +63,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 +92,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;
     }
 
@@ -146,6 +148,52 @@ async function chat(provider, systemPrompt) {
       continue;
     }
 
+    if (cmd.startsWith('/search ') || cmd === '/search') {
+      const query = cmd === '/search' ? '' : trimmed.slice(8).trim();
+      if (!query) {
+        Terminal.printWarning('Usage: /search <query>');
+        continue;
+      }
+
+      Terminal.printInfo('Searching the web...');
+      const results = await WebSearch.search(query);
+
+      if (results.length === 0) {
+        Terminal.printError('No results found for the query.');
+        continue;
+      }
+
+      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.`,
+      });
+
+      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 });
+      }
+      continue;
+    }
+
     messages.push({ role: 'user', content: trimmed });
     Terminal.printAIStart();
 
@@ -198,7 +246,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/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.2',
 
   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/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) {
+    return String(text)
+      .replace(/<[^>]*>/g, '')
+      .replace(/&amp;/g, '&')
+      .replace(/&lt;/g, '<')
+      .replace(/&gt;/g, '>')
+      .replace(/&quot;/g, '"')
+      .replace(/&#x27;/g, "'")
+      .replace(/&#39;/g, "'")
+      .replace(/&nbsp;/g, ' ')
+      .trim();
+  }
+}