pi-canary 1.0.2 → 1.1.0

package/extensions/canary.ts

@@ -15,6 +15,9 @@
  */
 
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { readFileSync } from "node:fs";
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
 
 // Loaded from sibling JSON at startup; /set overrides for the current session only
 const cfg = (() => {
@@ -25,8 +28,8 @@ const cfg = (() => {
     FAIL_COMPACT: 0,
   };
   try {
-    const url = new URL("canary.json", import.meta.url);
-    return { ...defaults, ...JSON.parse((globalThis as any).Deno.readTextFileSync(url)) };
+    const extDir = dirname(fileURLToPath(import.meta.url));
+    return { ...defaults, ...JSON.parse(readFileSync(join(extDir, "canary.json"), "utf-8")) };
   } catch {
     return defaults;
   }
@@ -89,6 +92,12 @@ export default function (pi: ExtensionAPI) {
   let verifyResponseTimestamp: number | null = null;
 
   pi.on("before_agent_start", (_event, _ctx) => {
+    // COUNT=0 disables the canary check entirely
+    if (cfg.COUNT === 0) {
+      phase = "idle";
+      currentTokens = null;
+      return;
+    }
     if (cfg.VARIANT === "fixed") {
       if (!fixedTokens || fixedTokens.length !== cfg.COUNT) {
         fixedTokens = Array.from({ length: cfg.COUNT }, generateToken);
@@ -270,7 +279,7 @@ export default function (pi: ExtensionAPI) {
           if (eq > 0 && val !== "") {
             if (key === "COUNT") {
               const n = parseInt(val, 10);
-              if (n > 0) { cfg.COUNT = n; fixedTokens = null; results.push(`COUNT=${cfg.COUNT}`); }
+              if (n >= 0) { cfg.COUNT = n; fixedTokens = null; results.push(`COUNT=${cfg.COUNT}`); }
               else results.push(`invalid COUNT: ${val}`);
             } else if (key === "POSITION") {
               if (val === "start" || val === "equidistant" || val === "end") {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-canary",
-  "version": "1.0.2",
+  "version": "1.1.0",
   "description": "Pi extension: silently verifies agent context awareness every turn using hidden canary tokens. KV-cache friendly.",
   "keywords": ["pi-package", "pi", "pi-coding-agent", "extension", "context-awareness", "canary", "safety", "verification", "local-llm"],
   "license": "MIT",