@aliou/pi-neuralwatt 0.8.0 → 0.8.1

package/extensions/provider/index.ts

@@ -119,14 +119,17 @@ export default async function (pi: ExtensionAPI) {
   let hiddenModelsLoaded = false;
   let hiddenModelsAbort: AbortController | undefined;
 
+  let lastSseEmitAt = 0;
+
   const handleSseQuota = (line: string) => {
+    const now = Date.now();
+    if (now - lastSseEmitAt < HEADER_EMIT_THROTTLE_MS) return;
+
     const quotas = updateQuotasFromSseComment(latestQuotas, line);
     if (!quotas || quotas === latestQuotas) return;
-    latestQuotas = quotas;
-    pi.events.emit(NEURALWATT_QUOTAS_UPDATED_EVENT, {
-      quotas,
-      source: "sse",
-    });
+
+    lastSseEmitAt = now;
+    emitQuotas(quotas, "sse");
   };
 
   registerNeuralwattProvider(pi, handleSseQuota, hiddenModels);

package/extensions/quota-warnings/index.ts

@@ -17,7 +17,6 @@ export default async function (pi: ExtensionAPI) {
   await configLoader.load();
 
   let enabled = configLoader.getConfig().quotaWarnings.enabled;
-  let currentProvider: string | undefined;
   let currentContext: ExtensionContext | undefined;
 
   // Listen for config changes at runtime
@@ -33,36 +32,29 @@ export default async function (pi: ExtensionAPI) {
   pi.events.on(NEURALWATT_QUOTAS_UPDATED_EVENT, (data: unknown) => {
     if (!enabled) return;
     if (!data || typeof data !== "object") return;
-    if (currentProvider !== "neuralwatt" || !currentContext) return;
-    const { quotas, source } = data as NeuralwattQuotasUpdatedPayload;
-    checkQuotas(currentContext, quotas, source === "header");
+    if (!currentContext) return;
+    if (currentContext.model?.provider !== "neuralwatt") return;
+
+    const { quotas } = data as NeuralwattQuotasUpdatedPayload;
+    checkQuotas(currentContext, quotas);
   });
 
   pi.on("session_start", async (_event, ctx) => {
     currentContext = ctx;
-    currentProvider = ctx.model?.provider;
     if (ctx.model?.provider !== "neuralwatt") return;
     clearAlertState();
   });
 
   pi.on("model_select", (_event, ctx) => {
     currentContext = ctx;
-    currentProvider = ctx.model?.provider;
-    if (ctx.model?.provider !== "neuralwatt") {
-      clearAlertState();
-      return;
-    }
-    clearAlertState();
   });
 
   pi.on("session_before_switch", (_event, ctx) => {
     currentContext = ctx;
-    currentProvider = ctx.model?.provider;
   });
 
   pi.on("session_shutdown", () => {
     currentContext = undefined;
-    currentProvider = undefined;
     clearAlertState();
   });
 

package/extensions/quota-warnings/notifier.ts

@@ -11,6 +11,8 @@ interface AlertState {
   lastNotifiedAt: number;
 }
 
+// Module-level state so cooldowns survive across invocations of checkQuotas()
+// within the same Pi runtime.
 const alerts = new Map<string, AlertState>();
 
 export function clearAlertState(): void {
@@ -22,9 +24,9 @@ function shouldNotify(key: string, severity: WarningSeverity): boolean {
   if (!state) return true;
 
   const order: WarningSeverity[] = ["warning", "critical"];
-  if (order.indexOf(severity) > order.indexOf(state.lastSeverity)) return true;
-
-  if (severity === "critical") return true;
+  const currentIndex = order.indexOf(severity);
+  const lastIndex = order.indexOf(state.lastSeverity);
+  if (currentIndex > lastIndex) return true;
 
   return Date.now() - state.lastNotifiedAt >= COOLDOWN_MS;
 }
@@ -40,7 +42,6 @@ function markNotified(key: string, severity: WarningSeverity): void {
 export function checkQuotas(
   ctx: ExtensionContext,
   quotas: NeuralwattQuotas,
-  skipAlreadyWarned: boolean,
 ): void {
   if (!ctx.hasUI) return;
 
@@ -55,7 +56,7 @@ export function checkQuotas(
       if (pct <= 25) {
         const severity: WarningSeverity = pct <= 10 ? "critical" : "warning";
         const key = "credits";
-        if (!skipAlreadyWarned || shouldNotify(key, severity)) {
+        if (shouldNotify(key, severity)) {
           markNotified(key, severity);
           warnings.push(
             `Credits: ${pct.toFixed(0)}% remaining (${formatUsd(credits_remaining_usd)} of ${formatUsd(total_credits_usd)})`,
@@ -77,7 +78,7 @@ export function checkQuotas(
             ? "critical"
             : "warning";
         const key = "energy";
-        if (!skipAlreadyWarned || shouldNotify(key, severity)) {
+        if (shouldNotify(key, severity)) {
           markNotified(key, severity);
           const tag = in_overage ? " [OVERAGE]" : "";
           warnings.push(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aliou/pi-neuralwatt",
-  "version": "0.8.0",
+  "version": "0.8.1",
   "license": "MIT",
   "type": "module",
   "private": false,