pi-telegram-notifier 0.1.0 → 0.1.2

package/README.md

@@ -1,21 +1,20 @@
 # pi-telegram-notifier
 
-Send the **final** response of each Pi agent run to a paired private Telegram chat. Each notification contains a UTC timestamp, provider, full response text, and the **actual** remaining provider balance when it can be queried. Telegram metadata (provider and quota) is in English; the agent's response is forwarded verbatim in its original language. Intermediate tool calls do not trigger messages.
+Send the **final** response of each Pi agent run to a paired private Telegram chat. Each notification contains a UTC timestamp, device hostname, provider, full response text, and the **actual** remaining provider balance when it can be queried. Telegram metadata (provider and quota) is in English; the agent's response is forwarded verbatim in its original language. Intermediate tool calls do not trigger messages.
 
 ## Install and configure
 
-Requires Node.js 22.19+ and [Pi](https://pi.dev). Create a Telegram bot using [@BotFather](https://t.me/BotFather), then set its token **outside the repository** before starting Pi:
+Requires Node.js 22.19+ and [Pi](https://pi.dev). Install the published package, create a Telegram bot using [@BotFather](https://t.me/BotFather), and set its token **outside the repository** before starting Pi:
 
 ```sh
+pi install npm:pi-telegram-notifier
 export PI_TELEGRAM_BOT_TOKEN='your-bot-token'
-pi -e .                       # test this checkout as a Pi package
-# Later, after public release: pi install npm:pi-telegram-notifier
-# Or: pi install git:github.com/Gybra/pi-telegram-notifier
+pi
 ```
 
-In Pi run `/telegram-pair`. Within five minutes send **`/start <code>`** to your bot in a private Telegram chat using the displayed one-time code. The extension stores only the numeric chat ID in `~/.pi/agent/pi-telegram-notifier.json` (or `$PI_CODING_AGENT_DIR/pi-telegram-notifier.json`), with mode 0600. To switch chats, run `/telegram-pair` again; the existing chat stays paired if pairing fails. The bot needs to receive messages via `getUpdates` during pairing: a webhook or another updates consumer must be disconnected first. There is no always-on polling after pairing.
+If Pi is already running, run `/reload` after installing the package (restart Pi if you just set the token). If you previously installed a local checkout, run `pi list`, then `pi remove /absolute/path/to/pi-telegram-notifier` to avoid loading both copies. To get future releases, run `pi update npm:pi-telegram-notifier`. In Pi run `/telegram-pair`. Within five minutes send **`/start <code>`** to your bot in a private Telegram chat using the displayed one-time code. The extension stores only the numeric chat ID in `~/.pi/agent/pi-telegram-notifier.json` (or `$PI_CODING_AGENT_DIR/pi-telegram-notifier.json`), with mode 0600. To switch chats, run `/telegram-pair` again; the existing chat stays paired if pairing fails. The bot needs to receive messages via `getUpdates` during pairing: a webhook or another updates consumer must be disconnected first. There is no always-on polling after pairing.
 
-**Privacy:** final agent responses are sent in full to Telegram. They can contain sensitive text; only use this package if that is acceptable for your projects and Telegram chat. Bot and provider tokens are not stored in the package or sent as message text. Telegram Bot API uses the bot token in the request URL; do not log outgoing URLs. Keep your shell environment and Pi credentials private. A lost or compromised Telegram bot token must be revoked with BotFather.
+**Privacy:** final agent responses and the device hostname are sent to Telegram. Responses can contain sensitive text; only use this package if that is acceptable for your projects and Telegram chat. Bot and provider tokens are not stored in the package or sent as message text. Telegram Bot API uses the bot token in the request URL; do not log outgoing URLs. Keep your shell environment and Pi credentials private. A lost or compromised Telegram bot token must be revoked with BotFather.
 
 ## Quota coverage
 
@@ -44,7 +43,7 @@ Telegram has a message-length limit; long responses are delivered in consecutive
 ```sh
 npm test
 npm pack --dry-run --json
-pi -e . --help
+pi --no-extensions -e . --help
 ```
 
-Read [AGENTS.md](AGENTS.md) before editing. Open contributions with tests for behavior changes and official provider documentation for balance integrations. MIT licensed; see [LICENSE](LICENSE). CI runs tests and packaging checks. Publishing is deliberately manual: create the public GitHub repository, then publish the npm package with the `pi-package` keyword to make it eligible for the [Pi package gallery](https://pi.dev/packages). No action here pushes, publishes, or handles real credentials.
+Read [AGENTS.md](AGENTS.md) before editing. Open contributions with tests for behavior changes and official provider documentation for balance integrations. MIT licensed; see [LICENSE](LICENSE). CI runs tests and packaging checks. The published npm package includes the `pi-package` keyword for the [Pi package gallery](https://pi.dev/packages). New versions are released manually; CI does not publish packages or handle real credentials.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-telegram-notifier",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Telegram notifications for final Pi agent responses and provider balances",
   "type": "module",
   "license": "MIT",

package/src/notification.js

@@ -1,3 +1,4 @@
+import { hostname } from 'node:os';
 import { loadChatId, sendTelegram } from './telegram.js';
 import { getQuota } from './quota.js';
 
@@ -12,7 +13,7 @@ export function finalResponse(branch, startLeafId) {
 }
 
 export function formatNotification(timestamp, response, quota) {
-  return `${timestamp}\nProvider: ${response.provider}\n${quota}\n\n${response.text}`;
+  return `${timestamp}\nDevice: ${hostname()}\nProvider: ${response.provider}\n${quota}\n\n${response.text}`;
 }
 
 export function registerNotifications(pi, agentDir, token, fetchImpl = fetch) {
@@ -26,9 +27,14 @@ export function registerNotifications(pi, agentDir, token, fetchImpl = fetch) {
     startLeafId = undefined;
     const chatId = loadChatId(agentDir);
     if (!chatId || !response) return;
+    let quota;
     try {
       const model = ctx.modelRegistry?.find(response.provider, response.modelId);
-      const quota = await getQuota(model, ctx.modelRegistry, fetchImpl);
+      quota = await getQuota(model, ctx.modelRegistry, fetchImpl);
+    } catch {
+      quota = 'Quota unavailable';
+    }
+    try {
       await sendTelegram(token, chatId,
         formatNotification(new Date().toISOString(), response, quota), fetchImpl);
     } catch {

package/src/quota.js

@@ -1,15 +1,15 @@
 const UNAVAILABLE = 'Quota unavailable';
 
 const providers = {
-  openrouter: ['https://openrouter.ai', 'https://openrouter.ai/api/v1/credits'],
-  'openai-codex': ['https://chatgpt.com', 'https://chatgpt.com/backend-api/wham/usage'],
-  deepseek: ['https://api.deepseek.com', 'https://api.deepseek.com/user/balance'],
-  moonshotai: ['https://api.moonshot.ai', 'https://api.moonshot.ai/v1/users/me/balance'],
-  'moonshotai-cn': ['https://api.moonshot.cn', 'https://api.moonshot.cn/v1/users/me/balance'],
-  zai: ['https://api.z.ai', 'https://api.z.ai/api/monitor/usage/quota/limit'],
-  'zai-coding-cn': ['https://open.bigmodel.cn', 'https://open.bigmodel.cn/api/monitor/usage/quota/limit'],
-  anthropic: ['https://api.anthropic.com', 'https://api.anthropic.com/api/oauth/usage'],
-  xai: ['https://api.x.ai', 'https://cli-chat-proxy.grok.com/v1/billing?format=credits'],
+  openrouter: { origin: 'https://openrouter.ai', url: 'https://openrouter.ai/api/v1/credits' },
+  'openai-codex': { origin: 'https://chatgpt.com', url: 'https://chatgpt.com/backend-api/wham/usage' },
+  deepseek: { origin: 'https://api.deepseek.com', url: 'https://api.deepseek.com/user/balance' },
+  moonshotai: { origin: 'https://api.moonshot.ai', url: 'https://api.moonshot.ai/v1/users/me/balance' },
+  'moonshotai-cn': { origin: 'https://api.moonshot.cn', url: 'https://api.moonshot.cn/v1/users/me/balance' },
+  zai: { origin: 'https://api.z.ai', url: 'https://api.z.ai/api/monitor/usage/quota/limit' },
+  'zai-coding-cn': { origin: 'https://open.bigmodel.cn', url: 'https://open.bigmodel.cn/api/monitor/usage/quota/limit' },
+  anthropic: { origin: 'https://api.anthropic.com', url: 'https://api.anthropic.com/api/oauth/usage' },
+  xai: { origin: 'https://api.x.ai', url: 'https://cli-chat-proxy.grok.com/v1/billing?format=credits' },
 };
 
 function remainingPercent(used) {
@@ -20,10 +20,10 @@ export async function getQuota(model, modelRegistry, fetchImpl = fetch) {
   const provider = model && providers[model.provider];
   if (!provider) return UNAVAILABLE;
   try {
-    if (new URL(model.baseUrl).origin !== provider[0]) return UNAVAILABLE;
+    if (new URL(model.baseUrl).origin !== provider.origin) return UNAVAILABLE;
     if (['anthropic', 'xai'].includes(model.provider) && !modelRegistry.isUsingOAuth(model)) return UNAVAILABLE;
     const auth = await modelRegistry.getApiKeyAndHeaders(model);
-    if (!auth.ok || !auth.apiKey || (auth.baseUrl && new URL(auth.baseUrl).origin !== provider[0])) return UNAVAILABLE;
+    if (!auth.ok || !auth.apiKey || (auth.baseUrl && new URL(auth.baseUrl).origin !== provider.origin)) return UNAVAILABLE;
     const headers = { Authorization: ['zai', 'zai-coding-cn'].includes(model.provider) ? auth.apiKey : `Bearer ${auth.apiKey}` };
     if (model.provider === 'anthropic') {
       headers['anthropic-beta'] = 'oauth-2025-04-20';
@@ -35,7 +35,7 @@ export async function getQuota(model, modelRegistry, fetchImpl = fetch) {
       if (typeof accountId !== 'string' || !accountId) return UNAVAILABLE;
       headers['ChatGPT-Account-Id'] = accountId;
     }
-    const response = await fetchImpl(provider[1], {
+    const response = await fetchImpl(provider.url, {
       headers,
       signal: AbortSignal.timeout(10_000),
     });
@@ -62,7 +62,7 @@ export async function getQuota(model, modelRegistry, fetchImpl = fetch) {
       if (remaining === undefined && config.creditUsagePercent === undefined) {
         const { currentPeriod, productUsage } = config;
         if (Date.parse(currentPeriod.start) < Date.parse(currentPeriod.end) &&
-            (productUsage === undefined || (Array.isArray(productUsage) && productUsage.every((item) => item?.usagePercent === 0)))) remaining = 100;
+            Array.isArray(productUsage) && productUsage.length > 0 && productUsage.every((item) => item?.usagePercent === 0)) remaining = 100;
       }
       return remaining === undefined ? UNAVAILABLE : `Remaining quota: 7d ${remaining}%`;
     }