agent-usage-all-in-one 0.8.0 → 0.8.1

package/dist/cli.js

@@ -1490,6 +1490,11 @@ function createProcessingStatus(startedAt, hardRebuild) {
 function yieldToEventLoop() {
   return new Promise((resolve5) => setImmediate(resolve5));
 }
+function yieldForPendingHttp() {
+  return new Promise((resolve5) => {
+    setTimeout(resolve5, PENDING_HTTP_YIELD_MS);
+  });
+}
 function providerForConnector(id, definitions) {
   return definitions.find((definition) => definition.id === id)?.target.provider.id ?? id;
 }
@@ -1562,7 +1567,7 @@ function pickAgentProviderIndex(overview) {
     providers: overview.providers.filter((provider) => provider.id !== "opencode" && provider.id !== "dsh").map(({ id, displayName }) => ({ id, displayName }))
   };
 }
-var UsageApplication;
+var UsageApplication, PENDING_HTTP_YIELD_MS;
 var init_usage_application = __esm({
   "src/core/usage-application.ts"() {
     "use strict";
@@ -1762,15 +1767,15 @@ var init_usage_application = __esm({
         const runtimeStates = new Map(
           this.#repository.getConnectorRuntimeStates().map((state) => [state.id, state])
         );
-        await Promise.all(
+        const collected = await Promise.all(
           this.#connectors.map(async (connector) => {
             if (connector.consentId && connectorStates.get(connector.consentId) !== "connected") {
-              return;
+              return null;
             }
             const now = this.#clock();
             const runtime = runtimeStates.get(connector.id);
             if (!options.userInitiated && runtime?.nextAllowedAt && new Date(runtime.nextAllowedAt).getTime() > now.getTime()) {
-              return;
+              return null;
             }
             const policy = this.#connectorPolicies[connector.id] ?? {
               minimumIntervalMs: 0,
@@ -1782,54 +1787,27 @@ var init_usage_application = __esm({
                 policy.timeoutMs,
                 `${connector.id} timed out`
               );
-              this.#repository.saveSnapshot(this.#withRetailCosts(snapshot2));
-              if (snapshot2.warnings && snapshot2.warnings.length > 0) {
-                const failure = redactFailure(combineFailures(snapshot2.warnings));
-                this.#repository.recordFailure(snapshot2.provider, this.#clock().toISOString(), failure);
-                this.#saveConnectorDiagnostic(
-                  connector,
-                  snapshot2.provider.id,
-                  snapshot2.billingDomains[0]?.id ?? defaultBillingDomain(connector.id, this.#connectorDefinitions),
-                  failure
-                );
-                this.#recordConnectorOutcome(
-                  connector.id,
-                  now,
-                  policy,
-                  false,
-                  runtime?.failureCount ?? 0
-                );
-              } else {
-                this.#saveHealthyConnectorDiagnostic(
-                  connector,
-                  snapshot2.provider.id,
-                  snapshot2.billingDomains[0]?.id ?? defaultBillingDomain(connector.id, this.#connectorDefinitions),
-                  snapshot2.observedAt
-                );
-                this.#recordConnectorOutcome(
-                  connector.id,
-                  now,
-                  policy,
-                  true,
-                  runtime?.failureCount ?? 0
-                );
-              }
+              return { ok: true, connector, now, runtime, policy, snapshot: snapshot2 };
             } catch (error) {
-              const providerId = providerForConnector(connector.id, this.#connectorDefinitions);
-              const failure = redactFailure(safeConnectorFailure(error));
-              this.#repository.recordFailure(
-                {
-                  id: providerId,
-                  displayName: displayNameForProvider(providerId, connector.displayName)
-                },
-                this.#clock().toISOString(),
-                failure
-              );
+              return { ok: false, connector, now, runtime, policy, error };
+            }
+          })
+        );
+        if (collected.some((result) => result !== null)) await yieldForPendingHttp();
+        for (const result of collected) {
+          if (!result) continue;
+          const { connector, now, runtime, policy } = result;
+          if (result.ok) {
+            const snapshot2 = result.snapshot;
+            await this.#repository.saveSnapshot(this.#withRetailCosts(snapshot2));
+            if (snapshot2.warnings && snapshot2.warnings.length > 0) {
+              const failure2 = redactFailure(combineFailures(snapshot2.warnings));
+              this.#repository.recordFailure(snapshot2.provider, this.#clock().toISOString(), failure2);
               this.#saveConnectorDiagnostic(
                 connector,
-                providerId,
-                defaultBillingDomain(connector.id, this.#connectorDefinitions),
-                failure
+                snapshot2.provider.id,
+                snapshot2.billingDomains[0]?.id ?? defaultBillingDomain(connector.id, this.#connectorDefinitions),
+                failure2
               );
               this.#recordConnectorOutcome(
                 connector.id,
@@ -1838,9 +1816,35 @@ var init_usage_application = __esm({
                 false,
                 runtime?.failureCount ?? 0
               );
+            } else {
+              this.#saveHealthyConnectorDiagnostic(
+                connector,
+                snapshot2.provider.id,
+                snapshot2.billingDomains[0]?.id ?? defaultBillingDomain(connector.id, this.#connectorDefinitions),
+                snapshot2.observedAt
+              );
+              this.#recordConnectorOutcome(connector.id, now, policy, true, runtime?.failureCount ?? 0);
             }
-          })
-        );
+            continue;
+          }
+          const providerId = providerForConnector(connector.id, this.#connectorDefinitions);
+          const failure = redactFailure(safeConnectorFailure(result.error));
+          this.#repository.recordFailure(
+            {
+              id: providerId,
+              displayName: displayNameForProvider(providerId, connector.displayName)
+            },
+            this.#clock().toISOString(),
+            failure
+          );
+          this.#saveConnectorDiagnostic(
+            connector,
+            providerId,
+            defaultBillingDomain(connector.id, this.#connectorDefinitions),
+            failure
+          );
+          this.#recordConnectorOutcome(connector.id, now, policy, false, runtime?.failureCount ?? 0);
+        }
         if (this.#repository.getMonitoringSettings().notificationsEnabled) {
           await this.#sendNotificationTransitions();
         }
@@ -2179,9 +2183,9 @@ var init_usage_application = __esm({
         if (snapshot2.usage.length === 0 && snapshot2.costs.length === 0) {
           throw new Error("Telemetry payload contained no supported metrics");
         }
-        await this.#queueDatabaseWrite(() => {
-          this.#repository.saveSnapshot(this.#withRetailCosts(snapshot2), { preserveFailure: true });
-        });
+        await this.#queueDatabaseWrite(
+          () => this.#repository.saveSnapshot(this.#withRetailCosts(snapshot2), { preserveFailure: true })
+        );
       }
       #queueDatabaseWrite(action) {
         const result = this.#databaseWriteQueue.then(action, action);
@@ -2296,6 +2300,7 @@ var init_usage_application = __esm({
         return status;
       }
     };
+    PENDING_HTTP_YIELD_MS = 20;
   }
 });
 
@@ -8879,7 +8884,7 @@ var init_sqlite_usage_repository = __esm({
         this.#database.exec("PRAGMA journal_mode = WAL; PRAGMA foreign_keys = ON;");
         this.#migrate();
       }
-      saveSnapshot(snapshot2, options = {}) {
+      async saveSnapshot(snapshot2, options = {}) {
         this.#database.exec("BEGIN IMMEDIATE");
         try {
           const providerSql = options.preserveFailure ? `INSERT INTO providers (