@bytesbrains/pi-tool-awareness-gate 1.0.3 → 1.0.4

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 BytesBrains Pte Ltd
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -5,14 +5,14 @@
 ## Install
 
 ```bash
-pi install npm:pi-tool-awareness-gate
+pi install npm:@bytesbrains/pi-tool-awareness-gate
 ```
 
 Or add to `.pi/settings.json`:
 
 ```json
 {
-  "packages": ["npm:pi-tool-awareness-gate"]
+  "packages": ["npm:@bytesbrains/pi-tool-awareness-gate"]
 }
 ```
 
@@ -146,7 +146,7 @@ tool-awareness-gate/
 When a gate wants to provide domain-specific quality signals (richer than what inference can detect), it imports the envelope types:
 
 ```typescript
-import type { ToolResultEnvelope } from "pi-tool-awareness-gate/src/types";
+import type { ToolResultEnvelope } from "@bytesbrains/pi-tool-awareness-gate/src/types";
 
 // Gate enriches its own result with domain-specific quality signals
 return {
@@ -174,3 +174,8 @@ The awareness gate will **merge** gate-provided signals with its own inference,
 ## License
 
 MIT © [nandal](https://github.com/nandal)
+
+---
+
+Built and maintained by [BytesBrains](https://bytesbrains.com) — AI automation & agents, engineered to production standards.
+*The model proposes, code guarantees.*

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytesbrains/pi-tool-awareness-gate",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Tool limitation awareness layer for AI agents — intercepts tool results, infers quality signals, injects limitation reminders, and logs structured envelopes for evaluation.",
   "keywords": [
     "pi-package",
@@ -11,15 +11,14 @@
     "agent-protocol",
     "observability"
   ],
-  "author": "nandal <nandal@users.noreply.github.com>",
+  "author": "BytesBrains Pte Ltd (https://bytesbrains.com)",
   "repository": {
     "type": "git",
-    "url": "https://github.com/nandal/pi-ext",
-    "directory": "tool-awareness-gate"
+    "url": "git+https://github.com/bytesbrains/pi-tool-awareness-gate.git"
   },
-  "homepage": "https://github.com/nandal/pi-ext/tree/main/tool-awareness-gate",
+  "homepage": "https://github.com/bytesbrains/pi-tool-awareness-gate#readme",
   "bugs": {
-    "url": "https://github.com/nandal/pi-ext/issues"
+    "url": "https://github.com/bytesbrains/pi-tool-awareness-gate/issues"
   },
   "license": "MIT",
   "main": "./src/index.ts",

package/src/index.ts

@@ -75,6 +75,12 @@ export default function (pi: ExtensionAPI) {
     const latencyMs = startTime ? Date.now() - startTime : 0;
     invocationStarts.delete(event.toolCallId);
 
+    // Prune stale entries (>5 min old) to prevent memory leak from orphaned starts
+    const now = Date.now();
+    for (const [key, ts] of invocationStarts) {
+      if (now - ts > 300000) invocationStarts.delete(key);
+    }
+
     // Build raw input for inference
     const raw: RawToolResult = {
       toolName: event.toolName,
@@ -155,6 +161,7 @@ export default function (pi: ExtensionAPI) {
 
     const startTime = invocationStarts.get(event.toolCallId);
     const latencyMs = startTime ? Date.now() - startTime : 0;
+    invocationStarts.delete(event.toolCallId);
 
     const envelope: ToolResultEnvelope = {
       status: "failure",