@prestyj/core 5.28.1 → 5.29.0

package/dist/index.cjs

@@ -69,7 +69,7 @@ __export(index_exports, {
   getContextWindow: () => getContextWindow,
   getDefaultModel: () => getDefaultModel,
   getDefaultThinkingLevel: () => getDefaultThinkingLevel,
-  getFastModel: () => getFastModel,
+  getLowestThinkingLevel: () => getLowestThinkingLevel,
   getMaxThinkingLevel: () => getMaxThinkingLevel,
   getModel: () => getModel,
   getModelsForProvider: () => getModelsForProvider,
@@ -152,6 +152,12 @@ function getAppPaths() {
   };
 }
 
+// src/oauth/request-timeout.ts
+var OAUTH_REQUEST_TIMEOUT_MS = 3e4;
+function oauthRequestSignal() {
+  return AbortSignal.timeout(OAUTH_REQUEST_TIMEOUT_MS);
+}
+
 // src/oauth/kimi.ts
 var CLIENT_ID = "17e5f671-d194-4dfb-9706-5516cb48c098";
 var DEFAULT_OAUTH_HOST = "https://auth.kimi.com";
@@ -239,7 +245,8 @@ async function postForm(endpoint, params) {
       "Content-Type": "application/x-www-form-urlencoded",
       Accept: "application/json"
     },
-    body: new URLSearchParams(params).toString()
+    body: new URLSearchParams(params).toString(),
+    signal: oauthRequestSignal()
   });
   let data = {};
   try {
@@ -713,7 +720,12 @@ async function postTokenRequest(body, label) {
   for (const url of TOKEN_URLS) {
     let response;
     try {
-      response = await fetch(url, { method: "POST", headers, body: encoded });
+      response = await fetch(url, {
+        method: "POST",
+        headers,
+        body: encoded,
+        signal: oauthRequestSignal()
+      });
     } catch (err) {
       lastError = err instanceof Error ? err : new Error(String(err));
       continue;
@@ -962,7 +974,8 @@ async function exchangeOpenAICode(code, verifier) {
       code,
       redirect_uri: REDIRECT_URI2,
       code_verifier: verifier
-    })
+    }),
+    signal: oauthRequestSignal()
   });
   if (!response.ok) {
     const text = await response.text();
@@ -983,7 +996,8 @@ async function refreshOpenAIToken(refreshToken) {
       grant_type: "refresh_token",
       refresh_token: refreshToken,
       client_id: CLIENT_ID3
-    })
+    }),
+    signal: oauthRequestSignal()
   });
   if (!response.ok) {
     const text = await response.text();
@@ -1193,7 +1207,8 @@ async function postTokenRequest2(body) {
   const response = await fetch(TOKEN_URL2, {
     method: "POST",
     headers: { "Content-Type": "application/x-www-form-urlencoded" },
-    body: new URLSearchParams(body)
+    body: new URLSearchParams(body),
+    signal: oauthRequestSignal()
   });
   if (!response.ok) {
     const text = await response.text();
@@ -1322,7 +1337,8 @@ async function codeAssistGet(accessToken, operationName) {
 async function codeAssistRequest(url, accessToken, label, init) {
   const response = await fetch(url, {
     ...init,
-    headers: codeAssistHeaders(accessToken)
+    headers: codeAssistHeaders(accessToken),
+    signal: oauthRequestSignal()
   });
   if (!response.ok) {
     const text = await response.text();
@@ -1408,7 +1424,8 @@ async function postForm2(endpoint, params) {
       Accept: "application/json",
       "User-Agent": `grok-cli/${grokCliVersion()}`
     },
-    body: new URLSearchParams(params).toString()
+    body: new URLSearchParams(params).toString(),
+    signal: oauthRequestSignal()
   });
   let data = {};
   try {
@@ -2430,9 +2447,7 @@ var MODELS = [
     // Sent over our Code Assist (OAuth) transport ahead of gemini-cli — upstream
     // hasn't listed 3.7 yet (google-gemini/gemini-cli#28802, still open) — so
     // free/personal accounts 404 (entitlement-gated) while Code Assist
-    // Standard/Enterprise accounts get it. Kept after the working flash-lite:
-    // getFastModel picks the first low-tier entry, and flash-lite is the one
-    // that works on every account.
+    // Standard/Enterprise accounts get it.
     id: "gemini-3.7-flash",
     name: "Gemini 3.7 Flash",
     provider: "gemini",
@@ -2843,19 +2858,17 @@ function getDefaultThinkingLevel(modelId, options) {
   if (model?.id === "kimi-k3" && isKimiCodingEndpoint(options?.baseUrl)) return "high";
   return model?.defaultThinkingLevel ?? model?.maxThinkingLevel ?? "high";
 }
+var SUMMARY_COST_TIER = {
+  anthropic: "medium",
+  openai: "low",
+  glm: "low",
+  deepseek: "low",
+  huggingface: "low"
+};
 function getSummaryModel(provider, currentModelId) {
-  if (provider === "anthropic") {
-    return MODELS.find((m) => m.id === "claude-sonnet-5-5");
-  }
-  if (provider === "openai" || provider === "glm" || provider === "deepseek" || provider === "huggingface") {
-    const low = getModelsForProvider(provider).find((m) => m.costTier === "low");
-    if (low) return low;
-  }
-  return getModel(currentModelId) ?? getDefaultModel(provider);
-}
-function getFastModel(provider, currentModelId) {
-  const low = getModelsForProvider(provider).find((m) => m.costTier === "low");
-  return low ?? getModel(currentModelId) ?? getDefaultModel(provider);
+  const tier = SUMMARY_COST_TIER[provider];
+  const tagged = tier ? getModelsForProvider(provider).find((m) => m.costTier === tier) : void 0;
+  return tagged ?? getModel(currentModelId) ?? getDefaultModel(provider);
 }
 
 // src/thinking-level.ts
@@ -2945,6 +2958,10 @@ function getSupportedThinkingLevels(provider, model) {
   if (maxIndex === -1) return ["medium"];
   return levels.slice(0, maxIndex + 1);
 }
+function getLowestThinkingLevel(provider, model) {
+  if (getModel(model)?.supportsThinking === false) return void 0;
+  return getSupportedThinkingLevels(provider, model)[0];
+}
 function isThinkingLevelSupported(provider, model, level) {
   return getSupportedThinkingLevels(provider, model).includes(level);
 }
@@ -3492,12 +3509,16 @@ async function fetchSubscriptionUsage(provider, credentials, options = {}) {
 
 // src/telegram.ts
 var TELEGRAM_API = "https://api.telegram.org";
+var LONG_POLL_SECONDS = 30;
+var API_TIMEOUT_MS = 3e4;
 var MAX_MESSAGE_LENGTH = 4096;
 var TelegramBot = class {
   token;
   allowedUserId;
   offset = 0;
   running = false;
+  /** Aborts the in-flight long poll so stop() takes effect immediately. */
+  pollAbort = null;
   onMessage = null;
   onVoiceMessage = null;
   onCallback = null;
@@ -3550,6 +3571,7 @@ var TelegramBot = class {
   /** Stop long polling. */
   stop() {
     this.running = false;
+    this.pollAbort?.abort();
   }
   /** Send a text message to a specific chat. Converts markdown and splits long messages. */
   async send(chatId, text, buttons) {
@@ -3596,12 +3618,26 @@ var TelegramBot = class {
   }
   // ── Private ───────────────────────────────────────────
   async getUpdates() {
-    const result = await this.apiCall("getUpdates", {
-      offset: this.offset,
-      timeout: 30,
-      allowed_updates: ["message", "callback_query", "my_chat_member"]
-    });
-    if (!result.ok || !Array.isArray(result.result)) return [];
+    const poll = new AbortController();
+    this.pollAbort = poll;
+    let result;
+    try {
+      result = await this.apiCall(
+        "getUpdates",
+        {
+          offset: this.offset,
+          timeout: LONG_POLL_SECONDS,
+          allowed_updates: ["message", "callback_query", "my_chat_member"]
+        },
+        // A half-open socket would otherwise hang the poll forever.
+        AbortSignal.any([poll.signal, AbortSignal.timeout((LONG_POLL_SECONDS + 15) * 1e3)])
+      );
+    } finally {
+      if (this.pollAbort === poll) this.pollAbort = null;
+    }
+    if (!result.ok || !Array.isArray(result.result)) {
+      throw new Error(`getUpdates failed: ${JSON.stringify(result)}`);
+    }
     const updates = result.result;
     if (updates.length > 0) {
       this.offset = updates[updates.length - 1].update_id + 1;
@@ -3649,12 +3685,13 @@ var TelegramBot = class {
       }
     }
   }
-  async apiCall(method, body) {
+  async apiCall(method, body, signal = AbortSignal.timeout(API_TIMEOUT_MS)) {
     const url = `${TELEGRAM_API}/bot${this.token}/${method}`;
     const response = await fetch(url, {
       method: "POST",
       headers: { "Content-Type": "application/json" },
-      body: body ? JSON.stringify(body) : void 0
+      body: body ? JSON.stringify(body) : void 0,
+      signal
     });
     if (!response.ok) {
       return { ok: false };
@@ -3787,7 +3824,7 @@ function isModelLoaded() {
   return transcriber !== null;
 }
 async function transcribeVoice(fileUrl) {
-  const response = await fetch(fileUrl);
+  const response = await fetch(fileUrl, { signal: AbortSignal.timeout(6e4) });
   if (!response.ok) throw new Error(`Failed to download voice file: ${response.status}`);
   const buffer = new Uint8Array(await response.arrayBuffer());
   const pcm = await decodeOggOpus(buffer);
@@ -4006,7 +4043,7 @@ function createAutoUpdater(config) {
   getContextWindow,
   getDefaultModel,
   getDefaultThinkingLevel,
-  getFastModel,
+  getLowestThinkingLevel,
   getMaxThinkingLevel,
   getModel,
   getModelsForProvider,