build: token-monitor v0.1.0 — modular LLM API quota visibility

Implements modular provider probing with two distinct header schemas:
- Teams direct (unified schema): 5h/7d utilization floats, status, reset countdown
- Shelley proxy (classic schema): token/request counts + Exedev-Gateway-Cost (USD/call)
- api-ateam: reports no billing data (confirmed non-existent by recon)

Key: uses claude-haiku-4-5-20251001 for minimal probe calls (1 token).
Rate-limit headers present on ALL responses (200 and 429).

113/113 tests passing.

Built from Face recon (trentuna/a-team#91) — live header capture confirmed
unified schema with utilization floats replaces old per-count schema.
This commit is contained in:
Hannibal Smith 2026-04-04 17:01:05 +00:00
parent 760049a25e
commit 07a544c50d
Signed by: hannibal
GPG key ID: 6EB37F7E6190AF1C
10 changed files with 1093 additions and 1 deletions

47
providers/index.js Normal file
View file

@ -0,0 +1,47 @@
/**
* providers/index.js provider registry
*
* Reads ~/.pi/agent/models.json and returns typed provider config for all
* providers we know how to probe. Non-anthropic-messages providers (e.g. zai)
* are silently skipped.
*/
import { readFileSync } from 'fs';
import { homedir } from 'os';
/**
* Classify a provider by name and config.
* @returns {'teams-direct'|'shelley-proxy'|'api-direct'|null}
*/
function classifyProvider(name, config) {
if (name === 'shelley-proxy') return 'shelley-proxy';
if (name === 'api-ateam') return 'api-direct';
if (config.api === 'anthropic-messages' && name.startsWith('team-')) return 'teams-direct';
return null; // skip (zai, etc.)
}
/**
* Load and classify all providers from models.json.
* @returns {Object} map of provider name { name, type, baseUrl, apiKey }
*/
export function getProviders() {
const modelsJson = JSON.parse(
readFileSync(`${homedir()}/.pi/agent/models.json`, 'utf-8')
);
const providers = {};
for (const [name, config] of Object.entries(modelsJson.providers)) {
const type = classifyProvider(name, config);
if (!type) continue;
providers[name] = {
name,
type,
baseUrl: config.baseUrl,
apiKey: config.apiKey || null,
};
}
return providers;
}
// Alias for backwards compatibility
export const loadProviders = getProviders;