jaitus 0.3.2 → 0.3.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jaitus",
-  "version": "0.3.2",
+  "version": "0.3.3",
   "description": "Jaitus — a coding agent in your terminal.",
   "type": "module",
   "bin": {
@@ -16,7 +16,7 @@
     "node": ">=20"
   },
   "scripts": {
-    "test": "node tests/source-sanity.js && node tests/rules.js && node tests/run.js && node tests/commands.js && node tests/login.js && node tests/session.js && node tests/openUrl.js && node tests/ink-tui.js && node tests/tui-render.js && node tests/mouse-e2e.js && node tests/environment.js && node tests/update.js && node tests/continuation.js",
+    "test": "node tests/source-sanity.js && node tests/rules.js && node tests/run.js && node tests/commands.js && node tests/login.js && node tests/session.js && node tests/openUrl.js && node tests/ink-tui.js && node tests/tui-render.js && node tests/mouse-e2e.js && node tests/environment.js && node tests/update.js && node tests/continuation.js && node tests/context-budget.js",
     "preview": "node scripts/tui-preview.js",
     "prepublishOnly": "npm test"
   },

package/src/api.js

@@ -10,6 +10,10 @@ import { fromGatewayBody, fromNetwork, JaitusError } from './errors.js';
 
 export class NotSignedIn extends Error {}
 
+// The context window the gateway last advertised, or null before it has been asked.
+let contextWindowSeen = null;
+export const advertisedContextWindow = () => contextWindowSeen;
+
 export function createClient() {
   const { baseUrl } = endpoints();
   if (!signedIn()) throw new NotSignedIn('Not signed in. Run `jaitus login` first.');
@@ -63,6 +67,9 @@ export function createClient() {
       const body = await res.json().catch(() => ({}));
       if (!res.ok) throw fromGatewayBody(res.status, body, { requestId: res.headers.get('x-request-id') || '' });
       const models = (body?.data || []).map((m) => m?.id).filter(Boolean);
+      // How much conversation fits, as the gateway computed it for this account.
+      const ctx = (body?.data || []).map((m) => Number(m?.context_window)).find((n) => n > 0);
+      if (ctx) contextWindowSeen = ctx;
       // An empty list is a service fault, not something the user can fix — and naming
       // the gateway's model map to them would be naming our internals.
       if (!models.length) throw new JaitusError({ code: 'unavailable', status: 502 });

package/src/index.js

@@ -9,7 +9,7 @@
 import path from 'node:path';
 import fs from 'node:fs';
 import { spawn, spawnSync } from 'node:child_process';
-import { createClient, NotSignedIn } from './api.js';
+import { createClient, NotSignedIn, advertisedContextWindow } from './api.js';
 import { fromResponse, isJaitusError } from './errors.js';
 import { signedIn } from './session.js';
 import {
@@ -300,10 +300,18 @@ function gitBranch(cwd) {
   }
 }
 
-// Used only to fill the context meter. The real ceiling is the model's, which the
-// gateway does not advertise; this is the conservative floor of what any model it
-// serves accepts, so the bar reads "getting full" before anything actually breaks.
-const CONTEXT_BUDGET = 180_000;
+// How much conversation a session carries before it is compacted. The gateway works it
+// out from the providers' real limits and the admin's cap, and advertises it on its
+// model list; this floor is only used until that answer arrives, or if it never does.
+const CONTEXT_FALLBACK = 128_000;
+let CONTEXT_BUDGET = CONTEXT_FALLBACK;
+async function loadContextBudget(client) {
+  try {
+    if (!advertisedContextWindow()) await client.models();
+    CONTEXT_BUDGET = advertisedContextWindow() || CONTEXT_FALLBACK;
+  } catch { /* keep the floor */ }
+  return CONTEXT_BUDGET;
+}
 
 // Compact at this fraction of the budget. Low enough that the summarisation request
 // itself still fits comfortably, high enough that a normal session never sees it.
@@ -548,6 +556,7 @@ async function session({ flags }) {
   let [model, device] = await Promise.all([
     resolveModel(client, flags.model),
     probeDevice({}).catch(() => ({ vt: !!process.stdout.isTTY, sixel: false })),
+    loadContextBudget(client),
   ]);
   // Degrade on EVIDENCE, not on silence.
   //