agent-relay-server 0.89.0 → 0.89.2

package/docs/openapi.json

@@ -2,7 +2,7 @@
   "openapi": "3.1.0",
   "info": {
     "title": "Agent Relay API",
-    "version": "0.89.0",
+    "version": "0.89.2",
     "description": "Real-time message bus for inter-agent communication. Agent-first: this spec is designed for machine consumption — agents can self-discover the full API surface via GET /api/spec.",
     "license": {
       "name": "MIT",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-relay-server",
-  "version": "0.89.0",
+  "version": "0.89.2",
   "description": "Lightweight HTTP message relay for inter-agent communication across machines",
   "module": "src/index.ts",
   "type": "module",

package/src/db/provider-quotas.ts

@@ -5,6 +5,9 @@ import { parseStringArray } from "./shared.ts";
 import type { AgentCard, ProviderQuotaError, ProviderQuotaMatrix, ProviderQuotaRecord, ProviderQuotaSnapshot, ProviderQuotaUpdateInput, QuotaState } from "../types";
 
 const PROVIDER_QUOTA_STALE_AFTER_MS = 10 * 60_000;
+const QUOTA_UTILIZATION_EPSILON = 1e-9;
+const QUOTA_RESET_BOUNDARY_ADVANCE_MS = 5_000;
+const QUOTA_RESET_UTILIZATION_MAX = 0.01;
 
 export interface ProviderQuotaAdvisoryInput {
   provider: string;
@@ -193,6 +196,30 @@ function quotaWindowKey(window: QuotaState["windows"][number]): string {
   return `${window.name}:${window.unit}`;
 }
 
+function quotaResetBoundaryAdvanced(existing: QuotaState["windows"][number], incoming: QuotaState["windows"][number]): boolean {
+  return quotaResetBoundaryMoved(existing, incoming)
+    && incoming.resetsAt! - existing.resetsAt! > QUOTA_RESET_BOUNDARY_ADVANCE_MS;
+}
+
+function quotaResetBoundaryMoved(existing: QuotaState["windows"][number], incoming: QuotaState["windows"][number]): boolean {
+  return typeof existing.resetsAt === "number"
+    && Number.isFinite(existing.resetsAt)
+    && typeof incoming.resetsAt === "number"
+    && Number.isFinite(incoming.resetsAt)
+    && incoming.resetsAt > existing.resetsAt;
+}
+
+function quotaWindowLooksReset(existing: QuotaState["windows"][number], incoming: QuotaState["windows"][number]): boolean {
+  return incoming.utilization <= QUOTA_RESET_UTILIZATION_MAX
+    && quotaResetBoundaryMoved(existing, incoming);
+}
+
+function quotaWindowUtilizationRegressed(existing: QuotaState["windows"][number], incoming: QuotaState["windows"][number]): boolean {
+  return incoming.utilization + QUOTA_UTILIZATION_EPSILON < existing.utilization
+    && !quotaResetBoundaryAdvanced(existing, incoming)
+    && !quotaWindowLooksReset(existing, incoming);
+}
+
 function publicQuotaState(quota: StoredQuotaState): QuotaState {
   return {
     windows: quota.windows.map((window) => ({ ...window })),
@@ -283,6 +310,10 @@ function mergeQuotaWindows(existing: StoredQuotaState | undefined, incoming: Quo
       mergedTimes.set(key, existingUpdatedAt);
       return window;
     }
+    if (quotaWindowUtilizationRegressed(window, incomingWindow)) {
+      mergedTimes.set(key, existingUpdatedAt);
+      return window;
+    }
     acceptedIncomingWindow = true;
     updatedAt = Math.max(updatedAt, incoming.updatedAt);
     source = incoming.source;
@@ -427,7 +458,7 @@ export function upsertProviderQuota(input: ProviderQuotaUpdateInput, now = Date.
       provider,
       accountKey,
       previousQuota: existingQuota ? publicQuotaState(existingQuota) : undefined,
-      quota: input.quota,
+      quota: publicQuotaState(mergedQuota.quota),
       sourceAgentIds,
       previousAdvisoryState: existing?.quota_advisory_state,
       now,

package/src/quota-advisory.ts

@@ -32,6 +32,7 @@ type AgentRecipientRow = {
 
 const RESET_ROUNDING_PERCENT = 99;
 const QUOTA_THRESHOLD_EPSILON = 1e-9;
+const QUOTA_RESET_BOUNDARY_ADVANCE_MS = 5_000;
 
 export function installQuotaAdvisoryHandler(): void {
   setProviderQuotaAdvisoryHandler(handleProviderQuotaAdvisory);
@@ -46,7 +47,7 @@ function handleProviderQuotaAdvisory(input: ProviderQuotaAdvisoryInput): string
     const previousWindow = input.previousQuota?.windows.find((candidate) => candidate.name === window.name);
     const remaining = quotaRemaining(window.utilization);
     const remainingPercent = percent(remaining);
-    const previous = nextState[window.name];
+    let previous = nextState[window.name];
     const reset = hasLastQuotaAdvisory(previous) && isQuotaReset(remaining, previousWindow, window, config.resetRemaining);
     if (reset) {
       notifyQuotaAdvisory({
@@ -65,8 +66,12 @@ function handleProviderQuotaAdvisory(input: ProviderQuotaAdvisoryInput): string
       continue;
     }
 
+    if (previous && hasLastQuotaAdvisory(previous) && quotaAdvisoryResetBoundaryAdvanced(previous, window)) {
+      delete nextState[window.name];
+      previous = undefined;
+    }
+
     if (remaining - config.warnRemaining > QUOTA_THRESHOLD_EPSILON) {
-      if (previous) delete nextState[window.name];
       continue;
     }
 
@@ -101,6 +106,17 @@ function hasLastQuotaAdvisory(previous: QuotaAdvisoryWindowState | undefined): b
   return previous?.lastQuotaAdvisoryStage !== undefined || previous?.lastQuotaAdvisoryBand !== undefined;
 }
 
+function quotaAdvisoryResetBoundaryAdvanced(
+  previous: QuotaAdvisoryWindowState,
+  window: QuotaState["windows"][number],
+): boolean {
+  return typeof previous.lastResetsAt === "number"
+    && Number.isFinite(previous.lastResetsAt)
+    && typeof window.resetsAt === "number"
+    && Number.isFinite(window.resetsAt)
+    && window.resetsAt - previous.lastResetsAt > QUOTA_RESET_BOUNDARY_ADVANCE_MS;
+}
+
 function lastQuotaAdvisoryStage(
   previous: QuotaAdvisoryWindowState | undefined,
   criticalRemaining: number,