@llblab/pi-telegram 0.51.6 → 0.52.1

package/docs/updates.md

@@ -156,7 +156,7 @@ This means:
 
 ## Ownership semantics
 
-The handler registry is ownership-agnostic and does not interact with the extension-local transport owner slots documented in [Architecture](./architecture.md#configuration-and-ownership). When the polling runtime loses its `owners.json` slot and stops `getUpdates`, handlers stop receiving updates because no updates are being fetched; they are not unregistered.
+The handler registry is ownership-agnostic and does not interact with the extension-local transport owner slots documented in [Architecture](./architecture.md#configuration-and-ownership). When the polling runtime loses its transport ownership and stops `getUpdates`, handlers stop receiving updates because no updates are being fetched; they are not unregistered.
 
 If a layered extension needs to react to ownership changes, it should observe `pi-telegram` lifecycle events through the standard pi extension hooks rather than through the handler registry.
 

package/lib/activity-verbosity.ts

@@ -17,16 +17,16 @@ import type {
   TelegramSendRichMessageBody,
   TelegramSentMessage,
 } from "./telegram-api.ts";
-import type { TelegramTarget } from "./target.ts";
+import { areTelegramTargetsEqual as targetEquals, type TelegramTarget } from "./target.ts";
 
-export const TELEGRAM_ACTIVITY_DETAIL_MAX_CHARS = 1_200;
-export const TELEGRAM_ACTIVITY_MESSAGE_MAX_CHARS = 3_900;
+const TELEGRAM_ACTIVITY_DETAIL_MAX_CHARS = 1_200;
+const TELEGRAM_ACTIVITY_MESSAGE_MAX_CHARS = 3_900;
 export const TELEGRAM_ACTIVITY_MESSAGE_MAX_TOOLS = 6;
-export const TELEGRAM_REASONING_MESSAGE_MAX_FRAMES = 24;
+const TELEGRAM_REASONING_MESSAGE_MAX_FRAMES = 24;
 export const TELEGRAM_REASONING_BUFFER_MAX_CHARS = 1_200;
 // Match native answer drafts: accumulate the opening frame for one full
 // interval, then publish at most one updated frame per interval.
-export const TELEGRAM_REASONING_MIN_INTERVAL_MS = 2_000;
+const TELEGRAM_REASONING_MIN_INTERVAL_MS = 2_000;
 export const TELEGRAM_TOOL_UPDATE_MAX_ENTRIES = 4;
 
 interface ToolActivity {
@@ -52,10 +52,6 @@ interface ReasoningMessage {
   target: TelegramTarget;
 }
 
-function targetEquals(left: TelegramTarget, right: TelegramTarget): boolean {
-  return left.chatId === right.chatId && left.threadId === right.threadId;
-}
-
 function redactActivityText(text: string): string {
   return text
     .replace(/\b\d{8,12}:[A-Za-z0-9_-]{30,}\b/g, "[REDACTED_BOT_TOKEN]")

package/lib/bindings.ts

@@ -116,6 +116,7 @@ export function createTelegramQueueBindingRuntime<TContext>(deps: {
   };
   const mutation = Queue.createTelegramQueueMutationController({
     ...deps.store,
+    hasPendingDispatch: deps.lifecycle.hasDispatchPending,
     allocateLaneOrder: deps.queue.allocateItemOrder,
     onItemsDiscarded(items, ctx) {
       if (!settleDiscardedItems(items, ctx)) {
@@ -835,6 +836,7 @@ interface TelegramLifecycleBindingDeps {
   publicationRuntime: TelegramBridgePublicationRuntime;
   activityRuntime: Activity.TelegramActivityRuntime;
   activityVerbosityRuntime?: ActivityVerbosity.TelegramActivityVerbosityRuntime;
+  diagnostics?: { onSessionStart(): void; onSessionShutdown(): Promise<void> };
   assistantOutputRuntime: Pick<
     Activity.TelegramAssistantOutputRuntime,
     "start" | "beginTurn" | "hasAdmittedTelegramIntermediate" | "waitForIdle" | "stop"
@@ -941,6 +943,7 @@ export function registerTelegramLifecycleRuntimeHooks({
   publicationRuntime,
   activityRuntime,
   activityVerbosityRuntime,
+  diagnostics,
   assistantOutputRuntime,
   sessionLifecycleRuntime,
   configStore,
@@ -1278,6 +1281,7 @@ export function registerTelegramLifecycleRuntimeHooks({
       activityRuntime.recordInputSource(event.source ?? "unknown");
     },
     async onSessionStart(event, ctx) {
+      diagnostics?.onSessionStart();
       cancelPendingFinalPublication();
       previewRuntime.invalidate();
       assistantOutputRuntime.start();
@@ -1289,6 +1293,7 @@ export function registerTelegramLifecycleRuntimeHooks({
     },
     async onSessionShutdown(event, ctx) {
       if (!isSessionContextActive(ctx)) return;
+      const diagnosticsStopped = diagnostics?.onSessionShutdown();
       shutdownGenerativeAppLiveSurfaces?.();
       agentLifecycleHooks.clearRetainedAgentEnd();
       activityRuntime.onSessionShutdown();
@@ -1299,6 +1304,8 @@ export function registerTelegramLifecycleRuntimeHooks({
       cancelPendingFinalPublication();
       uiPromptActive = false;
       compactionObserver.onSessionShutdown();
+      await diagnosticsStopped;
+      if (!isSessionContextActive(ctx)) return;
       if (event.reason === "quit" && disconnectOnQuit) {
         try {
           const automaticCleanupEnabled =

package/lib/bus-api.ts

@@ -10,8 +10,10 @@ import {
 } from "./bus.ts";
 import {
   buildTelegramAnswerGuestQueryBody,
+  formatTelegramCallbackAnswerText,
   isTelegramMessageNotModifiedError,
 } from "./telegram-api.ts";
+import type { TelegramAttachmentSource } from "./media.ts";
 import type {
   TelegramAnswerGuestQueryOptions,
   TelegramApiCallOptions,
@@ -125,12 +127,14 @@ export function createTelegramBusAwareApiRuntime(
             options,
           ]) as Promise<TResponse>);
     },
-    downloadFile(fileId: string, suggestedName: string): Promise<string> {
+    downloadFile(fileId: string, suggestedName: string, source?: TelegramAttachmentSource): Promise<string> {
+      // The source names the file (kind-scope-message); dropping it falls back to the bare generated name.
       return deps.ownsDirect()
-        ? deps.directRuntime.downloadFile(fileId, suggestedName)
+        ? deps.directRuntime.downloadFile(fileId, suggestedName, source)
         : (deps.callFollowerApi("downloadFile", [
             fileId,
             suggestedName,
+            ...(source ? [source] : []),
           ]) as Promise<string>);
     },
     deleteWebhook(signal?: AbortSignal): Promise<boolean> {
@@ -318,15 +322,16 @@ export function createTelegramBusAwareApiRuntime(
       callbackQueryId: string,
       text?: string,
     ): Promise<void> {
+      const answer = formatTelegramCallbackAnswerText(text);
       if (deps.ownsDirect()) {
-        await deps.directRuntime.answerCallbackQuery(callbackQueryId, text);
+        await deps.directRuntime.answerCallbackQuery(callbackQueryId, answer);
         return;
       }
       await deps.callFollowerApi("call", [
         "answerCallbackQuery",
         {
           callback_query_id: callbackQueryId,
-          ...(text !== undefined ? { text } : {}),
+          ...(answer !== undefined ? { text: answer } : {}),
         },
       ]);
     },