arisa 4.0.3 → 4.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "arisa",
-  "version": "4.0.3",
+  "version": "4.0.5",
   "description": "Telegram + Pi Agent modular assistant",
   "type": "module",
   "main": "src/index.js",

package/src/core/agent/agent-manager.js

@@ -23,6 +23,25 @@ function requiresProviderAuth(model) {
   return !isLocalBaseUrl(model?.baseUrl);
 }
 
+async function promptAndThrowOnAssistantError(session, prompt) {
+  let assistantErrorMessage = "";
+  const unsubscribe = session.subscribe((event) => {
+    if (event.type === "message_end" && event.message?.stopReason === "error") {
+      assistantErrorMessage = event.message.errorMessage || "assistant message ended with error";
+    }
+  });
+
+  try {
+    await session.prompt(prompt);
+  } finally {
+    unsubscribe();
+  }
+
+  if (assistantErrorMessage) {
+    throw new Error(assistantErrorMessage);
+  }
+}
+
 function mimeMatches(pattern, mimeType = "") {
   if (!pattern || !mimeType) return false;
   if (pattern === mimeType) return true;
@@ -72,6 +91,10 @@ export class AgentManager {
     this.pendingNewSessions.add(sessionKey);
   }
 
+  clearSessionCache(chatId) {
+    this.sessions.delete(String(chatId));
+  }
+
   createSessionManager(chatId) {
     const sessionKey = String(chatId);
     const sessionDir = getChatPiSessionsDir(sessionKey);
@@ -103,7 +126,7 @@ export class AgentManager {
       model,
       sessionManager: SessionManager.inMemory(),
     });
-    await withTimeout(session.prompt("Reply with exactly: OK"), {
+    await withTimeout(promptAndThrowOnAssistantError(session, "Reply with exactly: OK"), {
       timeoutMs: piValidationTimeoutMs,
       label: "Pi validation prompt"
     });

package/src/runtime/create-app.js

@@ -3,6 +3,7 @@ import { ArtifactStore } from "../core/artifacts/artifact-store.js";
 import { ToolRegistry } from "../core/tools/tool-registry.js";
 import { TaskStore } from "../core/tasks/task-store.js";
 import { AgentManager } from "../core/agent/agent-manager.js";
+import { getErrorMessage, getPiAuthIssue } from "../core/agent/auth-flow.js";
 import { createTelegramBot } from "../transport/telegram/bot.js";
 import { createToolProcessSupervisor } from "./tool-process-supervisor.js";
 
@@ -64,16 +65,22 @@ export async function createApp({ logger, runtimeOverrides, webhookUrl, setHttpR
   return {
     async start() {
       logger?.log("app", `validating Pi model ${config.pi.provider}/${config.pi.model}`);
+      let skipAgentStartupPrompts = false;
       try {
         await agentManager.validatePiAgent();
       } catch (error) {
+        const issue = getPiAuthIssue(error);
+        if (!issue) {
+          throw error;
+        }
+        skipAgentStartupPrompts = true;
+        logger?.error("app", `Pi auth validation failed; starting Telegram in auth recovery mode: ${getErrorMessage(error)}`);
         await bot.notifyPiAuthIssue?.(error);
-        throw error;
       }
       await toolProcessSupervisor.start();
       logger?.log("app", "starting Telegram bot");
       try {
-        await bot.start();
+        await bot.start({ skipAgentStartupPrompts });
       } catch (error) {
         await toolProcessSupervisor.stop();
         throw error;

package/src/transport/telegram/bot.js

@@ -532,6 +532,7 @@ export async function createTelegramBot({ config, artifactStore, toolRegistry, t
     await withTyping(ctx, async () => {
       try {
         await agentManager.validatePiAgent();
+        agentManager.clearSessionCache(ctx.chat.id);
         await ctx.reply(buildPiAuthTelegramMessage({ config, verified: true }));
       } catch (error) {
         const issue = getPiAuthIssue(error) || { kind: "validation-failed", message: getErrorMessage(error) };
@@ -561,26 +562,30 @@ export async function createTelegramBot({ config, artifactStore, toolRegistry, t
   });
 
   return {
-    async start() {
+    async start({ skipAgentStartupPrompts = false } = {}) {
       config.telegram.chatMeta ||= {};
-      for (const chatId of config.telegram.authorizedChatIds || []) {
-        try {
-          logger?.log("telegram", `generating startup message for chat ${chatId}`);
-          const chatMeta = config.telegram.chatMeta[chatId] || {};
-          const welcomePrompt = [
-            "System event: Arisa has just started.",
-            `chatId: ${chatId}`,
-            `preferredTelegramLanguageCode: ${chatMeta.languageCode || "unknown"}`,
-            chatMeta.username ? `username: ${chatMeta.username}` : null,
-            chatMeta.firstName ? `firstName: ${chatMeta.firstName}` : null,
-            "Send a short welcome-back message for Telegram.",
-            "Keep it brief, warm, and natural.",
-            "Use the user's Telegram language when possible.",
-            "Do not mention internal implementation details."
-          ].filter(Boolean).join("\n");
-          await enqueuePrompt({ chatId, prompt: welcomePrompt, label: "startup message" });
-        } catch (error) {
-          logger?.log("telegram", `startup message failed for chat ${chatId}: ${error instanceof Error ? error.message : String(error)}`);
+      if (skipAgentStartupPrompts) {
+        logger?.log("telegram", "skipping agent startup messages because Pi auth needs attention");
+      } else {
+        for (const chatId of config.telegram.authorizedChatIds || []) {
+          try {
+            logger?.log("telegram", `generating startup message for chat ${chatId}`);
+            const chatMeta = config.telegram.chatMeta[chatId] || {};
+            const welcomePrompt = [
+              "System event: Arisa has just started.",
+              `chatId: ${chatId}`,
+              `preferredTelegramLanguageCode: ${chatMeta.languageCode || "unknown"}`,
+              chatMeta.username ? `username: ${chatMeta.username}` : null,
+              chatMeta.firstName ? `firstName: ${chatMeta.firstName}` : null,
+              "Send a short welcome-back message for Telegram.",
+              "Keep it brief, warm, and natural.",
+              "Use the user's Telegram language when possible.",
+              "Do not mention internal implementation details."
+            ].filter(Boolean).join("\n");
+            await enqueuePrompt({ chatId, prompt: welcomePrompt, label: "startup message" });
+          } catch (error) {
+            logger?.log("telegram", `startup message failed for chat ${chatId}: ${error instanceof Error ? error.message : String(error)}`);
+          }
         }
       }
       await bot.api.setMyCommands([