@selesai/code 0.5.13 → 0.5.14

package/dist/extensions/tps-tracker.ts

@@ -1,280 +0,0 @@
-/**
- * TPS Tracker Extension
- *
- * Tracks tokens per second during model generation and reports
- * final TPS statistics at the end of each agent run.
- */
-
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-
-export default function (pi: ExtensionAPI) {
-	/** Timestamp when the current assistant message event started. Used as a fallback. */
-	let messageStart: number | null = null;
-	/** Timestamp of the first streamed output delta for the current assistant message. */
-	let streamStart: number | null = null;
-	/** Estimated streamed output tokens for live display before providers report final usage. */
-	let estimatedStreamedTokens = 0;
-	/** Cumulative official output tokens across all assistant messages in this agent run. */
-	let totalOutputTokens = 0;
-	/** Cumulative time (ms) spent actually streaming output deltas (excludes tool execution and first-token latency). */
-	let totalStreamMs = 0;
-	/** Subagent tool-call start times, used to include pi-subagents child output in aggregate TPS. */
-	const subagentStarts = new Map<string, number>();
-	/** Latest cumulative foreground subagent tokens seen during tool_execution_update. */
-	const subagentLiveTokens = new Map<string, number>();
-	/** Cumulative subagent output tokens in this agent run. */
-	let totalSubagentTokens = 0;
-	/** Cumulative subagent wall time (ms) in this agent run. */
-	let totalSubagentMs = 0;
-	/** Completed subagent call summaries in this agent run. */
-	const subagentSummaries: Array<{ tokens: number; ms: number; tps: number }> = [];
-
-	function asRecord(value: unknown): Record<string, unknown> | null {
-		return value && typeof value === "object" && !Array.isArray(value) ? value as Record<string, unknown> : null;
-	}
-
-	function numberValue(value: unknown): number {
-		return typeof value === "number" && Number.isFinite(value) ? value : 0;
-	}
-
-	function getDetailsResult(value: unknown): Record<string, unknown> {
-		const result = asRecord(value);
-		const details = asRecord(result?.details);
-		return details ?? result ?? {};
-	}
-
-	function getResults(value: unknown): Record<string, unknown>[] {
-		const details = getDetailsResult(value);
-		return Array.isArray(details.results)
-			? details.results.map(asRecord).filter((item): item is Record<string, unknown> => !!item)
-			: [];
-	}
-
-	function generatedTokensFromUsage(usage: Record<string, unknown> | null): number {
-		return numberValue(usage?.output ?? usage?.outputTokens ?? usage?.completionTokens ?? usage?.completion_tokens);
-	}
-
-	function generatedTokensFromMessage(message: { usage?: unknown; content?: unknown }): number {
-		const usageTokens = generatedTokensFromUsage(asRecord(message.usage));
-		if (usageTokens > 0) return usageTokens;
-
-		const content = Array.isArray(message.content) ? message.content : [];
-		const chars = content.reduce((sum, block) => {
-			const item = asRecord(block);
-			if (!item) return sum;
-			if (item.type === "text") return sum + String(item.text ?? "").length;
-			if (item.type === "thinking") return sum + String(item.thinking ?? "").length;
-			if (item.type === "toolCall") return sum + String(item.name ?? "").length + JSON.stringify(item.arguments ?? {}).length;
-			return sum;
-		}, 0);
-		return Math.ceil(chars / 4);
-	}
-
-	function outputTokensFromResult(value: unknown): number {
-		const results = getResults(value);
-		return results.reduce((sum, result) => sum + generatedTokensFromUsage(asRecord(result.usage)), 0);
-	}
-
-	function liveTokensFromResult(value: unknown): number {
-		return subagentProgressItems(value).reduce((sum, item) => sum + item.tokens, 0);
-	}
-
-	function subagentProgressItems(value: unknown): Array<{ name: string; tokens: number; status?: string }> {
-		const details = getDetailsResult(value);
-		const progress = Array.isArray(details.progress) ? details.progress : undefined;
-		const resultItems = getResults(value).map((result, index) => {
-			const progress = asRecord(result.progress);
-			const usage = asRecord(result.usage);
-			return {
-				name: String(result.agent || progress?.agent || `agent${index + 1}`),
-				tokens: generatedTokensFromUsage(usage) || generatedTokensFromUsage(progress),
-				status: typeof progress?.status === "string" ? progress.status : undefined,
-			};
-		});
-		if (!progress || resultItems.some((item) => item.tokens > 0)) return resultItems;
-
-		return progress.map((entry, index) => {
-			const item = asRecord(entry) ?? {};
-			return {
-				name: String(item.agent || `agent${index + 1}`),
-				tokens: generatedTokensFromUsage(item),
-				status: typeof item.status === "string" ? item.status : undefined,
-			};
-		});
-	}
-
-	function subagentModeLabel(value: unknown): string {
-		const details = getDetailsResult(value);
-		const mode = typeof details.mode === "string" ? details.mode : "subagents";
-		return mode === "parallel" ? "parallel" : mode === "chain" ? "chain" : "subagent";
-	}
-
-	pi.on("agent_start", async (_event, ctx) => {
-		totalOutputTokens = 0;
-		totalStreamMs = 0;
-		totalSubagentTokens = 0;
-		totalSubagentMs = 0;
-		subagentSummaries.length = 0;
-		messageStart = null;
-		streamStart = null;
-		estimatedStreamedTokens = 0;
-		const theme = ctx.ui.theme;
-		subagentStarts.clear();
-		subagentLiveTokens.clear();
-		ctx.ui.setStatus("tps", theme.fg("success", "⏱ generating..."));
-	});
-
-	pi.on("message_start", async (event) => {
-		if (event.message.role !== "assistant") return;
-		messageStart = Date.now();
-		streamStart = null;
-		estimatedStreamedTokens = 0;
-	});
-
-	pi.on("message_update", async (event, ctx) => {
-		if (event.message.role !== "assistant") return;
-
-		const streamEvent = event.assistantMessageEvent;
-		const isOutputDelta =
-			streamEvent.type === "text_delta" ||
-			streamEvent.type === "thinking_delta" ||
-			streamEvent.type === "toolcall_delta";
-
-		if (!isOutputDelta) return;
-
-		const now = Date.now();
-		streamStart ??= now;
-		estimatedStreamedTokens += Math.max(0, streamEvent.delta.length / 4);
-
-		const elapsed = (now - streamStart) / 1000;
-		const officialTokens = generatedTokensFromUsage(asRecord(event.message.usage));
-		const currentTokens = officialTokens > 0 ? officialTokens : estimatedStreamedTokens;
-
-		if (elapsed > 0 && currentTokens > 0) {
-			const tps = Math.round(currentTokens / elapsed);
-			const tokenLabel = officialTokens > 0
-				? `${officialTokens} tok`
-				: `~${Math.round(estimatedStreamedTokens)} tok`;
-			const theme = ctx.ui.theme;
-			ctx.ui.setStatus(
-				"tps",
-				`${theme.fg("accent", `${tps} tok/s`)} ${theme.fg("success", `(${tokenLabel} / ${elapsed.toFixed(1)}s)`)}`,
-			);
-		}
-	});
-
-	pi.on("message_end", async (event) => {
-		if (event.message.role !== "assistant") return;
-
-		const messageTokens = generatedTokensFromMessage(event.message);
-		const timingStart = streamStart ?? messageStart;
-		if (!timingStart || messageTokens <= 0) {
-			messageStart = null;
-			streamStart = null;
-			estimatedStreamedTokens = 0;
-			return;
-		}
-
-		totalOutputTokens += messageTokens;
-		totalStreamMs += Math.max(0, Date.now() - timingStart);
-
-		messageStart = null;
-		streamStart = null;
-		estimatedStreamedTokens = 0;
-	});
-
-	pi.on("tool_execution_start", async (event) => {
-		if (event.toolName !== "subagent") return;
-		subagentStarts.set(event.toolCallId, Date.now());
-	});
-
-	pi.on("tool_execution_update", async (event, ctx) => {
-		if (event.toolName !== "subagent") return;
-
-		const start = subagentStarts.get(event.toolCallId);
-		if (!start) return;
-
-		const items = subagentProgressItems(event.partialResult);
-		const tokens = items.reduce((sum, item) => sum + item.tokens, 0);
-		subagentLiveTokens.set(event.toolCallId, tokens);
-		const elapsed = (Date.now() - start) / 1000;
-		if (tokens <= 0 || elapsed <= 0) return;
-
-		const tps = Math.round(tokens / elapsed);
-		const mode = subagentModeLabel(event.partialResult);
-		const agentSummary = items
-			.filter((item) => item.tokens > 0 || item.status === "running")
-			.map((item) => `${item.name} ${Math.round(item.tokens / elapsed)} t/s`)
-			.join(" ");
-		const theme = ctx.ui.theme;
-		ctx.ui.setStatus("tps", `${theme.fg("accent", `main+${mode} ${tps} t/s`)} ${theme.fg("success", `(${agentSummary || `${tps} t/s`})`)}`);
-	});
-
-	pi.on("tool_execution_end", async (event, ctx) => {
-		if (event.toolName !== "subagent") return;
-
-		const start = subagentStarts.get(event.toolCallId);
-		subagentStarts.delete(event.toolCallId);
-		const liveTokens = subagentLiveTokens.get(event.toolCallId) ?? liveTokensFromResult(event.result);
-		subagentLiveTokens.delete(event.toolCallId);
-		if (!start || event.isError) return;
-
-		const outputTokens = outputTokensFromResult(event.result);
-		const displayTokens = outputTokens || liveTokens;
-		const mode = subagentModeLabel(event.result);
-		const elapsedMs = Math.max(0, Date.now() - start);
-		if (displayTokens <= 0 || elapsedMs <= 0) return;
-		const elapsedSeconds = elapsedMs / 1000;
-		const items = subagentProgressItems(event.result);
-		const agentSummary = items
-			.filter((item) => item.tokens > 0)
-			.map((item) => `${item.name} ${Math.round(item.tokens / elapsedSeconds)} t/s`)
-			.join(" ");
-
-		totalOutputTokens += displayTokens;
-		totalStreamMs += elapsedMs;
-		totalSubagentTokens += displayTokens;
-		totalSubagentMs += elapsedMs;
-
-		const tps = Math.round(displayTokens / elapsedSeconds);
-		subagentSummaries.push({ tokens: displayTokens, ms: elapsedMs, tps });
-		const theme = ctx.ui.theme;
-		const detail = theme.fg("success", `${mode}: ${displayTokens} tokens in ${(elapsedMs / 1000).toFixed(1)}s${agentSummary ? ` (${agentSummary})` : ""}`);
-		ctx.ui.notify(`${theme.fg("success", "✓")} ${theme.fg("accent", `${tps} t/s`)} ${detail}`, "info");
-		ctx.ui.setStatus(
-			"tps",
-			`${theme.fg("accent", `main+${mode} ${tps} t/s`)} ${theme.fg("success", `(${agentSummary || `${tps} t/s`})`)}`,
-		);
-	});
-
-	pi.on("agent_end", async (_event, ctx) => {
-		const elapsed = totalStreamMs / 1000;
-		const tps = totalOutputTokens > 0 && elapsed > 0 ? Math.round(totalOutputTokens / elapsed) : 0;
-
-		const theme = ctx.ui.theme;
-		const icon = theme.fg("success", "✓");
-		const tpsLabel = tps > 0
-			? theme.fg("accent", `${tps} t/s`)
-			: theme.fg("success", "N/A");
-		const mainTokens = totalOutputTokens - totalSubagentTokens;
-		const mainMs = totalStreamMs - totalSubagentMs;
-		const mainTps = mainTokens > 0 && mainMs > 0 ? Math.round(mainTokens / (mainMs / 1000)) : 0;
-		const subagentTps = totalSubagentTokens > 0 && totalSubagentMs > 0
-			? Math.round(totalSubagentTokens / (totalSubagentMs / 1000))
-			: 0;
-		const mainDetail = mainTps > 0 ? theme.fg("success", ` | main ${mainTps} t/s`) : "";
-		const subagentDetail = subagentTps > 0
-			? theme.fg("success", ` | subagents ${subagentTps} t/s (${subagentSummaries.length} call${subagentSummaries.length === 1 ? "" : "s"})`)
-			: "";
-
-		ctx.ui.notify(`${icon} total ${tpsLabel}${mainDetail}${subagentDetail}`, "info");
-		const finalMainStatus = mainTps > 0 ? `main ${mainTps} t/s` : "main N/A";
-		const finalSubagentStatus = totalSubagentTokens > 0 && totalSubagentMs > 0
-			? `subagents ${Math.round(totalSubagentTokens / (totalSubagentMs / 1000))} t/s`
-			: "";
-		ctx.ui.setStatus(
-			"tps",
-			`${theme.fg("accent", `done ${tps} t/s`)} ${theme.fg("success", `(${[finalMainStatus, finalSubagentStatus].filter(Boolean).join(" ")})`)}`,
-		);
-	});
-}

package/dist/extensions/workflow/task-validators.ts

@@ -1,69 +0,0 @@
-// ponytail: task-specific validators enforce the acceptance/verification
-// contract used only by task mode. Pure string functions; no fs/pi.
-
-import { markerValidator } from "./validators.ts";
-
-const PLAN_READY = markerValidator("WORKFLOW_PLAN_STATUS", "ready");
-const REVIEW_CLEAN = markerValidator("WORKFLOW_REVIEW_STATUS", "clean");
-
-const AC_ID_RE = /\bAC-(\d+)\b/gi;
-const ACCEPTANCE_CRITERIA_HEADING_RE = /^##\s+Acceptance Criteria\b/im;
-const IMPLEMENTATION_HEADING_RE = /^##\s+Implementation Outline\b/im;
-const VERIFICATION_RE = /\bVerification\s*:/gi;
-
-function extractACIds(text: string): string[] {
-  const ids: string[] = [];
-  let m: RegExpExecArray | null;
-  AC_ID_RE.lastIndex = 0;
-  while ((m = AC_ID_RE.exec(text)) !== null) {
-    ids.push(m[0]!.toUpperCase());
-  }
-  return ids;
-}
-
-export const taskPlanValidator = (content: string) => {
-  const marker = PLAN_READY(content);
-  if (!marker.ok) return marker;
-
-  if (!ACCEPTANCE_CRITERIA_HEADING_RE.test(content)) {
-    return { ok: false as const, reason: "missing `## Acceptance Criteria` section" };
-  }
-  const ids = extractACIds(content);
-  if (ids.length === 0) {
-    return { ok: false as const, reason: "acceptance criteria section must contain at least one AC-<n> ID" };
-  }
-  const unique = new Set(ids);
-  if (unique.size !== ids.length) {
-    return { ok: false as const, reason: "acceptance criteria AC-<n> IDs must be unique" };
-  }
-  if (!IMPLEMENTATION_HEADING_RE.test(content)) {
-    return { ok: false as const, reason: "missing `## Implementation Outline` section" };
-  }
-  const verifications = (content.match(VERIFICATION_RE) ?? []).length;
-  if (verifications < unique.size) {
-    return { ok: false as const, reason: "each AC-<n> must have a Verification: step" };
-  }
-  return { ok: true as const };
-};
-
-export function taskReviewValidator(
-  planContent: string | undefined,
-  reviewText: string,
-): { ok: true } | { ok: false; reason: string } {
-  const marker = REVIEW_CLEAN(reviewText);
-  if (!marker.ok) return marker;
-
-  const planIds = extractACIds(planContent ?? "");
-  if (planIds.length === 0) {
-    return { ok: false as const, reason: "plan contains no AC-<n> criteria; cannot verify coverage" };
-  }
-  const reviewIds = extractACIds(reviewText);
-  const missing = planIds.filter((id) => !reviewIds.includes(id));
-  if (missing.length > 0) {
-    return { ok: false as const, reason: `review is missing coverage for ${missing.join(", ")}` };
-  }
-  if (!/\bpassed\b/i.test(reviewText) || !/\bverification\b/i.test(reviewText)) {
-    return { ok: false as const, reason: "review must report that verification passed" };
-  }
-  return { ok: true as const };
-}

package/dist/skills/selesai-default/SKILL.md

@@ -1,16 +0,0 @@
----
-name: selesai-default
-description: Default Selesai skill. A minimal example showing how bundled skills are shipped in src/skills/ and loaded at boot via additionalSkillPaths. Safe to delete or replace.
----
-
-# Selesai Default Skill
-
-This skill ships inside `@selesai/code` under `src/skills/` and is loaded
-automatically at boot (same hook as `--skill`), so every Selesai install gets
-it without users copying anything to `~/.selesai/agent/skills/`.
-
-Add more bundled skills by creating sibling directories here, each with its own
-`SKILL.md` (YAML frontmatter `name` + `description`, then the body). The build
-copies the whole `src/skills/` tree into `dist/skills/`.
-
-To disable all skills at runtime: `selesai --no-skills`.