antigravity-usage 0.2.8 → 0.2.9

package/README.md

@@ -174,10 +174,27 @@ Quickly check if your auth tokens are valid or expired.
 antigravity-usage wakeup config     # Interactive setup (takes 30 seconds)
 antigravity-usage wakeup install    # Install to native system cron
 antigravity-usage wakeup status     # Check configuration & next run
-antigravity-usage wakeup test       # Test trigger manually
+antigravity-usage wakeup test       # Test trigger manually (interactive)
 antigravity-usage wakeup history    # View trigger history
 ```
 
+**Quick Test with Flags:**
+```bash
+# Test with specific email and model
+antigravity-usage wakeup test -e your@gmail.com -m claude-sonnet-4-5
+
+# Full command with custom prompt
+antigravity-usage wakeup test --email your@gmail.com --model gemini-3-flash --prompt "hello"
+
+# Mix flags (missing ones will be prompted)
+antigravity-usage wakeup test -e your@gmail.com
+```
+
+**Available Options:**
+- `-e, --email <email>` - Account email to use for testing
+- `-m, --model <model>` - Model ID to test (e.g., claude-sonnet-4-5, gemini-3-flash)
+- `-p, --prompt <prompt>` - Test prompt to send (default: "hi")
+
 **Why This Matters:**
 Your Antigravity quota resets every ~5 hours, but if you don't use it, you lose it. The wakeup feature ensures you **automatically trigger** both Claude and Gemini models to keep your quota flowing.
 

package/dist/index.js

@@ -1679,15 +1679,16 @@ var CloudCodeClient = class {
           if (jsonStr.trim() === "[DONE]") continue;
           try {
             const data = JSON.parse(jsonStr);
-            const candidateText = data.candidates?.[0]?.content?.parts?.[0]?.text;
+            const responseData = data.response || data;
+            const candidateText = responseData.candidates?.[0]?.content?.parts?.[0]?.text;
             if (candidateText) {
               fullText += candidateText;
             }
-            if (data.usageMetadata) {
+            if (responseData.usageMetadata) {
               tokensUsed = {
-                prompt: data.usageMetadata.promptTokenCount || 0,
-                completion: data.usageMetadata.candidatesTokenCount || 0,
-                total: data.usageMetadata.totalTokenCount || 0
+                prompt: responseData.usageMetadata.promptTokenCount || 0,
+                completion: responseData.usageMetadata.candidatesTokenCount || 0,
+                total: responseData.usageMetadata.totalTokenCount || 0
               };
             }
           } catch {
@@ -3952,7 +3953,7 @@ async function wakeupCommand(subcommand, args, options) {
       await uninstallSchedule();
       break;
     case "test":
-      await runTestTrigger();
+      await runTestTrigger(options);
       break;
     case "history":
       await showHistory(options);
@@ -4187,7 +4188,7 @@ async function uninstallSchedule() {
   }
   console.log("");
 }
-async function runTestTrigger() {
+async function runTestTrigger(options = {}) {
   console.log("\n\u{1F9EA} Test Trigger\n");
   const accountManager = getAccountManager();
   const accounts = accountManager.getAccountEmails();
@@ -4195,8 +4196,17 @@ async function runTestTrigger() {
     console.log("\u274C No accounts available. Please login first.");
     return;
   }
-  let accountEmail = accounts[0];
-  if (accounts.length > 1) {
+  let accountEmail;
+  if (options.email) {
+    if (!accounts.includes(options.email)) {
+      console.log(`\u274C Account "${options.email}" not found.`);
+      console.log(`   Available accounts: ${accounts.join(", ")}`);
+      return;
+    }
+    accountEmail = options.email;
+  } else if (accounts.length === 1) {
+    accountEmail = accounts[0];
+  } else {
     const { selectedAccount } = await inquirer2.prompt([{
       type: "list",
       name: "selectedAccount",
@@ -4205,19 +4215,20 @@ async function runTestTrigger() {
     }]);
     accountEmail = selectedAccount;
   }
-  const config = loadWakeupConfig();
-  const { modelId } = await inquirer2.prompt([{
-    type: "input",
-    name: "modelId",
-    message: "Model ID to test:",
-    default: config?.selectedModels[0] || "claude-sonnet-4-5"
-  }]);
-  const { prompt } = await inquirer2.prompt([{
-    type: "input",
-    name: "prompt",
-    message: "Test prompt:",
-    default: "hi"
-  }]);
+  let modelId;
+  if (options.model) {
+    modelId = options.model;
+  } else {
+    const config = loadWakeupConfig();
+    const { selectedModel } = await inquirer2.prompt([{
+      type: "input",
+      name: "selectedModel",
+      message: "Model ID to test:",
+      default: config?.selectedModels[0] || "claude-sonnet-4-5"
+    }]);
+    modelId = selectedModel;
+  }
+  const prompt = options.prompt || "hi";
   console.log("\n\u23F3 Triggering...");
   try {
     const result = await testTrigger(modelId, accountEmail, prompt);
@@ -4351,7 +4362,7 @@ wakeupCmd.command("config").description("Configure auto wake-up schedule").actio
 wakeupCmd.command("trigger").description("Execute one trigger cycle (called by cron)").option("--scheduled", "Mark as scheduled trigger").action((options) => wakeupCommand("trigger", [], options));
 wakeupCmd.command("install").description("Install wake-up schedule to system cron").action(() => wakeupCommand("install", [], {}));
 wakeupCmd.command("uninstall").description("Remove wake-up schedule from system cron").action(() => wakeupCommand("uninstall", [], {}));
-wakeupCmd.command("test").description("Test trigger manually").action(() => wakeupCommand("test", [], {}));
+wakeupCmd.command("test").description("Test trigger manually").option("-e, --email <email>", "Account email to use for testing").option("-m, --model <model>", "Model ID to test").option("-p, --prompt <prompt>", "Test prompt to send", "hi").action((options) => wakeupCommand("test", [], options));
 wakeupCmd.command("history").description("View trigger history").option("--limit <n>", "Number of records to show", "10").option("--json", "Output as JSON").action((options) => wakeupCommand("history", [], options));
 wakeupCmd.command("status").description("Show wake-up status and configuration").action(() => wakeupCommand("status", [], {}));
 wakeupCmd.action(() => wakeupCommand("status", [], {}));