@fragno-dev/pi-harness 0.0.0-stage → 0.0.3

package/dist/node/pi/compact-command-outcome-projection.js

@@ -0,0 +1,41 @@
+import { PiSessionDataIntegrityError } from "./types.js";
+import { asPersistedPiHarnessStepResult } from "./session-entry-projection.js";
+import { z } from "zod";
+//#region src/pi/compact-command-outcome-projection.ts
+const compactCommandOutcomeSchema = z.discriminatedUnion("status", [z.object({
+	kind: z.literal("compact"),
+	commandId: z.string(),
+	status: z.literal("succeeded")
+}), z.object({
+	kind: z.literal("compact"),
+	commandId: z.string(),
+	status: z.literal("rejected"),
+	code: z.enum(["nothing_to_compact", "compaction_failed"]),
+	message: z.string()
+})]);
+const parseCompactCommandOutcome = (value, identity) => {
+	if (typeof value !== "object" || value === null || value.kind !== "compact") return null;
+	const outcome = compactCommandOutcomeSchema.safeParse(value);
+	if (!outcome.success) throw new PiSessionDataIntegrityError(identity.workflowName, identity.sessionId, new Error("Persisted Pi compact command outcome is invalid.", { cause: outcome.error }));
+	return outcome.data;
+};
+const projectPiCompactCommandOutcomes = (steps, identity) => {
+	const byCommandId = {};
+	let latestCommandCompactOutcome = null;
+	for (const step of steps) {
+		if (step.status !== "completed" || !step.stepKey.startsWith("command:")) continue;
+		const result = asPersistedPiHarnessStepResult(step.result);
+		if (!result) throw new PiSessionDataIntegrityError(identity.workflowName, identity.sessionId, /* @__PURE__ */ new Error(`Pi command step ${step.stepKey} does not contain a harness result.`));
+		if (result.outcome !== "completed") continue;
+		latestCommandCompactOutcome = parseCompactCommandOutcome(result.value, identity);
+		if (latestCommandCompactOutcome) byCommandId[latestCommandCompactOutcome.commandId] = latestCommandCompactOutcome;
+	}
+	return {
+		byCommandId,
+		latestCommandCompactOutcome
+	};
+};
+//#endregion
+export { projectPiCompactCommandOutcomes };
+
+//# sourceMappingURL=compact-command-outcome-projection.js.map

package/dist/node/pi/compact-command-outcome-projection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compact-command-outcome-projection.js","names":[],"sources":["../../../src/pi/compact-command-outcome-projection.ts"],"sourcesContent":["import { z } from \"zod\";\n\nimport {\n  asPersistedPiHarnessStepResult,\n  type PiProjectionSessionIdentity,\n} from \"./session-entry-projection\";\nimport {\n  PI_SESSION_COMMAND_STEP_PREFIX,\n  PiSessionDataIntegrityError,\n  type PiCompactCommandOutcome,\n} from \"./types\";\n\nconst compactCommandOutcomeSchema = z.discriminatedUnion(\"status\", [\n  z.object({\n    kind: z.literal(\"compact\"),\n    commandId: z.string(),\n    status: z.literal(\"succeeded\"),\n  }),\n  z.object({\n    kind: z.literal(\"compact\"),\n    commandId: z.string(),\n    status: z.literal(\"rejected\"),\n    code: z.enum([\"nothing_to_compact\", \"compaction_failed\"]),\n    message: z.string(),\n  }),\n]) satisfies z.ZodType<PiCompactCommandOutcome>;\n\nconst parseCompactCommandOutcome = (\n  value: unknown,\n  identity: PiProjectionSessionIdentity,\n): PiCompactCommandOutcome | null => {\n  if (\n    typeof value !== \"object\" ||\n    value === null ||\n    (value as { kind?: unknown }).kind !== \"compact\"\n  ) {\n    return null;\n  }\n\n  const outcome = compactCommandOutcomeSchema.safeParse(value);\n  if (!outcome.success) {\n    throw new PiSessionDataIntegrityError(\n      identity.workflowName,\n      identity.sessionId,\n      new Error(\"Persisted Pi compact command outcome is invalid.\", { cause: outcome.error }),\n    );\n  }\n  return outcome.data;\n};\n\nexport const projectPiCompactCommandOutcomes = (\n  steps: readonly { stepKey: string; status: string; result: unknown }[],\n  identity: PiProjectionSessionIdentity,\n): {\n  byCommandId: Readonly<Record<string, PiCompactCommandOutcome>>;\n  latestCommandCompactOutcome: PiCompactCommandOutcome | null;\n} => {\n  const byCommandId: Record<string, PiCompactCommandOutcome> = {};\n  let latestCommandCompactOutcome: PiCompactCommandOutcome | null = null;\n\n  for (const step of steps) {\n    if (step.status !== \"completed\" || !step.stepKey.startsWith(PI_SESSION_COMMAND_STEP_PREFIX)) {\n      continue;\n    }\n\n    const result = asPersistedPiHarnessStepResult(step.result);\n    if (!result) {\n      throw new PiSessionDataIntegrityError(\n        identity.workflowName,\n        identity.sessionId,\n        new Error(`Pi command step ${step.stepKey} does not contain a harness result.`),\n      );\n    }\n    if (result.outcome !== \"completed\") {\n      continue;\n    }\n\n    latestCommandCompactOutcome = parseCompactCommandOutcome(result.value, identity);\n    if (latestCommandCompactOutcome) {\n      byCommandId[latestCommandCompactOutcome.commandId] = latestCommandCompactOutcome;\n    }\n  }\n\n  return { byCommandId, latestCommandCompactOutcome };\n};\n"],"mappings":";;;;AAYA,MAAM,8BAA8B,EAAE,mBAAmB,UAAU,CACjE,EAAE,OAAO;CACP,MAAM,EAAE,QAAQ,SAAS;CACzB,WAAW,EAAE,OAAO;CACpB,QAAQ,EAAE,QAAQ,WAAW;AAC/B,CAAC,GACD,EAAE,OAAO;CACP,MAAM,EAAE,QAAQ,SAAS;CACzB,WAAW,EAAE,OAAO;CACpB,QAAQ,EAAE,QAAQ,UAAU;CAC5B,MAAM,EAAE,KAAK,CAAC,sBAAsB,mBAAmB,CAAC;CACxD,SAAS,EAAE,OAAO;AACpB,CAAC,CACH,CAAC;AAED,MAAM,8BACJ,OACA,aACmC;CACnC,IACE,OAAO,UAAU,YACjB,UAAU,QACT,MAA6B,SAAS,WAEvC,OAAO;CAGT,MAAM,UAAU,4BAA4B,UAAU,KAAK;CAC3D,IAAI,CAAC,QAAQ,SACX,MAAM,IAAI,4BACR,SAAS,cACT,SAAS,WACT,IAAI,MAAM,oDAAoD,EAAE,OAAO,QAAQ,MAAM,CAAC,CACxF;CAEF,OAAO,QAAQ;AACjB;AAEA,MAAa,mCACX,OACA,aAIG;CACH,MAAM,cAAuD,CAAC;CAC9D,IAAI,8BAA8D;CAElE,KAAK,MAAM,QAAQ,OAAO;EACxB,IAAI,KAAK,WAAW,eAAe,CAAC,KAAK,QAAQ,WAAA,UAAyC,GACxF;EAGF,MAAM,SAAS,+BAA+B,KAAK,MAAM;EACzD,IAAI,CAAC,QACH,MAAM,IAAI,4BACR,SAAS,cACT,SAAS,2BACT,IAAI,MAAM,mBAAmB,KAAK,QAAQ,oCAAoC,CAChF;EAEF,IAAI,OAAO,YAAY,aACrB;EAGF,8BAA8B,2BAA2B,OAAO,OAAO,QAAQ;EAC/E,IAAI,6BACF,YAAY,4BAA4B,aAAa;CAEzD;CAEA,OAAO;EAAE;EAAa;CAA4B;AACpD"}

package/dist/node/pi/context-usage.d.ts

@@ -0,0 +1,20 @@
+import { PiHarnessFrontendAgentMessage } from "./harness/agent-harness-event-protocol.js";
+//#region src/pi/context-usage.d.ts
+type PiContextUsageEstimate = {
+  tokens: number;
+  usageTokens: number;
+  trailingTokens: number;
+  lastUsageIndex: number | null;
+};
+/** Estimate one message using the same conservative four-characters-per-token heuristic as Pi. */
+declare const estimatePiMessageTokens: (message: PiHarnessFrontendAgentMessage) => number;
+/**
+ * Estimate the context represented by an ordered message list.
+ *
+ * Provider usage is authoritative through the latest completed assistant message. Messages after
+ * that point use the same character heuristic Pi applies while preparing compaction.
+ */
+declare const estimatePiContextUsage: (messages: readonly PiHarnessFrontendAgentMessage[]) => PiContextUsageEstimate;
+//#endregion
+export { PiContextUsageEstimate, estimatePiContextUsage, estimatePiMessageTokens };
+//# sourceMappingURL=context-usage.d.ts.map

package/dist/node/pi/context-usage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-usage.d.ts","names":[],"sources":["../../../src/pi/context-usage.ts"],"mappings":";;KAIY;EACV;EACA;EACA;EACA;;;cAqDW,0BAA2B,SAAS;;;;;;;cAwCpC,yBACX,mBAAmB,oCAClB"}

package/dist/node/pi/context-usage.js

@@ -0,0 +1,84 @@
+//#region src/pi/context-usage.ts
+const ESTIMATED_IMAGE_CHARS = 4800;
+const contextTokensFromUsage = (usage) => usage.totalTokens || usage.input + usage.output + usage.cacheRead + usage.cacheWrite;
+const validAssistantUsage = (message) => {
+	if (message.role !== "assistant") return;
+	const assistant = message;
+	if (assistant.stopReason === "aborted" || assistant.stopReason === "error" || !assistant.usage || contextTokensFromUsage(assistant.usage) === 0) return;
+	return assistant.usage;
+};
+const textAndImageContentChars = (content) => {
+	if (typeof content === "string") return content.length;
+	return content.reduce((chars, block) => {
+		if (block.type === "text" && block.text) return chars + block.text.length;
+		if (block.type === "image") return chars + ESTIMATED_IMAGE_CHARS;
+		return chars;
+	}, 0);
+};
+const readableJsonLength = (value) => {
+	try {
+		return (JSON.stringify(value) ?? "undefined").length;
+	} catch {
+		return 16;
+	}
+};
+/** Estimate one message using the same conservative four-characters-per-token heuristic as Pi. */
+const estimatePiMessageTokens = (message) => {
+	let chars = 0;
+	switch (message.role) {
+		case "user":
+			chars = textAndImageContentChars(message.content);
+			break;
+		case "assistant":
+			for (const block of message.content) if (block.type === "text") chars += block.text.length;
+			else if (block.type === "thinking") chars += block.thinking.length;
+			else if (block.type === "toolCall") chars += block.name.length + readableJsonLength(block.arguments);
+			break;
+		case "custom":
+		case "toolResult":
+			chars = textAndImageContentChars(message.content);
+			break;
+		case "bashExecution":
+			chars = message.command.length + message.output.length;
+			break;
+		case "branchSummary":
+		case "compactionSummary":
+			chars = message.summary.length;
+			break;
+	}
+	return Math.ceil(chars / 4);
+};
+/**
+* Estimate the context represented by an ordered message list.
+*
+* Provider usage is authoritative through the latest completed assistant message. Messages after
+* that point use the same character heuristic Pi applies while preparing compaction.
+*/
+const estimatePiContextUsage = (messages) => {
+	let lastUsageIndex = null;
+	let usageTokens = 0;
+	let latestPrefixTimestamp = Number.NEGATIVE_INFINITY;
+	for (let index = 0; index < messages.length; index += 1) {
+		const message = messages[index];
+		const usage = validAssistantUsage(message);
+		const timestampIsFinite = Number.isFinite(message.timestamp);
+		if (usage && (!timestampIsFinite || message.timestamp >= latestPrefixTimestamp)) {
+			lastUsageIndex = index;
+			usageTokens = contextTokensFromUsage(usage);
+		}
+		if (timestampIsFinite) latestPrefixTimestamp = Math.max(latestPrefixTimestamp, message.timestamp);
+	}
+	const trailingStart = lastUsageIndex === null ? 0 : lastUsageIndex + 1;
+	let trailingTokens = 0;
+	for (let index = trailingStart; index < messages.length; index += 1) trailingTokens += estimatePiMessageTokens(messages[index]);
+	return {
+		tokens: usageTokens + trailingTokens,
+		usageTokens,
+		trailingTokens,
+		lastUsageIndex
+	};
+};
+//#endregion
+export { estimatePiContextUsage, estimatePiMessageTokens };
+
+//# sourceMappingURL=context-usage.js.map

package/dist/node/pi/context-usage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-usage.js","names":[],"sources":["../../../src/pi/context-usage.ts"],"sourcesContent":["import type { Usage } from \"@earendil-works/pi-ai\";\n\nimport type { PiHarnessFrontendAgentMessage } from \"./harness/agent-harness-event-protocol\";\n\nexport type PiContextUsageEstimate = {\n  tokens: number;\n  usageTokens: number;\n  trailingTokens: number;\n  lastUsageIndex: number | null;\n};\n\nconst ESTIMATED_IMAGE_CHARS = 4_800;\n\nconst contextTokensFromUsage = (usage: Usage): number =>\n  usage.totalTokens || usage.input + usage.output + usage.cacheRead + usage.cacheWrite;\n\nconst validAssistantUsage = (message: PiHarnessFrontendAgentMessage): Usage | undefined => {\n  if (message.role !== \"assistant\") {\n    return undefined;\n  }\n\n  const assistant = message;\n  if (\n    assistant.stopReason === \"aborted\" ||\n    assistant.stopReason === \"error\" ||\n    !assistant.usage ||\n    contextTokensFromUsage(assistant.usage) === 0\n  ) {\n    return undefined;\n  }\n\n  return assistant.usage;\n};\n\nconst textAndImageContentChars = (\n  content: string | Array<{ type: string; text?: string }>,\n): number => {\n  if (typeof content === \"string\") {\n    return content.length;\n  }\n\n  return content.reduce((chars, block) => {\n    if (block.type === \"text\" && block.text) {\n      return chars + block.text.length;\n    }\n    if (block.type === \"image\") {\n      return chars + ESTIMATED_IMAGE_CHARS;\n    }\n    return chars;\n  }, 0);\n};\n\nconst readableJsonLength = (value: unknown): number => {\n  try {\n    return (JSON.stringify(value) ?? \"undefined\").length;\n  } catch {\n    return \"[unserializable]\".length;\n  }\n};\n\n/** Estimate one message using the same conservative four-characters-per-token heuristic as Pi. */\nexport const estimatePiMessageTokens = (message: PiHarnessFrontendAgentMessage): number => {\n  let chars = 0;\n\n  switch (message.role) {\n    case \"user\":\n      chars = textAndImageContentChars(message.content);\n      break;\n    case \"assistant\":\n      for (const block of message.content) {\n        if (block.type === \"text\") {\n          chars += block.text.length;\n        } else if (block.type === \"thinking\") {\n          chars += block.thinking.length;\n        } else if (block.type === \"toolCall\") {\n          chars += block.name.length + readableJsonLength(block.arguments);\n        }\n      }\n      break;\n    case \"custom\":\n    case \"toolResult\":\n      chars = textAndImageContentChars(message.content);\n      break;\n    case \"bashExecution\":\n      chars = message.command.length + message.output.length;\n      break;\n    case \"branchSummary\":\n    case \"compactionSummary\":\n      chars = message.summary.length;\n      break;\n  }\n\n  return Math.ceil(chars / 4);\n};\n\n/**\n * Estimate the context represented by an ordered message list.\n *\n * Provider usage is authoritative through the latest completed assistant message. Messages after\n * that point use the same character heuristic Pi applies while preparing compaction.\n */\nexport const estimatePiContextUsage = (\n  messages: readonly PiHarnessFrontendAgentMessage[],\n): PiContextUsageEstimate => {\n  let lastUsageIndex: number | null = null;\n  let usageTokens = 0;\n  let latestPrefixTimestamp = Number.NEGATIVE_INFINITY;\n\n  for (let index = 0; index < messages.length; index += 1) {\n    const message = messages[index];\n    const usage = validAssistantUsage(message);\n    const timestampIsFinite = Number.isFinite(message.timestamp);\n    if (usage && (!timestampIsFinite || message.timestamp >= latestPrefixTimestamp)) {\n      lastUsageIndex = index;\n      usageTokens = contextTokensFromUsage(usage);\n    }\n    if (timestampIsFinite) {\n      latestPrefixTimestamp = Math.max(latestPrefixTimestamp, message.timestamp);\n    }\n  }\n\n  const trailingStart = lastUsageIndex === null ? 0 : lastUsageIndex + 1;\n  let trailingTokens = 0;\n  for (let index = trailingStart; index < messages.length; index += 1) {\n    trailingTokens += estimatePiMessageTokens(messages[index]);\n  }\n\n  return {\n    tokens: usageTokens + trailingTokens,\n    usageTokens,\n    trailingTokens,\n    lastUsageIndex,\n  };\n};\n"],"mappings":";AAWA,MAAM,wBAAwB;AAE9B,MAAM,0BAA0B,UAC9B,MAAM,eAAe,MAAM,QAAQ,MAAM,SAAS,MAAM,YAAY,MAAM;AAE5E,MAAM,uBAAuB,YAA8D;CACzF,IAAI,QAAQ,SAAS,aACnB;CAGF,MAAM,YAAY;CAClB,IACE,UAAU,eAAe,aACzB,UAAU,eAAe,WACzB,CAAC,UAAU,SACX,uBAAuB,UAAU,KAAK,MAAM,GAE5C;CAGF,OAAO,UAAU;AACnB;AAEA,MAAM,4BACJ,YACW;CACX,IAAI,OAAO,YAAY,UACrB,OAAO,QAAQ;CAGjB,OAAO,QAAQ,QAAQ,OAAO,UAAU;EACtC,IAAI,MAAM,SAAS,UAAU,MAAM,MACjC,OAAO,QAAQ,MAAM,KAAK;EAE5B,IAAI,MAAM,SAAS,SACjB,OAAO,QAAQ;EAEjB,OAAO;CACT,GAAG,CAAC;AACN;AAEA,MAAM,sBAAsB,UAA2B;CACrD,IAAI;EACF,QAAQ,KAAK,UAAU,KAAK,KAAK,YAAA,CAAa;CAChD,QAAQ;EACN,OAAO;CACT;AACF;;AAGA,MAAa,2BAA2B,YAAmD;CACzF,IAAI,QAAQ;CAEZ,QAAQ,QAAQ,MAAhB;EACE,KAAK;GACH,QAAQ,yBAAyB,QAAQ,OAAO;GAChD;EACF,KAAK;GACH,KAAK,MAAM,SAAS,QAAQ,SAC1B,IAAI,MAAM,SAAS,QACjB,SAAS,MAAM,KAAK;QACf,IAAI,MAAM,SAAS,YACxB,SAAS,MAAM,SAAS;QACnB,IAAI,MAAM,SAAS,YACxB,SAAS,MAAM,KAAK,SAAS,mBAAmB,MAAM,SAAS;GAGnE;EACF,KAAK;EACL,KAAK;GACH,QAAQ,yBAAyB,QAAQ,OAAO;GAChD;EACF,KAAK;GACH,QAAQ,QAAQ,QAAQ,SAAS,QAAQ,OAAO;GAChD;EACF,KAAK;EACL,KAAK;GACH,QAAQ,QAAQ,QAAQ;GACxB;CACJ;CAEA,OAAO,KAAK,KAAK,QAAQ,CAAC;AAC5B;;;;;;;AAQA,MAAa,0BACX,aAC2B;CAC3B,IAAI,iBAAgC;CACpC,IAAI,cAAc;CAClB,IAAI,wBAAwB,OAAO;CAEnC,KAAK,IAAI,QAAQ,GAAG,QAAQ,SAAS,QAAQ,SAAS,GAAG;EACvD,MAAM,UAAU,SAAS;EACzB,MAAM,QAAQ,oBAAoB,OAAO;EACzC,MAAM,oBAAoB,OAAO,SAAS,QAAQ,SAAS;EAC3D,IAAI,UAAU,CAAC,qBAAqB,QAAQ,aAAa,wBAAwB;GAC/E,iBAAiB;GACjB,cAAc,uBAAuB,KAAK;EAC5C;EACA,IAAI,mBACF,wBAAwB,KAAK,IAAI,uBAAuB,QAAQ,SAAS;CAE7E;CAEA,MAAM,gBAAgB,mBAAmB,OAAO,IAAI,iBAAiB;CACrE,IAAI,iBAAiB;CACrB,KAAK,IAAI,QAAQ,eAAe,QAAQ,SAAS,QAAQ,SAAS,GAChE,kBAAkB,wBAAwB,SAAS,MAAM;CAG3D,OAAO;EACL,QAAQ,cAAc;EACtB;EACA;EACA;CACF;AACF"}