@llblab/pi-kit 0.27.6 → 0.28.0

package/node_modules/@llblab/pi-telegram/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/node_modules/@llblab/pi-telegram/lib/bindings.ts

@@ -835,6 +835,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 +942,7 @@ export function registerTelegramLifecycleRuntimeHooks({
   publicationRuntime,
   activityRuntime,
   activityVerbosityRuntime,
+  diagnostics,
   assistantOutputRuntime,
   sessionLifecycleRuntime,
   configStore,
@@ -1278,6 +1280,7 @@ export function registerTelegramLifecycleRuntimeHooks({
       activityRuntime.recordInputSource(event.source ?? "unknown");
     },
     async onSessionStart(event, ctx) {
+      diagnostics?.onSessionStart();
       cancelPendingFinalPublication();
       previewRuntime.invalidate();
       assistantOutputRuntime.start();
@@ -1289,6 +1292,7 @@ export function registerTelegramLifecycleRuntimeHooks({
     },
     async onSessionShutdown(event, ctx) {
       if (!isSessionContextActive(ctx)) return;
+      const diagnosticsStopped = diagnostics?.onSessionShutdown();
       shutdownGenerativeAppLiveSurfaces?.();
       agentLifecycleHooks.clearRetainedAgentEnd();
       activityRuntime.onSessionShutdown();
@@ -1299,6 +1303,8 @@ export function registerTelegramLifecycleRuntimeHooks({
       cancelPendingFinalPublication();
       uiPromptActive = false;
       compactionObserver.onSessionShutdown();
+      await diagnosticsStopped;
+      if (!isSessionContextActive(ctx)) return;
       if (event.reason === "quit" && disconnectOnQuit) {
         try {
           const automaticCleanupEnabled =

package/node_modules/@llblab/pi-telegram/lib/bus-api.ts

@@ -12,6 +12,7 @@ import {
   buildTelegramAnswerGuestQueryBody,
   isTelegramMessageNotModifiedError,
 } from "./telegram-api.ts";
+import type { TelegramAttachmentSource } from "./media.ts";
 import type {
   TelegramAnswerGuestQueryOptions,
   TelegramApiCallOptions,
@@ -125,12 +126,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> {