agent-relay-server 0.78.0 → 0.78.2

package/docs/openapi.json

@@ -2,7 +2,7 @@
   "openapi": "3.1.0",
   "info": {
     "title": "Agent Relay API",
-    "version": "0.78.0",
+    "version": "0.78.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.78.0",
+  "version": "0.78.2",
   "description": "Lightweight HTTP message relay for inter-agent communication across machines",
   "module": "src/index.ts",
   "type": "module",
@@ -33,7 +33,7 @@
     "CONTRIBUTING.md"
   ],
   "dependencies": {
-    "agent-relay-sdk": "0.2.53",
+    "agent-relay-sdk": "0.2.54",
     "ajv": "^8.20.0"
   },
   "scripts": {

package/src/db/provider-quotas.ts

@@ -59,6 +59,10 @@ function canonicalProviderQuotaAccountKey(provider: string, value: string): stri
     : accountKey;
 }
 
+function unstableProviderQuotaAccountKey(accountKey: string): boolean {
+  return accountKey.startsWith("host:") || accountKey.startsWith("home:");
+}
+
 function sourceAgentHost(sourceAgentId: string | undefined): string | undefined {
   if (!sourceAgentId) return undefined;
   const row = getDb().query("SELECT machine, rig FROM agents WHERE id = ?").get(sourceAgentId) as { machine?: string | null; rig?: string | null } | undefined;
@@ -291,6 +295,9 @@ export function upsertProviderQuota(input: ProviderQuotaUpdateInput, now = Date.
       getDb().query("UPDATE provider_quotas SET quota_advisory_state = ? WHERE provider = ? AND account_key = ?").run(quotaAdvisoryState, provider, accountKey);
     }
   }
+  if (!unstableProviderQuotaAccountKey(accountKey)) {
+    pruneUnstableProviderQuotaRecords(provider);
+  }
   pruneProviderQuotaRecords(provider, undefined, now - DAY_MS);
 
   return listProviderQuotas({ provider, accountKey, historyLimit: 24, now }).quotas[0]!;
@@ -331,19 +338,22 @@ function pruneProviderQuotaRecords(provider?: string, accountKey?: string, older
   return result.changes;
 }
 
+function pruneUnstableProviderQuotaRecords(provider?: string): number {
+  const result = provider
+    ? getDb().query("DELETE FROM provider_quotas WHERE provider = ? AND (account_key LIKE 'host:%' OR account_key LIKE 'home:%')").run(provider)
+    : getDb().query("DELETE FROM provider_quotas WHERE account_key LIKE 'host:%' OR account_key LIKE 'home:%'").run();
+  return result.changes;
+}
+
 export function normalizeProviderQuotaAccountKeys(now = Date.now()): number {
   let changes = 0;
   for (const row of getDb().query("SELECT * FROM provider_quotas").all() as ProviderQuotaRow[]) {
-    let accountKey = canonicalProviderQuotaAccountKey(row.provider, row.account_key);
-    if (accountKey.startsWith("credential:")) {
-      const sourceAgentId = normalizeSourceAgentIds(row.source_agent_ids)[0];
-      const host = sourceAgentHost(sourceAgentId);
-      if (host) accountKey = `host:${host}`;
-    }
+    const accountKey = canonicalProviderQuotaAccountKey(row.provider, row.account_key);
     if (accountKey !== row.account_key) {
       changes += mergeProviderQuotaRows(row.provider, accountKey, [row.account_key], now);
     }
   }
+  changes += pruneUnstableProviderQuotaRecords();
   changes += pruneProviderQuotaRecords(undefined, undefined, now - DAY_MS);
   return changes;
 }