@adaptiqai/pi 0.1.0 → 0.1.1

package/lib/index.js

@@ -100,7 +100,7 @@ function classifyFailure(status, raw, headers, model2, options = {}) {
   if (effective === 503 && detail.code !== null && MODEL_OFFLINE_CODES.includes(detail.code)) {
     return make(CODES.server, `${model2 ?? "This model"} is offline right now. It will retry automatically.`, "retrying");
   }
-  if (has(detail, "AdaptiqNativePreflightBusy", "session_route_unavailable", "conversation_preparing", "preparing", "upload_capacity_exhausted") || effective === 503 && !has(detail, "offline")) {
+  if (has(detail, "AdaptiqNativePreflightBusy", "session_route_unavailable", "conversation_preparing", "preparing", "upload_capacity_exhausted", "AdaptiqMemoryNotReady") || effective === 425 || effective === 503 && !has(detail, "offline")) {
     return make(CODES.preparing, "AdaptiqAI is getting your conversation ready. It will retry automatically.", "preparing");
   }
   if (effective === 409 && has(detail, "AdaptiqNativePreflight", "conversation")) {
@@ -317,6 +317,91 @@ function detailMessage(raw) {
   }
   return null;
 }
+var EXPIRED = "The sign-in link expired. Start again.";
+async function startDeviceCode(client, request) {
+  let response;
+  try {
+    response = await client.fetch(`${client.apiBase}/v1/cli-auth/device`, {
+      method: "POST",
+      headers: { "content-type": "application/json", accept: "application/json" },
+      body: JSON.stringify({
+        client_app: request.app,
+        client_name: request.app,
+        device_name: request.device,
+        client_binding: request.binding,
+        ...request.tokenLifetime === void 0 ? {} : { token_lifetime: request.tokenLifetime }
+      }),
+      signal: client.signal === void 0 ? AbortSignal.timeout(2e4) : AbortSignal.any([client.signal, AbortSignal.timeout(2e4)]),
+      redirect: "error"
+    });
+  } catch (error) {
+    if (client.signal?.aborted === true)
+      throw error;
+    throw new Error(UNREACHABLE, { cause: error });
+  }
+  if (!response.ok) {
+    const message = detailMessage(await readJson(response));
+    if (response.status >= 500)
+      throw new Error("AdaptiqAI isn't answering right now. Try again in a minute.");
+    throw new Error(message ?? `AdaptiqAI sign-in could not start (HTTP ${response.status}).`);
+  }
+  const body = await readJson(response);
+  const deviceCode = String(body?.device_code ?? "");
+  const userCode = String(body?.user_code ?? "");
+  const url = String(body?.verification_uri_complete ?? body?.verification_uri ?? "");
+  if (deviceCode === "" || userCode === "" || !/^https?:\/\//u.test(url))
+    throw new Error("AdaptiqAI returned an incomplete sign-in.");
+  return {
+    deviceCode,
+    userCode,
+    url,
+    expiresAt: client.now() + Math.min(DEVICE_TIMEOUT_MS, Math.max(60, Number(body?.expires_in ?? 600)) * 1e3),
+    interval: Math.max(1, Number(body?.interval ?? 2)) * 1e3
+  };
+}
+async function pollDeviceCode(client, code, extra = {}) {
+  let response;
+  try {
+    response = await client.fetch(`${client.apiBase}/v1/cli-auth/device/token`, {
+      method: "POST",
+      headers: { "content-type": "application/json", accept: "application/json" },
+      body: JSON.stringify({ ...extra, device_code: code.deviceCode }),
+      signal: client.signal === void 0 ? AbortSignal.timeout(2e4) : AbortSignal.any([client.signal, AbortSignal.timeout(2e4)]),
+      redirect: "error"
+    });
+  } catch {
+    return { state: "waiting", delayMs: code.interval * 2 };
+  }
+  if (response.status === 202)
+    return { state: "waiting", delayMs: code.interval };
+  if (response.status === 429) {
+    const retry = Number(response.headers.get("retry-after") ?? 0) * 1e3;
+    return { state: "waiting", delayMs: Math.max(code.interval, retry) + 250 };
+  }
+  if (response.status === 403) {
+    const raw = await readJson(response);
+    return { state: "denied", message: detailCode(raw) === "access_denied" ? "The connection was declined in the browser." : detailMessage(raw) ?? "The connection was declined." };
+  }
+  if (response.status === 410 || response.status === 404)
+    return { state: "expired", message: EXPIRED };
+  if (!response.ok) {
+    if (response.status >= 500)
+      return { state: "waiting", delayMs: code.interval * 2 };
+    return { state: "failed", message: detailMessage(await readJson(response)) ?? `Sign-in failed (HTTP ${response.status}).` };
+  }
+  const body = await readJson(response);
+  const access = String(body?.access_token ?? "");
+  const refresh = String(body?.refresh_token ?? "");
+  if (!access.startsWith("ak_") || refresh !== "" && !refresh.startsWith("akr_"))
+    return { state: "failed", message: "AdaptiqAI returned an invalid credential." };
+  return {
+    state: "approved",
+    access,
+    refresh: refresh === "" ? null : refresh,
+    expiresIn: Math.max(60, Number(body?.expires_in ?? 3600)),
+    workspaceLabel: typeof body?.workspace_label === "string" && body.workspace_label !== "" ? body.workspace_label : "Your workspace"
+  };
+}
 var AuthManager = class {
   options;
   fetch;
@@ -359,6 +444,9 @@ var AuthManager = class {
     this.state = phase;
     this.options.onChange?.();
   }
+  client(signal) {
+    return { apiBase: this.options.apiBase, fetch: this.fetch, now: this.now, signal };
+  }
   /** Read the stored state once at startup. */
   async init() {
     const stored = await this.options.store.read();
@@ -492,39 +580,11 @@ var AuthManager = class {
     const clientProof = `${randomUUID()}${randomUUID()}`;
     const binding = createHash("sha256").update(clientProof).digest("hex");
     const app = this.options.appName ?? "DeepSeek Harness";
-    let response;
-    try {
-      response = await this.fetch(`${this.options.apiBase}/v1/cli-auth/device`, {
-        method: "POST",
-        headers: { "content-type": "application/json", accept: "application/json" },
-        body: JSON.stringify({
-          client_app: app,
-          client_name: app,
-          device_name: deviceLabel(this.options.deviceName ?? hostname()),
-          client_binding: binding
-        }),
-        signal: AbortSignal.any([flow.signal, AbortSignal.timeout(2e4)]),
-        redirect: "error"
-      });
-    } catch (error) {
-      if (flow.signal.aborted)
-        throw error;
-      throw new Error(UNREACHABLE, { cause: error });
-    }
-    if (!response.ok) {
-      const message = detailMessage(await readJson(response));
-      if (response.status >= 500)
-        throw new Error("AdaptiqAI isn't answering right now. Try again in a minute.");
-      throw new Error(message ?? `AdaptiqAI sign-in could not start (HTTP ${response.status}).`);
-    }
-    const body = await readJson(response);
-    const deviceCode = String(body?.device_code ?? "");
-    const userCode = String(body?.user_code ?? "");
-    const url = String(body?.verification_uri_complete ?? body?.verification_uri ?? "");
-    if (deviceCode === "" || userCode === "" || !/^https?:\/\//u.test(url))
-      throw new Error("AdaptiqAI returned an incomplete sign-in.");
-    const expiresAt = this.now() + Math.min(DEVICE_TIMEOUT_MS, Math.max(60, Number(body?.expires_in ?? 600)) * 1e3);
-    const interval = Math.max(1, Number(body?.interval ?? 2)) * 1e3;
+    const { deviceCode, userCode, url, expiresAt, interval } = await startDeviceCode(this.client(flow.signal), {
+      app,
+      device: deviceLabel(this.options.deviceName ?? hostname()),
+      binding
+    });
     this.pendingSignIn = { url, userCode, expiresAt };
     this.set("signing-in");
     void this.poll(flow, deviceCode, clientProof, interval, expiresAt);
@@ -536,61 +596,27 @@ var AuthManager = class {
       while (this.now() < expiresAt) {
         await this.sleep(delay, flow.signal);
         delay = interval;
-        let response;
-        try {
-          response = await this.fetch(`${this.options.apiBase}/v1/cli-auth/device/token`, {
-            method: "POST",
-            headers: { "content-type": "application/json", accept: "application/json" },
-            body: JSON.stringify({ device_code: deviceCode }),
-            signal: AbortSignal.any([flow.signal, AbortSignal.timeout(2e4)]),
-            redirect: "error"
-          });
-        } catch {
-          if (flow.signal.aborted)
-            return;
-          delay = interval * 2;
-          continue;
-        }
-        if (response.status === 202)
-          continue;
-        if (response.status === 429) {
-          const retry = Number(response.headers.get("retry-after") ?? 0) * 1e3;
-          delay = Math.max(interval, retry) + 250;
+        const answer = await pollDeviceCode(this.client(flow.signal), { deviceCode, interval });
+        if (answer.state === "waiting") {
+          delay = answer.delayMs;
           continue;
         }
-        if (response.status === 403) {
-          const raw = await readJson(response);
-          this.fail(flow, { code: "denied", message: detailCode(raw) === "access_denied" ? "The connection was declined in the browser." : detailMessage(raw) ?? "The connection was declined." });
+        if (answer.state !== "approved") {
+          this.fail(flow, { code: answer.state, message: answer.message });
           return;
         }
-        if (response.status === 410 || response.status === 404) {
-          this.fail(flow, { code: "expired", message: "The sign-in link expired. Start again." });
-          return;
-        }
-        if (!response.ok) {
-          if (response.status >= 500) {
-            delay = interval * 2;
-            continue;
-          }
-          this.fail(flow, { code: "failed", message: detailMessage(await readJson(response)) ?? `Sign-in failed (HTTP ${response.status}).` });
-          return;
-        }
-        const body = await readJson(response);
-        const access = String(body?.access_token ?? "");
-        const refresh = String(body?.refresh_token ?? "");
-        if (!access.startsWith("ak_") || !refresh.startsWith("akr_")) {
+        if (answer.refresh === null) {
           this.fail(flow, { code: "failed", message: "AdaptiqAI returned an invalid credential." });
           return;
         }
-        const workspaceLabel = typeof body?.workspace_label === "string" && body.workspace_label !== "" ? body.workspace_label : "Your workspace";
         const grant = {
           schema: "adaptiq-dsh-grant-v1",
           apiBase: this.options.apiBase,
-          access,
-          refresh,
+          access: answer.access,
+          refresh: answer.refresh,
           clientProof,
-          expiresAt: this.now() + Math.max(60, Number(body?.expires_in ?? 3600)) * 1e3,
-          workspaceLabel,
+          expiresAt: this.now() + answer.expiresIn * 1e3,
+          workspaceLabel: answer.workspaceLabel,
           obtainedAt: this.now()
         };
         await this.options.store.modify(async () => grant);
@@ -599,12 +625,12 @@ var AuthManager = class {
         this.flow = null;
         this.pendingSignIn = null;
         this.source = "device";
-        this.label = workspaceLabel;
+        this.label = answer.workspaceLabel;
         this.set("connected");
-        this.options.onSignedIn?.(workspaceLabel);
+        this.options.onSignedIn?.(answer.workspaceLabel);
         return;
       }
-      this.fail(flow, { code: "expired", message: "The sign-in link expired. Start again." });
+      this.fail(flow, { code: "expired", message: EXPIRED });
     } catch (error) {
       if (flow.signal.aborted)
         return;
@@ -729,6 +755,20 @@ var obj = (value) => typeof value === "object" && value !== null && !Array.isArr
 var str = (value) => typeof value === "string" && value.trim() !== "" ? value.trim() : null;
 var num = (value) => typeof value === "number" && Number.isFinite(value) && value > 0 ? value : null;
 var count = (value) => typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : null;
+function pricesFromRow(raw) {
+  const pricing = obj(raw);
+  const price = (value) => {
+    const parsed = typeof value === "string" && value.trim() !== "" ? Number(value) : typeof value === "number" ? value : Number.NaN;
+    return Number.isFinite(parsed) && parsed >= 0 ? parsed : null;
+  };
+  const input = price(pricing?.prompt);
+  const memoryWrite = price(pricing?.input_cache_write);
+  const memoryRead = price(pricing?.input_cache_read);
+  const output = price(pricing?.completion);
+  if (input === null || memoryWrite === null || memoryRead === null || output === null)
+    return null;
+  return { input, memoryWrite, memoryRead, output };
+}
 var LEVEL_NAMES = {
   off: "Off",
   minimal: "Minimal",
@@ -817,17 +857,17 @@ function placementsFromRoutes(routes) {
     const model2 = str(route?.model);
     if (route === null || model2 === null || route.enabled !== true)
       continue;
-    const add = (a, b) => a === null ? b : b === null ? a : a + b;
+    const add2 = (a, b) => a === null ? b : b === null ? a : a + b;
     const current = out.get(model2);
     const running = count(route.running);
     const waiting = count(route.waiting);
     out.set(model2, {
       name: current?.name ?? (str(route.name) !== null ? routeModelName(str(route.name)) : null),
       status: "running",
-      lanes: add(current?.lanes ?? null, count(route.lanes)),
-      maxParallel: add(current?.maxParallel ?? null, count(route.max_parallel_agents)),
-      runningNow: add(current?.runningNow ?? null, running),
-      waitingNow: add(current?.waitingNow ?? null, waiting),
+      lanes: add2(current?.lanes ?? null, count(route.lanes)),
+      maxParallel: add2(current?.maxParallel ?? null, count(route.max_parallel_agents)),
+      runningNow: add2(current?.runningNow ?? null, running),
+      waitingNow: add2(current?.waitingNow ?? null, waiting),
       region: current?.region ?? str(route.region),
       card: null,
       contextWindow: null,
@@ -888,7 +928,8 @@ function model(id, row, placement) {
     maxParallel: placement?.maxParallel ?? count(gateway?.max_parallel_agents),
     runningNow: placement?.runningNow ?? null,
     waitingNow: placement?.waitingNow ?? null,
-    region: placement?.region ?? str(gateway?.region)
+    region: placement?.region ?? str(gateway?.region),
+    prices: pricesFromRow(row?.pricing)
   };
 }
 function selectableLevels(reasoning) {
@@ -957,6 +998,60 @@ function laneLimit(offer) {
   return free < offer.minLanes ? 0 : Math.min(offer.maxLanes, free);
 }
 
+// ../core/lib/processed.js
+var NOTHING_PROCESSED = { requests: 0, newInput: 0, fromMemory: 0, written: 0, output: 0, reasoning: 0, costUsd: 0 };
+function stepCost(usage, prices) {
+  if (prices === null)
+    return null;
+  return usage.inputTokens * prices.input + (usage.cacheWriteTokens ?? 0) * prices.memoryWrite + (usage.cacheReadTokens ?? 0) * prices.memoryRead + usage.outputTokens * prices.output;
+}
+function add(view, step) {
+  return {
+    requests: view.requests + 1,
+    newInput: view.newInput + step.usage.inputTokens,
+    fromMemory: view.fromMemory + (step.usage.cacheReadTokens ?? 0),
+    written: view.written + (step.usage.cacheWriteTokens ?? 0),
+    output: view.output + step.usage.outputTokens,
+    reasoning: view.reasoning + (step.usage.reasoningTokens ?? 0),
+    costUsd: view.costUsd === null || step.costUsd === null ? null : view.costUsd + step.costUsd
+  };
+}
+function addStep(record, step, now) {
+  const turnKey = step.turn ?? record?.turnKey ?? null;
+  const fresh = record == null || step.turn !== void 0 && step.turn !== record.turnKey;
+  return {
+    since: record?.since ?? now,
+    at: now,
+    turnKey,
+    turn: add(fresh ? NOTHING_PROCESSED : record.turn, step),
+    session: add(record?.session ?? NOTHING_PROCESSED, step)
+  };
+}
+function readProcessed(raw) {
+  const record = typeof raw === "object" && raw !== null ? raw : null;
+  const view = (value) => {
+    const fields = typeof value === "object" && value !== null ? value : null;
+    if (fields === null)
+      return null;
+    const count3 = (name) => {
+      const field = fields[name];
+      return typeof field === "number" && Number.isFinite(field) && field >= 0 ? field : null;
+    };
+    const numbers = { requests: count3("requests"), newInput: count3("newInput"), fromMemory: count3("fromMemory"), written: count3("written"), output: count3("output"), reasoning: count3("reasoning") };
+    if (Object.values(numbers).some((value2) => value2 === null))
+      return null;
+    return { ...numbers, costUsd: count3("costUsd") };
+  };
+  const turn = view(record?.turn);
+  const session = view(record?.session);
+  const since = record?.since;
+  const at = record?.at;
+  if (turn === null || session === null || typeof since !== "number" || typeof at !== "number")
+    return null;
+  const turnKey = typeof record?.turnKey === "number" ? record.turnKey : null;
+  return { since, at, turnKey, turn, session };
+}
+
 // ../core/lib/relations.js
 var PARENT_SESSION_HEADER = "X-Adaptiq-Parent-Session-Id";
 var SESSION_ROLE_HEADER = "X-Adaptiq-Session-Role";
@@ -1045,6 +1140,22 @@ var en = {
   popKnowledge: "Knowledge",
   addKnowledge: "Add knowledge",
   popWorkspace: "All memory on {model}",
+  popSeesNow: "What the model sees now",
+  procTitle: "Processed",
+  procTurn: "Last turn",
+  procSession: "This conversation",
+  procSteps: "{count} steps",
+  procStepsOne: "1 step",
+  procNewInput: "New input",
+  procFromMemory: "Read from memory",
+  procWritten: "Saved to memory",
+  procOutput: "Output",
+  procThinking: "Thinking, part of output",
+  procCost: "Cost",
+  procExplain: "Each step reads the whole memory again. {percent} came from memory.",
+  procExplainFree: "Each step reads the whole memory again. {percent} came from memory, which is free.",
+  procExplainCheaper: "Each step reads the whole memory again. {percent} came from memory, at a fraction of the input price.",
+  procSince: "Counted on this computer since {date}",
   openInKnowledge: "Open in Knowledge",
   kTitle: "Knowledge",
   kSearch: "Search your knowledge",
@@ -1529,6 +1640,21 @@ function button(id, label, extra = {}) {
   return { id, label: en[label], labelKey: label, ...extra };
 }
 var count2 = (value) => typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : null;
+var PROCESSED_SAVE_MS = 2e3;
+function savedProcessed(raw) {
+  const out = {};
+  if (typeof raw !== "object" || raw === null)
+    return out;
+  for (const [id, value] of Object.entries(raw)) {
+    const record = readProcessed(value);
+    if (record !== null)
+      out[id] = record;
+  }
+  return out;
+}
+function processedSummary(record) {
+  return record === void 0 ? null : { turn: record.turn, session: record.session, since: record.since };
+}
 var PRIVATE_HELD = "A private conversation is not kept on AdaptiqAI, so it can't use this before it exists there. Turn Remember on for this conversation to add it.";
 var ATTACH_TRIES = 3;
 function transient(error) {
@@ -1894,6 +2020,7 @@ var Workspace = class {
   listeners = /* @__PURE__ */ new Set();
   state = { preferences: DEFAULT_PREFERENCES, sessions: {} };
   saving = Promise.resolve();
+  persistTimer;
   refreshing = null;
   timer;
   pendingAttach = /* @__PURE__ */ new Map();
@@ -1953,7 +2080,8 @@ var Workspace = class {
           preferences: { ...DEFAULT_PREFERENCES, ...saved.preferences },
           sessions: typeof saved.sessions === "object" && saved.sessions !== null ? saved.sessions : {},
           purchase: saved.purchase ?? null,
-          ...saved.defaultModel === "pending" || saved.defaultModel === "done" ? { defaultModel: saved.defaultModel } : {}
+          ...saved.defaultModel === "pending" || saved.defaultModel === "done" ? { defaultModel: saved.defaultModel } : {},
+          processed: savedProcessed(saved.processed)
         };
         this.purchases.restore(saved.purchase);
       }
@@ -1969,6 +2097,10 @@ var Workspace = class {
     this.purchases.dispose();
     clearTimeout(this.timer);
     clearTimeout(this.attachTimer);
+    if (this.persistTimer !== void 0) {
+      clearTimeout(this.persistTimer);
+      void this.persist();
+    }
     for (const session of this.sessions.values()) {
       clearTimeout(session.timer);
       clearTimeout(session.waitTimer);
@@ -2036,7 +2168,8 @@ var Workspace = class {
       region: model2.region,
       nativeMemory: model2.nativeMemory,
       tools: model2.tools,
-      memory: this.modelMemory.get(model2.id) ?? null
+      memory: this.modelMemory.get(model2.id) ?? null,
+      prices: model2.prices
     };
   }
   /**
@@ -2646,12 +2779,16 @@ var Workspace = class {
     return next;
   }
   async persist() {
+    clearTimeout(this.persistTimer);
+    this.persistTimer = void 0;
     const entries = Object.entries(this.state.sessions).sort((a, b) => b[1].at - a[1].at).slice(0, 500);
+    const processed = Object.entries(this.state.processed ?? {}).sort((a, b) => b[1].at - a[1].at).slice(0, 500);
     const snapshot = {
       preferences: this.state.preferences,
       sessions: Object.fromEntries(entries),
       purchase: this.state.purchase ?? null,
-      ...this.state.defaultModel === void 0 ? {} : { defaultModel: this.state.defaultModel }
+      ...this.state.defaultModel === void 0 ? {} : { defaultModel: this.state.defaultModel },
+      ...processed.length === 0 ? {} : { processed: Object.fromEntries(processed) }
     };
     this.saving = this.saving.then(() => this.deps.store.save(snapshot)).catch((error) => {
       this.deps.log?.("warn", `adaptiq: could not save preferences: ${String(error)}`);
@@ -2762,6 +2899,7 @@ var Workspace = class {
       saved: session.saved,
       optimizing: session.optimizing,
       lastTurn: session.lastTurn,
+      processed: processedSummary(this.state.processed?.[id]),
       knowledge: [...session.knowledge, ...[...session.pendingKnowledge.values()].filter((item) => !session.knowledge.some((entry) => entry.id === item.id))],
       conversations: [...session.conversations, ...[...session.pendingConversations.values()].filter((item) => !session.conversations.some((entry) => entry.id === item.id))],
       known: session.exists,
@@ -3141,10 +3279,19 @@ var Workspace = class {
     this.setPhase(session, "responding");
     this.clearNotices("busy", "preparing", "offline", "retrying");
   }
-  usage(sessionId, model2, usage) {
+  usage(sessionId, model2, usage, step = {}) {
     if (sessionId === void 0)
       return;
     const session = this.session(sessionId);
+    const costUsd = step.costUsd ?? stepCost(usage, this.catalog.find((entry) => entry.id === model2)?.prices ?? null);
+    const counted = addStep(this.state.processed?.[sessionId], { usage, costUsd, ...step.turn === void 0 ? {} : { turn: step.turn } }, this.now());
+    this.state = { ...this.state, processed: { ...this.state.processed, [sessionId]: counted } };
+    if (this.persistTimer === void 0) {
+      this.persistTimer = setTimeout(() => {
+        void this.persist();
+      }, PROCESSED_SAVE_MS);
+      this.persistTimer.unref?.();
+    }
     session.lastTurn = {
       fromMemory: usage.cacheReadTokens ?? 0,
       newInput: usage.inputTokens,
@@ -4949,7 +5096,7 @@ function userTexts(messages) {
 }
 
 // src/index.ts
-var VERSION = true ? "0.1.0" : "0.0.0";
+var VERSION = true ? "0.1.1" : "0.0.0";
 var ROUTE = "adaptiqai";
 var PARENT_ENV = "ADAPTIQ_PARENT_SESSION_ID";
 var SESSION_ID2 = /^[A-Za-z0-9][A-Za-z0-9_.:-]{0,127}$/u;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adaptiqai/pi",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "AdaptiqAI for Pi: your workspace's models, conversations AdaptiqAI remembers, and your saved knowledge, in one Pi package.",
   "type": "module",
   "keywords": [
@@ -37,7 +37,7 @@
     "esbuild": "~0.25.12",
     "typescript": "~5.9.3",
     "vitest": "~3.2.7",
-    "@adaptiqai/core": "0.2.10"
+    "@adaptiqai/core": "0.2.11"
   },
   "scripts": {
     "build": "node scripts/build.mjs",