@cortexkit/pi-openai-auth 1.0.2 → 1.0.3

package/README.md

@@ -13,7 +13,7 @@ Requires Pi 1.0.1 or newer.
 Install with Pi's package manager:
 
 ```bash
-pi install npm:@cortexkit/pi-openai-auth@1.0.2
+pi install npm:@cortexkit/pi-openai-auth@1.0.3
 ```
 
 For an unpinned install:

package/dist/index.d.ts

@@ -2183,6 +2183,10 @@ interface ClaustrumEnrollmentPaths {
   tokenPath: string;
 }
 //#endregion
+//#region ../core/dist/pool-routing.d.ts
+/** Existing callbacks may return a promise for an in-progress quota poll. */
+type QuotaReadingRequest = ((id: string) => void) | ((id: string) => Promise<unknown> | undefined);
+//#endregion
 //#region ../core/dist/vault.d.ts
 type VaultHost = 'opencode' | 'pi';
 interface VaultPaths extends ClaustrumEnrollmentPaths {
@@ -2326,7 +2330,7 @@ declare class OpenAiVault {
    * the background; a failure is logged.
    */
   recordSnapshot(routeId: string, snapshot: unknown, complete: boolean, receipt: QuotaReceipt): Promise<void>;
-  /** Takes one quota reading for a vault account through the usage endpoint. */
+  /** Joins an existing quota poll, or starts one through the usage endpoint. */
   pollQuota(routeId: string): Promise<{
     ok: boolean;
     error?: string;
@@ -2336,7 +2340,7 @@ declare class OpenAiVault {
    * because it has no reading yet. At most one reading per account is in
    * flight.
    */
-  requestReading(routeId: string): void;
+  requestReading(routeId: string): ReturnType<QuotaReadingRequest>;
   /** Readings for the routable vault accounts whose last one is older than `maxAgeMs`. */
   pollStale(maxAgeMs: number): Promise<Array<{
     id: string;
@@ -2424,9 +2428,9 @@ declare class PiMainAccount {
   /**
    * Asks for a quota poll, at most once per `MAIN_PULL_RETRY_MS` unless
    * `force`. Admission calls this for every refusal for want of a reading, so
-   * it never waits.
+   * callers may join the returned poll for their first reading.
    */
-  requestReading(force?: boolean): void;
+  requestReading(force?: boolean): Promise<MainPollResult> | undefined;
   /** The poll under way, if any, so a caller may wait a bounded time for it. */
   pending(): Promise<MainPollResult> | undefined;
   /** Polls now (joining a poll already under way) and reports how it ended. */
@@ -2500,6 +2504,7 @@ declare class PiPoolSource {
   private readonly pending;
   private readonly writes;
   private readonly lastPull;
+  private readonly quotaPulls;
   private readonly polled;
   /** The latest full poll snapshot per row id, for the `/openai` quota check. */
   private readonly polledSnapshots;
@@ -2537,7 +2542,6 @@ declare class PiPoolSource {
    * observed whose store write has not landed.
    */
   private withLocalState;
-  private merged;
   private replaceRow;
   /**
    * Polls every row this process has not polled for its current credential:
@@ -2548,9 +2552,11 @@ declare class PiPoolSource {
   /**
    * Asks the store for a quota poll of one row, at most once per
    * `POOL_PULL_RETRY_MS` unless `force`. Admission calls this for every row
-   * it refuses for want of a reading, so it never waits.
+   * it refuses for want of a reading. First-reading admission may join the
+   * returned promise; callers with an existing reading remain nonblocking.
    */
-  requestReading(id: string, force?: boolean): void;
+  requestReading(id: string, force?: boolean): Promise<void> | undefined;
+  private coalescedPull;
   private pull;
   private applyObservation;
   /**