@danypops/pi-jittor 0.1.0 → 0.1.1

package/extension/src/capabilities/context-hub.ts

@@ -0,0 +1,40 @@
+import {
+	CONTEXT_HUB_CONTRIBUTION_DEDUP_LIMIT,
+	validateContextContribution,
+	type ContextSegment,
+} from "@danypops/jittor";
+
+/**
+ * Merges Jittor's own directly-computed segments (tool ledger, real usage) with whatever
+ * segments other extensions contributed on CONTEXT_HUB_CONTRIBUTION_CHANNEL for the current
+ * session. Keeps only the latest contribution per producer -- a producer re-emits every turn
+ * (mirroring Papyrus's own context-injection.v1 cadence), so an older segment from the same
+ * producer is stale, not a second real contributor.
+ */
+export class ContextHubCapability {
+	private readonly latestByProducer = new Map<string, ContextSegment>();
+	private readonly seen = new Set<string>();
+
+	/** Validates and records one contribution; silently drops a malformed, stale, or duplicate one without retaining its payload or crashing the caller. */
+	observe(payload: unknown, now = Date.now()): void {
+		try {
+			const contribution = validateContextContribution(payload, now);
+			const key = `${contribution.producerName}:${contribution.sequence}`;
+			if (this.seen.has(key)) return;
+			this.seen.add(key);
+			if (this.seen.size > CONTEXT_HUB_CONTRIBUTION_DEDUP_LIMIT) this.seen.delete(this.seen.values().next().value!);
+			this.latestByProducer.set(contribution.producerName, contribution.segment);
+		} catch {
+			// Reject malformed or stale cross-extension contributions without retaining payloads.
+		}
+	}
+
+	contributedSegments(): ContextSegment[] {
+		return [...this.latestByProducer.values()];
+	}
+
+	reset(): void {
+		this.latestByProducer.clear();
+		this.seen.clear();
+	}
+}

package/extension/src/context-report.ts

@@ -0,0 +1,46 @@
+import type { ContextSegment, ContextSegmentItem } from "@danypops/jittor";
+
+export interface ContextReportUsage {
+	tokens: number | null;
+	contextWindow: number;
+	percent: number | null;
+}
+
+/** Bounds how many items render per segment -- a report is a scan-at-a-glance summary, not a full dump. */
+const MAX_ITEMS_PER_SEGMENT_LINE = 5;
+
+function formatTokens(tokens: number): string {
+	return tokens >= 1_000 ? `${(tokens / 1_000).toFixed(1)}k` : String(tokens);
+}
+
+function topItems(items: ContextSegmentItem[] | undefined): ContextSegmentItem[] {
+	return [...(items ?? [])].sort((left, right) => right.estimatedTokens - left.estimatedTokens).slice(0, MAX_ITEMS_PER_SEGMENT_LINE);
+}
+
+/**
+ * Plain-text Context Hub report: real provider-reported usage first (never a character-based
+ * estimate when real usage is available, matching Papyrus's own real-vs-estimate honesty
+ * accounting), then every segment heaviest-first with an explicit confidence tag
+ * (exact-tool/exact-cooperative/correlated/audited) so a reader never mistakes one attribution
+ * tier's certainty for another's.
+ */
+export function buildContextReport(segments: readonly ContextSegment[], usage: ContextReportUsage | undefined): string {
+	const lines: string[] = [];
+	if (usage?.tokens !== null && usage?.tokens !== undefined) {
+		const percent = usage.percent !== null && usage.percent !== undefined ? ` (${usage.percent.toFixed(1)}%)` : "";
+		lines.push(`Real usage: ${formatTokens(usage.tokens)} / ${formatTokens(usage.contextWindow)} tokens${percent}`);
+	} else {
+		lines.push("Real usage: not yet reported -- sizes below are estimates only");
+	}
+	const sorted = [...segments].sort((left, right) => right.estimatedTokens - left.estimatedTokens);
+	if (sorted.length === 0) {
+		lines.push("", "(no segments observed yet)");
+		return lines.join("\n");
+	}
+	lines.push("");
+	for (const segment of sorted) {
+		lines.push(`${segment.label} — ${formatTokens(segment.estimatedTokens)} tok [${segment.confidence}]`);
+		for (const item of topItems(segment.items)) lines.push(`  ${formatTokens(item.estimatedTokens)} tok  ${item.label}`);
+	}
+	return lines.join("\n");
+}

package/extension/src/footer.ts

@@ -1,6 +1,7 @@
 import { isAbsolute, relative, resolve, sep } from "node:path";
 import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+import { ProgressBar } from "malevich-tui-components";
 import {
 	FOOTER_BAR_MAX_WIDTH,
 	FOOTER_BAR_MIN_WIDTH,
@@ -119,11 +120,11 @@ function barWidth(width: number): number {
 	return width >= FOOTER_WIDE_TERMINAL_WIDTH ? FOOTER_BAR_MAX_WIDTH : FOOTER_BAR_MIN_WIDTH;
 }
 
+// Thin wrapper over Malevich's ProgressBar.format() -- keeps this file's own call sites
+// (fraction-first, width-second, tolerant of null/non-finite) unchanged.
 function progressBar(fraction: number | null, width: number): string {
-	if (fraction === null || !Number.isFinite(fraction)) return "░".repeat(width);
-	const clamped = Math.min(1, Math.max(0, fraction));
-	const filled = Math.round(width * clamped);
-	return "█".repeat(filled) + "░".repeat(width - filled);
+	const value = fraction === null || !Number.isFinite(fraction) ? 0 : fraction;
+	return new ProgressBar({ value, max: 1, width }).format(width);
 }
 
 function fillColor(fraction: number | null): FooterColor {

package/extension/src/index.ts

@@ -1,5 +1,6 @@
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import {
+	CONTEXT_HUB_CONTRIBUTION_CHANNEL,
 	FOOTER_COMPACTION_RENDER_INTERVAL_MS,
 	MAX_DYNAMIC_ROUTES,
 	PAPYRUS_CONTEXT_INJECTION_CHANNEL,
@@ -10,6 +11,7 @@ import {
 	validatePapyrusContextInjection,
 	applyTaskFocusEvent,
 	validateTaskFocusEvent,
+	toolLedgerSegment,
 	TASK_DOMAINS,
 	TASK_TYPES,
 	USAGE_PERIODS,
@@ -35,6 +37,8 @@ import { showUsagePanel } from "./usage.ts";
 import { CodexRecoveryCapability, SYSTEM_RECOVERY_RUNTIME, type CodexRecoveryRuntime } from "./capabilities/codex-recovery.ts";
 import { ProviderResponseTelemetry } from "./capabilities/provider-response-telemetry.ts";
 import { LocalRunTelemetry } from "./capabilities/local-run-telemetry.ts";
+import { ContextHubCapability } from "./capabilities/context-hub.ts";
+import { buildContextReport } from "./context-report.ts";
 
 export { formatFooterStatus } from "./tui.ts";
 export type { CodexRecoveryRuntime } from "./capabilities/codex-recovery.ts";
@@ -253,6 +257,8 @@ export function registerJittorExtension(
 	const localRunTelemetry = new LocalRunTelemetry();
 	const providerResponseTelemetry = new ProviderResponseTelemetry();
 	const codexRecoveryCapability = new CodexRecoveryCapability(pi, codexRecovery, recoveryRuntime);
+	const contextHub = new ContextHubCapability();
+	const stopContextHub = pi.events?.on?.(CONTEXT_HUB_CONTRIBUTION_CHANNEL, (payload) => contextHub.observe(payload));
 	const contextObservations = new Set<string>();
 	const stopPapyrusContext = pi.events?.on?.(PAPYRUS_CONTEXT_INJECTION_CHANNEL, (payload) => {
 		try {
@@ -458,6 +464,15 @@ export function registerJittorExtension(
 		},
 	});
 
+	pi.registerCommand("context", {
+		description: "Context Hub: real usage plus every segment's estimated size (tool schemas by owning extension, and whatever other extensions contributed), each tagged with how it was attributed",
+		handler: async (_args, ctx) => {
+			const toolSegment = toolLedgerSegment(pi.getAllTools());
+			const segments = [toolSegment, ...contextHub.contributedSegments()];
+			ctx.ui.notify(buildContextReport(segments, ctx.getContextUsage()), "info");
+		},
+	});
+
 	pi.registerCommand("usage", {
 		description: "Cumulative token/cost usage graph with hourly/daily/weekly/monthly/quarterly views",
 		handler: async (args, ctx) => {
@@ -502,6 +517,7 @@ export function registerJittorExtension(
 		finishCompactionUi();
 		compactionTelemetry = new CompactionTelemetry();
 		localRunTelemetry.reset();
+		contextHub.reset();
 		cancelRecovery(true);
 		providerResponseTelemetry.resetTurn();
 		ctx.ui.setStatus("jittor", undefined);
@@ -642,6 +658,7 @@ export function registerJittorExtension(
 		if (compactionTelemetry.hasOpenCompaction()) await recordMetrics(client, [compactionTelemetry.abort(Date.now(), "session-shutdown")]).catch(() => undefined);
 		stopPapyrusContext?.();
 		stopPapyrusTaskFocus?.();
+		stopContextHub?.();
 		cancelRecovery(true);
 		localRunTelemetry.reset();
 		const session_id = ctx.sessionManager.getSessionId();

package/extension/src/service-client.ts

@@ -1,4 +1,4 @@
-import { createRetryingClient, type RetryingClient } from "@danypops/daemon-kit/pi-client";
+import { createRetryingClient, type RetryingClient } from "@danypops/vehicle-client/daemon-client";
 import { connectJittorClient, type JittorClient, type OperationInputs, type OperationName, type OperationOutputs } from "@danypops/jittor";
 
 type JittorConnector = () => Promise<JittorClient>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@danypops/pi-jittor",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Pi extension for Jittor: native routing enforcement, footer, settings, usage graphs, and benchmark panels backed by the @danypops/jittor daemon",
   "type": "module",
   "keywords": ["pi-package", "llm-router", "token-budget"],
@@ -12,8 +12,9 @@
     "extensions": ["extension/src/index.ts"]
   },
   "dependencies": {
-    "@danypops/daemon-kit": "^0.4.0",
-    "@danypops/jittor": "^0.12.0"
+    "@danypops/vehicle-client": "^0.1.1",
+    "@danypops/jittor": "^0.13.0",
+    "malevich-tui-components": "^0.2.0"
   },
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*",