@kenkaiiii/gg-core 5.21.0 → 5.22.0

package/dist/index.cjs

@@ -2322,12 +2322,97 @@ async function fetchCodexUsage(credentials, fetchFn, signal, now) {
     fetchedAt: now()
   };
 }
+function kimiNumber(value) {
+  if (typeof value === "string" && value.trim()) {
+    const parsed = Number(value);
+    return Number.isFinite(parsed) ? parsed : void 0;
+  }
+  return finiteNumber(value);
+}
+function kimiUsagePercent(detail) {
+  const used = kimiNumber(detail?.used);
+  const limit = kimiNumber(detail?.limit);
+  if (used === void 0 || limit === void 0 || limit <= 0) return void 0;
+  return Math.min(100, Math.max(0, used / limit * 100));
+}
+function kimiWindowMinutes(window) {
+  const duration = kimiNumber(window?.duration);
+  if (duration === void 0 || duration <= 0) return void 0;
+  switch (window?.timeUnit) {
+    case "TIME_UNIT_SECOND":
+      return duration / 60;
+    case "TIME_UNIT_MINUTE":
+      return duration;
+    case "TIME_UNIT_HOUR":
+      return duration * 60;
+    case "TIME_UNIT_DAY":
+      return duration * 24 * 60;
+    default:
+      return void 0;
+  }
+}
+async function fetchKimiUsage(credentials, fetchFn, signal, now) {
+  const response = await fetchFn(`${kimiCodeBaseUrl()}/usages`, {
+    method: "GET",
+    signal,
+    headers: {
+      Authorization: "Bearer " + credentials.accessToken,
+      Accept: "application/json",
+      // The managed coding endpoint gates on the kimi-code-cli client identity,
+      // same as model requests.
+      ...kimiCodingHeaders()
+    }
+  });
+  const data = await readUsageResponse(response);
+  const windows = [];
+  const weeklyPercent = kimiUsagePercent(data.usage);
+  if (weeklyPercent !== void 0) {
+    windows.push({
+      kind: "weekly",
+      label: "Weekly",
+      usedPercent: weeklyPercent,
+      resetsAt: isoTimestamp(data.usage?.resetTime)
+    });
+  }
+  for (const limit of data.limits ?? []) {
+    const minutes = kimiWindowMinutes(limit?.window);
+    const percent = kimiUsagePercent(limit?.detail);
+    if (minutes === void 0 || percent === void 0) continue;
+    if (minutes >= 6 * 24 * 60) {
+      if (!windows.some((window) => window.kind === "weekly")) {
+        windows.push({
+          kind: "weekly",
+          label: "Weekly",
+          usedPercent: percent,
+          resetsAt: isoTimestamp(limit?.detail?.resetTime)
+        });
+      }
+      continue;
+    }
+    windows.push({
+      kind: "current",
+      label: `${Math.max(1, Math.round(minutes / 60))}-hour`,
+      usedPercent: percent,
+      resetsAt: isoTimestamp(limit?.detail?.resetTime)
+    });
+  }
+  windows.sort((left, right) => {
+    if (left.kind === right.kind) return 0;
+    return left.kind === "current" ? -1 : 1;
+  });
+  return {
+    provider: "moonshot",
+    displayName: "Kimi",
+    windows,
+    fetchedAt: now()
+  };
+}
 async function fetchSubscriptionUsage(provider, credentials, options = {}) {
   const fetchFn = options.fetchFn ?? fetch;
   const now = options.now ?? Date.now;
   const signal = AbortSignal.timeout(options.timeoutMs ?? 8e3);
   try {
-    return provider === "anthropic" ? await fetchAnthropicUsage(credentials, fetchFn, signal, now) : await fetchCodexUsage(credentials, fetchFn, signal, now);
+    return provider === "anthropic" ? await fetchAnthropicUsage(credentials, fetchFn, signal, now) : provider === "openai" ? await fetchCodexUsage(credentials, fetchFn, signal, now) : await fetchKimiUsage(credentials, fetchFn, signal, now);
   } catch (error) {
     if (error instanceof SubscriptionUsageError) throw error;
     const message = error instanceof Error ? error.message : String(error);