pi-nebius 0.3.1 → 0.3.2

package/CHANGELOG.md

@@ -3,6 +3,14 @@
 Versions use Semantic Versioning. Entries describe changes included in the named version;
 a version is released only when its matching Git tag and GitHub release are published.
 
+## [0.3.2](https://github.com/PeterHdd/pi-nebius/compare/v0.3.1...v0.3.2) (2026-09-17)
+
+
+### Bug Fixes
+
+* launch benchmark workers from installed packages ([f7e717a](https://github.com/PeterHdd/pi-nebius/commit/f7e717a3454ae700451e63f54d78bda1d155d1dd))
+* launch benchmark workers from installed packages ([cf516f7](https://github.com/PeterHdd/pi-nebius/commit/cf516f724d0d85ce921e4b0a6347bdcb6b251493))
+
 ## [0.3.1](https://github.com/PeterHdd/pi-nebius/compare/v0.3.0...v0.3.1) (2026-09-17)
 
 

package/dist/benchmark/command.js

@@ -5,6 +5,7 @@ import { tmpdir } from "node:os";
 import { dirname, join } from "node:path";
 import { fileURLToPath, pathToFileURL } from "node:url";
 import { parseArgs } from "node:util";
+import { Text } from "@earendil-works/pi-tui";
 import { MISSING_KEY } from "../discovery.js";
 import { applyModelSettings } from "../model-settings.js";
 import { loadDefinition, positive } from "./definition.js";
@@ -33,6 +34,15 @@ export async function hostPiEntry() {
     return join(dirname(manifestPath), manifest.main);
 }
 export function registerBenchmarkCommand(pi, getSettings = () => ({})) {
+    pi.registerMessageRenderer("nebius-benchmark", (message, options, theme) => {
+        const content = typeof message.content === "string"
+            ? message.content
+            : message.content
+                .filter((part) => part.type === "text")
+                .map((part) => part.text)
+                .join("\n");
+        return new Text(`${theme.fg("customMessageLabel", "[nebius-benchmark]")}\n\n${content}`, options.outputPad ?? 1, 0);
+    });
     let starting = false;
     let active;
     const show = (content) => pi.sendMessage({ customType: "nebius-benchmark", content, display: true }, { triggerTurn: false });
@@ -149,7 +159,7 @@ export function registerBenchmarkCommand(pi, getSettings = () => ({})) {
                     apiKey: key,
                     signal: controller.signal,
                     piEntry,
-                    workerPath: fileURLToPath(new URL(import.meta.url.endsWith(".ts") ? "./host-worker.ts" : "./host-worker.js", import.meta.url)),
+                    workerPath: fileURLToPath(new URL("./host-worker.mjs", import.meta.url)),
                     workerArgs: [piEntry],
                     onRun: (run) => {
                         ctx.ui.setStatus("nebius-benchmark", `Benchmark: ${task} (${++completed}/${total})`);

package/dist/benchmark/host-worker.mjs

@@ -0,0 +1,28 @@
+import { readFileSync } from "node:fs";
+import { registerHooks, stripTypeScriptTypes } from "node:module";
+import { pathToFileURL } from "node:url";
+// Reuse the installed host's SDK; Git installs intentionally omit Pi dev dependencies.
+const hostEntry = process.argv[2];
+if (!hostEntry)
+    throw new Error("Missing host Pi entry");
+const sourceRoot = new URL("../", import.meta.url).href;
+registerHooks({
+    // Node's automatic type stripping rejects installed sources under node_modules.
+    // This JavaScript bootstrap explicitly strips only this package's own sources.
+    load(url, context, nextLoad) {
+        if (url.startsWith(sourceRoot) && url.endsWith(".ts")) {
+            return {
+                format: "module",
+                source: stripTypeScriptTypes(readFileSync(new URL(url), "utf8"), { sourceUrl: url }),
+                shortCircuit: true,
+            };
+        }
+        return nextLoad(url, context);
+    },
+    resolve(specifier, context, nextResolve) {
+        if (specifier.startsWith("@earendil-works/"))
+            return nextResolve(specifier, { ...context, parentURL: pathToFileURL(hostEntry).href });
+        return nextResolve(specifier, context);
+    },
+});
+await import("./worker.js");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-nebius",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "description": "Nebius Token Factory provider and agentic coding benchmarks for Pi",
   "repository": {
     "type": "git",
@@ -42,13 +42,15 @@
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": "*",
-    "@earendil-works/pi-coding-agent": "*"
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
   },
   "devDependencies": {
+    "@biomejs/biome": "^2.2.0",
     "@earendil-works/pi-ai": "0.85.1",
     "@earendil-works/pi-coding-agent": "0.85.1",
+    "@earendil-works/pi-tui": "0.85.1",
     "@types/node": "^22.0.0",
-    "@biomejs/biome": "^2.2.0",
     "tsx": "^4.20.0",
     "typescript": "^5.9.0"
   },

package/src/benchmark/command.ts

@@ -6,6 +6,7 @@ import { dirname, join } from "node:path";
 import { fileURLToPath, pathToFileURL } from "node:url";
 import { parseArgs } from "node:util";
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { Text } from "@earendil-works/pi-tui";
 import { MISSING_KEY } from "../discovery.ts";
 import { applyModelSettings, type ModelSettingsMap } from "../model-settings.ts";
 import type { NebiusModel } from "../models.ts";
@@ -42,6 +43,20 @@ export function registerBenchmarkCommand(
   pi: ExtensionAPI,
   getSettings: () => ModelSettingsMap = () => ({}),
 ) {
+  pi.registerMessageRenderer("nebius-benchmark", (message, options, theme) => {
+    const content =
+      typeof message.content === "string"
+        ? message.content
+        : message.content
+            .filter((part) => part.type === "text")
+            .map((part) => part.text)
+            .join("\n");
+    return new Text(
+      `${theme.fg("customMessageLabel", "[nebius-benchmark]")}\n\n${content}`,
+      options.outputPad ?? 1,
+      0,
+    );
+  });
   let starting = false;
   let active: { controller: AbortController; done: Promise<void> } | undefined;
   const show = (content: string) =>
@@ -168,12 +183,7 @@ export function registerBenchmarkCommand(
           apiKey: key,
           signal: controller.signal,
           piEntry,
-          workerPath: fileURLToPath(
-            new URL(
-              import.meta.url.endsWith(".ts") ? "./host-worker.ts" : "./host-worker.js",
-              import.meta.url,
-            ),
-          ),
+          workerPath: fileURLToPath(new URL("./host-worker.mjs", import.meta.url)),
           workerArgs: [piEntry],
           onRun: (run) => {
             ctx.ui.setStatus("nebius-benchmark", `Benchmark: ${task} (${++completed}/${total})`);

package/src/benchmark/host-worker.mjs

@@ -0,0 +1,28 @@
+import { readFileSync } from "node:fs";
+import { registerHooks, stripTypeScriptTypes } from "node:module";
+import { pathToFileURL } from "node:url";
+
+// Reuse the installed host's SDK; Git installs intentionally omit Pi dev dependencies.
+const hostEntry = process.argv[2];
+if (!hostEntry) throw new Error("Missing host Pi entry");
+const sourceRoot = new URL("../", import.meta.url).href;
+registerHooks({
+  // Node's automatic type stripping rejects installed sources under node_modules.
+  // This JavaScript bootstrap explicitly strips only this package's own sources.
+  load(url, context, nextLoad) {
+    if (url.startsWith(sourceRoot) && url.endsWith(".ts")) {
+      return {
+        format: "module",
+        source: stripTypeScriptTypes(readFileSync(new URL(url), "utf8"), { sourceUrl: url }),
+        shortCircuit: true,
+      };
+    }
+    return nextLoad(url, context);
+  },
+  resolve(specifier, context, nextResolve) {
+    if (specifier.startsWith("@earendil-works/"))
+      return nextResolve(specifier, { ...context, parentURL: pathToFileURL(hostEntry).href });
+    return nextResolve(specifier, context);
+  },
+});
+await import("./worker.ts");

package/dist/benchmark/host-worker.js

@@ -1,14 +0,0 @@
-import { registerHooks } from "node:module";
-import { pathToFileURL } from "node:url";
-// Reuse the installed host's SDK; Git installs intentionally omit Pi dev dependencies.
-const hostEntry = process.argv[2];
-if (!hostEntry)
-    throw new Error("Missing host Pi entry");
-registerHooks({
-    resolve(specifier, context, nextResolve) {
-        if (specifier.startsWith("@earendil-works/"))
-            return nextResolve(specifier, { ...context, parentURL: pathToFileURL(hostEntry).href });
-        return nextResolve(specifier, context);
-    },
-});
-await import("./worker.js");

package/src/benchmark/host-worker.ts

@@ -1,14 +0,0 @@
-import { registerHooks } from "node:module";
-import { pathToFileURL } from "node:url";
-
-// Reuse the installed host's SDK; Git installs intentionally omit Pi dev dependencies.
-const hostEntry = process.argv[2];
-if (!hostEntry) throw new Error("Missing host Pi entry");
-registerHooks({
-  resolve(specifier, context, nextResolve) {
-    if (specifier.startsWith("@earendil-works/"))
-      return nextResolve(specifier, { ...context, parentURL: pathToFileURL(hostEntry).href });
-    return nextResolve(specifier, context);
-  },
-});
-await import("./worker.ts");