@sebastianandreasson/pi-autonomous-agents 0.14.0 → 0.14.1

package/README.md

@@ -80,7 +80,7 @@ Typical scripts:
 
 Start from [templates/pi.config.example.json](./templates/pi.config.example.json), [templates/DEVELOPER.md](./templates/DEVELOPER.md), [templates/TESTER.md](./templates/TESTER.md), and [templates/gitignore.fragment](./templates/gitignore.fragment).
 
-Request telemetry is enabled by default for SDK runs. `pi-harness` writes a managed Pi extension shim to `.pi/extensions/pi-harness-request-telemetry/index.mjs` in the consuming repo, and Pi auto-discovers it on the next resource reload. Disable that with `PI_REQUEST_TELEMETRY_ENABLED=0` or `"piRequestTelemetryEnabled": false`.
+Request telemetry is enabled by default for SDK runs. `pi-harness` writes a managed Pi extension package under `.pi/extensions/pi-harness-request-telemetry/` in the consuming repo, with a `package.json` manifest and `index.mjs` shim that Pi auto-discovers on the next resource reload. Disable that with `PI_REQUEST_TELEMETRY_ENABLED=0` or `"piRequestTelemetryEnabled": false`.
 
 ## CLI
 

package/docs/PI_REQUEST_TELEMETRY_EXTENSION.md

@@ -2,7 +2,7 @@
 
 This document describes the repo-local Pi extension prototype under [pi-extensions/request-telemetry](../pi-extensions/request-telemetry/README.md).
 
-In normal `pi-harness` SDK runs, this extension is auto-enabled by installing a managed shim under `.pi/extensions/pi-harness-request-telemetry/` in the consuming repo before Pi reloads resources. Opt out with `PI_REQUEST_TELEMETRY_ENABLED=0` or `"piRequestTelemetryEnabled": false`.
+In normal `pi-harness` SDK runs, this extension is auto-enabled by installing a managed extension package under `.pi/extensions/pi-harness-request-telemetry/` in the consuming repo before Pi reloads resources. That package contains a `package.json` Pi manifest plus the generated `index.mjs` shim. Opt out with `PI_REQUEST_TELEMETRY_ENABLED=0` or `"piRequestTelemetryEnabled": false`.
 
 The purpose of this extension is to gather request-level data directly from Pi extension hooks before we decide whether to patch `@mariozechner/pi-coding-agent` or `pi-ai`.
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@sebastianandreasson/pi-autonomous-agents",
   "private": false,
-  "version": "0.14.0",
+  "version": "0.14.1",
   "type": "module",
   "description": "Portable unattended PI harness for developer/tester/visual-review loops.",
   "license": "MIT",

package/src/pi-request-telemetry.mjs

@@ -761,6 +761,7 @@ export function getManagedRequestTelemetryExtensionPaths({ cwd } = {}) {
     extensionRoot,
     extensionDir,
     entryFile: path.join(extensionDir, 'index.mjs'),
+    manifestFile: path.join(extensionDir, 'package.json'),
     sourceFile: bundledRequestTelemetryExtensionFile,
   }
 }
@@ -775,6 +776,17 @@ function renderRequestTelemetryExtensionShim(sourceFile) {
   ].join('\n')
 }
 
+function renderRequestTelemetryExtensionManifest() {
+  return `${JSON.stringify({
+    name: REQUEST_TELEMETRY_EXTENSION_DIRNAME,
+    private: true,
+    type: 'module',
+    pi: {
+      extensions: ['./index.mjs'],
+    },
+  }, null, 2)}\n`
+}
+
 export async function ensureBundledRequestTelemetryExtension({ cwd, enabled = true } = {}) {
   const paths = getManagedRequestTelemetryExtensionPaths({ cwd })
 
@@ -792,15 +804,23 @@ export async function ensureBundledRequestTelemetryExtension({ cwd, enabled = tr
   await fs.access(paths.sourceFile)
   await fs.mkdir(paths.extensionDir, { recursive: true })
 
-  const content = renderRequestTelemetryExtensionShim(paths.sourceFile)
-  let existing = ''
+  const entryContent = renderRequestTelemetryExtensionShim(paths.sourceFile)
+  const manifestContent = renderRequestTelemetryExtensionManifest()
+  let existingEntry = ''
+  let existingManifest = ''
   try {
-    existing = await fs.readFile(paths.entryFile, 'utf8')
+    existingEntry = await fs.readFile(paths.entryFile, 'utf8')
+  } catch {}
+  try {
+    existingManifest = await fs.readFile(paths.manifestFile, 'utf8')
   } catch {}
 
-  const updated = existing !== content
-  if (updated) {
-    await fs.writeFile(paths.entryFile, content, 'utf8')
+  const updated = existingEntry !== entryContent || existingManifest !== manifestContent
+  if (existingEntry !== entryContent) {
+    await fs.writeFile(paths.entryFile, entryContent, 'utf8')
+  }
+  if (existingManifest !== manifestContent) {
+    await fs.writeFile(paths.manifestFile, manifestContent, 'utf8')
   }
 
   return {

package/src/pi-visualizer-server.mjs

@@ -445,6 +445,7 @@ npm run dev:visualizer:ui</pre>
 export async function startVisualizerServer(config, overrides = {}) {
   const host = String(overrides.host ?? readVisualizerHost()).trim() || '127.0.0.1'
   const port = Number.isFinite(Number(overrides.port)) ? Number(overrides.port) : readVisualizerPort()
+  const sockets = new Set()
 
   const server = http.createServer(async (req, res) => {
     try {
@@ -494,6 +495,13 @@ export async function startVisualizerServer(config, overrides = {}) {
     }
   })
 
+  server.on('connection', (socket) => {
+    sockets.add(socket)
+    socket.on('close', () => {
+      sockets.delete(socket)
+    })
+  })
+
   await new Promise((resolve, reject) => {
     server.once('error', reject)
     server.listen(port, host, () => {
@@ -509,14 +517,44 @@ export async function startVisualizerServer(config, overrides = {}) {
     port,
     url,
     async close() {
-      await new Promise((resolve, reject) => {
+      await new Promise((resolve) => {
+        const finish = () => {
+          for (const socket of sockets) {
+            socket.destroy()
+          }
+          sockets.clear()
+          resolve()
+        }
+
+        const timeout = setTimeout(() => {
+          if (typeof server.closeIdleConnections === 'function') {
+            server.closeIdleConnections()
+          }
+          if (typeof server.closeAllConnections === 'function') {
+            server.closeAllConnections()
+          }
+          finish()
+        }, 1000)
+
         server.close((error) => {
+          clearTimeout(timeout)
           if (error) {
-            reject(error)
+            finish()
             return
           }
-          resolve()
+          finish()
         })
+
+        if (typeof server.closeIdleConnections === 'function') {
+          server.closeIdleConnections()
+        }
+        if (typeof server.closeAllConnections === 'function') {
+          server.closeAllConnections()
+        }
+
+        for (const socket of sockets) {
+          socket.end()
+        }
       })
     },
   }