@narumitw/pi-usage 0.60.7 → 0.60.9

package/src/codex-fast.ts

@@ -5,120 +5,110 @@ export const CODEX_FAST_SERVICE_TIER = "priority";
 export const CODEX_STANDARD_SERVICE_TIER = "default";
 
 export const CODEX_FAST_MODEL_IDS: ReadonlySet<string> = new Set([
-	"gpt-5.4",
-	"gpt-5.5",
-	"gpt-5.6-luna",
-	"gpt-5.6-sol",
-	"gpt-5.6-terra",
+  "gpt-5.5",
+  "gpt-5.6-luna",
+  "gpt-5.6-sol",
+  "gpt-5.6-terra",
 ]);
 
 export type CodexFastAvailability =
-	| { kind: "available"; enabled: boolean }
-	| { kind: "not-codex" }
-	| { kind: "unavailable"; reason: string };
+  | { kind: "available"; enabled: boolean }
+  | { kind: "not-codex" }
+  | { kind: "unavailable"; reason: string };
 
-export function codexFastAvailability(
-	model: PiModel | undefined,
-	enabled: boolean,
-): CodexFastAvailability {
-	if (model?.provider !== "openai-codex") return { kind: "not-codex" };
-	if (!isOfficialCodexModel(model)) {
-		return {
-			kind: "unavailable",
-			reason: "Fast mode requires the official OpenAI Codex Responses endpoint.",
-		};
-	}
-	if (!CODEX_FAST_MODEL_IDS.has(model.id)) {
-		return {
-			kind: "unavailable",
-			reason: `${model.id} does not advertise Codex Fast support.`,
-		};
-	}
-	return { kind: "available", enabled };
+export function codexFastAvailability(model: PiModel | undefined, enabled: boolean): CodexFastAvailability {
+  if (model?.provider !== "openai-codex") return { kind: "not-codex" };
+  if (!isOfficialCodexModel(model)) {
+    return {
+      kind: "unavailable",
+      reason: "Fast mode requires the official OpenAI Codex Responses endpoint.",
+    };
+  }
+  if (!CODEX_FAST_MODEL_IDS.has(model.id)) {
+    return {
+      kind: "unavailable",
+      reason: `${model.id} does not advertise Codex Fast support.`,
+    };
+  }
+  return { kind: "available", enabled };
 }
 
 export function codexFastIsEffective(model: PiModel | undefined, enabled: boolean): boolean {
-	return codexFastAvailability(model, enabled).kind === "available" && enabled;
+  return codexFastAvailability(model, enabled).kind === "available" && enabled;
 }
 
 export function codexFastRequestTier(
-	model: PiModel | undefined,
-	enabled: boolean,
+  model: PiModel | undefined,
+  enabled: boolean,
 ): typeof CODEX_FAST_SERVICE_TIER | typeof CODEX_STANDARD_SERVICE_TIER | undefined {
-	if (!isOfficialCodexModel(model)) return undefined;
-	return enabled && CODEX_FAST_MODEL_IDS.has(model.id)
-		? CODEX_FAST_SERVICE_TIER
-		: CODEX_STANDARD_SERVICE_TIER;
+  if (!isOfficialCodexModel(model)) return undefined;
+  return enabled && CODEX_FAST_MODEL_IDS.has(model.id) ? CODEX_FAST_SERVICE_TIER : CODEX_STANDARD_SERVICE_TIER;
 }
 
 export function rewriteCodexFastPayload(
-	payload: unknown,
-	model: PiModel | undefined,
-	enabled: boolean,
+  payload: unknown,
+  model: PiModel | undefined,
+  enabled: boolean,
 ): unknown | undefined {
-	const serviceTier = codexFastRequestTier(model, enabled);
-	if (!serviceTier || !isRecord(payload)) return undefined;
-	return { ...payload, service_tier: serviceTier };
+  const serviceTier = codexFastRequestTier(model, enabled);
+  if (!serviceTier || !isRecord(payload)) return undefined;
+  return { ...payload, service_tier: serviceTier };
 }
 
 export function correctCodexFastMessageCost(
-	message: unknown,
-	model: PiModel | undefined,
-	fastRequested: boolean,
+  message: unknown,
+  model: PiModel | undefined,
+  fastRequested: boolean,
 ): unknown | undefined {
-	if (
-		!codexFastIsEffective(model, fastRequested) ||
-		!isRecord(message) ||
-		message.role !== "assistant" ||
-		message.provider !== model?.provider ||
-		message.model !== model?.id
-	) {
-		return undefined;
-	}
-	const usage = isRecord(message.usage) ? message.usage : undefined;
-	const cost = usage && isRecord(usage.cost) ? usage.cost : undefined;
-	if (!usage || !cost || !hasCompleteUsage(usage) || !isOfficialCodexModel(model)) return undefined;
-	const correctedUsage = structuredClone(usage) as typeof usage;
-	calculateCost(model, correctedUsage as never);
-	const multiplier = model.id === "gpt-5.5" ? 2.5 : 2;
-	const correctedCost = correctedUsage.cost as Record<string, number>;
-	for (const key of ["input", "output", "cacheRead", "cacheWrite", "total"] as const) {
-		correctedCost[key] *= multiplier;
-	}
-	if (costsEqual(cost, correctedCost)) return undefined;
-	return { ...message, usage: correctedUsage };
+  if (
+    !codexFastIsEffective(model, fastRequested) ||
+    !isRecord(message) ||
+    message.role !== "assistant" ||
+    message.provider !== model?.provider ||
+    message.model !== model?.id
+  ) {
+    return undefined;
+  }
+  const usage = isRecord(message.usage) ? message.usage : undefined;
+  const cost = usage && isRecord(usage.cost) ? usage.cost : undefined;
+  if (!usage || !cost || !hasCompleteUsage(usage) || !isOfficialCodexModel(model)) return undefined;
+  const correctedUsage = structuredClone(usage) as typeof usage;
+  calculateCost(model, correctedUsage as never);
+  const multiplier = model.id === "gpt-5.5" ? 2.5 : 2;
+  const correctedCost = correctedUsage.cost as Record<string, number>;
+  for (const key of ["input", "output", "cacheRead", "cacheWrite", "total"] as const) {
+    correctedCost[key] *= multiplier;
+  }
+  if (costsEqual(cost, correctedCost)) return undefined;
+  return { ...message, usage: correctedUsage };
 }
 
 export function codexFastStatusLabel(status: string, enabled: boolean): string {
-	if (!enabled || !/^codex(?:\s|$)/u.test(status)) return status;
-	return status === "codex" ? "codex fast" : `codex fast${status.slice("codex".length)}`;
+  if (!enabled || !/^codex(?:\s|$)/u.test(status)) return status;
+  return status === "codex" ? "codex fast" : `codex fast${status.slice("codex".length)}`;
 }
 
-function isOfficialCodexModel(
-	model: PiModel | undefined,
-): model is PiModel & { api: "openai-codex-responses" } {
-	if (model?.provider !== "openai-codex" || !hasApi(model, "openai-codex-responses")) {
-		return false;
-	}
-	try {
-		return new URL(model.baseUrl).origin === "https://chatgpt.com";
-	} catch {
-		return false;
-	}
+function isOfficialCodexModel(model: PiModel | undefined): model is PiModel & { api: "openai-codex-responses" } {
+  if (model?.provider !== "openai-codex" || !hasApi(model, "openai-codex-responses")) {
+    return false;
+  }
+  try {
+    return new URL(model.baseUrl).origin === "https://chatgpt.com";
+  } catch {
+    return false;
+  }
 }
 
 function isRecord(value: unknown): value is Record<string, unknown> {
-	return typeof value === "object" && value !== null && !Array.isArray(value);
+  return typeof value === "object" && value !== null && !Array.isArray(value);
 }
 
 function hasCompleteUsage(value: Record<string, unknown>): boolean {
-	return ["input", "output", "cacheRead", "cacheWrite"].every(
-		(key) => typeof value[key] === "number" && Number.isFinite(value[key]),
-	);
+  return ["input", "output", "cacheRead", "cacheWrite"].every(
+    (key) => typeof value[key] === "number" && Number.isFinite(value[key]),
+  );
 }
 
 function costsEqual(left: Record<string, unknown>, right: Record<string, number>): boolean {
-	return ["input", "output", "cacheRead", "cacheWrite", "total"].every(
-		(key) => left[key] === right[key],
-	);
+  return ["input", "output", "cacheRead", "cacheWrite", "total"].every((key) => left[key] === right[key]);
 }