@skydiveai/pi-extensions 0.1.0-beta.2610 → 0.1.0-beta.2611

package/dist/index.mjs

@@ -26,6 +26,7 @@ import { promisify } from "node:util";
 import { parse } from "yaml";
 import { getModels } from "@earendil-works/pi-ai";
 import { quote } from "shell-quote";
+import { createHash } from "node:crypto";
 import { createWriteStream } from "node:fs";
 import { finished } from "node:stream/promises";
 import { createLocalBashOperations } from "@earendil-works/pi-coding-agent";
@@ -4477,6 +4478,33 @@ const toolCallEnvExtension = (pi) => {
 	});
 };
 //#endregion
+//#region src/extensions/tool-history.ts
+function historyToolName(name) {
+	if (/^[a-zA-Z0-9_-]{1,64}$/.test(name)) return name;
+	return `invalid_tool_${createHash("sha256").update(name).digest("hex").slice(0, 32)}`;
+}
+/** Repair provider-bound history, never tool execution or persisted messages. */
+function repairToolHistory(messages) {
+	if (!messages.some((message) => message.role === "assistant" && message.content.some((block) => block.type === "toolCall" && historyToolName(block.name) !== block.name) || message.role === "toolResult" && historyToolName(message.toolName) !== message.toolName)) return messages;
+	return messages.map((message) => {
+		if (message.role === "assistant") return {
+			...message,
+			content: message.content.map((block) => block.type === "toolCall" ? {
+				...block,
+				name: historyToolName(block.name)
+			} : block)
+		};
+		if (message.role === "toolResult") return {
+			...message,
+			toolName: historyToolName(message.toolName)
+		};
+		return message;
+	});
+}
+const toolHistoryExtension = (pi) => {
+	pi.on("context", ({ messages }) => ({ messages: repairToolHistory(messages) }));
+};
+//#endregion
 //#region src/extensions/background-tasks.ts
 /**
 * Background bash tasks as a pi extension.
@@ -5186,7 +5214,8 @@ function platformExtensions({ sessionId, channelContext }) {
 		createSubagentExtension({ channelContext }),
 		createProactiveResultExtension({ channelContext }),
 		createContextManagementExtension(),
-		resourcePressureWarningExtension
+		resourcePressureWarningExtension,
+		toolHistoryExtension
 	];
 }
 //#endregion

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skydiveai/pi-extensions",
-  "version": "0.1.0-beta.2610",
+  "version": "0.1.0-beta.2611",
   "homepage": "https://skydive.com",
   "license": "MIT",
   "author": "Create, Inc.",