oira666_pi-limits-wait 0.6.8 → 0.6.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oira666_pi-limits-wait",
-  "version": "0.6.8",
+  "version": "0.6.10",
   "description": "Extension for PI that allows to automatically wait for limits to reset and retry the request",
   "type": "module",
   "exports": {
@@ -50,7 +50,7 @@
   "devDependencies": {
     "@earendil-works/pi-ai": "1.4.0",
     "@earendil-works/pi-coding-agent": "1.4.0",
-    "tsx": "^4.20.6",
-    "typescript": "^5.9.3"
+    "tsx": "^4.23.15",
+    "typescript": "^7.0.2"
   }
 }

package/src/header-dispatch.ts

@@ -21,7 +21,13 @@ export function installHeaderDispatch(): void {
   const runners = new WeakMap<ModelRegistry, ExtensionRunner>();
   const wrapper = function (this: ExtensionRunner) {
     const context = previous.call(this);
-    runners.set(context.modelRegistry, this);
+    // modelRegistry is a getter that throws once the runner is stale (after
+    // reload / session replacement). Never let that crash pi.
+    try {
+      runners.set(context.modelRegistry, this);
+    } catch {
+      // stale runner: skip registration
+    }
     return context;
   };
   Object.defineProperty(prototype, SLOT, { configurable: true, value: { runners, wrapper } });
@@ -29,7 +35,12 @@ export function installHeaderDispatch(): void {
 }
 
 export function fallbackHeaderTransform(): ModelsSimpleStreamOptions["transformHeaders"] {
-  const registry = state.sharedCtx?.modelRegistry;
+  let registry: ModelRegistry | undefined;
+  try {
+    registry = state.sharedCtx?.modelRegistry;
+  } catch {
+    return undefined; // stale ctx after reload / session replacement
+  }
   const runner = registry && (ExtensionRunner.prototype as RunnerPrototype)[SLOT]?.runners.get(registry);
   return runner ? (headers) => runner.emitBeforeProviderHeaders(headers) : undefined;
 }