@vierratale/ai 0.1.0-beta.15 → 0.1.0-beta.16

package/src/config.js

@@ -1,34 +1,24 @@
 import { readFileSync, existsSync, mkdirSync, writeFileSync } from 'fs';
 import { join } from 'path';
 import { homedir } from 'os';
-import { Catalog } from './catalog.js';
 
-function configPaths() {
-  const dir = process.env.VIERRATALE_CONFIG_DIR
-    ? join(process.env.VIERRATALE_CONFIG_DIR)
-    : join(homedir(), '.config', 'vierrataleai');
-  return { dir, file: join(dir, 'config.json') };
-}
+const CONFIG_DIR = join(homedir(), '.config', 'vierrataleai');
+const CONFIG_FILE = join(CONFIG_DIR, 'config.json');
 
 const DEFAULTS = {
   provider: 'auto',
-  model: 'VTL-2.7-Flash',
+  model: 'vierratale-fast',
   engineHost: 'http://127.0.0.1:11434',
   numCtx: 2048,
   temperature: 0.7,
   maxTokens: 4096,
   keepAlive: '5m',
-  numThreads: 8,
-  commandTimeoutMs: 60000,
-  planTimeoutMs: 45000,
-  downloadDir: join(homedir(), 'Downloads'),
 };
 
 function loadConfigFile() {
   try {
-    const { file } = configPaths();
-    if (existsSync(file)) {
-      const raw = readFileSync(file, 'utf-8');
+    if (existsSync(CONFIG_FILE)) {
+      const raw = readFileSync(CONFIG_FILE, 'utf-8');
       return JSON.parse(raw);
     }
   } catch {}
@@ -54,15 +44,11 @@ export const Config = {
       temperature: parseFloat(getEnv('VIERRATALE_TEMPERATURE', String(file.temperature ?? DEFAULTS.temperature))),
       maxTokens: parseInt(getEnv('VIERRATALE_MAX_TOKENS', String(file.maxTokens ?? DEFAULTS.maxTokens))),
       keepAlive: getEnv('VIERRATALE_KEEP_ALIVE', file.keepAlive ?? DEFAULTS.keepAlive),
-      numThreads: parseInt(getEnv('VIERRATALE_NUM_THREADS', String(file.numThreads ?? DEFAULTS.numThreads))),
       openaiApiKey: getEnv('OPENAI_API_KEY', file.openaiApiKey || ''),
       anthropicApiKey: getEnv('ANTHROPIC_API_KEY', file.anthropicApiKey || ''),
       geminiApiKey: getEnv('GEMINI_API_KEY', file.geminiApiKey || ''),
       systemPrompt: file.systemPrompt || '',
       providerModels: file.providerModels || {},
-      downloadDir: getEnv('VIERRATALE_DOWNLOAD_DIR', file.downloadDir ?? DEFAULTS.downloadDir),
-      commandTimeoutMs: parseInt(getEnv('VIERRATALE_COMMAND_TIMEOUT_MS', String(file.commandTimeoutMs ?? DEFAULTS.commandTimeoutMs))),
-      planTimeoutMs: parseInt(getEnv('VIERRATALE_PLAN_TIMEOUT_MS', String(file.planTimeoutMs ?? DEFAULTS.planTimeoutMs))),
     };
     return this._config;
   },
@@ -90,22 +76,20 @@ export const Config = {
   },
 
   getEffectiveModel(provider) {
-    const raw = this.getProviderModel(provider) || this.get('model');
-    return Catalog.normalize(raw);
+    return this.getProviderModel(provider) || this.get('model');
   },
 
   save(overrides = {}) {
-    const { dir, file } = configPaths();
-    if (!existsSync(dir)) {
-      mkdirSync(dir, { recursive: true });
+    if (!existsSync(CONFIG_DIR)) {
+      mkdirSync(CONFIG_DIR, { recursive: true });
     }
     const current = this._config || DEFAULTS;
     const merged = { ...current, ...overrides };
-    writeFileSync(file, JSON.stringify(merged, null, 2));
+    writeFileSync(CONFIG_FILE, JSON.stringify(merged, null, 2));
     this._config = merged;
   },
 
   getConfigDir() {
-    return configPaths().dir;
+    return CONFIG_DIR;
   },
 };

package/src/index.js

@@ -1,7 +1,6 @@
 export { Catalog } from './catalog.js';
 export { Config } from './config.js';
-export { Engine } from './engine.js';
+export { Installer } from './installer.js';
 export { CortexProvider } from './providers/cortex.js';
 export { OpenAIProvider } from './providers/openai.js';
 export { Branding } from './ui/branding.js';
-export { FileWriter } from './utils/filewriter.js';

package/src/installer.js

@@ -0,0 +1,105 @@
+import { execSync, exec } from 'child_process';
+import { existsSync } from 'fs';
+import { Catalog } from './catalog.js';
+import { Config } from './config.js';
+
+function runSilent(cmd) {
+  return new Promise((resolve) => {
+    exec(cmd, { stdio: 'ignore' }, (err) => resolve(!err));
+  });
+}
+
+function isEngineRunning(host) {
+  return fetch(`${host}/api/tags`, { signal: AbortSignal.timeout(3000) })
+    .then((r) => r.ok)
+    .catch(() => false);
+}
+
+function isEngineInstalled() {
+  try {
+    execSync('which ollama 2>/dev/null', { stdio: 'ignore' });
+    return true;
+  } catch {}
+  const paths = ['/usr/local/bin/ollama', '/usr/bin/ollama'];
+  return paths.some((p) => existsSync(p));
+}
+
+async function installEngine() {
+  await runSilent('curl -fsSL https://ollama.com/install.sh | sh');
+  return isEngineInstalled();
+}
+
+async function pullModel(host, model) {
+  try {
+    const resp = await fetch(`${host}/api/pull`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ name: model, stream: false }),
+      signal: AbortSignal.timeout(600000),
+    });
+    return resp.ok;
+  } catch {
+    return false;
+  }
+}
+
+async function getInstalledModels(host) {
+  try {
+    const resp = await fetch(`${host}/api/tags`);
+    if (!resp.ok) return [];
+    const data = await resp.json();
+    return (data.models || []).map((m) => m.name);
+  } catch {
+    return [];
+  }
+}
+
+export const Installer = {
+  async pullModel(host, model) {
+    return pullModel(host, model);
+  },
+
+  async getInstalledModels(host) {
+    return getInstalledModels(host);
+  },
+
+  async ensure() {
+    const host = Config.get('engineHost');
+
+    if (!isEngineInstalled()) {
+      await installEngine();
+    }
+
+    const running = await isEngineRunning(host);
+    if (!running) {
+      try {
+        // No systemd available here, so launch the daemon directly, detached,
+        // and poll until it responds instead of relying on a service manager.
+        const child = exec('ollama serve', {
+          stdio: 'ignore',
+          detached: true,
+        });
+        if (child.unref) child.unref();
+        for (let i = 0; i < 30; i++) {
+          await new Promise((r) => setTimeout(r, 1000));
+          if (await isEngineRunning(host)) break;
+        }
+      } catch {}
+    }
+
+    const realModel = Catalog.getRealModel(Config.get('model'));
+    const installed = await getInstalledModels(host);
+    const needsPull = !installed.includes(realModel);
+
+    if (needsPull) {
+      await pullModel(host, realModel);
+    }
+
+    return true;
+  },
+
+  async isReady() {
+    const host = Config.get('engineHost');
+    return isEngineRunning(host);
+  },
+};

package/src/prompts/system.json

@@ -1,3 +1,3 @@
 {
-  "system": "You are VierrataleAI, an intelligent AI assistant created by Vierratale. You are helpful, concise, and accurate. You provide clear answers and help users with their questions and tasks. When asked about your identity, always respond that you are VierrataleAI, an AI assistant created by Vierratale. You CAN write files and folders directly to the user's disk. When the user asks you to create, save, or generate code, configs, scripts, projects, web pages, or any other files, DO NOT refuse — answer with the FILE:/FOLDER: format instead. Use this exact format for EACH file: a line \"FILE: relative/path/to/file\" on its own line, immediately followed by a markdown code fence containing the file content. This works for any file type, including JSON config files and HTML pages: write a config as \"FILE: config.json\" followed by a ```json fence, and a web page as \"FILE: index.html\" followed by a ```html fence with the full <!doctype html>... markup inside. Use paths relative to the user's working directory. Emit one FILE: block per file; nested folders are created automatically. If you need an empty folder, emit \"FOLDER: relative/path\" on its own line. When asked for a story, folk tale, legend, or myth, give the real, well-known version using the web search results you were given when available, and also save the full story to a .txt file using FILE: <title>.txt followed by a ```text fence. Keep any explanation short and outside of the FILE:/FOLDER: blocks."
-}
+  "system": "You are VierrataleAI, an intelligent AI assistant created by Vierratale. You are helpful, concise, and accurate. You provide clear answers and help users with their questions and tasks. When asked about your identity, always respond that you are VierrataleAI, an AI assistant created by Vierratale."
+}

package/src/prompts/system.md

@@ -13,45 +13,3 @@ You are **VierrataleAI**, an intelligent AI assistant created by Vierratale.
 - Help users with tasks including coding, writing, analysis, and general knowledge.
 - If you don't know something, say so honestly.
 - Use markdown formatting when appropriate for readability.
-
-## Writing Files and Folders
-
-You CAN write files and folders directly to the user's disk. When the user asks
-you to create, save, or generate code, configs, scripts, projects, or any other
-files, DO NOT refuse — answer with the format below. Use this exact format for
-EACH file:
-
-    FILE: relative/path/to/file.js
-    ```js
-    // file content here
-    ```
-
-This works for any file type, including JSON and HTML. For example, write a
-config file exactly like this:
-
-    FILE: config.json
-    ```json
-    {
-      "name": "my-app",
-      "port": 3000
-    }
-    ```
-
-And a web page like this:
-
-    FILE: index.html
-    ```html
-    <!doctype html>
-    <html>
-    <head><title>My Page</title></head>
-    <body><h1>Hello</h1></body>
-    </html>
-    ```
-
-Rules:
-- Put `FILE: <path>` on its own line, immediately followed by a markdown code fence containing the file content. No text between the header and the fence.
-- Use paths relative to the user's working directory (e.g. `src/app.js`, `config.json`).
-- Emit one `FILE:` block per file; nested folders are created automatically.
-- If you need an empty folder, emit `FOLDER: <path>` on its own line.
-- When asked for a story, folk tale, legend, or myth, give the real, well-known version — use the web search results you were given when available — and also save the full story to a .txt file using `FILE: <title>.txt` followed by a ```text fence.
-- Keep any explanation short and outside of the `FILE:`/`FOLDER:` blocks.

package/src/providers/anthropic.js

@@ -39,7 +39,7 @@ export class AnthropicProvider extends BaseProvider {
 
   async *stream(messages, options = {}) {
     const key = Config.get('anthropicApiKey');
-    if (!key) throw new Error('[ERR-0004] Cloud API key not configured. Add it to the config and try again.');
+    if (!key) throw new Error('Anthropic API key not configured');
 
     const model = this._resolveModel(options);
     const systemPrompt = options.systemPrompt || '';
@@ -51,80 +51,49 @@ export class AnthropicProvider extends BaseProvider {
       content: m.content,
     }));
 
-    const controller = new AbortController();
-    const connectTimer = setTimeout(() => controller.abort(), 240000);
-
-    let resp;
-    try {
-      resp = await fetch(ANTHROPIC_URL, {
-        method: 'POST',
-        headers: {
-          'Content-Type': 'application/json',
-          'x-api-key': key,
-          'anthropic-version': '2023-06-01',
-        },
-        body: JSON.stringify({
-          model,
-          system: system.length ? system : undefined,
-          messages: apiMessages,
-          max_tokens: options.maxTokens || Config.get('maxTokens'),
-          stream: true,
-          temperature: options.temperature || Config.get('temperature'),
-        }),
-        signal: controller.signal,
-      });
-    } catch (err) {
-      if (err.name === 'AbortError') {
-        throw new Error('[ERR-0001] Could not reach the API (no response within 60s). Check your network connection.');
-      }
-      throw new Error('[ERR-0001] Could not reach the API (no response). Check your network connection.');
-    } finally {
-      clearTimeout(connectTimer);
-    }
+    const resp = await fetch(ANTHROPIC_URL, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'x-api-key': key,
+        'anthropic-version': '2023-06-01',
+      },
+      body: JSON.stringify({
+        model,
+        system: system.length ? system : undefined,
+        messages: apiMessages,
+        max_tokens: options.maxTokens || Config.get('maxTokens'),
+        stream: true,
+        temperature: options.temperature || Config.get('temperature'),
+      }),
+    });
 
     if (!resp.ok) {
-      throw new Error('[ERR-0003] The cloud provider returned HTTP ' + resp.status + '. Check your API key in config.');
+      throw new Error('Failed to connect to cloud engine');
     }
 
     const reader = resp.body.getReader();
     const decoder = new TextDecoder();
     let buffer = '';
-    let idle = null;
-    const settle = () => {
-      if (idle) clearTimeout(idle);
-      idle = setTimeout(() => controller.abort(), 240000);
-    };
-    settle();
-    try {
-      while (true) {
-        let value, done;
+
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+
+      buffer += decoder.decode(value, { stream: true });
+      const lines = buffer.split('\n');
+      buffer = lines.pop() || '';
+
+      for (const line of lines) {
+        if (!line.startsWith('data: ')) continue;
+        const data = line.slice(6);
         try {
-          ({ done, value } = await reader.read());
-        } catch (err) {
-          if (err.name === 'AbortError') {
-            throw new Error('[ERR-0001] Lost the connection to the cloud API (no data for 60s).');
+          const json = JSON.parse(data);
+          if (json.type === 'content_block_delta' && json.delta?.text) {
+            yield json.delta.text;
           }
-          throw err;
-        }
-        if (done) break;
-        settle();
-        buffer += decoder.decode(value, { stream: true });
-        const lines = buffer.split('\n');
-        buffer = lines.pop() || '';
-
-        for (const line of lines) {
-          if (!line.startsWith('data: ')) continue;
-          const data = line.slice(6);
-          try {
-            const json = JSON.parse(data);
-            if (json.type === 'content_block_delta' && json.delta?.text) {
-              yield json.delta.text;
-            }
-          } catch {}
-        }
+        } catch {}
       }
-    } finally {
-      if (idle) clearTimeout(idle);
     }
   }
 }

package/src/providers/base.js

@@ -15,16 +15,6 @@ export class BaseProvider {
     throw new Error('stream() must be implemented');
   }
 
-  // Non-streaming convenience: collects the stream. Providers may override
-  // with a proper single-shot request (see CortexProvider).
-  async complete(messages, options = {}) {
-    let out = '';
-    for await (const chunk of this.stream(messages, options)) {
-      out += chunk;
-    }
-    return out;
-  }
-
   async listModels() {
     throw new Error('listModels() must be implemented');
   }