@aliou/pi-dev-kit 0.6.4 → 0.7.0

package/src/tools/utils.ts

@@ -1,38 +1,48 @@
 import * as fs from "node:fs";
+import { createRequire } from "node:module";
 import * as path from "node:path";
 
+const require = createRequire(import.meta.url);
+
+const PI_PACKAGE_NAMES = [
+  "@earendil-works/pi-coding-agent",
+  "@mariozechner/pi-coding-agent",
+] as const;
+
 /**
  * Find the currently running Pi installation directory by resolving the
  * pi-coding-agent package location.
  */
 export function findPiInstallation(): string | null {
-  try {
-    const piModulePath = require.resolve(
-      "@mariozechner/pi-coding-agent/package.json",
-    );
-    return path.dirname(piModulePath);
-  } catch (_error) {
-    const scriptPath = process.argv[1];
-    if (scriptPath) {
-      let currentDir = path.dirname(scriptPath);
+  for (const packageName of PI_PACKAGE_NAMES) {
+    try {
+      const piModulePath = require.resolve(`${packageName}/package.json`);
+      return path.dirname(piModulePath);
+    } catch {
+      // Try the next package namespace.
+    }
+  }
 
-      while (currentDir !== path.dirname(currentDir)) {
-        const packageJsonPath = path.join(currentDir, "package.json");
-        if (fs.existsSync(packageJsonPath)) {
-          try {
-            const packageContent = fs.readFileSync(packageJsonPath, "utf-8");
-            const packageJson = JSON.parse(packageContent);
-            if (packageJson.name === "@mariozechner/pi-coding-agent") {
-              return currentDir;
-            }
-          } catch {
-            // Continue searching
+  const scriptPath = process.argv[1];
+  if (scriptPath) {
+    let currentDir = path.dirname(scriptPath);
+
+    while (currentDir !== path.dirname(currentDir)) {
+      const packageJsonPath = path.join(currentDir, "package.json");
+      if (fs.existsSync(packageJsonPath)) {
+        try {
+          const packageContent = fs.readFileSync(packageJsonPath, "utf-8");
+          const packageJson = JSON.parse(packageContent);
+          if (PI_PACKAGE_NAMES.includes(packageJson.name)) {
+            return currentDir;
           }
+        } catch {
+          // Continue searching
         }
-        currentDir = path.dirname(currentDir);
       }
+      currentDir = path.dirname(currentDir);
     }
-
-    return null;
   }
+
+  return null;
 }

package/src/tools/version-tool.ts

@@ -2,69 +2,69 @@ 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 { Text } from "@mariozechner/pi-tui";
-import { Type } from "@sinclair/typebox";
+} from "@earendil-works/pi-coding-agent";
+import { defineTool, VERSION } from "@earendil-works/pi-coding-agent";
+import { Text } from "@earendil-works/pi-tui";
+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) {
+    if (options.isPartial) {
+      return new Text(theme.fg("dim", "Pi Version: loading..."), 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);
-      }
+    const { details } = result;
 
-      return new Text(
-        theme.fg("accent", `Pi version: ${details.version}`),
-        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);
+  },
+});
+
+export function setupVersionTool(pi: ExtensionAPI) {
+  pi.registerTool(versionTool);
 }

package/src/index.ts

@@ -1,8 +0,0 @@
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { registerCommands } from "./commands";
-import { setupTools } from "./tools";
-
-export default function (pi: ExtensionAPI) {
-  setupTools(pi);
-  registerCommands(pi);
-}