@aliou/pi-dev-kit 0.6.3 → 0.6.5

package/src/tools/version-tool.ts

@@ -2,69 +2,65 @@ import { ToolCallHeader } from "@aliou/pi-utils-ui";
 import type {
   AgentToolResult,
   ExtensionAPI,
-  ExtensionContext,
   Theme,
-  ToolRenderResultOptions,
 } from "@mariozechner/pi-coding-agent";
-import { VERSION } from "@mariozechner/pi-coding-agent";
+import { defineTool, VERSION } from "@mariozechner/pi-coding-agent";
 import { Text } from "@mariozechner/pi-tui";
-import { Type } from "@sinclair/typebox";
+import { type Static, Type } from "typebox";
 
 const VersionParams = Type.Object({});
-type VersionParamsType = Record<string, never>;
+type VersionParamsType = Static<typeof VersionParams>;
+
+function renderVersionCall(_args: VersionParamsType, theme: Theme) {
+  return new ToolCallHeader({ toolName: "Pi Version" }, theme);
+}
 
 interface VersionDetails {
   version?: string;
 }
 
-export function setupVersionTool(pi: ExtensionAPI) {
-  pi.registerTool<typeof VersionParams, VersionDetails>({
-    name: "pi_version",
-    label: "Pi Version",
-    description: "Get the version of the currently running Pi instance",
-    promptSnippet: "Check the current Pi version.",
-    promptGuidelines: [
-      "Use pi_version when the user asks about the Pi version or when a task depends on knowing the installed version.",
-    ],
+const versionTool = defineTool({
+  name: "pi_version",
+  label: "Pi Version",
+  description: "Get the version of the currently running Pi instance",
+  promptSnippet: "Check the current Pi version.",
+  promptGuidelines: [
+    "Use pi_version when the user asks about the Pi version or when a task depends on knowing the installed version.",
+  ],
 
-    parameters: VersionParams,
+  parameters: VersionParams,
 
-    async execute(
-      _toolCallId: string,
-      _params: VersionParamsType,
-      _signal: AbortSignal | undefined,
-      _onUpdate: unknown,
-      _ctx: ExtensionContext,
-    ): Promise<AgentToolResult<VersionDetails>> {
-      return {
-        content: [{ type: "text", text: `Pi version ${VERSION}` }],
-        details: { version: VERSION },
-      };
-    },
+  async execute(
+    _toolCallId,
+    _params,
+    _signal,
+    _onUpdate,
+    _ctx,
+  ): Promise<AgentToolResult<VersionDetails>> {
+    return {
+      content: [{ type: "text", text: `Pi version ${VERSION}` }],
+      details: { version: VERSION },
+    };
+  },
 
-    renderCall(_args: VersionParamsType, theme: Theme) {
-      return new ToolCallHeader({ toolName: "Pi Version" }, theme);
-    },
+  renderCall(args, theme) {
+    return renderVersionCall(args, theme);
+  },
 
-    renderResult(
-      result: AgentToolResult<VersionDetails>,
-      _options: ToolRenderResultOptions,
-      theme: Theme,
-    ): Text {
-      const { details } = result;
+  renderResult(result, _options, theme) {
+    const { details } = result;
 
-      if (!details?.version) {
-        const textBlock = result.content.find((c) => c.type === "text");
-        const msg =
-          (textBlock?.type === "text" && textBlock.text) || "Unknown version";
-        return new Text(theme.fg("error", msg), 0, 0);
-      }
+    if (!details?.version) {
+      const textBlock = result.content.find((c) => c.type === "text");
+      const msg =
+        (textBlock?.type === "text" && textBlock.text) || "Unknown version";
+      return new Text(theme.fg("error", msg), 0, 0);
+    }
 
-      return new Text(
-        theme.fg("accent", `Pi version: ${details.version}`),
-        0,
-        0,
-      );
-    },
-  });
+    return new Text(theme.fg("accent", `Pi version: ${details.version}`), 0, 0);
+  },
+});
+
+export function setupVersionTool(pi: ExtensionAPI) {
+  pi.registerTool(versionTool);
 }