@llblab/pi-kit 0.6.0 → 0.7.1

package/node_modules/@llblab/pi-telegram/lib/agent-messages.ts

@@ -18,6 +18,7 @@ export interface TelegramAgentMessageRuntimeDeps<TContext, TUpdate> {
   getAllowedChatId: () => number | undefined;
   getLeaderTarget: () => TelegramTarget | undefined;
   getLeaderThreadName: () => string | undefined;
+  getDisplayTitle?: (target: TelegramTarget) => string | undefined;
   followerRegistry: TelegramBusFollowerRegistry;
   getContext: () => TContext | undefined;
   handleUpdate: (update: TUpdate, ctx: TContext) => Promise<void>;
@@ -39,7 +40,7 @@ export function createTelegramAgentMessageRuntime<TContext, TUpdate>(
     if (leaderTarget?.threadId) {
       targets.push({
         target: { chatId: leaderTarget.chatId, threadId: leaderTarget.threadId },
-        threadName: deps.getLeaderThreadName(),
+        threadName: deps.getDisplayTitle?.(leaderTarget) ?? deps.getLeaderThreadName(),
       });
     }
     for (const follower of deps.followerRegistry.list()) {
@@ -49,7 +50,7 @@ export function createTelegramAgentMessageRuntime<TContext, TUpdate>(
           chatId: follower.target.chatId,
           threadId: follower.target.threadId,
         },
-        threadName: follower.threadName,
+        threadName: deps.getDisplayTitle?.(follower.target) ?? follower.threadName,
       });
     }
     return targets;
@@ -98,8 +99,10 @@ export function createTelegramAgentMessageRuntime<TContext, TUpdate>(
       if (allowedChatId === undefined || !ctx) {
         throw new Error("Telegram agent turn routing is unavailable.");
       }
+      const sourceTitle = (input.sourceTarget ? deps.getDisplayTitle?.(input.sourceTarget) : undefined)
+        ?? input.sourceThreadName;
       const sourceLabel =
-        input.sourceThreadName
+        sourceTitle
           ?.replace(/[\r\n\[\]]+/g, " ")
           .trim()
           .slice(0, 64) || "Pi";

package/node_modules/@llblab/pi-telegram/lib/bindings.ts

@@ -6,6 +6,7 @@
 
 import * as Activity from "./activity.ts";
 import * as ActivityVerbosity from "./activity-verbosity.ts";
+import * as ChannelPosts from "./channel-posts.ts";
 import * as CommandTemplates from "./command-templates.ts";
 import * as Commands from "./commands.ts";
 import * as Config from "./config.ts";
@@ -498,6 +499,10 @@ interface TelegramCommandsAndToolsBindingDeps {
   stopPolling?: () => Promise<void | string>;
   recoverPollingStart?: Commands.TelegramBridgeCommandRegistrationDeps["recoverPollingStart"];
   getDisconnectThreadName?: () => string | undefined;
+  setRequestedThreadNameForPollingStart?: (
+    threadName: string | undefined,
+  ) => void;
+  validateThreadName?: Commands.TelegramBridgeCommandRegistrationDeps["validateThreadName"];
   onTransportChanged?: () => Promise<void> | void;
   getStatusLines: (
     options?: Status.TelegramBridgeStatusLineOptions,
@@ -512,6 +517,15 @@ interface TelegramCommandsAndToolsBindingDeps {
       target?: { chatId: number; threadId?: number };
     },
   ) => Promise<number | undefined>;
+  sendChannelMarkdownMessage: NonNullable<
+    OutboundAttachments.TelegramOutboundMessageToolRegistrationDeps["sendChannelMarkdownMessage"]
+  >;
+  sendChannelMediaMessage: NonNullable<
+    OutboundAttachments.TelegramOutboundMessageToolRegistrationDeps["sendChannelMediaMessage"]
+  >;
+  listChannelPosts: ChannelPosts.TelegramChannelPostJournalStore["list"];
+  mutateChannelPost(input: { action: "edit" | "delete"; operationId: string;
+    mutationId: string; markdown?: string }): Promise<ChannelPosts.TelegramChannelPostRecord>;
   callMultipart: OutboundHandlers.TelegramVoiceReplySenderDeps["sendMultipart"];
   getDefaultChatId: () => number | undefined;
   getDefaultTarget?: () => OutboundAttachments.TelegramQueuedOutboundAttachmentTurnView["target"];
@@ -533,10 +547,16 @@ export function registerTelegramCommandsAndTools({
   stopPolling,
   recoverPollingStart,
   getDisconnectThreadName,
+  setRequestedThreadNameForPollingStart,
+  validateThreadName,
   onTransportChanged,
   getStatusLines,
   buttonActionStore,
   sendMarkdownReply,
+  sendChannelMarkdownMessage,
+  sendChannelMediaMessage,
+  listChannelPosts,
+  mutateChannelPost,
   callMultipart,
   getDefaultChatId,
   getDefaultTarget,
@@ -556,6 +576,8 @@ export function registerTelegramCommandsAndTools({
     sendMarkdownReply,
     recordRuntimeEvent,
   });
+  ChannelPosts.registerTelegramChannelPostMutationTool(pi, { mutate: mutateChannelPost });
+  ChannelPosts.registerTelegramChannelPostListTool(pi, { list: listChannelPosts });
   OutboundAttachments.registerTelegramOutboundAttachmentTool(pi, {
     getActiveTurn: activeTurnRuntime.get,
     getDefaultChatId,
@@ -578,6 +600,8 @@ export function registerTelegramCommandsAndTools({
       ),
     sendMarkdownMessage: (chatId, markdown, options) =>
       sendMarkdownReply(chatId, undefined, markdown, options),
+    sendChannelMarkdownMessage,
+    sendChannelMediaMessage,
     recordRuntimeEvent,
   });
   const queueAgentConnectionContext = (connected: boolean): void => {
@@ -636,7 +660,11 @@ export function registerTelegramCommandsAndTools({
               );
             }
             await configStore.load();
-            configStore.setProfile(profileName, profile);
+            const latestProfile = configStore.getStoredConfig().profiles?.[profileName];
+            configStore.setProfile(profileName, {
+              ...profile,
+              threadDisplayMode: latestProfile?.threadDisplayMode,
+            });
             configStore.activateProfile(profileName);
             await onTransportChanged?.();
             await persistConfig(configStore.get());
@@ -653,6 +681,10 @@ export function registerTelegramCommandsAndTools({
         setConfig: setupConfigStore.set,
         setupGuard: setup,
         getMe: TelegramApi.fetchTelegramBotIdentity,
+        resolveBotToken: (value) =>
+          Config.resolveTelegramBotToken(value, process.env),
+        describeBotToken: (value) =>
+          Config.getTelegramBotTokenDiagnostic(value, process.env),
         persistConfig: persistSetupConfig,
         startPolling: lockedPollingRuntime.start,
         updateStatus,
@@ -669,17 +701,22 @@ export function registerTelegramCommandsAndTools({
     getStatusLines,
     reloadConfig: configStore.load,
     hasBotToken: configStore.hasBotToken,
+    getBotTokenDiagnostic: configStore.getBotTokenDiagnostic,
     startPolling: async (ctx, options) => {
+      setRequestedThreadNameForPollingStart?.(options?.requestedThreadName);
       try {
         return await lockedPollingRuntime.start(ctx, options);
       } catch (error) {
         recordRuntimeEvent("recovery", error, { phase: "polling-start" });
         throw error;
+      } finally {
+        setRequestedThreadNameForPollingStart?.(undefined);
       }
     },
     stopPolling: stopPolling ?? lockedPollingRuntime.stop,
     recoverPollingStart,
     getDisconnectThreadName,
+    validateThreadName,
     queueAgentConnectionContext,
     updateStatus,
     getProfileNames: () =>
@@ -780,6 +817,12 @@ interface TelegramLifecycleBindingDeps {
       Keyboard.TelegramInlineKeyboardMarkup
     >["sendGuestReply"]
   >;
+  editGuestReply?: Queue.TelegramAgentEndHookRuntimeDeps<
+    Queue.PendingTelegramTurn,
+    Pi.ExtensionContext,
+    Pi.AgentEndEvent["messages"][number],
+    Keyboard.TelegramInlineKeyboardMarkup
+  >["editGuestReply"];
   preparePreviewDelivery?: Queue.TelegramAgentEndRuntimeDeps<Queue.PendingTelegramTurn>["preparePreviewDelivery"];
   finalizeMarkdownPreview: Queue.TelegramAgentEndHookRuntimeDeps<
     Queue.PendingTelegramTurn,
@@ -832,6 +875,7 @@ export function registerTelegramLifecycleRuntimeHooks({
   answerGuestQuery,
   deleteMessage,
   sendGuestReply,
+  editGuestReply,
   preparePreviewDelivery,
   finalizeMarkdownPreview,
   proactivePushTargetGetter,
@@ -1045,6 +1089,7 @@ export function registerTelegramLifecycleRuntimeHooks({
     sendRichAttachmentReply: richAttachmentSender,
     answerGuestQuery,
     sendGuestReply,
+    editGuestReply,
     sendGuestAttachment,
     sendGuestVoiceReply,
     planOutboundReply: outboundReplyPlanner,

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

@@ -8,11 +8,15 @@ import {
   markTelegramBusCrossTargetDelivery,
   stripTelegramBusApiMetadata,
 } from "./bus.ts";
-import { isTelegramMessageNotModifiedError } from "./telegram-api.ts";
+import {
+  buildTelegramAnswerGuestQueryBody,
+  isTelegramMessageNotModifiedError,
+} from "./telegram-api.ts";
 import type {
   TelegramAnswerGuestQueryOptions,
   TelegramApiCallOptions,
   TelegramBridgeApiRuntime,
+  TelegramEditGuestInlineMessageContent,
   TelegramEditMessageTextBody,
   TelegramSendMessageBody,
   TelegramSendMessageDraftBody,
@@ -342,24 +346,33 @@ export function createTelegramBusAwareApiRuntime(
         await deps.directRuntime.answerGuestQuery(guestQueryId, text, options);
         return;
       }
-      const body: Record<string, unknown> = { guest_query_id: guestQueryId };
-      if (options?.result) {
-        body.result = options.result;
-      } else if (text !== undefined || options?.richMessage) {
-        const inputContent: Record<string, unknown> = options?.richMessage
-          ? { rich_message: options.richMessage }
-          : { message_text: text };
-        if (!options?.richMessage && options?.parseMode) {
-          inputContent.parse_mode = options.parseMode;
-        }
-        body.result = {
-          type: "article",
-          id: "1",
-          title: "Response",
-          input_message_content: inputContent,
-        };
-      }
-      await deps.callFollowerApi("call", ["answerGuestQuery", body]);
+      await deps.callFollowerApi("call", [
+        "answerGuestQuery",
+        buildTelegramAnswerGuestQueryBody(guestQueryId, text, options),
+      ]);
+    },
+    answerGuestQueryForInlineMessage(
+      guestQueryId: string,
+      text?: string,
+      options?: TelegramAnswerGuestQueryOptions,
+    ): Promise<string | undefined> {
+      // Guest answers can only be edited while this instance owns direct
+      // transport; follower forwarding cannot preserve the inline message id.
+      return deps.ownsDirect()
+        ? deps.directRuntime.answerGuestQueryForInlineMessage(
+            guestQueryId,
+            text,
+            options,
+          )
+        : rejectTelegramDirectOwnership("answerGuestQueryForInlineMessage");
+    },
+    editGuestInlineMessage(
+      inlineMessageId: string,
+      content: TelegramEditGuestInlineMessageContent,
+    ): Promise<void> {
+      return deps.ownsDirect()
+        ? deps.directRuntime.editGuestInlineMessage(inlineMessageId, content)
+        : rejectTelegramDirectOwnership("editGuestInlineMessage");
     },
     async deleteMessage(chatId: number, messageId: number): Promise<void> {
       if (deps.ownsDirect())