@dyyz1993/pi-coding-agent 0.74.24 → 0.74.25

package/dist/extensions/subagent-shared/utils.ts

@@ -0,0 +1,112 @@
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import type { Message } from "@dyyz1993/pi-ai";
+import { withFileMutationQueue } from "@dyyz1993/pi-coding-agent";
+import type { DisplayItem, UsageStats } from "./types.js";
+
+export function formatTokens(count: number): string {
+	if (count < 1000) return count.toString();
+	if (count < 10000) return `${(count / 1000).toFixed(1)}k`;
+	if (count < 1000000) return `${Math.round(count / 1000)}k`;
+	return `${(count / 1000000).toFixed(1)}M`;
+}
+
+export function formatUsageStats(
+	usage: {
+		input: number;
+		output: number;
+		cacheRead: number;
+		cacheWrite: number;
+		cost: number;
+		contextTokens?: number;
+		turns?: number;
+	},
+	model?: string,
+): string {
+	const parts: string[] = [];
+	if (usage.turns) parts.push(`${usage.turns} turn${usage.turns > 1 ? "s" : ""}`);
+	if (usage.input) parts.push(`↑${formatTokens(usage.input)}`);
+	if (usage.output) parts.push(`↓${formatTokens(usage.output)}`);
+	if (usage.cacheRead) parts.push(`R${formatTokens(usage.cacheRead)}`);
+	if (usage.cacheWrite) parts.push(`W${formatTokens(usage.cacheWrite)}`);
+	if (usage.cost) parts.push(`$${usage.cost.toFixed(4)}`);
+	if (usage.contextTokens && usage.contextTokens > 0) parts.push(`ctx:${formatTokens(usage.contextTokens)}`);
+	if (model) parts.push(model);
+	return parts.join(" ");
+}
+
+export function getFinalOutput(messages: Message[]): string {
+	for (let i = messages.length - 1; i >= 0; i--) {
+		const msg = messages[i];
+		if (msg.role === "assistant") {
+			for (const part of msg.content) {
+				if (part.type === "text") return part.text;
+			}
+		}
+	}
+	return "";
+}
+
+export function getDisplayItems(messages: Message[]): DisplayItem[] {
+	const items: DisplayItem[] = [];
+	for (const msg of messages) {
+		if (msg.role === "assistant") {
+			for (const part of msg.content) {
+				if (part.type === "text") items.push({ type: "text", text: part.text });
+				else if (part.type === "toolCall") items.push({ type: "toolCall", name: part.name, args: part.arguments });
+			}
+		}
+	}
+	return items;
+}
+
+export async function writePromptToTempFile(
+	agentName: string,
+	prompt: string,
+	tmpPrefix = "pi-subagent-",
+): Promise<{ dir: string; filePath: string }> {
+	const tmpDir = await fs.promises.mkdtemp(path.join(os.tmpdir(), tmpPrefix));
+	const safeName = agentName.replace(/[^\w.-]+/g, "_");
+	const filePath = path.join(tmpDir, `prompt-${safeName}.md`);
+	await withFileMutationQueue(filePath, async () => {
+		await fs.promises.writeFile(filePath, prompt, { encoding: "utf-8", mode: 0o600 });
+	});
+	return { dir: tmpDir, filePath };
+}
+
+export function cleanupTempFiles(filePath: string | null, dir: string | null, logPrefix = "subagent"): void {
+	if (filePath)
+		try {
+			fs.unlinkSync(filePath);
+		} catch (err) {
+			console.debug(`[${logPrefix}] temp file cleanup failed:`, err instanceof Error ? err.message : err);
+		}
+	if (dir)
+		try {
+			fs.rmdirSync(dir);
+		} catch (err) {
+			console.debug(`[${logPrefix}] temp dir cleanup failed:`, err instanceof Error ? err.message : err);
+		}
+}
+
+export function makeUsage(): UsageStats {
+	return { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0 };
+}
+
+export function accumulateUsage(result: { usage: UsageStats; model?: string; stopReason?: string; errorMessage?: string }, msg: Message): void {
+	if (msg.role !== "assistant") return;
+	result.usage.turns++;
+	const usage = msg.usage;
+	if (usage) {
+		result.usage.input += usage.input || 0;
+		result.usage.output += usage.output || 0;
+		result.usage.cacheRead += usage.cacheRead || 0;
+		result.usage.cacheWrite += usage.cacheWrite || 0;
+		result.usage.cost += usage.cost?.total || 0;
+		result.usage.contextTokens = usage.totalTokens || 0;
+	}
+	if (!result.model && msg.model) result.model = msg.model;
+	if (msg.stopReason) result.stopReason = msg.stopReason;
+	if (msg.errorMessage) result.errorMessage = msg.errorMessage;
+}

package/dist/extensions/subagent-v2/contract.d.ts

@@ -0,0 +1,2 @@
+export { SUBAGENT_CHANNEL_NAME as SUBAGENT_V2_CHANNEL_NAME, type SubagentChannelContract as SubagentV2ChannelContract, type SubagentEventPayload as SubagentV2EventPayload } from "../subagent-shared/contract.js";
+//# sourceMappingURL=contract.d.ts.map

package/dist/extensions/subagent-v2/contract.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"contract.d.ts","sourceRoot":"","sources":["contract.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,IAAI,wBAAwB,EAAE,KAAK,uBAAuB,IAAI,yBAAyB,EAAE,KAAK,oBAAoB,IAAI,sBAAsB,EAAE,MAAM,gCAAgC,CAAC","sourcesContent":["export { SUBAGENT_CHANNEL_NAME as SUBAGENT_V2_CHANNEL_NAME, type SubagentChannelContract as SubagentV2ChannelContract, type SubagentEventPayload as SubagentV2EventPayload } from \"../subagent-shared/contract.js\";\n"]}

package/dist/extensions/subagent-v2/contract.js

@@ -0,0 +1,2 @@
+export { SUBAGENT_CHANNEL_NAME as SUBAGENT_V2_CHANNEL_NAME } from "../subagent-shared/contract.js";
+//# sourceMappingURL=contract.js.map

package/dist/extensions/subagent-v2/contract.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contract.js","sourceRoot":"","sources":["contract.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,IAAI,wBAAwB,EAAkH,MAAM,gCAAgC,CAAC","sourcesContent":["export { SUBAGENT_CHANNEL_NAME as SUBAGENT_V2_CHANNEL_NAME, type SubagentChannelContract as SubagentV2ChannelContract, type SubagentEventPayload as SubagentV2EventPayload } from \"../subagent-shared/contract.js\";\n"]}

package/dist/extensions/subagent-v2/contract.ts

@@ -0,0 +1 @@
+export { SUBAGENT_CHANNEL_NAME as SUBAGENT_V2_CHANNEL_NAME, type SubagentChannelContract as SubagentV2ChannelContract, type SubagentEventPayload as SubagentV2EventPayload } from "../subagent-shared/contract.js";