pi-canary 1.1.0 → 1.1.2

package/README.md

@@ -51,7 +51,7 @@ Persistent configuration lives in `extensions/canary.json`. You can ask the agen
 
 | Key | Default | Description |
 |-----|---------|-------------|
-| `COUNT` | `3` | Number of canary tokens injected per turn |
+| `COUNT` | `3` | Number of canary tokens injected per turn (`0` disables the canary check entirely) |
 | `POSITION` | `end` | Where tokens are injected: `start`, `equidistant`, or `end` |
 | `VARIANT` | `fixed` | `fixed` = same tokens every turn (preserves KV cache); `variant` = new tokens each turn |
 | `FAIL_COMPACT` | `0` | Compact context after N consecutive failures (`0` = disabled) |

package/extensions/canary.ts

@@ -116,10 +116,19 @@ export default function (pi: ExtensionAPI) {
       verifyContextSent = true;
       const messages = [...event.messages];
 
-      // Suppress the original user question so the agent focuses only on the canary check.
-      // The question remains in session history and reappears in Phase 2.
+      // Replace the original user question with a neutral prompt so the agent focuses
+      // only on the canary check. We keep the user role (replacing content, not the
+      // message) because some providers (e.g. llama-server) use Jinja2 chat templates
+      // that require a user message at the end of the conversation — removing it causes
+      // template parsing to fail with "No user query found in messages." The original
+      // question remains in session history and reappears in Phase 2.
       if (messages.length > 0 && (messages[messages.length - 1] as any).role === "user") {
-        messages.pop();
+        const lastMsg = messages[messages.length - 1] as any;
+        if (typeof lastMsg.content === "string") {
+          lastMsg.content = "Please return the canary tokens.";
+        } else if (Array.isArray(lastMsg.content) && lastMsg.content.length > 0) {
+          lastMsg.content = [{ type: "text", text: "Please return the canary tokens." }];
+        }
       }
 
       const histLen = messages.length;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-canary",
-  "version": "1.1.0",
+  "version": "1.1.2",
   "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",

package/skills/canary-help/SKILL.md

@@ -21,7 +21,7 @@ injected into the conversation history. Passed = proceed; failed = warning.
 
 | Key | Default | Valid values | What it controls |
 |-----|---------|-------------|-----------------|
-| `COUNT` | `3` | integer > 0 | Number of canary tokens per turn |
+| `COUNT` | `3` | integer ≥ 0 | Number of canary tokens per turn (`0` disables the canary check entirely) |
 | `POSITION` | `end` | `start` \| `equidistant` \| `end` | Where tokens are injected in context |
 | `VARIANT` | `fixed` | `fixed` \| `variant` | Fixed reuses same tokens (KV-cache friendly); variant regenerates each turn |
 | `FAIL_COMPACT` | `0` | integer ≥ 0 | Trigger compaction after N consecutive failures (0 = disabled) |