quota-keepalive 0.1.1 → 0.1.3

package/README.en.md

@@ -0,0 +1,284 @@
+<div align="center">
+
+# quota-keepalive
+
+Multi-provider quota check-in, real-chat keepalive, run history, and quota trends for Magpie.
+
+[![License](https://img.shields.io/badge/License-MIT-3B82F6?style=for-the-badge)](./LICENSE)
+[![npm](https://img.shields.io/npm/v/quota-keepalive?style=for-the-badge&logo=npm&logoColor=white)](https://www.npmjs.com/package/quota-keepalive)
+[![Magpie](https://img.shields.io/badge/Magpie-Plugin-0EA5A4?style=for-the-badge)](https://github.com/sljdxde/quota-keepalive)
+[![Unofficial](https://img.shields.io/badge/Status-Unofficial-F59E0B?style=for-the-badge)](#unofficial-plugin-notice)
+
+[中文](./README.md) · English
+
+</div>
+
+---
+
+quota-keepalive adds one provider-neutral page to Magpie for inspecting signed-in
+providers, running check-ins, sending configurable real-chat keepalive requests,
+reviewing run history, and watching quota changes.
+
+It does not take over provider sign-in. Magpie and the installed provider plugins
+keep owning credentials, model lists, quotas, and the provider's own `auth.checkin`
+hook. This plugin only coordinates those capabilities and presents the results.
+
+## Unofficial plugin notice
+
+> This project is not an official Magpie plugin.
+>
+> Check-in uses the provider plugin's public `auth.checkin` hook whenever possible.
+> Real-chat keepalive is not a provider-authorized or guaranteed feature. Keepalive
+> requests use the signed provider adapter on your machine and are sent directly to
+> the vendor, with no third-party relay and no token export. That reduces credential
+> exposure, but it does not make vendor-risk zero.
+>
+> Automated synthetic messages may be classified as abnormal traffic and can lead to
+> rate limits, credit freezes, account restrictions, or bans. Keepalive is disabled by
+> default. Enable it only if you understand the risk and accept the consequences.
+
+## Features
+
+- Detect every signed-in Magpie provider, account, model, and quota window.
+- Run check-in through each provider plugin's public `auth.checkin` hook.
+- Respect provider-level and account-level switches; unsupported or signed-out
+  accounts are reported explicitly instead of being silently skipped.
+- Send real-chat keepalive through each provider's signed adapter.
+- Use short probes by default. Long-context probes are opt-in and respect
+  `inputTokens`.
+- Prefer models that recently completed a real keepalive successfully, then fall
+  back by success rate, recency, and known failures. Provider model order is used
+  when there is no history.
+- Search and select a model per provider. If the selected model is unavailable,
+  the runner falls back to another model available to that account.
+- Schedule check-in, keepalive, or both from every 5 minutes to every 7 days.
+- Persist the next scheduled run so an overdue task can catch up after a Magpie
+  restart or computer sleep.
+- Show today's result, a five-day account history, run details, and quota trends.
+- Keep refresh-only events out of the check-in/keepalive run list.
+- Refresh quotas every minute for enabled providers, and support per-provider or
+  refresh-all actions.
+- Keep only hashed account tags, sanitized errors, model names, and status fields in
+  logs. Tokens, full account identifiers, enterprise identifiers, and model replies
+  are never stored.
+
+## Install
+
+Install from npm. This is the recommended path and supports Magpie update checks:
+
+```sh
+magpie plugin add quota-keepalive
+```
+
+Magpie checks installed npm/git plugins shortly after startup and then every hour.
+When a newer version is available, Plugins shows an update badge and an update action.
+
+Update every npm/git plugin:
+
+```sh
+magpie plugin update
+```
+
+Update only this plugin:
+
+```sh
+magpie plugin add quota-keepalive
+```
+
+Re-adding the same npm package replaces the old version while preserving plugin
+options, provider preferences, and enabled/disabled state. Updating also unpins an
+older `quota-keepalive@x.y.z` spec.
+
+Install from GitHub without npm:
+
+```sh
+magpie plugin add github:sljdxde/quota-keepalive
+```
+
+Install from a local folder only for development:
+
+```sh
+magpie plugin add "$(pwd)"
+```
+
+A folder plugin loads the working tree directly and is not included in Magpie's
+version checks or `magpie plugin update`. Switch back to npm before shipping.
+
+After installation, open **Plugins → Quota Keepalive** in Magpie. Reopen the page
+after a UI-only update; restarting all of Magpie is not required.
+
+## Quick start
+
+1. Open the **Quota Keepalive** page.
+2. Click **Detect** to load signed-in providers, accounts, models, and quotas.
+3. Keep provider and account automatic switches in the desired state.
+4. Use **Check in all** or **Send keepalive** for one manual run.
+5. Configure **Schedule** if you want automatic execution.
+6. Review **Recent runs**, account history, and quota changes.
+
+Keepalive sends a real provider request and may consume credits. The plugin does not
+enable automatic keepalive by default.
+
+## Page controls
+
+- **Detect**: refresh the provider/account/model/quota snapshot.
+- **Check in all**: run check-in for every eligible provider.
+- **Send keepalive**: confirm and send one keepalive request to every eligible account.
+- Provider switch: enable or disable all automatic tasks for that provider.
+- Account **Check-in** / **Keepalive** switches: control automatic participation only;
+  they do not send a request immediately.
+- Account **Run check-in** / **Run keepalive** buttons: execute one account now.
+  Manual keepalive asks for confirmation and still requires the provider to be enabled.
+- Circular refresh: refresh all enabled providers; each provider row also has a
+  provider-only refresh button.
+- **Save schedule**: set the shared interval and action type.
+- **Short probe / Long context**: choose the request shape used by keepalive.
+
+Saving a schedule starts the next run after the configured interval. The next run
+time is persisted, and a task already due when Magpie starts is run immediately.
+
+## Scheduling and failures
+
+Magpie's native daily check-in scheduler still runs independently. This plugin adds
+an optional shared scheduler for more frequent check-in or keepalive.
+
+- Default: disabled.
+- Interval: 5 minutes to 7 days.
+- Actions: `checkin`, `keepalive`, or `all`.
+- A successful check-in or keepalive for the same account and Beijing date is
+  idempotent and will not be sent again.
+- Keepalive retries are bounded by `maxAttempts`.
+- HTTP 402/429 credit exhaustion stops automatic retry for that day.
+- Manual actions can deliberately retry a failed or quota-exhausted account.
+- Network errors, expired sign-in, unavailable models, empty replies, and provider
+  business errors are recorded with sanitized details.
+
+## Model availability
+
+Keepalive does not blindly choose the cheapest or a free model:
+
+1. Intersect the provider's model list with the account's allowed models.
+2. Prefer a model that recently completed a real keepalive successfully.
+3. If the account has no history, prefer a model another account of the same
+   provider used successfully.
+4. If there is still no history, follow the provider's model order.
+5. Put known failures after unverified candidates.
+6. Fall back to the next candidate on a missing model, credit rejection, upstream
+   outage, or empty reply.
+
+For WorkBuddy CN, the keepalive path follows dsh probe semantics: an HTTP 2xx means
+the upstream accepted and recorded a real chat, even when a one-token response only
+contains `[DONE]`, usage data, or an empty visible body. JSON business errors, HTML
+login pages, and timeouts are failures.
+
+## Configuration
+
+The page writes plugin-level options. The same values can be changed from the CLI:
+
+```sh
+magpie plugin options quota-keepalive '{
+  "scheduleEnabled": true,
+  "scheduleMinutes": 60,
+  "scheduleAction": "all",
+  "modelPreferences": {
+    "workbuddy": "deepseek-v4.1-flash",
+    "codearts": "deepseek-v4.1-flash"
+  },
+  "modelCandidates": 4,
+  "probeMode": "short",
+  "probeMessage": "Magpie daily connectivity check. Reply with OK.",
+  "inputTokens": 25000,
+  "maxAttempts": 3,
+  "timeoutMs": 90000,
+  "logMaxBytes": 2097152
+}'
+```
+
+- `scheduleEnabled`: enable the plugin scheduler. Default `false`.
+- `scheduleMinutes`: shared interval, default `60`, range 5 minutes to 7 days.
+- `scheduleAction`: `checkin`, `keepalive`, or `all`. Default `checkin`.
+- `model`: optional preferred model; it is not a hard allowlist.
+- `modelPreferences`: preferred model per provider.
+- `modelCandidates`: maximum fallback candidates per account, default `4`.
+- `probeMode`: `short` by default; `context` sends the configured long input.
+- `probeMessage`: short probe text, up to 500 characters.
+- `inputTokens`: long-context input size, only used with `probeMode=context`.
+- `maxAttempts`: maximum keepalive attempts per account per Beijing day.
+- `timeoutMs`: overall timeout for one model request, default 90 seconds.
+- `logMaxBytes`: rotate one previous log after this size, default 2 MiB.
+- `statePath`, `logPath`, `snapshotPath`, `commandPath`, `schedulePath`: optional
+  absolute paths.
+
+## Logs
+
+The default log is:
+
+```text
+~/.config/magpie/workbuddy-daily.log.jsonl
+```
+
+Read it with the bundled command:
+
+```sh
+node ./bin/daily-keepalive-log.js --limit 50
+node ./bin/daily-keepalive-log.js --day 2026-10-30 --status failed
+node ./bin/daily-keepalive-log.js --json
+```
+
+Agents can use the `quota-keepalive_logs` tool. Logs contain stable hashes of account
+identifiers, provider IDs, status, model names, and sanitized errors. They never
+contain tokens, full account identifiers, enterprise identifiers, or model replies.
+
+## Security
+
+- State and log files are created with mode `0600`.
+- Writes use temporary files and atomic rename.
+- `plugin-auth.json` updates use a short lock, file fingerprint checks, and conflict
+  retries. The plugin re-reads and merges when Magpie or another provider plugin
+  changed the file after it was read.
+- Model replies are not written to state or logs.
+- Tokens, Bearer values, API keys, and account identifiers are redacted from page
+  output and logs.
+- Missing provider plugins, signed-out accounts, and model-less providers produce
+  explicit skip or failure results.
+
+## Development
+
+```sh
+bun test
+bun test --coverage
+npm pack --dry-run
+```
+
+Tests use mocked requests. They do not contact vendors or consume credits. The core
+JavaScript modules currently have 100% line and function coverage, including
+performance-boundary cases.
+
+## Publishing
+
+The repository uses npm Trusted Publishing through
+`.github/workflows/publish.yml`; no npm token is stored in the repository or in
+GitHub Secrets.
+
+1. Update the version in `package.json` and commit the changes.
+2. Create a matching tag, for example `v0.1.2`.
+3. Push `main` and the tag.
+4. Publish a GitHub Release for that tag.
+5. The workflow runs tests, syntax checks, `npm pack`, and publishes through OIDC.
+6. Installed users see the update through Magpie's hourly plugin update check.
+
+Manual npm publishing with two-factor authentication enabled requires the current
+one-time code:
+
+```sh
+npm publish --access public --otp=<current-one-time-code>
+```
+
+Never put an OTP, npm token, or recovery code in a repository, README, or chat.
+
+## Links
+
+- [GitHub](https://github.com/sljdxde/quota-keepalive)
+- [npm](https://www.npmjs.com/package/quota-keepalive)
+- [Releases](https://github.com/sljdxde/quota-keepalive/releases)
+- [License](./LICENSE)

package/README.md

@@ -1,8 +1,32 @@
+<div align="center">
+
 # quota-keepalive
 
 面向 Magpie 全部已登录供应商的额度签到、真实对话保活、执行留痕和额度趋势页面。
-插件不接管任何供应商的登录，也不会把页面伪装成单一供应商专属功能；登录、模型、
-额度和签到 Hook 仍由各供应商插件维护，本插件负责统一检测、调度、执行和展示。
+
+[![License](https://img.shields.io/badge/License-MIT-3B82F6?style=for-the-badge)](./LICENSE)
+[![npm](https://img.shields.io/npm/v/quota-keepalive?style=for-the-badge&logo=npm&logoColor=white)](https://www.npmjs.com/package/quota-keepalive)
+[![Magpie](https://img.shields.io/badge/Magpie-Plugin-0EA5A4?style=for-the-badge)](https://github.com/sljdxde/quota-keepalive)
+[![Unofficial](https://img.shields.io/badge/Status-Unofficial-F59E0B?style=for-the-badge)](#非官方插件声明)
+
+中文 · [English](./README.en.md)
+
+</div>
+
+---
+
+quota-keepalive 不接管任何供应商的登录，也不会把页面伪装成单一供应商专属功能。
+登录、模型、额度和签到 Hook 仍由各供应商插件维护；本插件负责统一检测、调度、
+执行和展示。
+
+> **非官方插件声明**
+>
+> 本项目不是 Magpie 官方插件。签到优先调用供应商插件公开的签到接口；真实对话
+> 保活不是供应商授权或保证支持的功能。保活请求通过本机已登录的供应商签名适配器
+> 直接发往供应商，不经过第三方服务器，也不会导出令牌，因此降低了凭据泄露风险，
+> 但不等于厂商风控风险为零。自动化合成消息仍可能被供应商识别为异常流量，并触发
+> 限流、额度冻结、账号限制或封禁。默认关闭保活，仅在理解风险后按需启用，并由
+> 使用者自行承担相关后果。
 
 ## 功能
 
@@ -10,6 +34,7 @@
 - 一键签到：通过各供应商登录插件公开的 `auth.checkin` 接口执行，尊重供应商、
   账号两级启停；页面会明确显示未适配或未登录的账号，不会静默跳过。
 - 真实对话保活：通过各供应商自己的登录插件取得签名请求，发送一条真实消息。
+  默认使用“短探活”，只发送一条短消息；需要长上下文时才显式切换“长上下文”模式。
 - 模型选择：只从该账号实际列出的模型中选择。默认优先近期真实保活成功过的模型，
   再按成功率、成功时间和失败次数排序；没有历史记录时沿用供应商自己的模型顺序，
   不再按免费或价格抢先选择。第一个模型不可用时自动换下一档，全部不可用才记失败。
@@ -39,18 +64,50 @@
 
 ## 安装
 
-从 GitHub 安装：
+npm 安装（推荐，支持 Magpie 自动检查更新）：
+
+```sh
+magpie plugin add quota-keepalive
+```
+
+安装后的插件规格会显示为 `quota-keepalive@latest`。Magpie 会在启动后和之后每小时
+检查一次更新，并在插件页面提示新版本。更新方式：
+
+```sh
+# 更新全部 npm/git 插件
+magpie plugin update
+
+# 只更新这个插件
+magpie plugin add quota-keepalive
+```
+
+页面中的 `Update all` 与 `magpie plugin update` 等价。重新添加同一个 npm 包会替换
+旧版本，并保留插件 options、启用/停用状态和供应商偏好。已固定版本
+`quota-keepalive@0.1.x` 也会在更新时升级到最新版并解除固定。
+
+从 GitHub 安装（不经过 npm）：
 
 ```sh
 magpie plugin add github:sljdxde/quota-keepalive
 ```
 
-从本地目录安装：
+从本地目录安装（仅用于开发调试）：
 
 ```sh
 magpie plugin add "$(pwd)"
 ```
 
+目录安装会直接读取当前源码，适合修改后立即调试，但不会进入 npm 的版本检查，也不会
+被 `magpie plugin update` 更新。开发完成后应切换到 npm 安装。切换前先保存 npm
+条目中的 options，再停用目录条目：
+
+```sh
+magpie plugin add quota-keepalive
+magpie plugin options quota-keepalive '<原来的 options JSON>'
+magpie plugin off '<目录插件的 spec>'
+magpie plugin rm '<目录插件的 spec>'
+```
+
 插件不需要接管 provider。供应商登录仍由 Magpie 已安装的对应社区插件提供；本插件
 直接从 Magpie 的插件目录加载这些签名适配器，因此 Magpie 或供应商插件更新后，
 只要其公开接口没有变化，本插件仍可继续使用。若供应商包改名为不兼容的接口，页面
@@ -59,12 +116,28 @@ magpie plugin add "$(pwd)"
 安装后在 Magpie 左侧 `Plugins` 分组打开“额度签到”。如果页面文件刚更新，
 重新打开一次插件页即可刷新清单，不需要重启 Magpie。
 
-发布到 npm 后也可以按包名安装：
+### 发布新版本
+
+仓库包含 `.github/workflows/publish.yml`，使用 npm Trusted Publishing，不需要把
+npm token 写进仓库或 GitHub Secrets：
+
+1. 更新 `package.json` 版本并提交改动。
+2. 创建同名 Git tag，例如版本 `0.1.2` 对应 `v0.1.2`。
+3. 推送 `main` 和 tag。
+4. 在 GitHub 创建并发布对应的 Release。
+5. Release 发布后，工作流会重新跑测试、语法检查和 `npm pack`，再通过 OIDC
+   发布到 npm。
+6. 本机等待 Magpie 的更新提示，或执行 `magpie plugin add quota-keepalive` /
+   `magpie plugin update`。
+
+本地手动发布时，npm 如果启用了双因素认证，需要附加当前动态码：
 
 ```sh
-magpie plugin add quota-keepalive
+npm publish --access public --otp=<6 位动态码>
 ```
 
+不要把动态码、npm token 或恢复码写进仓库、README 或聊天记录。
+
 ## 页面操作
 
 - `一键检测`：刷新供应商、账号、模型和额度快照。
@@ -76,7 +149,10 @@ magpie plugin add quota-keepalive
   不会因为快速连续点击触发“每秒只能写入一次”的报错。
 - 圆形刷新按钮：立即刷新全部启用供应商额度；供应商卡片内的刷新按钮只刷新该供应商。
 - `保存计划`：设置统一执行间隔和任务类型。计划默认关闭，防止无意间消耗积分。
-- 定时计划保存或修改后，从下一个间隔开始执行；需要立即执行时使用页面顶部的手动按钮。
+- `短探活/长上下文`：短探活风险、耗时和额度消耗更低；长上下文会按 `inputTokens`
+  填充请求，适合明确需要维持长上下文窗口的场景。
+- 定时计划保存或修改后，从下一个间隔开始执行；下次执行时间会持久化，Magpie 重启或
+  电脑休眠后会补跑已经到期的任务。需要立即执行时使用页面顶部的手动按钮。
 
 页面每 5 秒读取一次本地快照；插件后台每 1 分钟刷新启用中供应商的额度，停用项会
 跳过。额度刷新仍写入后台诊断日志，但不会出现在页面的签到/保活执行记录中。
@@ -120,6 +196,8 @@ magpie plugin options quota-keepalive '{
     "codearts": "deepseek-v4.1-flash"
   },
   "modelCandidates": 4,
+  "probeMode": "short",
+  "probeMessage": "Magpie daily connectivity check. Reply with OK.",
   "inputTokens": 25000,
   "maxAttempts": 3,
   "timeoutMs": 90000,
@@ -136,21 +214,25 @@ magpie plugin options quota-keepalive '{
   WorkBuddy CN 默认优先 `deepseek-v4.1-flash`，这是 dsh 实测较稳定的真实对话模型；
   其他供应商按近 5 天可用性排序，没有成功记录时按供应商模型顺序选择，不按免费或倍率抢先选择。
 - `modelCandidates`：每个账号每次最多尝试的模型数，默认 4，范围 1–12。
-- `inputTokens`：真实请求的输入规模，默认 25000，范围 1000–25000。
+- `probeMode`：`short`（默认）只发一条短消息；`context` 才使用长上下文填充。
+- `probeMessage`：短探活消息，最多 500 字符；留空时使用默认连通性检查文本。
+- `inputTokens`：仅 `probeMode=context` 时生效，默认 25000，范围 1000–25000。
 - `maxAttempts`：每个账号每个北京时间日最多保活尝试轮数，默认 3。
 - `timeoutMs`：单个模型请求的整体截止时间，默认 90 秒，范围 10–180 秒。WorkBuddy
   CN 的 25k 输入请求可能先返回响应头、再继续流式读取；90 秒比 25 秒更接近 dsh 的
   probe 容忍度，同时仍避免任务无限挂起。即使供应商
   登录插件内部忽略中止信号，外层也会结束该候选并继续回退；账号较多时不要调得过大。
 - `logMaxBytes`：单个日志上限，默认 2 MiB，超出后轮转一份 `.1`。
-- `statePath`、`logPath`、`snapshotPath`、`commandPath`：可选绝对路径。
+- `statePath`、`logPath`、`snapshotPath`、`commandPath`、`schedulePath`：可选绝对路径。
 
 ## 执行周期与失败
 
 Magpie 自己的每日签到调度仍会在启动后约 2 分钟进行首轮，之后约每 30 分钟巡检；
 失败约 30 分钟后重试。本插件的定时任务是额外的统一调度，默认关闭，适合需要更
 频繁签到或保活的场景。本插件的定时任务不会在保存后立即执行，也不会在修改后
-重启 Magpie；修改配置会替换运行中的定时器，并从新的间隔重新计时。
+重启 Magpie；修改配置会替换运行中的定时器，并从新的间隔重新计时。调度状态保存在
+`quota-keepalive-schedule.json`，Magpie 重启后会读取上次执行时间和下次执行时间，
+已到期的任务会补跑，不会因为反复重启而一直错过执行。
 
 保活失败会按 `maxAttempts` 重试；达到上限后当天不再自动发送。网络失败、登录过期、
 模型不可用和空回复都会进入日志。遇到明确的 HTTP 402/429 额度耗尽或免费额度用尽时，
@@ -180,7 +262,8 @@ node ./bin/daily-keepalive-log.js --json
 ## 安全
 
 - 状态和日志文件以 `0600` 创建，写入使用临时文件和原子 rename。
-- `plugin-auth.json` 的更新使用短锁，避免与 Magpie 或另一个任务并发写坏登录。
+- `plugin-auth.json` 的更新使用短锁、文件指纹校验和冲突重试。若 Magpie 或其他
+  供应商插件在读取后改动了凭据文件，本插件会重新读取并合并，不会直接覆盖新内容。
 - 模型回复不会写入状态或日志。
 - 日志和页面显示会对令牌、Bearer、API key 和账号标识做脱敏。
 - 未安装对应供应商插件、未登录或没有可用模型时，会明确记为跳过或失败。

package/index.js

@@ -11,6 +11,12 @@ import { snapshotWarning, writeDailySnapshot } from "./src/snapshot.js"
 import { createProviderCheckin } from "./src/checkin-runner.js"
 import { createWorkBuddyKeepalive } from "./src/workbuddy.js"
 import { readCommand } from "./src/commands.js"
+import {
+  nextRunAfter,
+  readSchedule,
+  scheduleSignature,
+  writeSchedule,
+} from "./src/schedule.js"
 
 const LOG_STATUSES = new Set(["started", "success", "failed", "skipped"])
 const SCHEDULER = Symbol.for("quota-keepalive.scheduler")
@@ -20,32 +26,85 @@ const REFRESH_INTERVAL_MS = 60_000
 
 export function stopScheduler() {
   const previous = globalThis[SCHEDULER]
-  if (previous?.timer) clearInterval(previous.timer)
+  if (previous?.timer) clearTimeout(previous.timer)
+  if (previous) previous.stopped = true
   if (previous) delete globalThis[SCHEDULER]
 }
 
-export function startScheduler({ config, checkin, keepalive }) {
+function warnSchedule(action, error) {
+  console.warn(`[quota-keepalive] ${action}: ${error?.message || error}`)
+}
+
+export async function startScheduler({
+  config,
+  checkin,
+  keepalive,
+  now = () => Date.now(),
+  read = readSchedule,
+  persist = writeSchedule,
+  run = runCommand,
+  log = appendDailyLog,
+  snapshot = writeDailySnapshot,
+}) {
   // Every behavior setting must participate in the signature. Otherwise a
   // model or provider/account preference change would leave the running timer
   // on the previous closures and configuration.
-  const signature = JSON.stringify(config)
+  const signature = scheduleSignature(config)
   const previous = globalThis[SCHEDULER]
   if (previous?.signature === signature) return
   stopScheduler()
   if (!config.scheduleEnabled) return
-  let running = false
-  const timer = setInterval(async () => {
-    if (running) return
-    running = true
+  const controller = {
+    signature,
+    timer: null,
+    running: false,
+    stopped: false,
+    nextRunAt: 0,
+  }
+  globalThis[SCHEDULER] = controller
+
+  const intervalMs = config.scheduleMinutes * 60_000
+  let saved
+  try {
+    saved = await read(config.schedulePath)
+  } catch (error) {
+    warnSchedule("cannot read the persisted schedule", error)
+    saved = {}
+  }
+  controller.nextRunAt = nextRunAfter({
+    now: now(),
+    lastRunAt: saved.lastRunAt,
+    nextRunAt: saved.nextRunAt,
+    intervalMs,
+    storedSignature: saved.signature,
+    signature,
+  })
+  await persist(config.schedulePath, {
+    ...saved,
+    signature,
+    nextRunAt: new Date(controller.nextRunAt).toISOString(),
+  }).catch((error) => warnSchedule("cannot save the schedule", error))
+
+  const arm = () => {
+    if (controller.stopped) return
+    const delay = Math.max(0, controller.nextRunAt - now())
+    controller.timer = setTimeout(tick, delay)
+    controller.timer?.unref?.()
+  }
+
+  async function tick() {
+    if (controller.stopped || controller.running) return
+    controller.running = true
+    let result = null
     try {
-      const result = await runCommand({
+      result = await run({
         action: config.scheduleAction,
         config,
         checkin,
         keepalive,
         force: false,
       })
-      await appendDailyLog({
+      await log({
         logPath: config.logPath,
         logMaxBytes: config.logMaxBytes,
         entry: {
@@ -58,15 +117,15 @@ export function startScheduler({ config, checkin, keepalive }) {
           outcome: result.status,
           message: result.summary,
         },
-      })
-      await writeDailySnapshot({
+      }).catch((error) => warnSchedule("cannot write the schedule log", error))
+      await snapshot({
         ...(config.configHome ? { configHome: config.configHome } : {}),
         logPath: config.logPath,
         snapshotPath: config.snapshotPath,
         commandPath: config.commandPath,
-      })
+      }).catch((error) => warnSchedule(snapshotWarning(error), error))
     } catch (error) {
-      await appendDailyLog({
+      await log({
         logPath: config.logPath,
         logMaxBytes: config.logMaxBytes,
         entry: {
@@ -79,13 +138,31 @@ export function startScheduler({ config, checkin, keepalive }) {
           outcome: "failed",
           error: error instanceof Error ? error.message : String(error),
         },
-      })
+      }).catch((logError) => warnSchedule("cannot write the failure log", logError))
     } finally {
-      running = false
+      controller.running = false
+      const finishedAt = now()
+      controller.nextRunAt = nextRunAfter({
+        now: finishedAt,
+        lastRunAt: new Date(finishedAt).toISOString(),
+        intervalMs,
+        storedSignature: signature,
+        signature,
+      })
+      await persist(config.schedulePath, {
+        ...saved,
+        signature,
+        lastRunAt: new Date(finishedAt).toISOString(),
+        nextRunAt: new Date(controller.nextRunAt).toISOString(),
+        lastStatus: result?.status || "failed",
+        lastSummary: result?.summary || "",
+      }).catch((error) => warnSchedule("cannot save the completed schedule", error))
+      arm()
     }
-  }, config.scheduleMinutes * 60_000)
-  timer.unref?.()
-  globalThis[SCHEDULER] = { timer, signature }
+  }
+
+  arm()
+  return controller
 }
 
 export function stopOptionsWatcher() {
@@ -205,7 +282,7 @@ export function startOptionsWatcher({
       const liveConfig = resolveKeepaliveOptions(liveRaw)
       const checkin = makeCheckin({ input, config: liveConfig })
       const keepalive = makeKeepalive({ input, config: liveConfig })
-      start({ config: liveConfig, checkin, keepalive })
+      await start({ config: liveConfig, checkin, keepalive })
       startRefresh({ config: liveConfig })
       await consume({
         config: liveConfig,
@@ -303,7 +380,7 @@ export async function QuotaKeepalivePlugin(input, rawOptions = {}) {
   const config = resolveKeepaliveOptions(rawOptions)
   const checkin = createProviderCheckin({ input, config })
   const keepalive = createWorkBuddyKeepalive({ input, config })
-  startScheduler({ config, checkin, keepalive })
+  await startScheduler({ config, checkin, keepalive })
   startQuotaRefresher({ config })
   startOptionsWatcher({ input, rawOptions, config })
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "quota-keepalive",
-  "version": "0.1.1",
-  "description": "Multi-provider quota inventory, check-in, keepalive scheduling, history, and quota trends for Magpie",
+  "version": "0.1.3",
+  "description": "Unofficial multi-provider quota inventory, check-in, keepalive scheduling, history, and quota trends for Magpie",
   "keywords": [
     "magpie",
     "magpie-plugin",
@@ -25,7 +25,7 @@
   "main": "./index.js",
   "exports": "./index.js",
   "bin": {
-    "daily-keepalive-log": "./bin/daily-keepalive-log.js"
+    "daily-keepalive-log": "bin/daily-keepalive-log.js"
   },
   "magpie": {
     "icon": "data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHZpZXdCb3g9IjAgMCA2NCA2NCI+CiAgPGRlZnM+CiAgICA8bGluZWFyR3JhZGllbnQgaWQ9ImJnIiB4MT0iOCIgeTE9IjQiIHgyPSI1NiIgeTI9IjYwIiBncmFkaWVudFVuaXRzPSJ1c2VyU3BhY2VPblVzZSI+CiAgICAgIDxzdG9wIHN0b3AtY29sb3I9IiMwQjIwMjkiLz4KICAgICAgPHN0b3Agb2Zmc2V0PSIxIiBzdG9wLWNvbG9yPSIjMTczQzQ2Ii8+CiAgICA8L2xpbmVhckdyYWRpZW50PgogICAgPGxpbmVhckdyYWRpZW50IGlkPSJnb2xkIiB4MT0iMzgiIHkxPSIzOCIgeDI9IjU2IiB5Mj0iNTYiIGdyYWRpZW50VW5pdHM9InVzZXJTcGFjZU9uVXNlIj4KICAgICAgPHN0b3Agc3RvcC1jb2xvcj0iI0ZGRDI3QSIvPgogICAgICA8c3RvcCBvZmZzZXQ9IjEiIHN0b3AtY29sb3I9IiNFNzlCMzgiLz4KICAgIDwvbGluZWFyR3JhZGllbnQ+CiAgPC9kZWZzPgogIDxyZWN0IHg9IjMiIHk9IjMiIHdpZHRoPSI1OCIgaGVpZ2h0PSI1OCIgcng9IjE4IiBmaWxsPSJ1cmwoI2JnKSIvPgogIDxjaXJjbGUgY3g9IjMwLjUiIGN5PSIzMSIgcj0iMTcuMiIgZmlsbD0ibm9uZSIgc3Ryb2tlPSIjRjhGMURFIiBzdHJva2Utd2lkdGg9IjcuNCIKICAgICAgICAgIHN0cm9rZS1saW5lY2FwPSJyb3VuZCIgc3Ryb2tlLWRhc2hhcnJheT0iODggMjEiIHRyYW5zZm9ybT0icm90YXRlKDM4IDMwLjUgMzEpIi8+CiAgPHBhdGggZD0iTTM0LjUgMzYuNWM1LjMgNC40IDEwLjIgOS4yIDE5LjUgMTguMyIKICAgICAgICBmaWxsPSJub25lIiBzdHJva2U9InVybCgjZ29sZCkiIHN0cm9rZS13aWR0aD0iNi4yIiBzdHJva2UtbGluZWNhcD0icm91bmQiLz4KPC9zdmc+Cg==",

package/src/config.js

@@ -7,16 +7,20 @@ import {
   MIN_LOG_MAX_BYTES,
   defaultLogPath,
 } from "./log.js"
+import { DEFAULT_PROBE_MESSAGE } from "./keepalive.js"
 import {
   normalizeAccountPreferences,
   normalizeProviderPreferences,
 } from "./preferences.js"
+import { defaultSchedulePath } from "./schedule.js"
 import { defaultSnapshotPath } from "./snapshot.js"
 
 export const DEFAULT_KEEPALIVE = Object.freeze({
   model: "",
   modelCandidates: 4,
   inputTokens: 25_000,
+  probeMode: "short",
+  probeMessage: DEFAULT_PROBE_MESSAGE,
   maxAttempts: 3,
   scheduleEnabled: false,
   scheduleAction: "checkin",
@@ -54,6 +58,9 @@ function absolutePath(value, fallback) {
 export function resolveKeepaliveOptions(raw) {
   const source = raw && typeof raw === "object" && !Array.isArray(raw) ? raw : {}
   const model = typeof source.model === "string" ? source.model.trim().slice(0, 200) : ""
+  const probeMessage = typeof source.probeMessage === "string"
+    ? source.probeMessage.trim().slice(0, 500)
+    : ""
   const scheduleAction = ["checkin", "keepalive", "all"].includes(source.scheduleAction)
     ? source.scheduleAction
     : DEFAULT_KEEPALIVE.scheduleAction
@@ -68,6 +75,8 @@ export function resolveKeepaliveOptions(raw) {
     scheduleMinutes: integerIn(source.scheduleMinutes, DEFAULT_KEEPALIVE.scheduleMinutes, 5, 10_080),
     configHome: absolutePath(source.configHome, undefined),
     inputTokens: integerIn(source.inputTokens, DEFAULT_KEEPALIVE.inputTokens, 1_000, 25_000),
+    probeMode: source.probeMode === "context" ? "context" : DEFAULT_KEEPALIVE.probeMode,
+    probeMessage: probeMessage || DEFAULT_PROBE_MESSAGE,
     maxAttempts: integerIn(source.maxAttempts, DEFAULT_KEEPALIVE.maxAttempts, 1, 10),
     timeoutMs: integerIn(source.timeoutMs, DEFAULT_KEEPALIVE.timeoutMs, 10_000, 180_000),
     lockWaitMs: integerIn(source.lockWaitMs, DEFAULT_KEEPALIVE.lockWaitMs, 0, 10_000),
@@ -82,5 +91,6 @@ export function resolveKeepaliveOptions(raw) {
       MAX_LOG_MAX_BYTES,
     ),
     snapshotPath: absolutePath(source.snapshotPath, defaultSnapshotPath()),
+    schedulePath: absolutePath(source.schedulePath, defaultSchedulePath()),
   }
 }

package/src/keepalive.js

@@ -4,20 +4,36 @@ const ORIGIN = "https://copilot.tencent.com"
 const CHAT_PATH = "/v2/chat/completions"
 const FILLER = "The quick brown fox jumps over the lazy dog. "
 const CHARS_PER_TOKEN = 4.5
+export const DEFAULT_PROBE_MESSAGE = "Magpie daily connectivity check. Reply with OK."
 
-export function buildKeepaliveBody({ model, inputTokens, day }) {
-  const targetChars = Math.ceil(inputTokens * CHARS_PER_TOKEN)
-  const filler = FILLER.repeat(Math.ceil(targetChars / FILLER.length))
+function probeMessage(value) {
+  return sanitize(value, 500).trim() || DEFAULT_PROBE_MESSAGE
+}
+
+export function buildKeepaliveBody({
+  model,
+  inputTokens,
+  day,
+  probeMode = "short",
+  probeMessage: message = "",
+}) {
+  const prompt = `${probeMessage(message)} Today is ${day}.`
+  let content = prompt
+  if (probeMode === "context") {
+    const targetChars = Math.ceil(inputTokens * CHARS_PER_TOKEN)
+    const filler = FILLER.repeat(Math.ceil(targetChars / FILLER.length))
+    content = `${filler}${prompt}`
+  }
   return {
     model,
     messages: [
       {
         role: "system",
-        content: "You are a connectivity check. Reply with a single word.",
+        content: "You are a concise connectivity check. Reply with a single word.",
       },
       {
         role: "user",
-        content: `${filler}Reply with a single word.`,
+        content,
       },
     ],
     max_tokens: 1,
@@ -152,6 +168,8 @@ export async function sendKeepalive({
   origin = ORIGIN,
   model,
   inputTokens,
+  probeMode,
+  probeMessage,
   timeoutMs,
   day,
   key,
@@ -167,7 +185,13 @@ export async function sendKeepalive({
         accept: "text/event-stream",
         "x-magpie-workbuddy-session": `workbuddy-keepalive:${key}:${day}`.slice(0, 200),
       },
-      body: JSON.stringify(buildKeepaliveBody({ model, inputTokens, day })),
+      body: JSON.stringify(buildKeepaliveBody({
+        model,
+        inputTokens,
+        day,
+        probeMode,
+        probeMessage,
+      })),
       signal: AbortSignal.timeout(timeoutMs),
     })
     return { response, raw: await response.text() }

package/src/log.js

@@ -227,6 +227,7 @@ export function formatLogRecord(record) {
     ["outcome", record.outcome],
     ["attempts", record.attempts],
     ["model", record.model],
+    ["probeMode", record.probeMode],
     ["inputTokens", record.inputTokens],
     ["credit", record.credit],
     ["streak", record.streak],

package/src/provider-chat.js

@@ -1,5 +1,9 @@
 import { sanitize, terminalQuotaFailure, text, withTimeout } from "./common.js"
-import { buildKeepaliveBody, sendKeepalive } from "./keepalive.js"
+import {
+  DEFAULT_PROBE_MESSAGE,
+  buildKeepaliveBody,
+  sendKeepalive,
+} from "./keepalive.js"
 
 const FILLER = "The quick brown fox jumps over the lazy dog. "
 const CHARS_PER_TOKEN = 4.5
@@ -42,15 +46,27 @@ export function endpointFor(baseURL, protocol) {
   return `${base}${protocol === "anthropic" ? "/messages" : "/chat/completions"}`
 }
 
-export function anthropicBody({ model, inputTokens, day }) {
-  const targetChars = Math.ceil(inputTokens * CHARS_PER_TOKEN)
-  const filler = FILLER.repeat(Math.ceil(targetChars / FILLER.length))
+export function anthropicBody({
+  model,
+  inputTokens,
+  day,
+  probeMode = "short",
+  probeMessage = "",
+}) {
+  const message = sanitize(probeMessage, 500).trim() || DEFAULT_PROBE_MESSAGE
+  const prompt = `${message} Today is ${day}.`
+  let content = prompt
+  if (probeMode === "context") {
+    const targetChars = Math.ceil(inputTokens * CHARS_PER_TOKEN)
+    const filler = FILLER.repeat(Math.ceil(targetChars / FILLER.length))
+    content = `${filler}${prompt}`
+  }
   return {
     model,
-    system: "You are a connectivity check. Reply with a short word.",
+    system: "You are a concise connectivity check. Reply with a short word.",
     messages: [{
       role: "user",
-      content: `${filler}Today is ${day}. Reply with a short word.`,
+      content,
     }],
     max_tokens: 8,
     stream: true,
@@ -126,6 +142,8 @@ export async function sendProviderChat({
   protocol,
   loader,
   inputTokens,
+  probeMode,
+  probeMessage,
   timeoutMs,
   day,
   key,
@@ -148,6 +166,8 @@ export async function sendProviderChat({
         origin,
         model,
         inputTokens,
+        probeMode,
+        probeMessage,
         timeoutMs,
         day,
         key,
@@ -169,8 +189,8 @@ export async function sendProviderChat({
 
   const url = endpointFor(loader.baseURL, protocol)
   const body = protocol === "anthropic"
-    ? anthropicBody({ model, inputTokens, day })
-    : buildKeepaliveBody({ model, inputTokens, day })
+    ? anthropicBody({ model, inputTokens, day, probeMode, probeMessage })
+    : buildKeepaliveBody({ model, inputTokens, day, probeMode, probeMessage })
 
   let response
   let raw

package/src/schedule.js

@@ -0,0 +1,93 @@
+import { randomBytes } from "node:crypto"
+import { chmod, mkdir, open, readFile, rename, unlink } from "node:fs/promises"
+import { dirname, join } from "node:path"
+import { errorText, magpieConfigHome, sanitize } from "./common.js"
+
+const VERSION = 1
+const FILE_OPS = { chmod, mkdir, open, readFile, rename, unlink }
+
+export function defaultSchedulePath() {
+  return join(magpieConfigHome(), "quota-keepalive-schedule.json")
+}
+
+export function emptySchedule() {
+  return {
+    version: VERSION,
+    signature: "",
+    lastRunAt: "",
+    nextRunAt: "",
+    lastStatus: "",
+    lastSummary: "",
+  }
+}
+
+function validDate(value) {
+  return typeof value === "string" && Number.isFinite(Date.parse(value))
+}
+
+export async function readSchedule(path, ops = FILE_OPS) {
+  let raw
+  try {
+    raw = await ops.readFile(path, "utf8")
+  } catch (error) {
+    if (error?.code === "ENOENT") return emptySchedule()
+    throw new Error(`cannot read ${path}: ${errorText(error)}`)
+  }
+  let value
+  try {
+    value = JSON.parse(raw)
+  } catch {
+    throw new Error(`${path} is not valid JSON; it was left unchanged`)
+  }
+  if (!value || typeof value !== "object" || Array.isArray(value) || value.version !== VERSION) {
+    throw new Error(`${path} has an unsupported shape; it was left unchanged`)
+  }
+  return {
+    version: VERSION,
+    signature: sanitize(value.signature, 2_000),
+    lastRunAt: validDate(value.lastRunAt) ? value.lastRunAt : "",
+    nextRunAt: validDate(value.nextRunAt) ? value.nextRunAt : "",
+    lastStatus: sanitize(value.lastStatus, 24),
+    lastSummary: sanitize(value.lastSummary, 480),
+  }
+}
+
+export async function writeSchedule(path, value, ops = FILE_OPS) {
+  await ops.mkdir(dirname(path), { recursive: true, mode: 0o700 })
+  const tmp = `${path}.tmp-${process.pid}-${randomBytes(6).toString("hex")}`
+  let handle
+  try {
+    handle = await ops.open(tmp, "wx", 0o600)
+    await handle.writeFile(`${JSON.stringify(value, null, 2)}\n`, "utf8")
+    await handle.sync()
+    await handle.close()
+    handle = null
+    await ops.rename(tmp, path)
+    await ops.chmod(path, 0o600)
+  } catch (error) {
+    if (handle) await handle.close().catch(() => {})
+    await ops.unlink(tmp).catch(() => {})
+    throw error
+  }
+}
+
+export function nextRunAfter({
+  now = Date.now(),
+  lastRunAt = "",
+  nextRunAt = "",
+  intervalMs,
+  storedSignature = "",
+  signature = "",
+}) {
+  const interval = Math.max(1_000, Number(intervalMs) || 60_000)
+  if (storedSignature !== signature) return now + interval
+  const saved = Date.parse(nextRunAt)
+  const last = Date.parse(lastRunAt)
+  if (Number.isFinite(saved)) return Math.max(saved, now)
+  if (Number.isFinite(last)) return Math.max(last + interval, now)
+  return now + interval
+}
+
+export function scheduleSignature(config = {}) {
+  return JSON.stringify(config)
+}

package/src/snapshot.js

@@ -31,6 +31,7 @@ const FIELD_LIMITS = {
   status: 12,
   outcome: 24,
   model: 160,
+  probeMode: 16,
   account: 24,
   run: 48,
 }
@@ -47,7 +48,7 @@ function safeField(record, name) {
 
 export function publicRecord(record) {
   const out = {}
-  for (const name of ["ts", "day", "provider", "event", "status", "outcome", "model", "account", "run"]) {
+  for (const name of ["ts", "day", "provider", "event", "status", "outcome", "model", "probeMode", "account", "run"]) {
     const value = safeField(record, name)
     if (value) out[name] = value
   }

package/src/state.js

@@ -232,8 +232,21 @@ export async function runDailyOnce({
       return { kind: "unavailable", error: errorText(error) }
     }
     prune(doc, day)
-    const previous = newer(doc.accounts[account], inMemory)
+    let previous = newer(doc.accounts[account], inMemory)
+    let interrupted = false
     if (previous?.day === day) {
+      const startedAt = Date.parse(previous.at || "")
+      const staleMs = Math.max(30_000, timeoutMs + 30_000)
+      if (previous.status === "running" && Number.isFinite(startedAt) &&
+          now.getTime() - startedAt > staleMs) {
+        interrupted = true
+        previous = {
+          ...previous,
+          status: "failed",
+          at: new Date(startedAt).toISOString(),
+          error: "the previous attempt was interrupted",
+        }
+      }
       remember(statePath, account, previous)
       if (previous.status === "ok") {
         return { kind: "skipped", reason: "done-today", attempts: previous.attempts }
@@ -246,7 +259,7 @@ export async function runDailyOnce({
           error: previous.error || "",
         }
       }
-      if (previous.attempts >= maxAttempts && !force) {
+      if (previous.attempts >= maxAttempts && !force && !interrupted) {
         return {
           kind: "skipped",
           reason: "exhausted",

package/src/workbuddy.js

@@ -181,31 +181,83 @@ export async function withAuthLock(authPath, work, waitMs = 2_000, ops = FILE_OP
 export async function saveAuthEntry(authPath, key, next, ops = FILE_OPS) {
   if (!text(key) || !object(next)) throw new Error("refusing to save an invalid provider sign-in")
   return withAuthLock(authPath, async () => {
-    let all
-    try {
-      all = JSON.parse(await ops.readFile(authPath, "utf8"))
-    } catch (error) {
-      if (error?.code !== "ENOENT") throw error
-      all = {}
+    for (let attempt = 0; attempt < 4; attempt++) {
+      const snapshot = await readAuthSnapshot(authPath, ops)
+      const all = snapshot.value
+      if (!object(all)) throw new Error("plugin-auth.json isn't an object")
+      if (JSON.stringify(all[key]) === JSON.stringify(next)) return true
+
+      all[key] = next
+      const serialized = `${JSON.stringify(all, null, 2)}\n`
+      const tmp = `${authPath}.tmp-${process.pid}-${randomBytes(6).toString("hex")}`
+      let handle
+      try {
+        handle = await ops.open(tmp, "wx", 0o600)
+        await handle.writeFile(serialized, "utf8")
+        await handle.sync()
+        await handle.close()
+        handle = null
+
+        // Magpie and provider plugins do not use this lock. Refuse to replace
+        // a file that changed after our read, then merge against the new copy.
+        const current = await readAuthStamp(authPath, ops)
+        if (current !== snapshot.stamp) {
+          await ops.unlink(tmp).catch(() => {})
+          await new Promise((resolve) => setTimeout(resolve, 20))
+          continue
+        }
+        await ops.rename(tmp, authPath)
+        await ops.chmod(authPath, 0o600)
+        return true
+      } catch (error) {
+        if (handle) await handle.close().catch(() => {})
+        await ops.unlink(tmp).catch(() => {})
+        throw error
+      }
     }
-    if (!object(all)) throw new Error("plugin-auth.json isn't an object")
-    all[key] = next
-    const tmp = `${authPath}.tmp-${process.pid}-${randomBytes(6).toString("hex")}`
-    let handle
+    throw new Error("magpie's sign-in file changed repeatedly; the update was not saved")
+  })
+}
+
+function authStamp(info) {
+  if (!info) return "missing"
+  return [
+    info.ino ?? "",
+    info.size ?? "",
+    info.mtimeMs ?? "",
+    info.ctimeMs ?? "",
+  ].join(":")
+}
+
+async function readAuthStamp(authPath, ops) {
+  try {
+    return authStamp(await ops.stat(authPath))
+  } catch (error) {
+    if (error?.code === "ENOENT") return "missing"
+    throw error
+  }
+}
+
+async function readAuthSnapshot(authPath, ops) {
+  for (let attempt = 0; attempt < 4; attempt++) {
+    const before = await readAuthStamp(authPath, ops)
+    if (before === "missing") {
+      return { value: {}, stamp: before }
+    }
+    let raw
     try {
-      handle = await ops.open(tmp, "wx", 0o600)
-      await handle.writeFile(`${JSON.stringify(all, null, 2)}\n`, "utf8")
-      await handle.sync()
-      await handle.close()
-      handle = null
-      await ops.rename(tmp, authPath)
-      await ops.chmod(authPath, 0o600)
+      raw = await ops.readFile(authPath, "utf8")
     } catch (error) {
-      if (handle) await handle.close().catch(() => {})
-      await ops.unlink(tmp).catch(() => {})
+      if (error?.code === "ENOENT") return { value: {}, stamp: "missing" }
       throw error
     }
-  })
+    const after = await readAuthStamp(authPath, ops)
+    if (before === after) {
+      return { value: JSON.parse(raw), stamp: after }
+    }
+    await new Promise((resolve) => setTimeout(resolve, 20))
+  }
+  throw new Error("magpie's sign-in file kept changing while it was being read")
 }
 
 export function authIdentity(auth, key) {
@@ -215,6 +267,7 @@ export function authIdentity(auth, key) {
 }
 
 export function resultFields(result, config, model = "") {
+  const inputTokens = config?.probeMode === "context" ? Number(config.inputTokens) || 0 : 0
   switch (result?.kind) {
     case "skipped":
       if (result.reason === "done-today") {
@@ -271,7 +324,7 @@ export function resultFields(result, config, model = "") {
               : "exhausted",
         attempts: result.attempts,
         model: result.result?.model || model,
-        inputTokens: config.inputTokens,
+        inputTokens,
         retry: !result.ok && !terminal && result.attempts < config.maxAttempts,
         persisted: result.persisted !== false,
         error: result.error,
@@ -364,6 +417,8 @@ export async function sendWithCandidates({
         protocol: candidate.npm === "@ai-sdk/anthropic" ? "anthropic" : adapter.protocol,
         loader,
         inputTokens: config.inputTokens,
+        probeMode: config.probeMode,
+        probeMessage: config.probeMessage,
         timeoutMs: config.timeoutMs,
         day,
         key: accountTag(identity),
@@ -473,7 +528,8 @@ export function createProviderKeepalive(options = {}) {
           run,
           model: candidates[0].id,
           models: candidates.map((candidate) => candidate.id).join(","),
-          inputTokens: config.inputTokens,
+          inputTokens: config.probeMode === "context" ? config.inputTokens : 0,
+          probeMode: config.probeMode,
           keepalive: true,
         })
 
@@ -482,7 +538,8 @@ export function createProviderKeepalive(options = {}) {
           account: accountTag(`${provider}|${identity}`),
           day,
           model: candidates[0].id,
-          inputTokens: config.inputTokens,
+          inputTokens: config.probeMode === "context" ? config.inputTokens : 0,
+          probeMode: config.probeMode,
           maxAttempts: config.maxAttempts,
           lockWaitMs: config.lockWaitMs,
           timeoutMs: config.timeoutMs,

package/ui/app.js

@@ -996,6 +996,7 @@
       ["供应商", record?.provider],
       ["账号", record?.account],
       ["模型", record?.model],
+      ["保活方式", record?.probeMode === "context" ? "长上下文" : record?.probeMode === "short" ? "短探活" : ""],
       ["输入", number(record?.inputTokens) ? `${formatNumber(record.inputTokens)} tokens` : ""],
       ["本次积分", number(record?.credit) ? `+${formatNumber(record.credit)}` : ""],
       ["连续签到", number(record?.streak) ? `${formatNumber(record.streak)} 天` : ""],
@@ -1192,7 +1193,8 @@
       keepalive: "真实保活",
       all: "签到 + 真实保活",
     }[options.scheduleAction] || "签到"
-    return `每 ${number(options.scheduleMinutes) || 60} 分钟执行一次${action}。`
+    const probe = options.probeMode === "context" ? "长上下文保活" : "短探活"
+    return `每 ${number(options.scheduleMinutes) || 60} 分钟执行一次${action}；${probe}。`
   }
 
   async function loadSchedule(showToast = false, refresh = false) {
@@ -1210,6 +1212,7 @@
       $("schedule-action").value = ["checkin", "keepalive", "all"].includes(options.scheduleAction)
         ? options.scheduleAction
         : "checkin"
+      $("probe-mode").value = options.probeMode === "context" ? "context" : "short"
       $("schedule-status").textContent = scheduleDescription(options)
       if (showToast) toast("定时计划已保存")
     } catch (error) {
@@ -1225,6 +1228,7 @@
       scheduleEnabled: $("schedule-enabled").checked,
       scheduleMinutes: minutes,
       scheduleAction: $("schedule-action").value,
+      probeMode: $("probe-mode").value,
     })))
     await loadSchedule(true)
   }

package/ui/index.html

@@ -108,6 +108,17 @@
             <path d="m4 6 4 4 4-4"></path>
           </svg>
         </label>
+        <label class="select-field probe-field"
+               title="短探活只发送一条短消息；长上下文会填充到配置的输入规模">
+          <select id="probe-mode" aria-label="保活方式">
+            <option value="short">短探活</option>
+            <option value="context">长上下文</option>
+          </select>
+          <svg viewBox="0 0 16 16" fill="none" stroke="currentColor" stroke-width="1.5"
+               stroke-linecap="round" stroke-linejoin="round" aria-hidden="true">
+            <path d="m4 6 4 4 4-4"></path>
+          </svg>
+        </label>
         <button class="button schedule-save" id="schedule-save" type="button">保存计划</button>
       </div>
     </section>

package/ui/style.css

@@ -384,6 +384,7 @@ h2 { margin-bottom: 2px; font-size: 16px; letter-spacing: -.015em; }
   appearance: none;
   -webkit-appearance: none;
 }
+.probe-field select { min-width: 118px; }
 .select-field svg {
   position: absolute;
   right: 8px;