aegis-desktop 0.4.3 → 0.4.4

package/lib/local/engine.js

@@ -473,6 +473,17 @@ function createLocalEngine({ aegis, settings, ollama, providers, tools, promptBu
 
   async function listModels(cls) {
     if (cls === 'aegis') {
+      // The catalog is behind the account's key: GET /api/v1/models answers
+      // `401 {"error":{"message":"No API key"}}` without one (verified against
+      // aegiscloud.org). Aegis Cloud is the *default* class, so firing the call
+      // anyway painted a raw "listModels failed: … 401" over the default model
+      // picker for every user who had just installed the app and not yet
+      // pasted a key — the one state where the UI must say what unblocks it and
+      // not what went wrong. Report the missing key as a state (`needsKey`) and
+      // let the renderer invite the user to connect; nothing else about the
+      // class changes, and the model dropdown keeps its "server default (auto)"
+      // entry so the class is usable the moment a key lands.
+      if (!aegis.apiKey) return { class: cls, models: [], needsKey: true };
       const data = await aegis.listModels();
       return { class: cls, models: filterAegisCatalog(normalizeCatalog(data && data.models)) };
     }
@@ -535,7 +546,7 @@ function createLocalEngine({ aegis, settings, ollama, providers, tools, promptBu
         model: opts.model,
         mode: opts.mode,
         maxTokens: opts.maxTokens,
-        stream: true,
+        stream: opts.stream !== false,
         // The pooled (Nexus) brain is streamed, and an OpenAI-compatible SSE
         // stream reports no token usage unless asked. Without this the Aegis
         // Cloud class — the desktop's default — was the one class that answered
@@ -699,6 +710,11 @@ function createLocalEngine({ aegis, settings, ollama, providers, tools, promptBu
         workers: payload && payload.workers,
         onReasoning,
         idleTimeoutMs: autonomous ? AUTONOMOUS_IDLE_TIMEOUT_MS : undefined,
+        // A caller with no live streaming surface (a `--no-stream` CLI flag, a
+        // one-shot script) can ask for the buffered non-stream wire form
+        // instead. Undefined/anything but `false` keeps every existing caller
+        // (the desktop renderer never sets this) on the streamed path.
+        stream: payload && payload.stream === false ? false : true,
       };
 
       // No round cap: a model that keeps calling tools keeps going for as
@@ -782,6 +798,15 @@ function createLocalEngine({ aegis, settings, ollama, providers, tools, promptBu
         }
         addUsage(res);
 
+        // A cancelled turn is over. Both recoveries below exist for "the model
+        // said nothing", and an aborted request resolves with exactly that —
+        // no text — so without this guard a user pressing Esc to stop a slow
+        // turn immediately issued ANOTHER billed provider call (and, because
+        // every transport concatenates the deltas it forwards, streamed the
+        // partial answer a second time onto the same bubble). Measured before
+        // the guard: one interrupted turn == two dispatches.
+        if (signal.aborted) return withTurnUsage(res);
+
         // Budget exhausted before the answer was written. Doubling it costs
         // one request and converts a dead turn into a real one; a second
         // 'length' result is accepted as-is so a hard-capped model can't

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "aegis-desktop",
   "productName": "AEGIS Desktop",
-  "version": "0.4.3",
+  "version": "0.4.4",
   "description": "Thin Electron host for AEGIS — a local chat UI over the shared client/aegis.js transport. Ships transport + UI only; engine logic stays server-side.",
   "author": {
     "name": "AEGIS Code",

package/renderer/app.js

@@ -171,6 +171,11 @@ const CUSTOM_MODEL_PRESETS = {
 // already filters out of settings.list(); never render it as a provider row
 // even if a stale store still surfaces it (defect #1).
 const RESERVED_PROVIDERS = new Set(['__aegis', 'aegis']);
+// Where a user without a key gets one. The class picker defaults to Aegis Cloud
+// and the catalog is key-gated, so this is the first thing a new install needs;
+// it lives here rather than inline so the Model hint and any future "connect"
+// affordance cannot drift to two different pages.
+const GET_AEGIS_KEY_URL = 'https://aegiscloud.org';
 
 let pendingEl = null;
 let pendingSessionId = null;
@@ -1858,6 +1863,11 @@ async function loadModels(cls) {
   try {
     const data = await models.listModels(cls);
     const list = Array.isArray(data && data.models) ? data.models : [];
+    // No key on the default class: the engine reports the missing credential as
+    // a state instead of letting the catalog call 401 (see engine.listModels).
+    // The hint is where a user finds out they can connect at all — a raw
+    // "listModels failed" told them only that something was broken.
+    const needsKey = Boolean(data && data.needsKey);
     for (const m of list) {
       modelMeta.set(m.id, m);
       const opt = document.createElement('option');
@@ -1866,12 +1876,28 @@ async function loadModels(cls) {
       els.modelSelect.appendChild(opt);
     }
     let hint;
-    if (!list.length) {
+    if (needsKey) {
+      hint = null; // carries a link, built below
+    } else if (!list.length) {
       hint = cls === 'ollama' ? 'Ollama not running or no models pulled.' : 'No models listed.';
     } else {
       hint = `${list.length} model${list.length === 1 ? '' : 's'} available.`;
     }
     const ceiling = applyMaxTokensClamp(els.modelSelect.value);
+    if (needsKey) {
+      // The hint elements are bare <p>s, so the link has to be a real child
+      // node — a text assignment would wipe it (same shape as capNotice).
+      els.modelHint.textContent =
+        'Connect Aegis Cloud to load its models: paste your AEGIS API key in the Status card above (';
+      const a = document.createElement('a');
+      a.href = GET_AEGIS_KEY_URL;
+      a.target = '_blank';
+      a.rel = 'noreferrer noopener';
+      a.textContent = 'free key at aegiscloud.org';
+      els.modelHint.appendChild(a);
+      els.modelHint.appendChild(document.createTextNode('), then Save.'));
+      return;
+    }
     els.modelHint.textContent =
       ceiling < FLAT_CEILING ? `${hint} · max output: ${ceiling.toLocaleString()}` : hint;
   } catch (err) {