kimiflare 0.36.0 → 0.36.1

package/dist/index.js

@@ -483,6 +483,9 @@ function isRetryable(err, attempt) {
   return false;
 }
 async function* runKimi(opts2) {
+  if (opts2.cloudMode && !opts2.cloudToken) {
+    throw new KimiApiError("kimiflare: cloud mode requires a cloud token. Run `kimiflare auth cloud` to authenticate.", void 0, 401);
+  }
   const { url, headers: gatewayHeaders } = buildKimiRequestTarget(opts2);
   const body = {
     messages: sanitizeMessagesForApi(opts2.messages),
@@ -4702,7 +4705,16 @@ async function fetchCloudUsage(token) {
     headers: { Authorization: `Bearer ${token}` }
   });
   if (!res.ok) return null;
-  return await res.json();
+  const data = await res.json();
+  if (typeof data.remaining !== "number" || typeof data.input_token_limit !== "number" || typeof data.input_tokens_used !== "number" || typeof data.expires_at !== "string") {
+    return null;
+  }
+  return {
+    input_token_limit: data.input_token_limit,
+    input_tokens_used: data.input_tokens_used,
+    remaining: data.remaining,
+    expires_at: data.expires_at
+  };
 }
 async function loadCloudCredentials() {
   try {
@@ -12977,6 +12989,7 @@ function App({
   const [cfg, setCfg] = useState10(initialCfg);
   const [lspScope, setLspScope] = useState10(initialLspScope);
   const [lspProjectPath, setLspProjectPath] = useState10(initialLspProjectPath);
+  const [cloudToken, setCloudToken] = useState10(initialCloudToken);
   const [events, setRawEvents] = useState10([]);
   const setEvents = useCallback2(
     (updater) => {
@@ -13953,7 +13966,7 @@ function App({
         memoryManager: memoryManagerRef.current,
         codeMode: effectiveCodeMode,
         cloudMode: cfg.cloudMode,
-        cloudToken: initialCloudToken,
+        cloudToken: cloudToken ?? initialCloudToken,
         onIterationEnd,
         onFileChange: (path, content) => {
           if (content) {
@@ -15163,7 +15176,7 @@ ${lines.join("\n")}` }]);
           keepLastImageTurns: cfg.imageHistoryTurns ?? 2,
           codeMode: effectiveCodeMode,
           cloudMode: cfg.cloudMode,
-          cloudToken: initialCloudToken,
+          cloudToken: cloudToken ?? initialCloudToken,
           onIterationEnd,
           intentClassification: classification,
           onFileChange: (path, content2) => {
@@ -15362,6 +15375,7 @@ ${lines.join("\n")}` }]);
             const { loadCloudCredentials: loadCloudCredentials2 } = await Promise.resolve().then(() => (init_auth(), auth_exports));
             const creds = await loadCloudCredentials2();
             if (creds) {
+              setCloudToken(creds.accessToken);
               setEvents((e) => [
                 ...e,
                 { kind: "info", key: mkKey(), text: "configuration saved \u2014 welcome to kimiflare! (cloud mode)" }
@@ -15886,14 +15900,12 @@ program.command("usage").description("Show Kimiflare Cloud token usage (requires
     console.error("Not authenticated with Kimiflare Cloud. Run: kimiflare auth cloud");
     process.exit(1);
   }
-  const res = await fetch("https://api.kimiflare.com/v1/usage", {
-    headers: { Authorization: `Bearer ${creds.accessToken}` }
-  });
-  if (!res.ok) {
-    console.error(`Failed to fetch usage: ${res.status} ${res.statusText}`);
+  const { fetchCloudUsage: fetchCloudUsage2 } = await Promise.resolve().then(() => (init_auth(), auth_exports));
+  const usage = await fetchCloudUsage2(creds.accessToken);
+  if (!usage) {
+    console.error("Failed to fetch usage: invalid response from server");
     process.exit(1);
   }
-  const usage = await res.json();
   console.log(`Token budget: ${usage.remaining.toLocaleString()} / ${usage.input_token_limit.toLocaleString()} remaining`);
   console.log(`Used: ${usage.input_tokens_used.toLocaleString()}`);
   console.log(`Grant expires: ${usage.expires_at}`);
@@ -15931,11 +15943,9 @@ Kimiflare Cloud Authentication`);
         }
       });
       console.log(`Authenticated! Token expires at ${new Date(creds.expiresAt * 1e3).toISOString()}`);
-      const usageRes = await fetch("https://api.kimiflare.com/v1/usage", {
-        headers: { Authorization: `Bearer ${creds.accessToken}` }
-      });
-      if (usageRes.ok) {
-        const usage = await usageRes.json();
+      const { fetchCloudUsage: fetchCloudUsage2 } = await Promise.resolve().then(() => (init_auth(), auth_exports));
+      const usage = await fetchCloudUsage2(creds.accessToken);
+      if (usage) {
         console.log(`
 Token budget: ${usage.remaining.toLocaleString()} / ${usage.input_token_limit.toLocaleString()} remaining`);
         console.log(`Grant expires: ${usage.expires_at}`);