replicate-token-verification 1.0.0 → 1.0.1

package/README.md

@@ -40,7 +40,7 @@ rtv
 rtv
 rtv --token <token>
 rtv <token>
-REPLICATE_API_TOKEN=<token> rtv
+rtv --token-from-env REPLICATE_API_TOKEN
 ```
 
 ## 可用选项
@@ -49,6 +49,7 @@ REPLICATE_API_TOKEN=<token> rtv
 --json      输出 JSON，适合脚本调用
 --quiet     仅输出最终结果
 --no-loop   交互模式下只校验一次后退出
+--token-from-env <name>   从指定环境变量读取 Token
 --no-model-test   跳过调用测试，只检查认证状态和账号状态
 ```
 
@@ -74,7 +75,7 @@ rtv --no-loop
 ```bash
 rtv --token r8_xxx
 rtv r8_xxx
-REPLICATE_API_TOKEN=r8_xxx rtv
+REPLICATE_API_TOKEN=r8_xxx rtv --token-from-env REPLICATE_API_TOKEN
 rtv --token r8_xxx --quiet
 ```
 
@@ -86,6 +87,12 @@ rtv --token r8_xxx --quiet
 rtv --token r8_xxx --json
 ```
 
+如果你明确希望从环境变量取值：
+
+```bash
+REPLICATE_API_TOKEN=r8_xxx rtv --token-from-env REPLICATE_API_TOKEN
+```
+
 `--json` 模式下会输出这些核心字段：
 
 - `usabilityStatus`：当前 Token 是否可直接使用

package/bin/replicate-token-verification.js

@@ -22,6 +22,7 @@ program
   .description("检查 Replicate API Token 当前是否可直接使用")
   .argument("[token]", "要检查的 Replicate API Token")
   .option("-t, --token <token>", "要检查的 Replicate API Token")
+  .option("--token-from-env <name>", "从指定环境变量读取 Token，例如 REPLICATE_API_TOKEN")
   .option("--json", "输出 JSON 结果，便于脚本调用")
   .option("-q, --quiet", "仅输出最终结果")
   .option("--no-loop", "交互模式下只校验一次后退出")
@@ -491,6 +492,7 @@ function printSummary(report, outputJson) {
 
   if (outputJson) {
     console.log(JSON.stringify({
+      token: report.token,
       usabilityStatus: usabilityClassification.usabilityStatus,
       authStatus: accountClassification.authStatus,
       accountStatus: accountClassification.accountStatus,
@@ -510,6 +512,7 @@ function printSummary(report, outputJson) {
   }
 
   console.log(paint("校验摘要", COLOR.bold));
+  console.log(`被测 Token：${paint(report.token, COLOR.cyan)}`);
   console.log(`${formatStatusLabel(usabilityClassification.usabilityStatus === "usable" || usabilityClassification.usabilityStatus === "probably_usable" ? "valid" : usabilityClassification.usabilityStatus === "unusable" ? "invalid" : "unknown")}：当前 Token ${formatUsabilitySummary(usabilityClassification)}`);
   console.log("");
   printAccountDetails(report.account, report.accountError);
@@ -526,8 +529,20 @@ function printSummary(report, outputJson) {
   }
 }
 
-function resolveToken(positionalToken, optionsToken) {
-  return optionsToken || positionalToken || process.env.REPLICATE_API_TOKEN || null;
+function resolveToken(positionalToken, optionsToken, tokenFromEnv) {
+  if (optionsToken) {
+    return optionsToken;
+  }
+
+  if (positionalToken) {
+    return positionalToken;
+  }
+
+  if (tokenFromEnv) {
+    return process.env[tokenFromEnv] || null;
+  }
+
+  return null;
 }
 
 function exitCodeForReport(report) {
@@ -564,6 +579,7 @@ async function runVerification(token, options) {
     const modelPromise = modelSkipped ? Promise.resolve(null) : verifyModelInvocation(token);
     const [accountResult, modelResult] = await Promise.allSettled([accountPromise, modelPromise]);
     const report = {
+      token,
       account: accountResult.status === "fulfilled" ? accountResult.value : null,
       accountError: accountResult.status === "rejected" ? accountResult.reason : null,
       model: modelResult.status === "fulfilled" ? modelResult.value : null,
@@ -636,7 +652,7 @@ async function main() {
 
   const options = program.opts();
   const positionalToken = program.args[0];
-  const token = resolveToken(positionalToken, options.token);
+  const token = resolveToken(positionalToken, options.token, options.tokenFromEnv);
 
   if (token) {
     if (!options.json && !options.quiet) {
@@ -650,7 +666,7 @@ async function main() {
   }
 
   if (!process.stdin.isTTY || !process.stdout.isTTY) {
-    console.error("缺少 Replicate API Token。请通过参数、选项或 REPLICATE_API_TOKEN 环境变量传入。");
+    console.error("缺少 Replicate API Token。请通过位置参数、--token、--token-from-env 或交互模式提供。");
     process.exitCode = 1;
     return;
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "replicate-token-verification",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "CLI to check whether a Replicate API token can be directly used",
   "type": "module",
   "author": "Duan <hello@duanjun.net>",