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

package/README.md

@@ -42,8 +42,26 @@ vierrataleai --model vierratale-pro   # Use specific model
 - `/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.2",
+  "version": "0.1.0-beta.3",
   "description": "VierrataleAI - Intelligent terminal assistant",
   "type": "module",
   "bin": {

package/src/cli.js

@@ -7,6 +7,8 @@ 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 };
@@ -40,14 +42,59 @@ Commands (in chat):
   /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,
@@ -101,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()
@@ -148,53 +202,25 @@ async function chat(provider, systemPrompt) {
       continue;
     }
 
-    if (cmd.startsWith('/search ') || cmd === '/search') {
-      const query = cmd === '/search' ? '' : trimmed.slice(8).trim();
+    if (cmd.startsWith('/search ')) {
+      const query = trimmed.slice(8).trim();
       if (!query) {
         Terminal.printWarning('Usage: /search <query>');
         continue;
       }
+      await answerWithSearch(messages, provider, 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 });
-      }
+    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 = '';
@@ -214,6 +240,7 @@ async function chat(provider, systemPrompt) {
     if (response) {
       messages.push({ role: 'assistant', content: response });
     }
+    Session.save(messages);
   }
 }
 

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/branding.js

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

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

@@ -79,15 +79,15 @@ export class WebSearch {
   }
 
   static _stripTags(text) {
+    const map = {
+      '&amp;': '&', '&lt;': '<', '&gt;': '>', '&quot;': '"',
+      '&#039;': "'", '&#39;': "'", '&apos;': "'", '&nbsp;': ' ',
+      '&#x27;': "'", '&hellip;': '...',
+    };
     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, ' ')
+      .replace(/&[a-zA-Z0-9#]+;/g, (m) => map[m] ?? '')
+      .replace(/\s{2,}/g, ' ')
       .trim();
   }
 }