@selesai/code 0.13.37 → 0.13.39

package/dist/extensions/pi-subagents/docs/agents.md

@@ -73,13 +73,7 @@ Native `oracle` runs inside Pi and can use its configured read tools. The Claude
 | `external-job-requests/` and `external-job-responses/` | Host-mediated provider bridge | pending request, terminal response | Host process writes a matching response and removes the request | Bridge timeout or malformed request response | Requests are operation-scoped. Recovery sends `reattach`/`result`, not `start` or `follow-up`, when job metadata exists. `start` and `follow-up` use durable dispatch claims | Provider not registered, host bridge not loaded, malformed request, provider exception, ambiguous dispatch without a provider job id |
 | Provider artifact path | External provider | provider-defined terminal artifact | Provider returns `artifactPath`, or Pi writes returned text to `external-job-<index>.result.md` | Provider reports failure or no result | Existing artifact path is retained in `status.json` | Missing artifact with no text output returns a terminal message instead of inventing content |
 
-The `researcher` builtin uses `web_search`, `fetch_content`, `get_search_content`, and selective `source_check` validation. Those require [pi-web-access](https://github.com/nicobailon/pi-web-access):
-
-```bash
-pi install npm:pi-web-access
-```
-
-The loaded provider must register all four tools, including `source_check`, before launch; a missing required tool prevents a successful run. Fetched-source inspection is a fallback for a registered `source_check` call failing, not for missing registration.
+The `researcher` builtin uses `web_explore` from the bundled `pi-web-agent` extension. Because the `tools` field is a strict allowlist, `web_explore` must be registered in the child; run `researcher` as a background child (`async: true`) so ambient extensions load.
 
 ## Overriding builtins and custom agents
 

package/dist/extensions/pi-subagents/test/unit/agent-frontmatter.test.ts

@@ -1904,7 +1904,7 @@ Do work
 				delegate: ["read", "grep", "find", "ls", "bash", "edit", "write", "contact_supervisor"],
 				reviewer: ["read", "grep", "find", "ls", "contact_supervisor"],
 				scout: ["read", "grep", "find", "ls", "bash", "write", "contact_supervisor"],
-				researcher: ["read", "write", "web_search", "fetch_content", "get_search_content", "source_check"],
+				researcher: ["read", "write", "web_explore"],
 			};
 			for (const [name, tools] of Object.entries(expectedTools)) {
 				const agent = agents.find((candidate) => candidate.name === name);
@@ -1914,12 +1914,11 @@ Do work
 
 			const researcherPrompt = agents.find((candidate) => candidate.name === "researcher")?.systemPrompt ?? "";
 			assert.match(researcherPrompt, /search-result summaries as discovery aids, not final evidence/);
-			assert.match(researcherPrompt, /source_check.*decision-critical or disputed claims/);
+			assert.match(researcherPrompt, /Verify decision-critical or disputed claims/);
 			assert.match(researcherPrompt, /direct evidence, source interpretation, and researcher inference distinctly/);
 			assert.match(researcherPrompt, /Record contradictions.*Record missing evidence/);
 			assert.match(researcherPrompt, /Never invent dates, quotations, citations, or unsupported precision/);
-			assert.match(researcherPrompt, /`source_check` must be registered by the loaded provider before launch/);
-			assert.match(researcherPrompt, /If a registered `source_check` call fails, continue/);
+			assert.match(researcherPrompt, /`web_explore`/);
 			assert.match(researcherPrompt, /\*\*Support:\*\* direct evidence \| interpretation\. \*\*Confidence:\*\* high \| medium \| low/);
 		} finally {
 			if (previousHome === undefined) delete process.env.HOME;

package/dist/extensions/pi-web-agent/src/extension.ts

@@ -1,3 +1,4 @@
+import { fileURLToPath } from 'node:url';
 import type { ExtensionAPI } from '@selesai/code';
 import { getDefaultBackendConfig, type BackendConfig } from './backends/config.js';
 import { Type } from 'typebox';
@@ -71,6 +72,16 @@ async function getEffectiveBackendConfig(pi: ExtensionAPI): Promise<BackendConfi
 export default function extension(pi: ExtensionAPI) {
   registerWebAgentConfigCommands(pi);
 
+  // Lets foreground `researcher` children load this extension (see pi-subagents builtin augmentations).
+  pi.events.on('pi-subagents:request-builtin-agent-augmentations', (request: unknown) => {
+    (request as { register?: (a: unknown) => void } | undefined)?.register?.({
+      id: 'pi-web-agent',
+      agentNames: ['researcher'],
+      addTools: ['web_explore'],
+      subagentOnlyExtensions: [fileURLToPath(import.meta.url)]
+    });
+  });
+
   const injectedWebExplore = (pi as ExtensionAPI & { __webExploreTool?: ReturnType<typeof createWebExploreTool> }).__webExploreTool;
   let cachedBackendKey: string | undefined;
   let cachedWebExplore: ReturnType<typeof createWebExploreTool> | undefined;

package/dist/extensions/tokenin-onboarding.ts

@@ -791,9 +791,141 @@ export function formatTokenInUsage(usage: TokenInUsage): string {
 	].join("\n");
 }
 
+export interface TokenInBalanceBucket {
+	kind: string;
+	planId?: string;
+	amountUsd: number;
+	usedUsd: number;
+	remainingUsd: number;
+	state: "active" | "upcoming" | "expired";
+	expiresAt?: string;
+}
+
+export interface TokenInBalance {
+	availableUsd: number;
+	fixedAvailableUsd: number;
+	paygAvailableUsd: number;
+	debtUsd: number;
+	asOf?: string;
+	buckets: TokenInBalanceBucket[];
+}
+
+const finiteNumber = (value: unknown): number | undefined => {
+	const n = typeof value === "number" ? value : Number.NaN;
+	return Number.isFinite(n) ? n : undefined;
+};
+
+/** Parse a /tokenin/account/me/balance payload. Exported for tests. */
+export function parseTokenInBalance(payload: unknown): TokenInBalance | undefined {
+	if (!payload || typeof payload !== "object" || Array.isArray(payload)) return undefined;
+	const obj = payload as Record<string, unknown>;
+	const availableUsd = finiteNumber(obj.available_usd);
+	if (availableUsd === undefined || !Array.isArray(obj.buckets)) return undefined;
+	const buckets: TokenInBalanceBucket[] = [];
+	for (const raw of obj.buckets) {
+		if (!raw || typeof raw !== "object") continue;
+		const b = raw as Record<string, unknown>;
+		const state = b.state;
+		if (state !== "active" && state !== "upcoming" && state !== "expired") continue;
+		const bucket: TokenInBalanceBucket = {
+			kind: typeof b.kind === "string" ? b.kind : "unknown",
+			amountUsd: finiteNumber(b.amount_usd) ?? 0,
+			usedUsd: finiteNumber(b.used_usd) ?? 0,
+			remainingUsd: finiteNumber(b.remaining_usd) ?? 0,
+			state,
+		};
+		if (typeof b.plan_id === "string") bucket.planId = b.plan_id;
+		if (typeof b.expires_at === "string") bucket.expiresAt = b.expires_at;
+		buckets.push(bucket);
+	}
+	const balance: TokenInBalance = {
+		availableUsd,
+		fixedAvailableUsd: finiteNumber(obj.fixed_available_usd) ?? 0,
+		paygAvailableUsd: finiteNumber(obj.payg_available_usd) ?? 0,
+		debtUsd: finiteNumber(obj.debt_usd) ?? 0,
+		buckets,
+	};
+	if (typeof obj.as_of === "string") balance.asOf = obj.as_of;
+	return balance;
+}
+
+function formatTimeUntil(iso: string): string | undefined {
+	const d = new Date(iso);
+	if (Number.isNaN(d.getTime())) return undefined;
+	const ms = d.getTime() - Date.now();
+	if (ms <= 0) return undefined;
+	const days = Math.floor(ms / 86_400_000);
+	const hours = Math.floor((ms % 86_400_000) / 3_600_000);
+	return days > 0 ? `${days}d ${hours}h` : `${Math.max(1, Math.ceil(ms / 3_600_000))}h`;
+}
+
+/** Format an account balance as a boxed display. Exported for tests. */
+export function formatTokenInBalance(balance: TokenInBalance): string {
+	const money = (v: number) => `$${v.toFixed(2)}`;
+	const active = balance.buckets.filter((b) => b.state === "active");
+	const total = active.reduce((sum, b) => sum + b.amountUsd, 0);
+	const used = active.reduce((sum, b) => sum + b.usedUsd, 0);
+	const pct = total > 0 ? Math.min(100, (used / total) * 100) : 0;
+	const barWidth = 40;
+	const filled = total > 0 ? Math.min(barWidth, Math.round((pct / 100) * barWidth)) : 0;
+	const bar = "█".repeat(filled) + "░".repeat(barWidth - filled);
+	const pctText = total > 0 ? `${pct.toFixed(1)}%` : "—";
+
+	const summary = `Available ${money(balance.availableUsd)}`;
+	let split = `Plan ${money(balance.fixedAvailableUsd)} · PAYG ${money(balance.paygAvailableUsd)}`;
+	if (balance.debtUsd > 0) split += ` · Debt ${money(balance.debtUsd)}`;
+	const usedLine = total > 0 ? `Used ${money(used)} of ${money(total)} active credit` : "No active credit";
+
+	const expiries = active
+		.map((b) => b.expiresAt)
+		.filter((v): v is string => typeof v === "string")
+		.sort((a, b) => new Date(a).getTime() - new Date(b).getTime());
+	const until = expiries.length > 0 ? formatTimeUntil(expiries[0]) : undefined;
+	const footer = until ? `Plan credit expires in ${until}` : "";
+
+	const W = 50;
+	const inner = W - 4;
+	const pad = (s: string) => s + " ".repeat(Math.max(0, inner - s.length));
+	const title = "╭─ Token-In balance ";
+	const top = `${title}${"─".repeat(W - title.length - 1)}╮`;
+	const bottom = `╰${"─".repeat(W - 2)}╯`;
+	const lines = [top, `│ ${pad(summary)} │`, `│ ${pad(split)} │`, `│ ${pad(usedLine)} │`, `│ ${bar} ${pctText.padStart(5)} │`];
+	if (footer) lines.push(`│ ${pad(footer)} │`);
+	lines.push(bottom);
+	return lines.join("\n");
+}
+
+function tokenInProxyRoot(baseUrl: string): string {
+	return baseUrl.replace(/\/v1\/?$/, "");
+}
+
+/**
+ * Query the self-scoped account balance with the user's own key. Returns undefined when
+ * the gateway has no account balance for this key (404: older gateway, accounts disabled,
+ * or a legacy key) or answers with something that is not a balance, so callers can fall
+ * back to /key/info. Other errors throw. Exported for tests.
+ */
+export async function fetchTokenInBalance(apiKey: string, baseUrl: string): Promise<TokenInBalance | undefined> {
+	const response = await fetch(`${tokenInProxyRoot(baseUrl)}/tokenin/account/me/balance`, {
+		headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+		signal: AbortSignal.timeout(5_000),
+	});
+	if (response.status === 404) return undefined;
+	if (!response.ok) {
+		throw new Error(`Token-In usage endpoint returned HTTP ${response.status}`);
+	}
+	let payload: unknown;
+	try {
+		payload = await response.json();
+	} catch {
+		return undefined;
+	}
+	return parseTokenInBalance(payload);
+}
+
 /** Query LiteLLM /key/info at the proxy root with the account key. Exported for tests. */
 export async function fetchTokenInUsage(apiKey: string, baseUrl: string): Promise<TokenInUsage> {
-	const root = baseUrl.replace(/\/v1\/?$/, "");
+	const root = tokenInProxyRoot(baseUrl);
 	const response = await fetch(`${root}/key/info`, {
 		headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
 		signal: AbortSignal.timeout(5_000),
@@ -906,6 +1038,11 @@ async function handleTokenInUsage(ctx: ExtensionCommandContext): Promise<void> {
 	const account = auth.accounts.find((a) => a.id === activeId);
 	const baseUrl = account?.baseUrl ?? getTokenInBaseUrl(readModelsJson()) ?? TOKEN_IN_DEFAULT_BASE_URL;
 	try {
+		const balance = await fetchTokenInBalance(apiKey, baseUrl);
+		if (balance) {
+			ctx.ui.notify(formatTokenInBalance(balance), "info");
+			return;
+		}
 		const usage = await fetchTokenInUsage(apiKey, baseUrl);
 		ctx.ui.notify(formatTokenInUsage(usage), "info");
 	} catch (err) {

package/docs/settings.md

@@ -67,12 +67,14 @@ routing. The host-side routes stay off until they are enabled in `jevAdvisory`;
   of the agent's loadout before each run (it returns after `/tokenin add`, no reload needed), and a call
   that slips through reads no file and runs no command.
 
-  The same route also offers `jev_find`, a file finder: ripgrep gathers up to 48 candidate files
-  (those matching `pattern`, or every file under `path` filtered by `glob`, ranked by question words),
-  Jev judges each file's best-matching lines against the agent's `question` in parallel batches of 16,
-  and the agent gets ranked `path:lines (relevance)` pointers — never file contents. It respects
-  `.gitignore`, stays inside the working directory, skips secret-named files, and still returns the
-  unranked candidates when Jev is unavailable.
+  The same route also offers `jev_find`, a behavior finder: ripgrep gathers candidate files (those
+  matching `pattern`, or every file under `path` filtered by `glob`); on large trees Jev first judges
+  directories and descends only into relevant ones, then judges files, then the declarations (source
+  units) inside relevant files. The agent gets a ranked file list with reading leads plus the accepted
+  units verbatim with original line numbers (at most 12 Jev requests and 16 KiB of source per call).
+  It respects `.gitignore`, stays inside the working directory, skips secret-named files, retries a
+  transport failure once, and still returns ripgrep-ranked candidates with keyword-window source when
+  Jev is unavailable.
 - `subagent` — sets up a single-child `subagent` call from the parent agent (off by default). Jev only
   fills what the parent left open, and can only narrow: (1) when `agent` is omitted or generic
   (`genericAgents`, default `["delegate"]`), it picks an agent by function from enabled native agents

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@selesai/code",
-  "version": "0.13.37",
+  "version": "0.13.39",
   "description": "Maintained, extension-first Pi coding agent with built-in workflows, subagents, web research, questions, skills, and an enhanced terminal UI.",
   "type": "module",
   "engines": {