@astrosheep/pi-quota 0.5.5 → 0.5.6

package/README.md

@@ -36,7 +36,7 @@ Add bindings in global `~/.pi/agent/settings.json` under `quotaUsage` (not proje
 }
 ```
 
-Use the **exact Pi provider ID**. Built-in providers and explicitly bound providers are queried; explicit bindings replace a built-in adapter with the same ID. Unknown keys or adapter options are rejected. The API key and model `baseUrl` come from Pi's active provider auth; do not put model API keys in `quotaUsage`.
+Use the **exact Pi provider ID**. Built-in providers and explicitly bound providers are queried; explicit bindings replace a built-in adapter with the same ID. Unknown keys or adapter options are rejected. The API key comes from Pi's provider auth, and the endpoint comes from the provider's registered model; do not put model API keys in `quotaUsage`.
 
 | Adapter | Endpoint | Result |
 |---|---|---|

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@astrosheep/pi-quota",
-  "version": "0.5.5",
+  "version": "0.5.6",
   "publishConfig": {
     "access": "public"
   },

package/src/extension.ts

@@ -85,8 +85,11 @@ export function createQuotaExtension(options: QuotaExtensionOptions = {}) {
         if (typeof value === 'string') headers[name] = value;
       }
       const auth: ProviderAuth = { ...resolved.auth, headers };
-      // A model-level custom endpoint must not be mistaken for an official account.
-      if (ctx.model?.provider === provider && ctx.model.baseUrl) auth.baseUrl = ctx.model.baseUrl;
+      // Provider auth often has no endpoint. /usage also queries providers other
+      // than the selected model, so resolve their configured model URL as well.
+      const model = ctx.model?.provider === provider ? ctx.model
+        : ctx.modelRegistry.getAll().find(model => model.provider === provider);
+      if (model?.baseUrl) auth.baseUrl = model.baseUrl;
       return auth;
     };