@caplets/pi 0.1.3 → 0.1.5

package/dist/index.js

@@ -1,26 +1,16 @@
+import { keyText } from "@earendil-works/pi-coding-agent";
+import { generatedToolInputJsonSchema } from "@caplets/core/generated-tool-input-schema";
 import { createNativeCapletsService, registerNativeCapletsProcessCleanup } from "@caplets/core/native";
-import { Type } from "@sinclair/typebox";
-import { operations } from "@caplets/core/generated-tool-input-schema";
-//#region src/schema.ts
-function capletsPiParameters() {
-	return Type.Object({
-		operation: Type.Union(operations.map((operation) => Type.Literal(operation))),
-		query: Type.Optional(Type.String()),
-		limit: Type.Optional(Type.Integer({ minimum: 1 })),
-		tool: Type.Optional(Type.String()),
-		arguments: Type.Optional(Type.Record(Type.String(), Type.Unknown())),
-		fields: Type.Optional(Type.Array(Type.String({ minLength: 1 }), { minItems: 1 }))
-	}, { additionalProperties: false });
-}
-//#endregion
 //#region src/index.ts
 function capletsPiExtension(pi, options = {}) {
 	const ownsService = !options.service;
 	const service = options.service ?? createNativeCapletsService();
 	if (ownsService) registerNativeCapletsProcessCleanup(service);
 	const registeredCapletToolSignatures = /* @__PURE__ */ new Map();
-	let knownCapletTools = new Set(pi.getActiveTools?.().filter((name) => name.startsWith("caplets_")) ?? []);
-	const syncTools = (caplets = service.listTools()) => {
+	let currentCapletTools = /* @__PURE__ */ new Set();
+	let knownCapletTools = /* @__PURE__ */ new Set();
+	let canSyncActiveTools = false;
+	const syncToolRegistrations = (caplets = service.listTools()) => {
 		const nextCapletTools = new Set(caplets.map((caplet) => caplet.toolName));
 		for (const [toolName] of registeredCapletToolSignatures) if (!nextCapletTools.has(toolName)) registeredCapletToolSignatures.delete(toolName);
 		for (const caplet of caplets) {
@@ -29,14 +19,24 @@ function capletsPiExtension(pi, options = {}) {
 			registeredCapletToolSignatures.set(caplet.toolName, signature);
 			pi.registerTool(createPiTool(service, caplet));
 		}
-		if (pi.getActiveTools && pi.setActiveTools) {
-			const activeNonCaplets = pi.getActiveTools().filter((name) => !knownCapletTools.has(name) && !nextCapletTools.has(name));
-			pi.setActiveTools([...activeNonCaplets, ...nextCapletTools]);
-		}
+		currentCapletTools = nextCapletTools;
+		return nextCapletTools;
+	};
+	const syncActiveTools = (nextCapletTools = currentCapletTools) => {
+		if (!canSyncActiveTools || !pi.getActiveTools || !pi.setActiveTools) return;
+		const activeNonCaplets = pi.getActiveTools().filter((name) => !knownCapletTools.has(name) && !nextCapletTools.has(name));
+		pi.setActiveTools([...activeNonCaplets, ...nextCapletTools]);
 		knownCapletTools = nextCapletTools;
 	};
-	syncTools();
-	const unsubscribe = service.onToolsChanged(syncTools);
+	currentCapletTools = syncToolRegistrations();
+	const unsubscribe = service.onToolsChanged((caplets) => {
+		syncActiveTools(syncToolRegistrations(caplets));
+	});
+	pi.on?.("session_start", () => {
+		canSyncActiveTools = true;
+		knownCapletTools = new Set(pi.getActiveTools?.().filter((name) => name.startsWith("caplets_")) ?? []);
+		syncActiveTools();
+	});
 	pi.on?.("session_shutdown", () => {
 		unsubscribe();
 		if (ownsService) service.close();
@@ -57,7 +57,7 @@ function createPiTool(service, caplet) {
 		description: caplet.description,
 		promptSnippet: `Use ${caplet.toolName} for the ${caplet.title} Caplet capability domain.`,
 		promptGuidelines: caplet.promptGuidance,
-		parameters: capletsPiParameters(),
+		parameters: generatedToolInputJsonSchema(),
 		async execute(_toolCallId, params) {
 			const result = await service.execute(caplet.caplet, params);
 			const serialized = serializeResult(result);
@@ -71,9 +71,35 @@ function createPiTool(service, caplet) {
 					serializationError: serialized.serializationError
 				} : { result }
 			};
+		},
+		renderCall(args, theme) {
+			const suffix = [stringProperty(args, "operation"), stringProperty(args, "tool")].filter(Boolean).join(" ");
+			return textComponent(theme.fg("toolTitle", theme.bold(caplet.title)) + (suffix ? ` ${theme.fg("muted", suffix)}` : ""));
+		},
+		renderResult(result, { expanded, isPartial }, theme) {
+			if (isPartial) return textComponent(theme.fg("warning", `${caplet.title} running...`));
+			const output = result.content.filter((item) => item.type === "text").map((item) => item.text).join("\n");
+			if (expanded) return textComponent(theme.fg("success", `✓ ${caplet.title} complete`) + theme.fg("dim", ` (${toolExpandKeyText()} to collapse)`) + (output ? `\n${theme.fg("toolOutput", output)}` : ""));
+			return textComponent(theme.fg("success", `✓ ${caplet.title} complete`) + theme.fg("dim", ` (${toolExpandKeyText()} to expand)`));
 		}
 	};
 }
+function toolExpandKeyText() {
+	return keyText("app.tools.expand") || "ctrl+o";
+}
+function textComponent(text) {
+	return {
+		render(_width) {
+			return text.split("\n");
+		},
+		invalidate() {}
+	};
+}
+function stringProperty(value, key) {
+	if (!value || typeof value !== "object" || !(key in value)) return;
+	const property = value[key];
+	return typeof property === "string" && property.length > 0 ? property : void 0;
+}
 function serializeResult(result) {
 	try {
 		return { text: JSON.stringify(result, null, 2) ?? "null" };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@caplets/pi",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "Native Pi extension for Caplets.",
   "homepage": "https://github.com/spiritledsoftware/caplets#readme",
   "bugs": {
@@ -26,8 +26,7 @@
     "access": "public"
   },
   "dependencies": {
-    "@sinclair/typebox": "^0.34.49",
-    "@caplets/core": "0.12.1"
+    "@caplets/core": "0.13.0"
   },
   "devDependencies": {
     "@types/node": "^25.7.0",
@@ -41,6 +40,11 @@
   "engines": {
     "node": ">=22"
   },
+  "pi": {
+    "extensions": [
+      "dist/index.js"
+    ]
+  },
   "scripts": {
     "build": "rm -rf dist && rolldown -c",
     "typecheck": "tsc --noEmit",