@madgagarin/pi-agentrouter 1.0.3 → 1.1.0

package/README.md

@@ -1,4 +1,4 @@
-# @madgagarin/pi-agentrouter
+# pi-agentrouter
 
 [![npm version](https://img.shields.io/npm/v/@madgagarin/pi-agentrouter.svg?color=blue)](https://www.npmjs.com/package/@madgagarin/pi-agentrouter)
 [![License: MIT](https://img.shields.io/badge/License-MIT-green.svg)](LICENSE)

package/index.ts

@@ -31,14 +31,7 @@ export function saveConfig(cfg: AgentRouterConfig): void {
 
 export function normalizeApiKey(key?: string): string {
   if (!key) return "";
-  let clean = key.trim().replace(/^["']|["']$/g, "").trim();
-  if (clean.toLowerCase().startsWith("bearer ")) {
-    clean = clean.slice(7).trim();
-  }
-  if (clean.startsWith("sk-")) {
-    clean = clean.slice(3).trim();
-  }
-  return clean;
+  return key.trim().replace(/^["']|["']$/g, "").trim();
 }
 
 const initialConfig = loadConfig();
@@ -117,8 +110,30 @@ export default function (pi: ExtensionAPI) {
 
   registerAgentRouterProviders(currentApiKey);
 
+  function updatePromptRewriteEnvForModel(model?: any): void {
+    if (isAgentRouter(model?.provider, model?.baseUrl)) {
+      process.env.PI_CACHE_OPTIMIZER_NO_PROMPT_REWRITE = "1";
+    } else {
+      delete process.env.PI_CACHE_OPTIMIZER_NO_PROMPT_REWRITE;
+    }
+  }
+
+  pi.on("session_start", async (_event, ctx) => {
+    updatePromptRewriteEnvForModel(ctx.model);
+    lastRequestTime = Date.now();
+  });
+
+  pi.on("model_select", async (event) => {
+    updatePromptRewriteEnvForModel(event.model);
+  });
+
+  pi.on("before_agent_start", async (_event, ctx) => {
+    updatePromptRewriteEnvForModel(ctx.model);
+  });
+
   pi.on("turn_start", async (_event, ctx) => {
     const model = ctx.model;
+    updatePromptRewriteEnvForModel(model);
     if (!isAgentRouter(model?.provider, (model as any)?.baseUrl)) {
       return;
     }
@@ -222,7 +237,7 @@ export default function (pi: ExtensionAPI) {
         currentApiKey = cleanKey;
         saveConfig({ apiKey: cleanKey, minIntervalMs });
         registerAgentRouterProviders(cleanKey);
-        ctx.ui.notify("AgentRouter API key updated and normalized successfully for all models.", "info");
+        ctx.ui.notify("AgentRouter API key updated successfully for all models.", "info");
         return;
       }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@madgagarin/pi-agentrouter",
-  "version": "1.0.3",
+  "version": "1.1.0",
   "description": "Pi coding agent extension for AgentRouter (GPT-5.6 Sol & Claude Opus 5) with rate-limit pacing and caching.",
   "publishConfig": {
     "access": "public"