@danypops/jittor 0.16.2 → 0.17.0

package/src/providers/telemetry-sources.ts

@@ -1,108 +0,0 @@
-import type { BudgetWindow } from "../policy.ts";
-import type { TelemetryBatch, TelemetrySource } from "../ports/telemetry-source.ts";
-import {
-	type CodexRateLimitSnapshot,
-	CodexSubscriptionTelemetryAdapter,
-	type CodexTransport,
-	type CodexWindow,
-	loadCodexFileCredentials,
-} from "./codex.ts";
-import type { GoogleAdcTokenProvider } from "./google-adc-auth.ts";
-import { GoogleVertexBudgetTelemetryAdapter, type GoogleVertexBudgetTransport } from "./google-vertex-budget.ts";
-import type { GoogleVertexMetricSource } from "./google-vertex-contracts.ts";
-import { OpenRouterTelemetryAdapter, type OpenRouterTransport } from "./openrouter.ts";
-
-function budgetWindow(limit: CodexRateLimitSnapshot, name: "primary" | "secondary", window: CodexWindow | null): BudgetWindow | null {
-	if (!window || window.windowSeconds === null || window.resetsAt === null) return null;
-	return {
-		id: `${limit.limitId}:${name}@${limit.observedAt}`,
-		source: "codex-subscription",
-		scope: limit.limitId === "codex" ? `codex:${name}` : `${limit.limitId}:${name}`,
-		usedFraction: window.usedPercent / 100,
-		windowSeconds: window.windowSeconds,
-		resetsAt: window.resetsAt * 1_000,
-		observedAt: limit.observedAt,
-		freshness: "fresh",
-		confidence: 0.8,
-	};
-}
-
-function windowsFromLimit(limit: CodexRateLimitSnapshot): BudgetWindow[] {
-	return [budgetWindow(limit, "primary", limit.primary), budgetWindow(limit, "secondary", limit.secondary)].filter(
-		(window): window is BudgetWindow => window !== null,
-	);
-}
-
-export class CodexTelemetrySource implements TelemetrySource {
-	readonly id = "codex-subscription";
-	readonly provider = "openai-codex";
-	readonly required = true;
-
-	constructor(
-		private readonly authFile: string,
-		private readonly transport: CodexTransport = fetch,
-		private readonly clock: () => number = Date.now,
-	) {}
-
-	async poll(): Promise<TelemetryBatch> {
-		const observedAt = this.clock();
-		const adapter = new CodexSubscriptionTelemetryAdapter(loadCodexFileCredentials(this.authFile), this.transport);
-		const snapshot = await adapter.readUsage(observedAt);
-		return {
-			observedAt,
-			metrics: snapshot.metrics,
-			windows: [snapshot.defaultLimit, ...snapshot.additionalLimits].flatMap(windowsFromLimit),
-		};
-	}
-}
-
-export class OpenRouterTelemetrySource implements TelemetrySource {
-	readonly id = "openrouter";
-	readonly provider = "openrouter";
-	readonly required = false;
-
-	constructor(
-		private readonly apiKey: string,
-		private readonly transport: OpenRouterTransport = fetch,
-		private readonly clock: () => number = Date.now,
-	) {}
-
-	async poll(): Promise<TelemetryBatch> {
-		const observedAt = this.clock();
-		const snapshot = await new OpenRouterTelemetryAdapter(this.apiKey, this.transport).readKey(observedAt);
-		return { observedAt, metrics: snapshot.metrics, windows: [] };
-	}
-}
-
-/**
- * Optional (never `required`): the one-time GCP setup (Pub/Sub topic + pull subscription
- * connected to the individual project's budget) lives entirely outside Jittor, so a subscription
- * that doesn't exist yet, or a project not yet migrated onto the individual-project model, must
- * not block every other route the way a missing required source would.
- */
-export class GoogleVertexBudgetTelemetrySource implements TelemetrySource {
-	readonly id: string;
-	readonly provider: GoogleVertexMetricSource;
-	readonly required = false;
-
-	private readonly adapter: GoogleVertexBudgetTelemetryAdapter;
-
-	constructor(
-		subscription: string,
-		tokenProvider: GoogleAdcTokenProvider,
-		private readonly clock: () => number = Date.now,
-		transport: GoogleVertexBudgetTransport = fetch,
-		source: GoogleVertexMetricSource = "google-vertex",
-	) {
-		this.id = `google-vertex-budget:${source}`;
-		this.provider = source;
-		this.adapter = new GoogleVertexBudgetTelemetryAdapter(subscription, tokenProvider, transport, source);
-	}
-
-	async poll(): Promise<TelemetryBatch> {
-		const observedAt = this.clock();
-		const snapshot = await this.adapter.pull(observedAt);
-		if (!snapshot) return { observedAt, metrics: [], windows: [] };
-		return { observedAt, metrics: snapshot.metrics, windows: snapshot.window ? [snapshot.window] : [] };
-	}
-}