@narumitw/pi-usage 0.60.7 → 0.60.8

package/src/types.ts

@@ -8,228 +8,228 @@ export type UsageUnit = "percent" | "usd" | "currency" | "count";
 export type UsageDisplayState = "current" | "configured";
 
 export interface UsageSemantics {
-	kind: UsageSemanticsKind;
-	label: string;
+  kind: UsageSemanticsKind;
+  label: string;
 }
 
 export interface UsageBucket {
-	id: string;
-	label: string;
-	groupId?: string;
-	groupLabel?: string;
-	modelKeys?: string[];
-	used?: number;
-	remaining?: number;
-	limit?: number;
-	unit: UsageUnit;
-	period?: string;
-	windowMinutes?: number;
-	resetsAt?: number;
+  id: string;
+  label: string;
+  groupId?: string;
+  groupLabel?: string;
+  modelKeys?: string[];
+  used?: number;
+  remaining?: number;
+  limit?: number;
+  unit: UsageUnit;
+  period?: string;
+  windowMinutes?: number;
+  resetsAt?: number;
 }
 
 export interface UsageMetric {
-	id: string;
-	label: string;
-	value: number | string;
-	unit?: UsageUnit;
-	currency?: string;
+  id: string;
+  label: string;
+  value: number | string;
+  unit?: UsageUnit;
+  currency?: string;
 }
 
 export interface UsageReport {
-	providerId: string;
-	providerName: string;
-	capturedAt: number;
-	source: string;
-	semantics: UsageSemantics;
-	accountLabel?: string;
-	buckets: UsageBucket[];
-	metrics: UsageMetric[];
-	notes?: string[];
+  providerId: string;
+  providerName: string;
+  capturedAt: number;
+  source: string;
+  semantics: UsageSemantics;
+  accountLabel?: string;
+  buckets: UsageBucket[];
+  metrics: UsageMetric[];
+  notes?: string[];
 }
 
 export interface ResolvedUsageAuth {
-	apiKey?: string;
-	headers: Record<string, string>;
-	fingerprint: string;
-	secrets: string[];
-	model: PiModel;
-	auth?: {
-		apiKey?: string;
-		headers?: Record<string, string | null>;
-		baseUrl?: string;
-	};
-	env?: Record<string, string>;
-	source?: string;
-	effectiveBaseUrl?: string;
+  apiKey?: string;
+  headers: Record<string, string>;
+  fingerprint: string;
+  secrets: string[];
+  model: PiModel;
+  auth?: {
+    apiKey?: string;
+    headers?: Record<string, string | null>;
+    baseUrl?: string;
+  };
+  env?: Record<string, string>;
+  source?: string;
+  effectiveBaseUrl?: string;
 }
 
 export interface UsageQuerySettings {
-	fireworksAccountId?: string;
+  fireworksAccountId?: string;
 }
 
 export type UsageRequestGuard = () => Promise<void>;
 
 export interface UsageProviderTarget {
-	id: string;
-	label: string;
-	description?: string;
+  id: string;
+  label: string;
+  description?: string;
 }
 
 export interface UsageTargetResolver {
-	singularLabel: string;
-	pluralLabel: string;
-	list(
-		auth: ResolvedUsageAuth,
-		signal: AbortSignal,
-		timeoutMs: number,
-		guard: UsageRequestGuard,
-	): Promise<readonly UsageProviderTarget[]>;
+  singularLabel: string;
+  pluralLabel: string;
+  list(
+    auth: ResolvedUsageAuth,
+    signal: AbortSignal,
+    timeoutMs: number,
+    guard: UsageRequestGuard,
+  ): Promise<readonly UsageProviderTarget[]>;
 }
 
 export interface UsageProviderAdapter {
-	id: string;
-	displayName: string;
-	semantics: UsageSemantics;
-	publishesStatusline?: boolean;
-	/** Invalidate the matching ready report when the latest query fails; default preserves it. */
-	invalidateCacheOnFailure?: boolean;
-	targets?: UsageTargetResolver;
-	query(
-		auth: ResolvedUsageAuth,
-		signal: AbortSignal,
-		timeoutMs: number,
-		guard?: UsageRequestGuard,
-		targetId?: string,
-	): Promise<UsageReport>;
+  id: string;
+  displayName: string;
+  semantics: UsageSemantics;
+  publishesStatusline?: boolean;
+  /** Invalidate the matching ready report when the latest query fails; default preserves it. */
+  invalidateCacheOnFailure?: boolean;
+  targets?: UsageTargetResolver;
+  query(
+    auth: ResolvedUsageAuth,
+    signal: AbortSignal,
+    timeoutMs: number,
+    guard?: UsageRequestGuard,
+    targetId?: string,
+  ): Promise<UsageReport>;
 }
 
 export type ProviderUsageState =
-	| {
-			providerId: string;
-			providerName: string;
-			displayState: UsageDisplayState;
-			status: "ready";
-			report: UsageReport;
-	  }
-	| {
-			providerId: string;
-			providerName: string;
-			displayState: UsageDisplayState;
-			status: "selection-required";
-			singularLabel: string;
-			pluralLabel: string;
-			choices: readonly UsageProviderTarget[];
-	  }
-	| {
-			providerId: string;
-			providerName: string;
-			displayState: UsageDisplayState;
-			status: "unsupported" | "auth-unavailable" | "query-failed";
-			message: string;
-	  };
+  | {
+      providerId: string;
+      providerName: string;
+      displayState: UsageDisplayState;
+      status: "ready";
+      report: UsageReport;
+    }
+  | {
+      providerId: string;
+      providerName: string;
+      displayState: UsageDisplayState;
+      status: "selection-required";
+      singularLabel: string;
+      pluralLabel: string;
+      choices: readonly UsageProviderTarget[];
+    }
+  | {
+      providerId: string;
+      providerName: string;
+      displayState: UsageDisplayState;
+      status: "unsupported" | "auth-unavailable" | "query-failed";
+      message: string;
+    };
 
 export type BasetenBillingUsagePayload = {
-	dedicated_usage?: unknown;
-	model_apis_usage?: unknown;
-	training_usage?: unknown;
+  dedicated_usage?: unknown;
+  model_apis_usage?: unknown;
+  training_usage?: unknown;
 };
 
 export type DeepSeekBalancePayload = {
-	is_available?: unknown;
-	balance_infos?: unknown;
+  is_available?: unknown;
+  balance_infos?: unknown;
 };
 
 export type FireworksAccountsPayload = {
-	accounts?: unknown;
-	nextPageToken?: unknown;
+  accounts?: unknown;
+  nextPageToken?: unknown;
 };
 
 export type FireworksBillingSummaryPayload = {
-	lineItems?: unknown;
+  lineItems?: unknown;
 };
 
 export type GitHubCopilotUsagePayload = {
-	login?: unknown;
-	copilot_plan?: unknown;
-	access_type_sku?: unknown;
-	limited_user_quotas?: unknown;
-	limited_user_reset_date?: unknown;
-	monthly_quotas?: unknown;
-	quota_reset_date?: unknown;
-	quota_reset_date_utc?: unknown;
-	quota_snapshots?: unknown;
+  login?: unknown;
+  copilot_plan?: unknown;
+  access_type_sku?: unknown;
+  limited_user_quotas?: unknown;
+  limited_user_reset_date?: unknown;
+  monthly_quotas?: unknown;
+  quota_reset_date?: unknown;
+  quota_reset_date_utc?: unknown;
+  quota_snapshots?: unknown;
 };
 
 export type OpenRouterKeyPayload = {
-	data?: unknown;
+  data?: unknown;
 };
 
 export type MiniMaxUsagePayload = {
-	available_amount?: unknown;
-	balance_alert_switch?: unknown;
-	balance_alert_threshold?: unknown;
-	base_resp?: unknown;
-	cash_balance?: unknown;
-	credit_balance?: unknown;
-	model_remains?: unknown;
-	owed_amount?: unknown;
-	voucher_balance?: unknown;
+  available_amount?: unknown;
+  balance_alert_switch?: unknown;
+  balance_alert_threshold?: unknown;
+  base_resp?: unknown;
+  cash_balance?: unknown;
+  credit_balance?: unknown;
+  model_remains?: unknown;
+  owed_amount?: unknown;
+  voucher_balance?: unknown;
 };
 
 export type MoonshotBalancePayload = {
-	code?: unknown;
-	data?: unknown;
-	scode?: unknown;
-	status?: unknown;
+  code?: unknown;
+  data?: unknown;
+  scode?: unknown;
+  status?: unknown;
 };
 
 export type VercelAIGatewayCreditsPayload = {
-	balance?: unknown;
-	total_used?: unknown;
+  balance?: unknown;
+  total_used?: unknown;
 };
 
 export type OpenCodeZenPayload = {
-	usage?: unknown;
+  usage?: unknown;
 };
 
 export type ZaiQuotaPayload = {
-	error?: unknown;
-	code?: unknown;
-	success?: unknown;
-	data?: unknown;
+  error?: unknown;
+  code?: unknown;
+  success?: unknown;
+  data?: unknown;
 };
 
 export type ZaiSubscriptionPayload = {
-	error?: unknown;
-	code?: unknown;
-	success?: unknown;
-	data?: unknown;
+  error?: unknown;
+  code?: unknown;
+  success?: unknown;
+  data?: unknown;
 };
 
 export interface ZaiPlanInfo {
-	name: string;
-	renewsAt?: string;
+  name: string;
+  renewsAt?: string;
 }
 
 export type KimiCodingUsagePayload = {
-	usage?: unknown;
-	limits?: unknown;
-	boosterWallet?: unknown;
+  usage?: unknown;
+  limits?: unknown;
+  boosterWallet?: unknown;
 };
 
 export type XaiUserPayload = {
-	userId?: unknown;
-	subscriptionTier?: unknown;
+  userId?: unknown;
+  subscriptionTier?: unknown;
 };
 
 export type XaiBillingPayload = {
-	config?: unknown;
+  config?: unknown;
 };
 
 export type CodexBackendPayload = {
-	plan_type?: unknown;
-	rate_limit?: unknown;
-	additional_rate_limits?: unknown;
-	credits?: unknown;
-	rate_limit_reset_credits?: unknown;
+  plan_type?: unknown;
+  rate_limit?: unknown;
+  additional_rate_limits?: unknown;
+  credits?: unknown;
+  rate_limit_reset_credits?: unknown;
 };

package/src/usage-helpers.ts

@@ -4,37 +4,37 @@ import { providerIsConfigured, usageAdapters } from "./query.js";
 import type { PiModel, UsageProviderAdapter } from "./types.js";
 
 export function configuredAdapters(ctx: ExtensionContext): UsageProviderAdapter[] {
-	return usageAdapters().filter(
-		(adapter) => adapter.id === ctx.model?.provider || providerIsConfigured(ctx, adapter.id),
-	);
+  return usageAdapters().filter(
+    (adapter) => adapter.id === ctx.model?.provider || providerIsConfigured(ctx, adapter.id),
+  );
 }
 
 export function providerDisplayName(ctx: ExtensionContext, providerId: string): string {
-	try {
-		return sanitizeDisplayText(ctx.modelRegistry.getProviderDisplayName(providerId), 80);
-	} catch {
-		return sanitizeDisplayText(providerId, 80);
-	}
+  try {
+    return sanitizeDisplayText(ctx.modelRegistry.getProviderDisplayName(providerId), 80);
+  } catch {
+    return sanitizeDisplayText(providerId, 80);
+  }
 }
 
 export function setBoundedMap<T>(map: Map<string, T>, key: string, value: T, limit: number): void {
-	map.delete(key);
-	while (map.size >= limit) {
-		const oldest = map.keys().next().value;
-		if (oldest === undefined) break;
-		map.delete(oldest);
-	}
-	map.set(key, value);
+  map.delete(key);
+  while (map.size >= limit) {
+    const oldest = map.keys().next().value;
+    if (oldest === undefined) break;
+    map.delete(oldest);
+  }
+  map.set(key, value);
 }
 
 export function modelIdentity(model: PiModel | undefined): string | undefined {
-	return model ? `${model.provider}/${model.id}` : undefined;
+  return model ? `${model.provider}/${model.id}` : undefined;
 }
 
 export function isAbortError(error: unknown): boolean {
-	return error instanceof Error && error.name === "AbortError";
+  return error instanceof Error && error.name === "AbortError";
 }
 
 export function isTimeoutError(error: unknown): boolean {
-	return error instanceof Error && error.name === "TimeoutError";
+  return error instanceof Error && error.name === "TimeoutError";
 }